Author: [log in to unmask]
Date: Wed May 27 19:05:29 2015
New Revision: 3048

Log:
Add command line tool for importing SVT timing constants from run spreadsheet CSV file.

Added:
    java/trunk/conditions/src/main/java/org/hps/conditions/svt/SvtTimingConstantsLoader.java

Added: java/trunk/conditions/src/main/java/org/hps/conditions/svt/SvtTimingConstantsLoader.java
 =============================================================================
--- java/trunk/conditions/src/main/java/org/hps/conditions/svt/SvtTimingConstantsLoader.java	(added)
+++ java/trunk/conditions/src/main/java/org/hps/conditions/svt/SvtTimingConstantsLoader.java	Wed May 27 19:05:29 2015
@@ -0,0 +1,168 @@
+package org.    hps.conditions.svt;
+
+import java.io.File;
+import java.util.ArrayList;
+import java.util.Collection;
+import java.util.Date;
+import java.util.HashSet;
+import java.util.Iterator;
+import java.util.List;
+import java.util.Set;
+
+import org.hps.conditions.api.ConditionsRecord;
+import org.hps.conditions.database.DatabaseConditionsManager;
+import org.hps.conditions.run.RunRange;
+import org.hps.conditions.run.RunSpreadsheet;
+import org.hps.conditions.svt.SvtTimingConstants.SvtTimingConstantsCollection;
+
+/**
+ * Load SVT timing constant data from the run spreadsheet and insert into the conditions database.
+ * <p>
+ * Be very careful about running this, because it will create many new conditions records that may 
+ * already be present in the database.  In fact, don't run this at all without talking to me first. :-) 
+ * 
+ * @author Jeremy McCormick
+ */
+public final class SvtTimingConstantsLoader {
+    
+    /**
+     * Setup conditions.     
+     */
+    private static final DatabaseConditionsManager MANAGER = DatabaseConditionsManager.getInstance();
+    
+    /**
+     * The fields from the run spreadsheet for SVT timing constants.
+     */
+    private static final Set<String> FIELDS = new HashSet<String>();
+    static {
+        FIELDS.add("svt_offset_phase");
+        FIELDS.add("svt_offset_time");
+    }
+    
+    /**
+     * Load the SVT timing constants for time and phase offsets into the conditions database.
+     * 
+     * @param args the command line arguments (requires one argument which is CSV file name)
+     */
+    public static void main(String[] args) {
+        
+        // Load in CSV records from the exported run spreadsheet.
+        String path = args[0];
+        RunSpreadsheet runSheet = new RunSpreadsheet(new File(path));
+        
+        // Find the run ranges that have the same fields values.
+        List<RunRange> ranges = RunRange.findRunRanges(runSheet, FIELDS);
+        
+        // Get the unique field values for inserting new conditions collections.
+        List<Collection<String>> uniqueValues = RunRange.getUniqueValues(ranges);
+        
+        /*
+        System.out.println("unique values ...");
+        for (Collection<String> collection : uniqueValues) {
+            for (String value : collection) {
+                System.out.print(value + " ");                
+            }
+            System.out.println();
+        } 
+        */      
+               
+        // Create a new collection for each unique combination set of timing constants.
+        List<SvtTimingConstantsCollection> collections = createCollections(uniqueValues); 
+        
+        // Create a new collection for each of the unique combinations of values.
+        for (SvtTimingConstantsCollection collection : collections) {
+            int collectionId = 0;
+            try {
+                collectionId = MANAGER.addCollection(
+                        "svt_timing_constants",
+                        "SVT timing constants added by " + System.getProperty("user.name"),
+                        "timing constants from run spreadsheet");
+                collection.setCollectionId(collectionId);
+                collection.insert();
+            } catch (Exception e) {
+                throw new RuntimeException(e);
+            }            
+        } 
+                
+        // Create the conditions records for the run ranges.
+        for (RunRange range : ranges) {
+            
+            System.out.println(range);
+            
+            // Get the data values from the run range.
+            int offsetPhase = Integer.parseInt(range.getValue("svt_offset_phase"));
+            int offsetTime = Integer.parseInt(range.getValue("svt_offset_time"));
+
+            // Find the matching timing constants collection to use.
+            SvtTimingConstantsCollection collection = findCollection(collections, offsetPhase, offsetTime);
+            if (collection != null) {
+                System.out.println("offset_phase : " + collection.get(0).getOffsetPhase() + ", offset_time: " + collection.get(0).getOffsetTime());
+            }
+                        
+            // Create a new conditions record with the run range.
+            ConditionsRecord condi = new ConditionsRecord();
+            condi.setFieldValue("run_start", range.getRunStart());
+            condi.setFieldValue("run_end", range.getRunEnd());
+            condi.setFieldValue("name", "svt_timing_constants");
+            condi.setFieldValue("table_name", "svt_timing_constants");
+            condi.setFieldValue("notes", "timing constants from run spreadsheet");
+            condi.setFieldValue("created", new Date());
+            condi.setFieldValue("created_by", System.getProperty("user.name"));
+            
+            condi.setFieldValue("collection_id", collection.getCollectionId());
+
+            try {
+                condi.insert();
+            } catch (Exception e) {
+                throw new RuntimeException(e);
+            }            
+            
+            System.out.print(condi);            
+            System.out.println();
+        }
+    }
+    
+    /**
+     * Create the conditions collections from the unique values found in the spreadsheet.
+     * 
+     * @param uniqueValues the list of unique raw values
+     * @return the new conditions collections
+     */
+    private static List<SvtTimingConstantsCollection> createCollections(List<Collection<String>> uniqueValues) {
+        List<SvtTimingConstantsCollection> collections = new ArrayList<SvtTimingConstantsCollection>();        
+        for (Collection<String> values : uniqueValues) {
+            
+            SvtTimingConstants timing = new SvtTimingConstants();
+            Iterator<String> it = values.iterator();            
+            Integer offsetPhase = Integer.parseInt(it.next());
+            Integer offsetTime = Integer.parseInt(it.next());
+            
+            timing.setFieldValue("offset_phase", offsetPhase);
+            timing.setFieldValue("offset_time", offsetTime);
+            
+            SvtTimingConstantsCollection collection = new SvtTimingConstantsCollection();
+            collection.add(timing);
+            collections.add(collection);
+        }
+        return collections;
+    }
+     
+    /**
+     * Find a timing constants collection from offset phase and time. 
+     * <p>
+     * Each collection has a single object in it.
+     * 
+     * @param timingConstantsList the list of collections
+     * @param offsetPhase the offset phase
+     * @param offsetTime the offset time
+     * @return the matching collection or <code>null</code> if not found
+     */
+    private static SvtTimingConstantsCollection findCollection(List<SvtTimingConstantsCollection> timingConstantsList, int offsetPhase, int offsetTime) {
+        for (SvtTimingConstantsCollection collection : timingConstantsList) {
+            if (collection.find(offsetPhase, offsetTime) != null) {
+                return collection;
+            }
+        }
+        return null;
+    }
+}