source: DCWoRMS/branches/coolemall/src/test/PhaseTimeEstimationPlugin.java @ 883

package test;

import java.util.Map;

import schedframe.events.scheduling.SchedulingEvent;
import schedframe.resources.units.PEUnit;
import schedframe.resources.units.ResourceUnit;
import schedframe.resources.units.ResourceUnitName;
import schedframe.resources.units.StandardResourceUnitName;
import dcworms.schedframe.scheduling.ExecTask;
import dcworms.schedframe.scheduling.Executable;
import example.timeestimation.BaseTimeEstimationPlugin;

public class PhaseTimeEstimationPlugin extends BaseTimeEstimationPlugin{

        /*
         * This method should return an estimation of time required to execute the task.
         * Requested calculation should be done based on the resources allocated for the task,
         * task description and task completion percentage.
         * 
         * Example implementation calculate the estimation based on cpu processing power.
         * There is also a simple assumption, that cpu processing power is a linear function
         * of number of allocated cpus and their speed.
         */
        public double execTimeEstimation(SchedulingEvent event, ExecTask task,
                        Map<ResourceUnitName, ResourceUnit> allocatedResources,
                        double completionPercentage) {

                
                Executable exec = (Executable) task;
        //      System.out.println(exec.getJobId() +":"+ exec.getApplicationName());
                // collect all information necessary to do the calculation
                PEUnit peUnit = (PEUnit) allocatedResources.get(StandardResourceUnitName.PE);
                
                // obtain single pe speed
                int speed = peUnit.getSpeed();

                // number of used pe  
                int cnt = peUnit.getUsedAmount();

                // estimate remainingTaskLength
                double remainingLength =  task.getLength() * (1 - completionPercentage/100);
                
                // do the calculation
                double execTime = (remainingLength / (cnt * speed));

                // if the result is very close to 0, but less then one millisecond then round this result to 0.001
                if (Double.compare(execTime, 0.001) < 0) {
                        execTime = 0.001;
                }

                // time is measured in integer units, so get the nearest execTime int value.
                execTime = Math.round(execTime);
                return execTime;
        }

}