praktikum-holons/src/holeg/model/GroupNode.java

package holeg.model;

import java.util.ArrayList;
import java.util.logging.Logger;
import java.util.stream.Stream;

public class GroupNode extends AbstractCanvasObject {
	private static final Logger log = Logger.getLogger(AbstractCanvasObject.class.getName());

	private ArrayList<HolonObject> objectList = new ArrayList<>();
	private ArrayList<HolonSwitch> switchList = new ArrayList<>();
	private ArrayList<Node> nodeList = new ArrayList<>();
	private ArrayList<GroupNode> groupNodeList = new ArrayList<>();

	public GroupNode(String nodeName) {
		super(nodeName);
		this.setImage("/Images/upper_node.png");
		this.setSav("CVS");
	}

	@Override
	public void initForReflection() {
		log.info("Executed");
		objectList = new ArrayList<>();
		switchList = new ArrayList<>();
		nodeList = new ArrayList<>();
		groupNodeList = new ArrayList<>();
	}

	
	public void add(AbstractCanvasObject object) {
		if (object instanceof HolonObject hObject) {
			objectList.add(hObject);
		}else if(object instanceof HolonSwitch hSwitch) {
			switchList.add(hSwitch);
		}else if(object instanceof Node node) {
			nodeList.add(node);
		}else if(object instanceof GroupNode groupNode) {
			groupNodeList.add(groupNode);
		}
		object.setGroupNode(this);
	}
	
	public void remove(AbstractCanvasObject object) {
		if (object instanceof HolonObject hObject) {
			objectList.remove(hObject);
		}else if(object instanceof HolonSwitch hSwitch) {
			switchList.remove(hSwitch);
		}else if(object instanceof Node node) {
			nodeList.remove(node);
		}else if(object instanceof GroupNode groupNode) {
			groupNodeList.remove(groupNode);
		}
		object.setGroupNode(null);
	}
	

	public Stream<AbstractCanvasObject> getObjectsInThisLayer() {
		return Stream.of(objectList.stream(), switchList.stream(), nodeList.stream(), groupNodeList.stream())
				.flatMap(s -> s);
	}

	public Stream<AbstractCanvasObject> getAllObjectsRecursive() {
		Stream<AbstractCanvasObject> objects = Stream.of(objectList.stream(), switchList.stream(), nodeList.stream())
				.flatMap(s -> s);
		return Stream.concat(objects,
				groupNodeList.stream().flatMap(groupnode -> groupnode.getObjectsInThisLayer()));

	}

	public Stream<HolonObject> getAllHolonObjectsRecursive() {
		return Stream.concat(objectList.stream(),
				groupNodeList.stream().flatMap(groupnode -> groupnode.getAllHolonObjectsRecursive()));
	}
	public Stream<HolonSwitch> getAllSwitchObjectsRecursive() {
		return Stream.concat(switchList.stream(),
				groupNodeList.stream().flatMap(groupnode -> groupnode.getAllSwitchObjectsRecursive()));
	}
	public Stream<Node> getAllNodeObjectsRecursive() {
		return Stream.concat(nodeList.stream(),
				groupNodeList.stream().flatMap(groupnode -> groupnode.getAllNodeObjectsRecursive()));
	}
	public Stream<GroupNode> getAllGroupNodeObjectsRecursive() {
		return Stream.concat(groupNodeList.stream(),
				groupNodeList.stream().flatMap(groupnode -> groupnode.getAllGroupNodeObjectsRecursive()));
	}
	
	public Stream<HolonObject> getHolonObjects() {
		return objectList.stream();
	}
	
	public Stream<HolonSwitch> getSwitches() {
		return switchList.stream();
	}
	public Stream<Node> getNodes() {
		return nodeList.stream();
	}
	
	public Stream<GroupNode> getGroupNodes() {
		return groupNodeList.stream();
	}
	
}