package org.jbpm.simulation.impl.ht;
import java.util.ArrayList;
import java.util.Calendar;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.TimeUnit;
import org.apache.commons.math3.random.RandomData;
import org.apache.commons.math3.random.RandomDataImpl;
import org.jbpm.simulation.SimulationContext;
import org.jbpm.simulation.SimulationDataProvider;
import org.jbpm.simulation.TimeGenerator;
import org.jbpm.simulation.TimeGeneratorFactory;
import org.jbpm.simulation.util.SimulationConstants;
import org.jbpm.simulation.util.SimulationUtils;
import org.kie.api.definition.process.Node;
public class StaffPoolImpl implements StaffPool {
private Map<String, Object> properties;
private int poolSize;
private long duration;
private List<Long> allocatedTill = new ArrayList<Long>();
private TimeUnit timeUnit = TimeUnit.MILLISECONDS;
//default working hours set to eight hours
private long workingHours = ( 8 * 60 * 60 * 1000);
private long poolCapacity = 0;
private long performedWork = 0;
private RandomData randomizer = new RandomDataImpl();
private double resourceCost;
private TimeUnit elementTimeUnit;
private TimeGenerator timeGenerator;
private Map<String, List<Long>> allocatedRanges = new HashMap<String, List<Long>>();
private RangeChain rangeChain = new RangeChain();
public StaffPoolImpl(String processId, Node element, double simulationDuration) {
SimulationDataProvider provider = SimulationContext.getContext().getDataProvider();
properties = provider.getSimulationDataForNode(element);
timeGenerator=TimeGeneratorFactory.newTimeGenerator(properties);
this.elementTimeUnit = SimulationUtils.getTimeUnit(properties);
this.poolSize = (int)SimulationUtils.asDouble(properties.get(SimulationConstants.STAFF_AVAILABILITY));
this.duration = timeGenerator.generateTime();
String workingHoursRange = (String) properties.get("working.hours.range");
if (workingHoursRange != null) {
String[] ranges = workingHoursRange.split(",");
for (String range : ranges) {
String[] rangeElems = range.split("-");
rangeChain.addRange(new Range(Integer.parseInt(rangeElems[0]), Integer.parseInt(rangeElems[1]), poolSize));
}
} else {
long workingHoursOpt = (long)SimulationUtils.asDouble(properties.get(SimulationConstants.WORKING_HOURS));
if (workingHoursOpt > 0) {
this.workingHours = timeUnit.convert(workingHoursOpt, TimeUnit.HOURS);
}
rangeChain.addRange(new Range(0, 24, poolSize));
}
this.poolCapacity = poolSize * this.workingHours;
// if simulation is estimated to more than one day multiply working hours by that factor
if (simulationDuration > 1) {
this.poolCapacity = (long) (this.poolCapacity * simulationDuration);
}
this.resourceCost = SimulationUtils.asDouble(properties.get(SimulationConstants.COST_PER_TIME_UNIT));
}
protected long allocate(long startTime, long duration) {
// long waitTime = 0;
performedWork += duration;
//
// List<Long> allocatedTill = findAllocatedRange(startTime);
// if(allocatedTill.size() < poolSize) {
//
// allocatedTill.add(startTime + duration);
//
// return waitTime;
// } else {
// Collections.sort(allocatedTill);
//
// long allocated = allocatedTill.get(0);
// if (allocated >= startTime) {
// waitTime = allocated - startTime;
// allocated += duration;
//
// } else {
// allocated = startTime + duration;
// }
// allocatedTill.set(0, allocated);
//
// return waitTime;
// }
return rangeChain.allocateWork(startTime, duration);
}
public long allocate(long startTime) {
return allocate(startTime, this.duration);
}
public long allocate(long startTime, Node element) {
long duration = this.duration = timeGenerator.generateTime();
return allocate(startTime, duration);
}
public double getResourceUtilization() {
if (poolCapacity == 0) {
return 0;
}
return performedWork * 100 / poolCapacity;
}
/* (non-Javadoc)
* @see org.onebpm.simulation.engine.api.StaffPool#getResourceCost()
*/
public double getResourceCost() {
return this.resourceCost;
}
public TimeUnit getElementTimeUnit() {
return elementTimeUnit;
}
protected List<Long> findAllocatedRange(long stringTime) {
Calendar c = Calendar.getInstance();
c.setTimeInMillis(stringTime);
int hour = c.get(Calendar.HOUR_OF_DAY);
Set<String> ranges = this.allocatedRanges.keySet();
for (String range : ranges) {
String[] elems = range.split("-");
int lower = Integer.parseInt(elems[0]);
int upper = Integer.parseInt(elems[1]);
if (hour >= lower && hour <= upper) {
return this.allocatedRanges.get(range);
}
}
return null;
}
}