Missing file

src/ui/controller/FlexManager.java

@@ -0,0 +1,139 @@
+package ui.controller;
+
+import java.util.ArrayList;
+import java.util.HashMap;
+import java.util.List;
+import java.util.stream.Collectors;
+
+import classes.AbstractCpsObject;
+import classes.CpsUpperNode;
+import classes.Flexibility;
+import classes.HolonElement;
+import classes.HolonObject;
+import ui.model.Model;
+
+/**
+ * Class to Manage to flexibilities.
+ * To order, view Flexibilities.
+ * @author tom
+ *
+ */
+public class FlexManager {
+	private int timeStep;
+	private  List<Flexibility> allFlexModels;
+	private List<Flexibility> allFlexesOrderedThisTimeStep = new ArrayList<Flexibility>();
+	private HashMap<Flexibility, FlexWrapper> accessFlexMap = new HashMap<Flexibility, FlexWrapper>();
+	public FlexManager(Model model, int timeStep, FlexManager timestepBefore){
+		this.timeStep = timeStep;
+		allFlexModels = getAllFlexFromModel(model);
+		//fill accessFlexMap
+		allFlexModels.stream().map(flex -> new FlexWrapper(flex, timeStep, (timestepBefore != null)?timestepBefore.getFlexWrapper(flex):null)).forEach(flexWrapper -> accessFlexMap.put(flexWrapper.getFlex(), flexWrapper));
+	}
+
+	
+	
+	
+	
+	private FlexWrapper getFlexWrapper(Flexibility flex) {
+		return accessFlexMap.getOrDefault(flex, null);
+	}
+
+	public List<FlexWrapper> getAllFlexWrapper() {
+		return accessFlexMap.values().stream().collect(Collectors.toList());
+	}
+	
+	public List<FlexWrapper> getAllFlexWrapperWithState(FlexState state) {
+		return accessFlexMap.values().stream().filter(flexWrapper -> (flexWrapper.getState() == state)).collect(Collectors.toList());
+	}
+
+
+
+	private List<Flexibility> getAllFlexFromModel(Model model) {
+		return createListOfHolonObjects(model.getObjectsOnCanvas()).stream().flatMap(hObject -> hObject.getElements().stream()).flatMap(hElement -> hElement.flexList.stream()).collect(Collectors.toList());
+	}
+
+	private List<HolonObject> createListOfHolonObjects(List<AbstractCpsObject> objectsOnCanvas) {
+		List<HolonObject> list = new ArrayList<HolonObject>();
+		for(AbstractCpsObject aCps :  objectsOnCanvas) {
+			if(aCps instanceof HolonObject) list.add((HolonObject) aCps);
+			else list.addAll(createListOfHolonObjects(((CpsUpperNode)aCps).getNodes()));
+		}
+		return list;
+	}
+
+
+	public int getTimeStep() {
+		return timeStep;
+	}
+	
+	public List<Flexibility> getAllFlexesOrderedThisTimeStep() {
+		return allFlexesOrderedThisTimeStep;
+	}
+
+	public void orderFlexFromList(List<Flexibility> flexList) {
+		flexList.stream().forEach(flex -> {
+			FlexWrapper flexToOrder = accessFlexMap.get(flex);
+			if(flexToOrder!=null)flexToOrder.order();
+		});
+	}
+	
+	public boolean isAFlexInUseOfHolonElement(HolonElement ele) {
+		return ele.flexList.stream().filter(flex -> this.accessFlexMap.containsKey(flex)).anyMatch(flex -> (this.accessFlexMap.get(flex).getState() == FlexState.IN_USE));
+	}
+	
+	
+	
+	public static enum FlexState{
+			IN_USE, ON_COOLDOWN, OFFERED, NOT_OFFERED, UNAVAILABLE
+		}
+	//Classes
+	public class FlexWrapper{
+		private Flexibility flex;
+		private FlexState state;
+		int timeStep;
+		int durationEndTime = -1;
+		int coolDownEndTime = -1;
+		public FlexWrapper(Flexibility flex, int timeStep, FlexWrapper old) {
+			this.flex = flex;
+			this.timeStep = timeStep;
+			if(old == null) {
+				state = flex.fulfillsConstrains()?(flex.offered?FlexState.OFFERED:FlexState.NOT_OFFERED):FlexState.UNAVAILABLE;
+			}else {
+				durationEndTime = old.durationEndTime;
+				coolDownEndTime = old.coolDownEndTime;
+				if(remainingTimeTillActivation() == 0) state = flex.fulfillsConstrains()?(flex.offered?FlexState.OFFERED:FlexState.NOT_OFFERED):FlexState.UNAVAILABLE;
+				else if(remainingDuration()== 0) state = FlexState.ON_COOLDOWN;
+				else state = FlexState.IN_USE;
+			}
+				
+			
+		}
+		public Flexibility getFlex() {
+			return flex;
+		}
+		public FlexState getState() {
+			return state;
+		}
+		public boolean canOrder() {
+			return (state == FlexState.OFFERED) &&  //Right state
+					!accessFlexMap.values().stream().anyMatch(flexWrapper -> (flexWrapper.getFlex().getElement() == flex.getElement() && flexWrapper != this && (flexWrapper.getState() == FlexState.IN_USE || flexWrapper.getState() == FlexState.ON_COOLDOWN))); //No other flex of this ele in use
+		}
+		public boolean order() {
+			if(canOrder()) {
+				state=FlexState.IN_USE;
+				allFlexesOrderedThisTimeStep.add(flex);
+				durationEndTime = timeStep + flex.getDuration();
+				coolDownEndTime = durationEndTime + flex.getCooldown(); 
+				return true;
+			}
+			return false;
+		}
+		public int remainingTimeTillActivation() {
+			return Math.max(0, coolDownEndTime - timeStep);
+		}
+		public int remainingDuration() {
+			return Math.max(0, durationEndTime - timeStep);
+		}
+		
+	}
+}