Author: [log in to unmask]
Date: Mon Oct 31 15:46:29 2016
New Revision: 4551

Log:
Analysis driver used to understand the hit performance of the SVT.

Added:
    java/trunk/users/src/main/java/org/hps/users/omoreno/SvtHitPerformance.java   (with props)

Added: java/trunk/users/src/main/java/org/hps/users/omoreno/SvtHitPerformance.java
 =============================================================================
--- java/trunk/users/src/main/java/org/hps/users/omoreno/SvtHitPerformance.java	(added)
+++ java/trunk/users/src/main/java/org/hps/users/omoreno/SvtHitPerformance.java	Mon Oct 31 15:46:29 2016
@@ -0,0 +1,170 @@
+package org.hps.users.omoreno;
+
+import java.io.IOException;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+
+import hep.aida.IHistogramFactory;
+import hep.aida.IAnalysisFactory;
+import hep.aida.IHistogram1D;
+import hep.aida.ITree;
+import hep.aida.ref.rootwriter.RootFileStore;
+
+import org.lcsim.detector.tracker.silicon.HpsSiSensor;
+import org.lcsim.event.EventHeader;
+import org.lcsim.event.RawTrackerHit;
+import org.lcsim.geometry.Detector;
+import org.lcsim.recon.tracking.digitization.sisim.SiTrackerHitStrip1D;
+
+/**
+ *
+ */
+public class SvtHitPerformance extends TriggerFilter {
+
+    //---------------//
+    //   Constants   //
+    //---------------//
+    private static final String SUBDETECTOR_NAME = "Tracker";
+
+    //--------------//
+    //--------------//
+
+    // Container for HPS sensor objects
+    private List<HpsSiSensor> sensors = null;
+    private Map<HpsSiSensor, MutableInt> hitMultiplicityMap     = new HashMap<HpsSiSensor, MutableInt>();
+    private Map<HpsSiSensor, MutableInt> clusterMultiplicityMap = new HashMap<HpsSiSensor, MutableInt>();
+
+    //-----------------//
+    //   Collections   //
+    //-----------------//
+    private static final String RAW_HIT_COLLECTION = "SVTRawTrackerHits";
+    private static final String CLUSTER_COLLECTION = "StripClusterer_SiTrackerHitStrip1D";
+
+    //--------------//
+    //--------------//
+
+    //--------------//
+    //   Plotting   //
+    //--------------//
+    private ITree tree = null; 
+    private IHistogramFactory histogramFactory = null; 
+
+    // Raw hits per sensor 
+    private Map<HpsSiSensor, IHistogram1D> rawHitPlots = new HashMap<HpsSiSensor, IHistogram1D>();
+    
+    // Clusters per sensor
+    private Map<HpsSiSensor, IHistogram1D> clusterMultPlots = new HashMap<HpsSiSensor, IHistogram1D>();
+
+    //--------------//
+    //--------------//
+
+    public void detectorChanged(Detector detector) { 
+
+        // Instantiate the tree and histogram factory
+        tree = IAnalysisFactory.create().createTreeFactory().create();
+        histogramFactory = IAnalysisFactory.create().createHistogramFactory(tree);
+
+        // Get the HpsSiSensor objects from the tracker detector element
+        sensors = detector.getSubdetector(SUBDETECTOR_NAME).getDetectorElement().findDescendants(HpsSiSensor.class);
+
+        // Associate plots with each of the sensors.
+        for (HpsSiSensor sensor : sensors) {
+
+            String sensorName = sensor.getName();
+
+            // Book a histogram of the raw hit multiplicity for each of the sensors. 
+            this.rawHitPlots.put(sensor, 
+                    histogramFactory.createHistogram1D(sensorName + " - Raw Hit Multiplicity", 101, 0, 100));
+            this.clusterMultPlots.put(sensor, 
+                    histogramFactory.createHistogram1D(sensorName + " - Cluster Multiplicity", 101, 0, 100));
+            
+            this.hitMultiplicityMap.put(sensor, new MutableInt());
+            this.clusterMultiplicityMap.put(sensor, new MutableInt());
+        }
+    }
+
+    public void process(EventHeader event) { 
+
+        super.process(event);
+        
+        if (!triggerFound) return; 
+        
+        // If the event doesn't have a collection of raw hits, skip it. This 
+        // should never happen.  If an event doesn't have any raw hits 
+        // from the SVT, an empty collection is still placed into the event.
+        if (!event.hasCollection(RawTrackerHit.class, RAW_HIT_COLLECTION)) return;
+
+        // Get the list of raw hits from the event. 
+        List<RawTrackerHit> rawHits = event.get(RawTrackerHit.class, RAW_HIT_COLLECTION);
+
+        // Reset the map keeping count of the hits per sensor.
+        this.resetMultiplicityMaps();
+
+        // Loop through all of the hits in the event and calculate various quantities.
+        for (RawTrackerHit rawHit : rawHits) { 
+
+            // Get the sensor associated with this hit.
+            HpsSiSensor sensor = (HpsSiSensor) rawHit.getDetectorElement();
+
+            // Increament the hit count on that sensor
+            this.hitMultiplicityMap.get(sensor).increment();
+        }
+
+        // Fill the hit multiplicity plots.
+        for (Map.Entry<HpsSiSensor, MutableInt> entry : this.hitMultiplicityMap.entrySet()) { 
+            rawHitPlots.get(entry.getKey()).fill(entry.getValue().get());
+        }
+        
+        // Check if the event has a collection of clusters.  If not, stop processing
+        // the event.
+        if (!event.hasCollection(SiTrackerHitStrip1D.class, CLUSTER_COLLECTION)) return;
+
+        // Get the collection of clusters from the event.
+        List<SiTrackerHitStrip1D> clusters = event.get(SiTrackerHitStrip1D.class, CLUSTER_COLLECTION);
+        
+        // Loop through all of the clusters in the event and calculate various quantities.
+        for (SiTrackerHitStrip1D cluster : clusters) { 
+
+            // Get the sensor associated with this cluster.
+            HpsSiSensor sensor = (HpsSiSensor) cluster.getRawHits().get(0).getDetectorElement();
+            
+            // Increment the cluster count on that sensor
+            this.clusterMultiplicityMap.get(sensor).increment();
+        }
+        
+        // Fill the cluster multiplicity plots.
+        for (Map.Entry<HpsSiSensor, MutableInt> entry : this.clusterMultiplicityMap.entrySet()) { 
+            clusterMultPlots.get(entry.getKey()).fill(entry.getValue().get());
+        }
+    }
+
+    @Override
+    public void endOfData(){
+        String rootFile = "hit_performance.root";
+        RootFileStore store = new RootFileStore(rootFile);
+        try {
+            store.open();
+            store.add(tree);
+            store.close(); 
+        } catch (IOException e) {
+            e.printStackTrace();
+        }
+    }
+
+    private void resetMultiplicityMaps() { 
+        for (Map.Entry<HpsSiSensor, MutableInt> entry : this.hitMultiplicityMap.entrySet()) { 
+            entry.getValue().reset();;
+        }
+        for (Map.Entry<HpsSiSensor, MutableInt> entry : this.clusterMultiplicityMap.entrySet()) { 
+            entry.getValue().reset();;
+        }
+    }
+
+    class MutableInt { 
+        int value = 0; 
+        public void increment() { ++value; }
+        public int get() { return value; }
+        public void reset() { value = 0; }
+    }
+}