Changeset 1710 for branches/autoquest-core-tasktrees-alignment/src/main

Timestamp:

09/01/14 22:05:22 (10 years ago)

Author:

rkrimmel

Message:

Better saving of intermediate results

File:

• branches/autoquest-core-tasktrees-alignment/src/main/java/de/ugoe/cs/autoquest/tasktrees/temporalrelation/SequenceForTaskDetectionRuleAlignment.java (modified) (5 diffs)

branches/autoquest-core-tasktrees-alignment/src/main/java/de/ugoe/cs/autoquest/tasktrees/temporalrelation/SequenceForTaskDetectionRuleAlignment.java

@@ -79 +79 @@
         public static int nThreads;
         public static boolean harmonizeSequences;
+        public static boolean loadSubstutionMatrix;
         
 
@@ -127 +128 @@
                 this.preparationTaskHandlingStrategy = new TaskHandlingStrategy(
                                 minimalTaskEquality);
-
         }
 
@@ -140 +140 @@
         }
 
+        public void saveAppData(String name) {
+                String objectName = name;
+                String filename = name + ".dat";
+                Object dataObject = GlobalDataContainer.getInstance().getData(
+                                objectName);
+                if (dataObject == null) {
+                        CommandHelpers.objectNotFoundMessage(objectName);
+                }
+                FileOutputStream fos = null;
+                ObjectOutputStream out = null;
+                try {
+                        fos = new FileOutputStream(filename);
+                        out = new ObjectOutputStream(fos);
+                        out.writeObject(dataObject);
+                        out.close();
+                } catch (IOException ex) {
+                        Console.logException(ex);
+                }
+        }
+        
+        public RuleApplicationData loadAppData(String name) {
+                String objectName = name;
+                String filename = name + ".dat";
+        
+                Object data = null;
+                FileInputStream fis = null;
+                ObjectInputStream in = null;
+                try {
+                        fis = new FileInputStream(filename);
+                        in = new ObjectInputStream(fis);
+                        data = in.readObject();
+                        in.close();
+                } catch (IOException ex) {
+                        Console.logException(ex);
+                } catch (ClassNotFoundException ex) {
+                        Console.logException(ex);
+                }
+                if (GlobalDataContainer.getInstance().addData(objectName, data)) {
+                        CommandHelpers.dataOverwritten(objectName);
+                }
+                return (RuleApplicationData) GlobalDataContainer.getInstance().getData(name);
+        }
+        
         /*
          * (non-Javadoc)
@@ -151 +194 @@
                 RuleApplicationData appData = new RuleApplicationData(sessions);
                 if(SequenceForTaskDetectionRuleAlignment.harmonizeSequences) {
+                        //appData.getStopWatch().start("harmonization");
                         harmonizeEventTaskInstancesModel(appData);
+                        //appData.getStopWatch().stop("harmonization");
+                        GlobalDataContainer.getInstance().addData("harmonized", appData);
+                        //Saving intermediate results to file
+                        Console.traceln(Level.INFO,"saving substitution matrix to file");
+                        saveAppData("harmonized");
+                }
+                else {
+                        Console.traceln(Level.INFO,"loading substitution matrix from file");
+                        appData = loadAppData("harmonized");
+                }
+        
+                if(!SequenceForTaskDetectionRuleAlignment.loadSubstutionMatrix) {       
                         Console.traceln(Level.INFO, "generating substitution matrix from " + appData.getUniqueTasks().size() + " unique tasks");
+                        appData.getStopWatch().start("substitution matrix");
                         appData.getSubmat().generate(appData.getUniqueTasks());
                         appData.getStopWatch().stop("substitution matrix");
-                        GlobalDataContainer.getInstance().addData("appData", appData);
-                        //Saving intermediate results to file
-                        Console.traceln(Level.INFO,"saving substitution matrix to file");
-                        String objectName = "appData";
-                        String filename = "appData.dat";
-                        Object dataObject = GlobalDataContainer.getInstance().getData(
-                                        objectName);
-                        if (dataObject == null) {
-                                CommandHelpers.objectNotFoundMessage(objectName);
-                        }
-                        FileOutputStream fos = null;
-                        ObjectOutputStream out = null;
-                        try {
-                                fos = new FileOutputStream(filename);
-                                out = new ObjectOutputStream(fos);
-                                out.writeObject(dataObject);
-                                out.close();
-                        } catch (IOException ex) {
-                                Console.logException(ex);
-                        }
+                        GlobalDataContainer.getInstance().addData("substitution", appData);
+                        saveAppData("substitution");
                 }
                 else {
-                        Console.traceln(Level.INFO,"loading substitution matrix from file");
-                        String objectName = "appData";
-                        String filename = "appData.dat";
-                        Object data = null;
-                        FileInputStream fis = null;
-                        ObjectInputStream in = null;
-                        try {
-                                fis = new FileInputStream(filename);
-                                in = new ObjectInputStream(fis);
-                                data = in.readObject();
-                                in.close();
-                        } catch (IOException ex) {
-                                Console.logException(ex);
-                        } catch (ClassNotFoundException ex) {
-                                Console.logException(ex);
-                        }
-                        if (GlobalDataContainer.getInstance().addData(objectName, data)) {
-                                CommandHelpers.dataOverwritten(objectName);
-                        }
-                        appData = (RuleApplicationData) GlobalDataContainer.getInstance().getData("appData");
+                        appData = loadAppData("substitution");
                 }
                 
@@ -204 +224 @@
                         Console.traceln(Level.INFO, "Iteration Number: " + iteration);
                         iteration++;
-                        
+        
                         appData.detectedAndReplacedTasks = false;
                         appData.getStopWatch().start("whole loop");