source: DCWoRMS/branches/coolemall/src/test/article/recs/utils/TaskToApp.java @ 1197

package test.article.recs.utils;

import java.util.HashMap;
import java.util.Map;
import java.util.Random;
import java.util.TreeMap;

import dcworms.schedframe.scheduling.Executable;

import schedframe.scheduling.tasks.TaskInterface;

public class TaskToApp {
        
        private static int ABINIT = 20;
        private static int C_RAY = 40;
        private static int TAR = 60;
        private static int LIN_3GB = 70;
        private static int LIN_TINY = 80;
        private static int FFT = 100;
        private static Random rand = new Random(5);
        private static Map<String, AppType> mapping = new TreeMap<String, AppType>();

        private static Random randLoad = new Random(5);
        private static Map<String, Double> loadMapping = new HashMap<String, Double>();
        private static double [] loadLevels = {0.25, 0.5, 0.75, 1};
        
        public AppType getAppType(TaskInterface<?> task){

                AppType appType = null;
                if(mapping.get(task.getJobId())!= null){
                        appType = mapping.get(task.getJobId());
                        Executable exec = (Executable) task;
                        exec.getDescription().getExecution().getExecutable().getApplication().setName(appType.toString());
                }else{
                        appType = randomAppType();
                        mapping.put(task.getJobId(), appType);
                }
                return appType;
        }
        
        private AppType randomAppType(){
                AppType appType = null;
                int n = rand.nextInt(100);
                if(n < ABINIT){
                        appType = AppType.abinit;
                }else if(n < C_RAY){
                        appType = AppType.c_ray;
                }else if(n < TAR){
                        appType = AppType.tar;
                }else if(n < LIN_3GB){
                        appType = AppType.lin_3gb;
                }else if(n < LIN_TINY){
                        appType = AppType.lin_tiny;
                }else if(n < FFT){
                        appType = AppType.fft;
                } 
                return appType;
        }
        
        public double getAppLoad(TaskInterface<?> task){
                double appLoad = 0;
                if(loadMapping.get(task.getJobId())!= null){
                        appLoad = loadMapping.get(task.getJobId());
                }else{
                        appLoad = randomAppLoad();
                        loadMapping.put(task.getJobId(), appLoad);
                }return appLoad;
        }
        
        private double randomAppLoad(){
                double appLoad = 0;
                int n = randLoad.nextInt(7);
                appLoad = loadLevels[n];
                return appLoad;
        }
        
}