Print

Print


Author: [log in to unmask]
Date: Mon Jul 13 09:25:55 2015
New Revision: 3257

Log:
Added ability to select on number of hits in event

Modified:
    java/trunk/analysis/src/main/java/org/hps/analysis/examples/StripEventDriver.java

Modified: java/trunk/analysis/src/main/java/org/hps/analysis/examples/StripEventDriver.java
 =============================================================================
--- java/trunk/analysis/src/main/java/org/hps/analysis/examples/StripEventDriver.java	(original)
+++ java/trunk/analysis/src/main/java/org/hps/analysis/examples/StripEventDriver.java	Mon Jul 13 09:25:55 2015
@@ -3,6 +3,7 @@
 import java.util.List;
 import org.lcsim.event.EventHeader;
 import org.lcsim.event.Track;
+import org.lcsim.event.TrackerHit;
 import org.lcsim.event.Vertex;
 import org.lcsim.util.Driver;
 
@@ -19,6 +20,8 @@
     private int _minNumberOfHitsOnTrack = 0;
     private int _numberOfEventsWritten = 0;
     private int _minNumberOfUnconstrainedV0Vertices = 0;
+    private int _minNumberOfStripHits = 0;
+    private int _maxNumberOfStripHits = Integer.MAX_VALUE;
 
     @Override
     protected void process(EventHeader event)
@@ -43,6 +46,15 @@
         if (event.hasCollection(Vertex.class, "UnconstrainedV0Vertices")) {
             int nVertices = event.get(Vertex.class, "UnconstrainedV0Vertices").size();
             if (nVertices < _minNumberOfUnconstrainedV0Vertices) {
+                skipEvent = true;
+            }
+        }
+        if(event.hasCollection(TrackerHit.class, "StripClusterer_SiTrackerHitStrip1D")) {
+            int nHits = event.get(TrackerHit.class, "StripClusterer_SiTrackerHitStrip1D").size();
+            if (nHits < _minNumberOfStripHits) {
+                skipEvent = true;
+            }
+            if (nHits > _maxNumberOfStripHits) {
                 skipEvent = true;
             }
         }
@@ -73,5 +85,15 @@
     {
         _minNumberOfUnconstrainedV0Vertices = nVertices;
     }
+    
+    public void setMinNumberOfStripHits(int n)
+    {
+        _minNumberOfStripHits = n;
+    }
+    
+    public void setMaxNumberOfStripHits(int n)
+    {
+        _maxNumberOfStripHits = n;
+    }
 
 }