source: DCWoRMS/branches/coolemall/src/test/testSOP/BaseTimeEstimationPlugin.java @ 1606

package test.testSOP;

import java.util.Map;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;

import dcworms.schedframe.scheduling.ExecTask;
import schedframe.Parameters;
import schedframe.PluginConfiguration;
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 schedframe.scheduling.plugin.estimation.ExecutionTimeEstimationPlugin;


public class BaseTimeEstimationPlugin implements ExecutionTimeEstimationPlugin {
        Log log = LogFactory.getLog(BaseTimeEstimationPlugin.class); 
        public PluginConfiguration getConfiguration() {
                return null;
        }

        public String getName() {
                return getClass().getName();
        }

        public void init(Parameters parameters) {
        }

        public double estimateMigrationTime(SchedulingEvent event, ExecTask task,
                        Map<ResourceUnitName, ResourceUnit> srcResources, Map<ResourceUnitName, ResourceUnit> dstResources) {
                double time = 0;
                //Test migration 10s
//              log.debug("Migration takes "+time+" seconds.");
                return time;
        }

        /*
         * 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) {
                
                // 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;
        }

        
}