praktikum-holons/src/holeg/ui/controller/CategoryController.java

package holeg.ui.controller;

import java.util.AbstractMap.SimpleEntry;

import holeg.model.AbstractCanvasObject;
import holeg.model.HolonElement;
import holeg.model.HolonObject;
import holeg.model.HolonSwitch;
import holeg.ui.model.GuiSettings;
import holeg.ui.view.main.Category;
import holeg.utility.events.Event;

import java.util.ArrayList;
import java.util.List;
import java.util.Optional;

/**
 * Controller for the Categories.
 * 
 * @author Gruppe14
 */
public class CategoryController {
	private final Event OnCategoryChanged;

	/**
	 * Constructor.
	 * 
	 * @param model
	 *            the Model
	 * @param mp
	 *            the MultiPurposeController
	 * @param control 
	 */
	public CategoryController(Control control) {
		this.OnCategoryChanged = control.OnCategoryChanged;
		initCategories();
	}

	/**
	 * init default category and objects.
	 */
	public void initCategories() {
		Category energy = createCategoryWithName("Energy");
		Category building = createCategoryWithName("Building");
		Category component = createCategoryWithName("Component");
		HolonObject powerPlant = addNewHolonObject(energy, "Power Plant", new ArrayList<HolonElement>(),
				"/Images/power-plant.png");
		HolonObject house = addNewHolonObject(building, "House", new ArrayList<HolonElement>(), "/Images/home-2.png");
		addNewHolonSwitch(component, "Switch", "/Images/switch-on.png");
		powerPlant.addElement(new HolonElement(null, "Power", 10000));
		energy.getObjects().add(powerPlant);
	
		house.addElement(new HolonElement(null, "TV", -250));
		house.addElement(new HolonElement(null, "TV", -250));
		house.addElement(new HolonElement(null, "Fridge", -500));
		house.addElement(new HolonElement(null, "Radio", -100));
		house.addElement(new HolonElement(null, "PC", -250));
		house.addElement(new HolonElement(null, "PC", -250));
		house.addElement(new HolonElement(null, "PC", -250));
		house.addElement(new HolonElement(null, "Light", -50));
		house.addElement(new HolonElement(null, "Light", -50));
		house.addElement(new HolonElement(null, "Light", -50));
		house.addElement(new HolonElement(null, "Light", -50));
		house.addElement(new HolonElement(null, "Light", -50));
		house.addElement(new HolonElement(null, "Solar Panel", 300));
		building.getObjects().add(house);
		OnCategoryChanged.broadcast();
	}

	
	public Category createCategoryWithName(String categoryName) {
		int i = 0;
		while (findCategoryWithName(categoryName).isPresent()) {
			if (categoryName.contains("_")) {
				categoryName = (categoryName.substring(0, categoryName.indexOf('_')));				
			}
			
			categoryName += "_" + i;
			i++;
		}
		
		Category cat = new Category(categoryName);
		GuiSettings.getCategories().add(cat);	
		OnCategoryChanged.broadcast();
		return cat;
	}


	/**
	 * remove a Category from Model.
	 * 
	 * @param c
	 *            Category
	 */
	public void removeCategory(Category c) {
		GuiSettings.getCategories().remove(c);
		OnCategoryChanged.broadcast();
	}
	

	/**
	 * Add Object into a Category.
	 * 
	 * @param category
	 *            Category
	 * @param object
	 *            Object
	 */
	public void addObject(Category category, AbstractCanvasObject object) {
		int i = 0;
		boolean updateElementSaves = false;
		String name = "";
		//TODO(Tom2021-12-1) remove/redo this search
		while (category.findObjectWithName(object.getName()).isPresent()) {
			updateElementSaves = true;
			if (object.getName().contains("_"))
				object.setName(object.getName().substring(0, object.getName().indexOf('_')));
			name = object.getName() + "_" + i;
			object.setName(name);
			i++;
		}
		if(updateElementSaves && object instanceof HolonObject hO&&hO.getElements()!=null){
			for(HolonElement e: hO.getElements()){
				e.setSaving(new SimpleEntry<String,String>(e.getSaving().getKey(), name));
			}
		}
		category.getObjects().add(object);
	}

	/**
	 * Add new Holon Object to a Category.
	 * 
	 * @param category
	 *            Category
	 * @param object
	 *            New Object Name
	 * @param list
	 *            Array of Elements
	 * @param image
	 *            the image Path
	 */
	public HolonObject addNewHolonObject(Category category, String object, List<HolonElement> list, String image) {
		HolonObject obj = new HolonObject(object);
		obj.setImage(image);
		obj.setElements(list);
		obj.setSav(category.getName());
		addObject(category, obj);
		return obj;
	}
	

	/**
	 * Add new Holon Switch.
	 * 
	 * @param cat
	 *            Category
	 * @param objName
	 *            New Object Name
	 * @param image
	 *            the Image Path
	 */
	public HolonSwitch addNewHolonSwitch(Category cat, String objName, String image) {
		HolonSwitch holonSwitch = new HolonSwitch(objName);
		holonSwitch.setImage(image);
		holonSwitch.setSav(cat.getName());
		addObject(cat, holonSwitch);
		return holonSwitch;
	}
	
	
	
	/**
	 * Removes an Object from a Category.
	 * @param category Category
	 * @param cps the Object
	 */
	public void removeObject(Category category, AbstractCanvasObject cps) {
		category.getObjects().remove(cps);
		OnCategoryChanged.broadcast();
	}

	
	
	public Optional<Category> findCategoryWithName(String categoryName) {
		return GuiSettings.getCategories().stream().filter(cat -> cat.getName().equals(categoryName)).findAny();
	}
	
	

}