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

package ui.controller;

import java.util.ArrayList;

import Interfaces.CategoryListener;
import classes.Category;
import classes.CpsObject;
import classes.HolonElement;
import classes.HolonObject;
import classes.HolonSwitch;
import classes.HolonTransformer;
import ui.model.Model;

public class CategoryController {
	private Model MODEL;
	private MultiPurposeController mpC;

	public CategoryController(Model model, MultiPurposeController mp) {
		this.MODEL = model;
		this.mpC = mp;
		initCategories();
	}

	/**
	 * init default category and objects
	 */
	public void initCategories() {

		addNewCategory("Energy");
		addNewCategory("Building");
		addNewCategory("Component");

		addNewHolonObject(mpC.searchCategory("Energy"), "Power Plant", new ArrayList<HolonElement>(), "/Images/power-plant.png");
		addNewHolonObject(mpC.searchCategory("Building"), "House", new ArrayList<HolonElement>(), "/Images/home-2.png");
		addNewHolonTransformer(mpC.searchCategory("Component"), "Transformer", "/Images/transformer-1.png");
		addNewHolonSwitch(mpC.searchCategory("Component"), "Switch", "/Images/switch-off.png");

	}

	/**
	 * Adds Category into Model if a Category with the same name already exists
	 * Add Category_+1
	 * 
	 * @param toAdd
	 *            neue Kategorie
	 */
	public void addCategory(Category category) {
		// int number = 0;
		// String name = toAdd.getName();
		// while (sC.containsInList(MODEL.getCategories(), toAdd)) {
		// number++;
		// toAdd.setName(name + "_" + number);
		// }
		int i = 0;
		while (mpC.searchCategory(category.getName()) != null) {
			if (category.getName().contains("_"))
				category.setName(category.getName().substring(0, category.getName().indexOf('_')));
			category.setName(category.getName() + "_" + i);
			i++;
		}
		MODEL.getCgIdx().put(category.getName(), MODEL.getCategories().size());
		MODEL.getCategories().add(category);
		notifyCatListeners();
	}

	/**
	 * Adds New Category into Model
	 * 
	 * @param name
	 *            Bezeichnung der neuen Kategorie
	 */
	public void addNewCategory(String name) {

		addCategory(new Category(name));
	}

	/**
	 * remove a Category from Model
	 */
	public void removeCategory(Category c) {
		mpC.decIdx(c.getName(), MODEL.getCgIdx());
		MODEL.getCgIdx().remove(c.getName());
		MODEL.getCategories().remove(c);

		notifyCatListeners();
	}

	/**
	 * delete a given Category
	 * 
	 * @param category
	 */
	public void deleteCategory(String category) {
		removeCategory(mpC.searchCategory(category));
	}

	/**
	 * Add Object into a Category
	 * 
	 * @param category
	 *            Category
	 * @param object
	 *            Object
	 */
	public void addObject(Category category, CpsObject object) {
		int i = 0;
		while (mpC.searchCategoryObject(category, object.getObjName()) != null) {
			if (object.getObjName().contains("_"))
				object.setObjName(object.getObjName().substring(0, object.getObjName().indexOf('_')));
			String name = object.getObjName() + "_" + i;
			object.setObjName(name);
			object.setName(name);
			i++;
		}
		category.getObjIdx().put(object.getObjName(), category.getObjects().size());
		category.getObjects().add(object);
		notifyCatListeners();
	}

	/**
	 * Add new Holon Object
	 * 
	 * @param cat
	 *            Category
	 * @param obj
	 *            New Object Name
	 */
	public void addNewHolonObject(Category category, String object, ArrayList<HolonElement> elements, String image) {
		HolonObject obj = new HolonObject(object);
		obj.setImage(image);
		obj.setElements(elements);
		obj.setSav(category.getName());
		addObject(category, obj);
	}

	/**
	 * Add new Holon Transformer
	 * 
	 * @param cat
	 *            Category
	 * @param obj
	 *            New Object Name
	 */
	public void addNewHolonTransformer(Category cat, String objName, String image) {
		HolonTransformer transformer = new HolonTransformer(objName);
		transformer.setImage(image);
		transformer.setSav(cat.getName());
		addObject(cat, transformer);
	}

	/**
	 * Add new Holon Switch
	 * 
	 * @param cat
	 *            Category
	 * @param obj
	 *            New Object Name
	 */
	public void addNewHolonSwitch(Category cat, String objName, String image) {
		HolonSwitch holonSwitch = new HolonSwitch(objName);

		holonSwitch.setImage(image);
		holonSwitch.setSav(cat.getName());
		addObject(cat, holonSwitch);
	}

	public void removeObject(Category category, CpsObject cps) {

		mpC.decIdx(cps.getObjName(), category.getObjIdx());
		category.getObjIdx().remove(cps.getObjName());
		category.getObjects().remove(cps);

		notifyCatListeners();
	}

	public void deleteObject(String category, String obj) {
		Category cat = mpC.searchCategory(category);
		removeObject(cat, mpC.searchCategoryObject(cat, obj));
	}

	/**
	 * 
	 * @param catLis
	 */
	public void addCategoryListener(CategoryListener catLis) {
		MODEL.getCategoryListeners().add(catLis);
	}

	/**
	 * notifies all listeners about changes in the Categories
	 */
	public void notifyCatListeners() {
		for (CategoryListener l : MODEL.getCategoryListeners()) {
			l.onChange(MODEL.getCategories());
		}
	}

}