lcsim/src/org/lcsim/contrib/Cassell/recon/Cheat

diff -u -r1.2 -r1.3
--- PPRParticleDriver.java	2 Mar 2007 16:58:30 -0000	1.2
+++ PPRParticleDriver.java	16 May 2007 18:54:00 -0000	1.3
@@ -9,6 +9,7 @@

  */
 package org.lcsim.contrib.Cassell.recon.Cheat;
 

+import org.lcsim.event.MCParticle;

 import org.lcsim.util.Driver;
 import java.util.ArrayList;
 import java.util.List;

@@ -25,6 +26,7 @@

 
     String inlist;
     String outlist;

+    String outlistMC;

     int minThits = 4;
     int minChits = 3;
     PPRReconParticleConstructor crpc;

@@ -33,6 +35,14 @@

     {
         inlist = in;
         outlist = out;

+	outlistMC = null;
+    }
+    /** Creates a new instance of PPRParticleDriver */
+    public PPRParticleDriver(String in,String out, String outMC)
+    {
+        inlist = in;
+        outlist = out;
+	outlistMC = outMC;

     }
     public void setMinTrackerHits(int n)
     {

@@ -46,6 +56,7 @@

     {
         List<MCReconstructedParticle> rl = event.get(MCReconstructedParticle.class,inlist);
         List<ReconstructedParticle>outrl = new ArrayList<ReconstructedParticle>();

+        List<MCParticle>outrlMC = new ArrayList<MCParticle>();

         crpc = new PPRReconParticleConstructor();
         crpc.setMinTrackerHits(minThits);
         crpc.setMinCalorimeterHits(minChits);

@@ -56,9 +67,13 @@

            {
                rp.addParticle(p);
                outrl.add(rp);

+	       outrlMC.add(p.getMCParticle());

            }
         }
         event.put(outlist,outrl,ReconstructedParticle.class,0);

+	if (outlistMC != null) {
+	    event.put(outlistMC,outrlMC,MCParticle.class,0);
+	}

     }
     
 }