source: DCWoRMS/branches/coolemall/src/test/drs_tst/recs/utils/TaskToApp.java @ 1120

package test.drs_tst.recs.utils;

import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.util.HashMap;
import java.util.Map;
import java.util.PropertyResourceBundle;
import java.util.Random;
import java.util.ResourceBundle;

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 HashMap<String, AppType>();
        
        private static String JOB_APPS_MAPPINGS_FILE_NAME = "src/test/drs_tst/recs/data/job_apps_mappings.properties";
        private ResourceBundle mappingsBundle;
        
        public TaskToApp(){
                loadMappings();
        }

        public AppType getAppType(TaskInterface<?> task){
                AppType appType = null;
                if(mapping.get(task.getJobId())!= null){
                        appType = mapping.get(task.getJobId());
                }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;
        }
        
        private void loadMappings(){
                try {
                        ResourceBundle rb = getMappingsBundle();
                        for(String key : rb.keySet()){
                                mapping.put(key, AppType.valueOf(rb.getString(key)));
                        }
                } catch (IOException e) {
                        e.printStackTrace();
                }
                
        }
        
        private ResourceBundle getMappingsBundle() throws FileNotFoundException, IOException{
                if(mappingsBundle == null){
                         mappingsBundle = new PropertyResourceBundle(new FileInputStream(JOB_APPS_MAPPINGS_FILE_NAME));
                }
                return mappingsBundle;
        }
}