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

package ui.controller;

import java.util.ArrayList;

import classes.Category;
import classes.HolonBattery;
import classes.AbstractCanvasObject;
import classes.HolonElement;
import classes.HolonObject;
import classes.HolonSwitch;
import classes.HolonTransformer;
import classes.Pair;
import interfaces.CategoryListener;
import ui.model.Model;

/**
 * Controller for the Categories.
 * 
 * @author Gruppe14
 */
public class CategoryController {
	private Model model;
	private MultiPurposeController mpC;

	/**
	 * Constructor.
	 * 
	 * @param model
	 *            the Model
	 * @param mp
	 *            the MultiPurposeController
	 */
	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");
		addNewCategory("Power Flow Analysis");
		addNewHolonObject(mpC.searchCat("Energy"), "Power Plant", new ArrayList<HolonElement>(),
				"/Images/power-plant.png");
		addNewHolonObject(mpC.searchCat("Building"), "House", new ArrayList<HolonElement>(), "/Images/home-2.png");
		addNewHolonSwitch(mpC.searchCat("Component"), "Switch", "/Images/switch-on.png");
		addNewHolonBattery(mpC.searchCat("Component"), "Battery", "/Images/battery.png");

		addNewHolonObject(mpC.searchCat("Power Flow Analysis"), "Slack", new ArrayList<HolonElement>(), "/Images/slack-node.png");
		addNewHolonObject(mpC.searchCat("Power Flow Analysis"), "Transformer", new ArrayList<HolonElement>(), "/Images/transformer-1.png");
	}

	/**
	 * Adds Category into Model if a Category with the same name already exists
	 * 
	 * @param category
	 *            the new Category
	 */
	public void addCategory(Category category) {
		int i = 0;
		while (mpC.searchCat(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.
	 * 
	 * @param c
	 *            Category
	 */
	public void removeCategory(Category c) {
		mpC.decIdx(c.getName(), model.getCgIdx());
		model.getCgIdx().remove(c.getName());
		model.getCategories().remove(c);

		notifyCatListeners();
	}
	
	/**
	 * get All Categories
	 * 
	 * @return the ArrayList of all Categories
	 */
	public ArrayList<Category> getCategories()
	{
		return model.getCategories();
	}
	
	
	/**
	 * delete a given Category.
	 * 
	 * @param category
	 *            the Category
	 */
	public void deleteCategory(String category) {
		removeCategory(mpC.searchCat(category));
	}

	/**
	 * 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 = "";
		while (mpC.searchCatObj(category, object.getObjName()) != null) {
			updateElementSaves = true;
			if (object.getObjName().contains("_"))
				object.setObjName(object.getObjName().substring(0, object.getObjName().indexOf('_')));
			name = object.getObjName() + "_" + i;
			object.setObjName(name);
			object.setName(name);
			i++;
		}
		if(updateElementSaves && object instanceof HolonObject &&((HolonObject)object).getElements()!=null){
			for(HolonElement e: ((HolonObject)object).getElements()){
				e.setSaving(new Pair<String,String>(e.getSaving().getKey(), name));
			}
		}
		category.getObjIdx().put(object.getObjName(), category.getObjects().size());
		category.getObjects().add(object);
		notifyCatListeners();
	}

	/**
	 * Add new Holon Object to a Category.
	 * 
	 * @param category
	 *            Category
	 * @param object
	 *            New Object Name
	 * @param elements
	 *            Array of Elements
	 * @param image
	 *            the image Path
	 */
	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 Switch.
	 * 
	 * @param cat
	 *            Category
	 * @param objName
	 *            New Object Name
	 * @param image
	 *            the Image Path
	 */
	public void addNewHolonSwitch(Category cat, String objName, String image) {
		HolonSwitch holonSwitch = new HolonSwitch(objName);
		
		holonSwitch.setImage(image);
		holonSwitch.setSav(cat.getName());
		addObject(cat, holonSwitch);
	}
	/**
	 * Add new Holon Battery
	 * @param cat 
	 * 			Categorx
	 * @param objName
	 * 			New Object Name
	 * @param image
	 * 			The Image Path
	 */
	public void addNewHolonBattery(Category cat, String objName, String image) {
		HolonBattery holonBat = new HolonBattery(objName);
		holonBat.setImage(image);
		holonBat.setSav(cat.getName());
		addObject(cat, holonBat);
	}
	
	public void addNewHolonBattery(Category cat, HolonBattery holonBat, String image) {
		holonBat.setImage(image);
		holonBat.setSav(cat.getName());
		addObject(cat, holonBat);
	}
	
	
	
	/**
	 * Removes an Object from a Category.
	 * @param category Category
	 * @param cps the Object
	 */
	public void removeObject(Category category, AbstractCanvasObject cps) {

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

		notifyCatListeners();
	}

	/**
	 * Delete an Object from a Category.
	 * 
	 * @param category
	 *            the Category
	 * @param obj
	 *            the Object
	 */
	public void deleteObject(String category, String obj) {
		Category cat = mpC.searchCat(category);
		removeObject(cat, mpC.searchCatObj(cat, obj));
	}

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

	/**
	 * notifies all listeners about changes in the Categories.
	 */
	public void notifyCatListeners() {
		ArrayList<Category> categories =  model.getCategories();
		for (CategoryListener l : model.getCategoryListeners()) {
			l.onChange(categories);
		}
	}

}