praktikum-holons/src/ui/controller/ObjectController.java

package ui.controller;

import java.util.ArrayList;
import java.util.Collection;

import classes.AbstractCanvasObject;
import classes.GroupNode;
import classes.HolonElement;
import classes.HolonObject;
import classes.Pair;
import ui.model.Model;

/**
 * Controller for Objects.
 *
 * @author Gruppe14
 */
public class ObjectController {

	private Model model;
	private MultiPurposeController mpC;

	/**
	 * Constructor.
	 *
	 * @param model Model
	 * @param mp    MultiPurposeController
	 */
	public ObjectController(Model model, MultiPurposeController mp) {
		this.model = model;
		this.mpC = mp;
		initHolonElements();
	}

	/**
	 * init default Power supply of the Power Plant.
	 */
	public void initHolonElements() {
		addNewElementIntoCategoryObject("Energy", "Power Plant", "Power", 10000);
		addNewElementIntoCategoryObject("Building", "House", "TV", -250);
		addNewElementIntoCategoryObject("Building", "House", "TV", -250);
		addNewElementIntoCategoryObject("Building", "House", "Fridge", -500);
		addNewElementIntoCategoryObject("Building", "House", "Radio", -100);
		addNewElementIntoCategoryObject("Building", "House", "PC", -250);
		addNewElementIntoCategoryObject("Building", "House", "PC", -250);
		addNewElementIntoCategoryObject("Building", "House", "PC", -250);
		addNewElementIntoCategoryObject("Building", "House", "Light", -50);
		addNewElementIntoCategoryObject("Building", "House", "Light", -50);
		addNewElementIntoCategoryObject("Building", "House", "Light", -50);
		addNewElementIntoCategoryObject("Building", "House", "Light", -50);
		addNewElementIntoCategoryObject("Building", "House", "Light", -50);
		addNewElementIntoCategoryObject("Building", "House", "Solar Panels", 300);
	}

	/**
	 * Adds Element into a Object.
	 *
	 * @param object  the Object
	 * @param element the Element
	 */
	public void addElement(HolonObject object, HolonElement element) {
		object.addElement(element);
	}

	/**
	 * Adds Element into a Object on the Canvas.
	 *
	 * @param object  the Object
	 * @param element the Element
	 */
	public void addElementIntoCanvasObject(HolonObject object, HolonElement element) {
		element.setSaving(null);
		addElement(object, element);
	}

	/**
	 * Add a new Element into a Object on the Canvas.
	 *
	 * @param objectId  the Object ID
	 * @param element   the Name of the Element
	 * @param amount    the Amount
	 * @param energy    the Energy
	 * @param elementId the Element ID
	 */
	public void addNewElementIntoCanvasObject(int objectId, String element, float energy, int elementId) {
		HolonObject hObject = (HolonObject) mpC.searchByID(objectId);
		if (hObject == null) {
			hObject = (HolonObject) model.getSelectedObjects().stream().findFirst().get();
		}
		HolonElement ele = new HolonElement(hObject, element, energy, elementId);
		addElementIntoCanvasObject(hObject, ele);
	}

	/**
	 * Add Element into a Object in Category.
	 *
	 * @param category the Category
	 * @param object   the Object
	 * @param element  the Element
	 */
	public void addElementIntoCategoryObject(String category, String object, HolonElement element) {
		element.setSaving(new Pair<>(category, object));
		addElement((HolonObject) mpC.searchCatObj(mpC.searchCat(category), object), element);
	}

	/**
	 * Add a new Element into a Object in Category.
	 *
	 * @param category the Category
	 * @param object   the Object
	 * @param element  the Element Name
	 * @param energy   the Energy
	 */
	public void addNewElementIntoCategoryObject(String category, String object, String element, float energy) {

		HolonElement ele = new HolonElement((HolonObject) mpC.searchCatObj(mpC.searchCat(category), object), element,
				energy);
		addElementIntoCategoryObject(category, object, ele);
	}

	/**
	 * deletes a Element from a given Object.
	 *
	 * @param obj the Oject
	 * @param ele the Element
	 */
	public void deleteElement(HolonObject obj, HolonElement ele) {
		obj.getElements().remove(ele);
	}

	/**
	 * deletes a selectedObject.
	 *
	 * @param obj Cpsobject
	 */
	public void removeObjectFromSelection(AbstractCanvasObject obj) {
		model.getSelectedObjects().remove(obj);
	}

	/**
	 * add an Object to selectedObject.
	 *
	 * @param obj AbstractCpsobject
	 */
	public void addSelectedObject(AbstractCanvasObject obj) {
		model.getSelectedObjects().add(obj);
	}


	/**
	 * add an Object to selectedObject.
	 *
	 * @param obj AbstractCpsobject
	 */
	public void addSelectedObjects(Collection<AbstractCanvasObject> objects) {
		model.getSelectedObjects().addAll(objects);
	}

	/**
	 * deletes a Element from a given Canvas Object.
	 *
	 * @param id    the ID
	 * @param eleid the Element ID
	 */
	public void deleteElementInCanvas(int id, int eleid) {
		HolonObject hObject = (HolonObject) mpC.searchByID(id);
		if (hObject == null) {
			hObject = (HolonObject) model.getSelectedObjects().stream().findFirst().get();
		}
		HolonElement element = mpC.searchEleById(hObject, eleid);
		deleteElement(hObject, element);
	}


	/**
	 * Get the number of HolonObjects in the given List
	 *
	 * @param list
	 */
	public int getNumberHolonObjects(ArrayList<AbstractCanvasObject> list) {
		int val = 0;

		for (AbstractCanvasObject obj : list) {
			if (obj instanceof HolonObject) {
				val++;
			} else if (obj instanceof GroupNode) {
				val += getNumberHolonObjects(((GroupNode) obj).getNodes());
			}
		}
		return val;
	}

}