--- java/trunk/recon/src/main/java/org/hps/recon/filtering/SimpleEventFilterDriver.java	                        (rev 0)
+++ java/trunk/recon/src/main/java/org/hps/recon/filtering/SimpleEventFilterDriver.java	2014-09-26 01:57:23 UTC (rev 1099)
@@ -0,0 +1,126 @@

+package org.hps.recon.filtering;
+
+import java.util.ArrayList;
+import java.util.List;
+
+import org.lcsim.event.Cluster;
+import org.lcsim.event.EventHeader;
+import org.lcsim.event.ReconstructedParticle;
+import org.lcsim.event.Track;
+import org.lcsim.util.Driver;
+
+/**
+ * This is a Driver for rejecting recon events using simple criteria on the output collections.
+ * 
+ * @author Jeremy McCormick <[log in to unmask]>
+ */
+public class SimpleEventFilterDriver extends Driver {
+        
+    int minTracks = 0;
+    int minReconParticles = 0;
+    int minClusters = 0;
+    
+    int nRejected = 0;
+    int nAccepted = 0;
+    
+    static final String trackCollectionName = "MatchedTracks";
+    static final String reconParticleCollectionName = "FinalStateParticles";
+    static final String clusterCollectionName = "EcalClusters";
+    
+    List<EventFilter> filters = new ArrayList<EventFilter>();
+    
+    public void startOfData() {
+        filters.add(new ClusterFilter());
+        filters.add(new TrackFilter());        
+        filters.add(new ReconstructedParticleFilter());
+    }
+    
+    public void process(EventHeader event) {
+        for (EventFilter filter : filters) {
+            boolean accept = filter.accept(event);
+            if (!accept) {  
+                System.out.println(filter.name() + " rejected event #" + event.getEventNumber());
+                skipEvent();
+            }
+        }               
+        ++nAccepted;
+    }
+    
+    public void endOfData() {
+        System.out.println(this.getClass().getName() + " - run summary");
+        System.out.println("  nRejected: " + nRejected);
+        System.out.println("  nAccepted: " + nAccepted);
+    } 
+    
+    public void skipEvent() {
+        ++nRejected;
+        throw new Driver.NextEventException();
+    }
+        
+    public void setMinTracks(int minTracks) {
+        this.minTracks = minTracks;
+    }
+    
+    public void setMinClusters(int minClusters) {
+        this.minClusters = minClusters;
+    }
+    
+    public void setMinReconParticles(int minReconParticles) {
+        this.minReconParticles = minReconParticles;
+    }
+    
+    static interface EventFilter {
+        
+        String name();
+        
+        boolean accept(EventHeader event);        
+    }
+        
+    class TrackFilter implements EventFilter {
+        
+        public String name() {
+            return "TrackFilter";
+        }
+        
+        public boolean accept(EventHeader event) {
+            if (event.hasCollection(Track.class, trackCollectionName)) {
+                if (event.get(Track.class, trackCollectionName).size() >= minTracks) {
+                    return true;
+                }
+            }
+            return false;
+        }        
+    }
+    
+    class ClusterFilter implements EventFilter {
+        
+        public String name() {
+            return "ClusterFilter";
+        }
+        
+        public boolean accept(EventHeader event) {
+            if (event.hasCollection(Cluster.class, clusterCollectionName)) {
+                if (event.get(Track.class, clusterCollectionName).size() >= minClusters) {
+                    return true;
+                }
+            }
+            return false;
+        }        
+    }
+    
+    class ReconstructedParticleFilter implements EventFilter {
+        
+        public String name() {
+            return "ReconstructedParticleFilter";
+        }
+        
+        public boolean accept(EventHeader event) {
+            if (event.hasCollection(ReconstructedParticle.class, reconParticleCollectionName)) {
+                if (event.get(Track.class, reconParticleCollectionName).size() >= minReconParticles) {
+                    return true;
+                }
+            }
+            return false;
+        }
+    }    
+}