lcsim-contrib/src/main/java/org/lcsim/contrib/Cassell/recon
diff -N FindFixedNumberOfJets.java
--- /dev/null 1 Jan 1970 00:00:00 -0000
+++ FindFixedNumberOfJets.java 24 Aug 2011 15:43:27 -0000 1.1
@@ -0,0 +1,63 @@
+/*
+ * To change this template, choose Tools | Templates
+ * and open the template in the editor.
+ */
+
+package org.lcsim.contrib.Cassell.recon;
+import org.lcsim.util.Driver;
+import org.lcsim.event.EventHeader;
+import org.lcsim.event.ReconstructedParticle;
+import org.lcsim.event.util.*;
+import hep.physics.jet.*;
+import java.util.*;
+
+/**
+ *
+ * @author cassell
+ */
+public class FindFixedNumberOfJets extends Driver
+{
+ boolean continueOnError;
+ int njets = 4;
+
+ public void setContinueOnError(boolean continueOnError)
+ {
+ this.continueOnError = continueOnError;
+ }
+ boolean first;
+
+ public void setNjets(int njets)
+ {
+ this.njets = njets;
+ }
+
+ public FindFixedNumberOfJets()
+ {
+ this(true);
+ }
+ public FindFixedNumberOfJets(boolean coe)
+ {
+ continueOnError = coe;
+ first = true;
+ }
+ protected void process(EventHeader event)
+ {
+ if(first)
+ {
+ List<List<ReconstructedParticle>> rll = event.get(ReconstructedParticle.class);
+ for(int i=0;i<rll.size();i++)
+ {
+ List<ReconstructedParticle> rl = rll.get(i);
+ String rps = event.getMetaData(rl).getName();
+ JetDriver jd = new JetDriver();
+ JetFinder jf = new FixNumberOfJetsFinder(njets);
+ jd.setInputCollectionName(rps);
+ jd.setOutputCollectionName(rps+njets+"Jets");
+ jd.setFinder(jf);
+ jd.setContinueOnError(continueOnError);
+ }
+ first = false;
+ }
+ super.process(event);
+ }
+}