no.oddsor.simulator3.json.TaskReader.java Source code

Introduction

Here is the source code for no.oddsor.simulator3.json.TaskReader.java

Source

/*
 * To change this license header, choose License Headers in Project Properties.
 * To change this template file, choose Tools | Templates
 * and open the template in the editor.
 */

package no.oddsor.simulator3.json;

import java.io.FileReader;
import java.util.ArrayList;
import java.util.Collection;
import java.util.HashSet;
import java.util.Iterator;
import java.util.List;
import java.util.Set;
import no.oddsor.simulator3.ITask;
import no.oddsor.simulator3.Need;
import no.oddsor.simulator3.Person;
import no.oddsor.simulator3.Task;
import org.json.simple.*;
import org.json.simple.parser.JSONParser;

/**
 *
 * @author Odd
 */
public class TaskReader {

    public JSONObject object;

    public TaskReader(String folder) throws Exception {
        JSONParser jp = new JSONParser();
        try {
            this.object = (JSONObject) jp.parse(new FileReader(folder + "/activities.json"));
        } catch (Exception e) {
            this.object = null;
        }
    }

    public Collection<ITask> getTasks() {
        Collection<ITask> tasks = new ArrayList<>();
        if (object == null)
            return tasks;
        JSONArray tasklist = (JSONArray) object.get("Activities");
        Iterator taskIterator = tasklist.iterator();
        while (taskIterator.hasNext()) {
            JSONObject nextTask = (JSONObject) taskIterator.next();
            Task newTask;
            if (nextTask.containsKey("Timespan")) {
                JSONArray timespan = (JSONArray) nextTask.get("Timespan");
                tasks.add(newTask = new Task((String) nextTask.get("Name"), (String) nextTask.get("Type"),
                        Integer.parseInt(nextTask.get("Duration").toString()),
                        Integer.parseInt(timespan.get(0).toString()), Integer.parseInt(timespan.get(1).toString()),
                        (String) nextTask.get("Appliance"),
                        (nextTask.containsKey("Label") ? (String) nextTask.get("Label") : null)));
            } else {
                tasks.add(newTask = new Task((String) nextTask.get("Name"), (String) nextTask.get("Type"),
                        Integer.parseInt(nextTask.get("Duration").toString()), (String) nextTask.get("Appliance"),
                        (nextTask.containsKey("Label") ? (String) nextTask.get("Label") : null)));
            }
            if (nextTask.containsKey("Cooldown")) {
                double d = (long) nextTask.get("Cooldown") * 3600.0;
                newTask.setCooldown(d);
            }
            if (nextTask.containsKey("IncreasesNeed")) {
                JSONArray need = (JSONArray) nextTask.get("IncreasesNeed");
                newTask.addResult((String) need.get(0), Integer.parseInt(need.get(1).toString()));
            }
            if (nextTask.containsKey("RequiresItem")) {
                if (!(nextTask.get("RequiresItem") instanceof JSONArray)) {
                    newTask.addRequiredItem((String) nextTask.get("RequiresItem"), 1);
                } else {
                    JSONArray requirements = (JSONArray) nextTask.get("RequiresItem");
                    if (requirements.get(1) instanceof Number) {
                        newTask.addRequiredItem((String) requirements.get(0),
                                Integer.parseInt(requirements.get(1).toString()));
                    } else {
                        for (Object item : requirements) {
                            newTask.addRequiredItem((String) item, 1);
                        }
                    }
                }
            }
            if (nextTask.containsKey("Creates")) {
                if (nextTask.get("Creates") instanceof JSONArray) {
                    JSONArray creates = (JSONArray) nextTask.get("Creates");
                    newTask.addResultingItem((String) creates.get(0), Integer.parseInt(creates.get(1).toString()));
                } else
                    newTask.addResultingItem((String) nextTask.get("Creates"), 1);
            }
            if (nextTask.containsKey("PoseFeatures")) {
                if (nextTask.get("PoseFeatures") instanceof JSONArray) {
                    JSONArray poses = (JSONArray) nextTask.get("PoseFeatures");
                    for (Object pose : poses) {
                        String poseString = (String) pose;
                        newTask.addPose(poseString);
                    }
                }
            }
            if (nextTask.containsKey("NeedsState")) {
                if (nextTask.get("NeedsState") instanceof JSONArray) {
                    JSONArray poses = (JSONArray) nextTask.get("NeedsState");
                    for (Object pose : poses) {
                        String poseString = (String) pose;
                        newTask.addNeededState(poseString);
                    }
                }
            }
            if (nextTask.containsKey("AddsState")) {
                if (nextTask.get("AddsState") instanceof JSONArray) {
                    JSONArray poses = (JSONArray) nextTask.get("AddsState");
                    for (Object pose : poses) {
                        String poseString = (String) pose;
                        if (poseString.charAt(0) == '+')
                            newTask.addPlusState(poseString.substring(1));
                        if (poseString.charAt(0) == '-')
                            newTask.addMinusState(poseString.substring(1));
                    }
                }
            }
        }
        return tasks;
    }

    public Set<String> getAppliances() {
        Set<String> appliances = new HashSet<>();
        if (object == null)
            return appliances;
        JSONArray tasklist = (JSONArray) object.get("Activities");
        Iterator tasks = tasklist.iterator();
        while (tasks.hasNext()) {
            JSONObject task = (JSONObject) tasks.next();
            if (task.containsKey("Appliance"))
                appliances.add((String) task.get("Appliance"));
            if (task.containsKey("Creates") && task.get("Creates") instanceof JSONArray) {
                JSONArray createArray = (JSONArray) task.get("Creates");
                if (createArray.get(1) instanceof String)
                    appliances.add(createArray.get(1).toString());
            }
        }
        return appliances;
    }

    public List<Need> getNeeds() {
        List<Need> needs = new ArrayList<>();
        if (object == null)
            return needs;
        JSONArray taskList = (JSONArray) object.get("Activities");
        Set<String> needNames = new HashSet<>();
        Iterator tasks = taskList.iterator();
        while (tasks.hasNext()) {
            JSONObject task = (JSONObject) tasks.next();
            if (task.containsKey("IncreasesNeed")) {
                JSONArray arr = (JSONArray) task.get("IncreasesNeed");
                needNames.add((String) arr.get(0));
            }
        }
        for (String name : needNames) {
            needs.add(new Need(name, 1.0));
        }
        return needs;
    }

    public Collection<Person> getPeople() {
        JSONArray peopleMap = (JSONArray) object.get("People");
        Collection<Person> people = new ArrayList<>();
        for (int i = 0; i < peopleMap.size(); i++) {
            JSONObject personObject = (JSONObject) peopleMap.get(i);
            Person p = new Person((String) personObject.get("Name"), (String) personObject.get("Image"), null,
                    getNeeds());
            people.add(p);
        }
        return people;
    }

}