source: DCWoRMS/trunk/src/simulator/utils/WorkloadCalculator.java @ 477

package simulator.utils;

import java.io.File;
import java.util.Calendar;
import java.util.HashMap;
import java.util.Iterator;
import java.util.Set;

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

import simulator.workload.reader.archive.swf.SWFFields;
import simulator.workload.reader.archive.swf.SWFParser;
import simulator.workload.writer.swf.QcgSWFJobWriter;

/**
 * 
 * @author Marcin Krystek
 *
 */
public class WorkloadCalculator {

        protected static Log log = LogFactory.getLog(WorkloadCalculator.class);
        
        public static void main(String args[]){
        
                WorkloadCalculator workloadCalculator = new WorkloadCalculator();
                try {
                        throw new RuntimeException("Nieladnie tak uruchamiac w ciemno ;). " +
                                        "Wybierz funkcje, ktora ma zostac wywolana");
                        
                        //workloadCalculator.normalizeSubmitTime(args[0]);
                        //workloadCalculator.removeIncorrectTasks(args[0]);
                        
                } catch (Exception e) {
                        e.printStackTrace();
                }
        }
        
        /**
         * Rewrites swf file. Submit time of all tasks will be scaled down.
         * @param swfFileName
         */
        public void normalizeSubmitTime(String swfFileName) throws Exception{
                log.info("Start workload submit time normalization for "+swfFileName);
                
                File file = new File(swfFileName);
                String dirName = file.getParent();
                String fileName = file.getName();
                
                File newFile = new File(dirName, Calendar.getInstance().getTimeInMillis()+"_"+fileName);
                file.renameTo(newFile);
                
                
                QcgSWFJobWriter writer = new QcgSWFJobWriter(dirName, fileName);
                // write swf filed description
                writer.writeFieldDescriptionHeader();
                
                SWFParser parser = new SWFParser(newFile.getAbsolutePath());
                
                // rewrite swf header
                HashMap<String, String> header = parser.loadHeader();
                Set<String> keys = header.keySet();
                Iterator<String> itr = keys.iterator();
                while(itr.hasNext()){
                        String key = itr.next();
                        writer.writeComment(key, header.get(key));
                }
                
                // rewrite tasks
                String task[] = parser.readTask();
                long submitTimeOffset = Long.valueOf(task[SWFFields.DATA_SUBMIT_TIME]);
                task[SWFFields.DATA_SUBMIT_TIME] = "0";
                writer.writeLine(task);
                
                while((task = parser.readTask()) != null){
                        long submitTime = Long.valueOf(task[SWFFields.DATA_SUBMIT_TIME]) - submitTimeOffset;
                        task[SWFFields.DATA_SUBMIT_TIME] = String.valueOf(submitTime);
                        writer.writeLine(task);
                }
                
                parser.close();
                writer.close();
                
                newFile.delete();
                
                log.info("End workload submit time normalization");
        }
        
        /**
         * Rewrites swf file and deletes all tasks which parameter RUN_TIME or REQUESTED_TIME or 
         * number of ALLOCATED PRCESSORS or number of REQUESTED PROCESSORS equals -1.
         * @param swf file name 
         */
        public void removeIncorrectTasks(String swfFileName) throws Exception{
                log.info("Start removing incorrect tasks for "+swfFileName);
                
                File file = new File(swfFileName);
                String dirName = file.getParent();
                String fileName = file.getName();
                
                File newFile = new File(dirName, Calendar.getInstance().getTimeInMillis()+"_"+fileName);
                file.renameTo(newFile);
                
                
                QcgSWFJobWriter writer = new QcgSWFJobWriter(dirName, fileName);
                // write swf filed description
                writer.writeFieldDescriptionHeader();
                
                SWFParser parser = new SWFParser(newFile.getAbsolutePath());
                
                // rewrite swf header
                HashMap<String, String> header = parser.loadHeader();
                Set<String> keys = header.keySet();
                Iterator<String> itr = keys.iterator();
                while(itr.hasNext()){
                        String key = itr.next();
                        writer.writeComment(key, header.get(key));
                }
                
                // rewrite tasks
                String task[] = null;
                
                while((task = parser.readTask()) != null){
                        if("-1".equals(task[SWFFields.DATA_RUN_TIME])) {
                                log.info("Removing task "+task[SWFFields.DATA_JOB_NUMBER] + ",  RUN_TIME = -1");
                                continue;       
                        }
                                
                        if("-1".equals(task[SWFFields.DATA_REQUESTED_TIME])) {
                                log.info("Removing task "+task[SWFFields.DATA_JOB_NUMBER] + ",  REQUESTED_TIME = -1");
                                continue;
                        }
                        if("-1".equals(task[SWFFields.DATA_NUMBER_OF_ALLOCATED_PROCESSORS])){
                                log.info("Removing task "+task[SWFFields.DATA_JOB_NUMBER] + ",  NUMBER_OF_ALLOCATED_PROCESSORS = -1");
                                continue;
                        }
                        if("-1".equals(task[SWFFields.DATA_REQUESTED_NUMBER_OF_PROCESSORS])){
                                log.info("Removing task "+task[SWFFields.DATA_JOB_NUMBER] + ",  REQUESTED_NUMBER_OF_PROCESSORS = -1");
                                continue;
                        }

                        writer.writeLine(task);
                }
                
                parser.close();
                writer.close();
                
                newFile.delete();
                
                log.info("End removing incorrect tasks");
        }
        
}