Merge pull request #43 from JeffersonLab/dcalign

New wagon for dc alignment

src/main/java/org/jlab/jnp/grapes/custom-services/DCAlignmentWagon.java

@@ -0,0 +1,78 @@
+package org.jlab.jnp.grapes.services;
+
+import java.util.ArrayList;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+import org.jlab.jnp.hipo4.data.Bank;
+import org.jlab.jnp.hipo4.data.Event;
+import org.jlab.jnp.hipo4.data.SchemaFactory;
+
+/**
+ * 
+ * Skim for DC alignment
+ *
+ * Determine whether an electron candidate, defined as a track with associated 
+ * Cherenkov and calorimter clusters, was detected.
+ *
+ * @author devita
+ */
+public class DCAlignmentWagon extends Wagon {
+
+    static final double VZMIN = -35;
+    static final double VZMAX =  35;
+    static final double ECALMIN = 0;
+    static final double NPHEMIN = 2.0;
+
+    public DCAlignmentWagon(){
+        super("DCAlignmentWagon","devita","0.9");
+    }
+
+    @Override
+    public boolean init(String jsonString) {
+        return true;
+    }
+
+    @Override
+    public boolean processDataEvent(Event event, SchemaFactory factory) {
+
+        Bank particles = new Bank(factory.getSchema("REC::Particle"));
+        Bank calorimeters = new Bank(factory.getSchema("REC::Calorimeter"));
+        Bank cherenkovs = new Bank(factory.getSchema("REC::Cherenkov"));
+        event.read(particles);
+        event.read(calorimeters);
+        event.read(cherenkovs);
+        if (particles.getRows()<1) return false;
+        if (calorimeters.getRows()<1) return false;
+        if (cherenkovs.getRows()<1) return false;
+
+        HashMap<Integer,ArrayList<Integer>> part2ecal = Util.mapByIndex(calorimeters);
+        HashMap<Integer,ArrayList<Integer>> part2cher = Util.mapByIndex(cherenkovs);
+
+        for(int ipart=0; ipart<particles.getRows(); ipart++) {
+            double beta = particles.getFloat("beta",ipart);
+            double vz   = particles.getFloat("vz",ipart);
+
+            double nphe = 0;
+            if(part2cher.containsKey(ipart))
+                nphe = cherenkovs.getFloat("nphe", part2cher.get(ipart).get(0));
+
+
+            double energy = 0;
+            if(part2ecal.containsKey(ipart)) {
+                for(int i=0; i<part2ecal.get(ipart).size(); i++) {
+                    energy+=calorimeters.getFloat("energy", part2ecal.get(ipart).get(i));
+                }
+            }
+
+            if( beta>0 && 
+                vz>VZMIN && vz<VZMAX &&
+                nphe>NPHEMIN && 
+                energy>ECALMIN ) { 
+                return true;
+            }
+        } 
+
+        return false;
+    }   
+}