praktikum-holons/src/holeg/ui/model/DecoratedState.java

package holeg.ui.model;

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

import holeg.model.Edge;
import holeg.model.Flexibility;

public class DecoratedState {
	int timestepOfState;
	ArrayList<DecoratedNetwork> networkList;
	ArrayList<Edge> leftOverEdges;
	ArrayList<DecoratedSwitch> decoratedSwitches;
	public DecoratedState(ArrayList<DecoratedNetwork> networkList, ArrayList<Edge> leftOverEdges, ArrayList<DecoratedSwitch> decoratedSwitches, int timestepOfState){
		this.networkList = networkList;
		this.leftOverEdges = leftOverEdges;
		this.decoratedSwitches = decoratedSwitches;
		this.timestepOfState = timestepOfState;
	}
	public ArrayList<DecoratedNetwork> getNetworkList() {
		return networkList;
	}
	public ArrayList<Edge> getLeftOverEdges() {
		return leftOverEdges;
	}
	public ArrayList<DecoratedSwitch> getDecoratedSwitches() {
		return decoratedSwitches;
	}
	public int getTimestepOfState() {
		return timestepOfState;
	}
	
	public Stream<Flexibility> getAllFlex(){
		return networkList.stream().flatMap(network -> network.getElementStream().flatMap(ele -> ele.flexList.stream()));
	}
}