Atof testing

etc/bankdefs/hipo4/README.md

@@ -386,6 +386,8 @@
 | 12 | `ATOF::tdc` | TDC bank for the ALERT TOF |
 | 21 | `ATOF::hits` | Reconstructed ATOF hits |
 | 22 | `ATOF::clusters` | Clusters in ATOF |
+| 31 | `ATOF::testhits` | Reconstructed ATOF hits |
+| 32 | `ATOF::testclusters` | Clusters in ATOF |
 
 ## Group 22600
 
@@ -407,3 +409,4 @@
 | 26 | `AHDC::KFTrack` | Reco Kalman Filter Tracks |
 | 30 | `AHDC_AI::Prediction` | Prediction given by AI |
 | 31 | `ALERT::Projections` | Track Projections to ATOF |
+| 35 | `ATOF::testVeff` | veff test for ATOF |

etc/bankdefs/hipo4/alert.json

@@ -49,10 +49,38 @@
         "name": "L_at_wedge",
         "type": "F",
         "info": "path length at atof wedge (inner surface) in mm"
+      }
+    ]
+  },{
+      "name": "ATOF::testVeff",
+      "group": 23000,
+      "item": 35,
+      "info": "veff test for ATOF",
+      "entries": [		
+	  {
+            "name": "id",
+            "type": "S",
+            "info": "calib id"
+      },{
+        "name": "iCluster",
+        "type": "S",
+        "info": "id of the cluster"
+      },{
+        "name": "iBar",
+        "type": "S",
+        "info": "id bar hit"
       },{
-        "name": "L_in_wedge",
+        "name": "ldiff",
         "type": "F",
-        "info": "path length inside atof wedge in mm"
+        "info": "length diff mm"
+      }, {
+        "name": "tdiff",
+        "type": "F",
+        "info": "time diff ns"			    
+      },{
+        "name": "module",
+        "type": "S",
+        "info": "module" 
       }
     ]
   },{
@@ -103,6 +131,62 @@
         "info": "id of cluster to which the hit was associated"
       }
     ]
+  },{
+    "name": "ATOF::testhits",
+    "group": 22500,
+    "item": 31,
+    "info": "Reconstructed ATOF hits",
+    "entries": [		
+      {
+        "name": "id",
+        "type": "S",
+        "info": "hit id"
+      }, {
+        "name": "sector",
+        "type": "I",
+        "info": "atof sector"
+      }, {
+        "name": "layer",
+        "type": "I",
+        "info": "atof layer"			    
+      },{
+        "name": "component",
+        "type": "I",
+        "info": "atof component" 
+      },{
+        "name": "time",
+        "type": "F",
+        "info": "time in ns"
+      },{
+        "name": "TDC",
+        "type": "I",
+        "info": "TDC"
+      },{
+        "name": "x",
+        "type": "F",
+        "info": "x position in mm"
+      }, {
+        "name": "y",
+        "type": "F",
+        "info": "y position in mm"
+      }, {
+        "name": "z",
+        "type": "F",
+        "info": "z position in mm"
+      },{
+        "name": "energy",
+        "type": "F",
+        "info": "deposited energy in MeV"
+      },{
+        "name": "TOT",
+        "type": "I",
+        "info": "TOT"
+      },{
+        "name": "clusterid",
+        "type": "S",
+        "info": "id of cluster to which the hit was associated"
+      }
+    ]
   },{
     "name": "ATOF::clusters",
     "group": 22500,
@@ -155,6 +239,54 @@
         "info": "id of the projected track matched. it is -1 if a straight track was assigned."
       }
     ]
+  },{
+    "name": "ATOF::testclusters",
+    "group": 22500,
+    "item": 32,
+    "info": "Clusters in ATOF",
+    "entries": [		
+      {
+        "name": "id",
+        "type": "S",
+        "info": "hit id"
+      }, {
+        "name": "N_bar",
+        "type": "I",
+        "info": "number of hits from the bars"
+      }, {
+        "name": "N_wedge",
+        "type": "I",
+        "info": "number of hits from the wedges"			    
+      },{
+        "name": "time",
+        "type": "F",
+        "info": "time in ns"
+      },{
+        "name": "TDC",
+        "type": "I",
+        "info": "TDC"
+      },{
+        "name": "x",
+        "type": "F",
+        "info": "x position in mm"
+      }, {
+        "name": "y",
+        "type": "F",
+        "info": "y position in mm"
+      }, {
+        "name": "z",
+        "type": "F",
+        "info": "z position in mm"
+      },{
+        "name": "energy",
+        "type": "F",
+        "info": "energy in MeV"
+      },{
+        "name": "TOT",
+        "type": "I",
+        "info": "TOT"
+      }
+    ]
   },{
     "name": "AHDC::Hits",
     "group": 23000,

reconstruction/alert/src/main/java/org/jlab/rec/atof/banks/TestBankWriter.java

@@ -0,0 +1,140 @@
+package org.jlab.rec.atof.banks;
+
+import java.util.ArrayList;
+import org.jlab.io.base.DataBank;
+import org.jlab.io.base.DataEvent;
+import org.jlab.rec.atof.cluster.ATOFCluster;
+import org.jlab.rec.atof.hit.ATOFHit;
+import org.jlab.rec.atof.hit.BarHit;
+import org.jlab.rec.atof.veff.VeffCalibration;
+
+/**
+ * The {@code RecoBankWriter} writes the banks needed for the atof testing: hits
+ * and clusters info.
+ *
+ * @author pilleux
+ */
+public class TestBankWriter {
+
+    public static DataBank fillATOFTestHitBank(DataEvent event, ArrayList<ATOFHit> wedgeHits, ArrayList<BarHit> barHits) {
+        ArrayList<ATOFHit> hitList = new ArrayList<>();
+        hitList.addAll(wedgeHits);
+        hitList.addAll(barHits);
+        DataBank bank = event.createBank("ATOF::testhits", hitList.size());
+        if (bank == null) {
+            System.err.println("COULD NOT CREATE A ATOF::testhits BANK!!!!!!");
+            return null;
+        }
+
+        for (int i = 0; i < hitList.size(); i++) {
+            bank.setShort("id", i, (short) (i + 1));
+            bank.setShort("clusterid", i, (short) hitList.get(i).getAssociatedClusterIndex());
+            bank.setInt("sector", i, (int) hitList.get(i).getSector());
+            bank.setInt("layer", i, (int) hitList.get(i).getLayer());
+            bank.setInt("component", i, (int) hitList.get(i).getComponent());
+            bank.setInt("TDC", i, (int) hitList.get(i).getTdc());
+            bank.setFloat("time", i, (float) hitList.get(i).getTime());
+            bank.setFloat("x", i, (float) (hitList.get(i).getX()));
+            bank.setFloat("y", i, (float) (hitList.get(i).getY()));
+            bank.setFloat("z", i, (float) (hitList.get(i).getZ()));
+            bank.setInt("TOT", i, (int) hitList.get(i).getTot());
+            bank.setFloat("energy", i, (float) hitList.get(i).getEnergy());
+        }
+        return bank;
+    }
+
+    public static DataBank fillATOFTestClusterBank(DataEvent event, ArrayList<ATOFCluster> clusterList) {
+
+        DataBank bank = event.createBank("ATOF::testclusters", clusterList.size());
+
+        if (bank == null) {
+            System.err.println("COULD NOT CREATE A ATOF::testclusters BANK!!!!!!");
+            return null;
+        }
+
+        for (int i = 0; i < clusterList.size(); i++) {
+            bank.setShort("id", i, (short) (i + 1));
+            bank.setInt("N_bar", i, (int) clusterList.get(i).getBarHits().size());
+            bank.setInt("N_wedge", i, (int) clusterList.get(i).getWedgeHits().size());
+            bank.setInt("TDC", i, (int) clusterList.get(i).getTdc());
+            bank.setFloat("time", i, (float) clusterList.get(i).getTime());
+            bank.setFloat("x", i, (float) (clusterList.get(i).getX()));
+            bank.setFloat("y", i, (float) (clusterList.get(i).getY()));
+            bank.setFloat("z", i, (float) (clusterList.get(i).getZ()));
+            bank.setInt("TOT", i, (int) clusterList.get(i).getTot());
+            bank.setFloat("energy", i, (float) clusterList.get(i).getEnergy());
+        }
+        return bank;
+    }
+
+    public static DataBank fillATOFTestVeff(DataEvent event, ArrayList<VeffCalibration> calibrations) {
+
+        DataBank bank = event.createBank("ATOF::testVeff", calibrations.size());
+
+        if (bank == null) {
+            System.err.println("COULD NOT CREATE A ATOF::testclusters BANK!!!!!!");
+            return null;
+        }
+        for (int i = 0; i < calibrations.size(); i++) {
+            bank.setShort("id", i, (short) (i + 1));
+            bank.setShort("iCluster", i, (short) calibrations.get(i).getICluster());
+            bank.setShort("iBar", i, (short) calibrations.get(i).getIBarHit());
+            bank.setFloat("ldiff", i, (float) calibrations.get(i).getLdiff());
+            bank.setFloat("tdiff", i, (float) calibrations.get(i).getTdiff());
+            bank.setShort("module", i, (short) calibrations.get(i).getModule());
+        }
+        return bank;
+    }
+
+    /**
+     * Appends the atof banks to an event.
+     *
+     * @param event the {@link DataEvent} in which to append the banks
+     * @param clusterList the {@link ArrayList} of {@link ATOFCluster}
+     * containing the clusters info to be added to the bank
+     * @param wedgeHits the {@link ArrayList} of {@link ATOFHit} containing the
+     * wedge hits info to be added
+     * @param barHits the {@link ArrayList} of {@link BarHit} containing the bar
+     * hits info to be added
+     *
+     * @return 0 if it worked, 1 if it failed
+     *
+     */
+    public int appendATOFBanks(DataEvent event, ArrayList<ATOFHit> wedgeHits, ArrayList<BarHit> barHits, ArrayList<ATOFCluster> clusterList) {
+
+        DataBank hitbank = this.fillATOFTestHitBank(event, wedgeHits, barHits);
+        if (hitbank != null) {
+            event.appendBank(hitbank);
+        } else {
+            return 1;
+        }
+
+        DataBank clusterbank = fillATOFTestClusterBank(event, clusterList);
+        if (clusterbank != null) {
+            event.appendBank(clusterbank);
+        } else {
+            return 1;
+        }
+
+        return 0;
+    }
+
+    public int appendVeffBanks(DataEvent event, ArrayList<VeffCalibration> calibrations) {
+
+        DataBank veffbank = fillATOFTestVeff(event, calibrations);
+        if (veffbank != null) {
+            event.appendBank(veffbank);
+        } else {
+            return 1;
+        }
+
+        return 0;
+    }
+
+    /**
+     * @param args the command line arguments
+     */
+    public static void main(String[] args) {
+    }
+
+}

reconstruction/alert/src/main/java/org/jlab/rec/atof/veff/VeffCalibration.java

@@ -0,0 +1,60 @@
+package org.jlab.rec.atof.veff;
+
+public class VeffCalibration {
+    double ldiff, tdiff;
+    int iCluster, iBarHit, module;
+
+    public double getLdiff() {
+        return ldiff;
+    }
+
+    public void setLdiff(double ldiff) {
+        this.ldiff = ldiff;
+    }
+
+    public double getTdiff() {
+        return tdiff;
+    }
+
+    public void setTdiff(double tdiff) {
+        this.tdiff = tdiff;
+    }
+
+    // Getter and Setter for iCluster
+    public int getICluster() {
+        return iCluster;
+    }
+
+    public void setICluster(int iCluster) {
+        this.iCluster = iCluster;
+    }
+
+    // Getter and Setter for iBarHit
+    public int getIBarHit() {
+        return iBarHit;
+    }
+
+    public void setIBarHit(int iBarHit) {
+        this.iBarHit = iBarHit;
+    }
+
+    // Getter and Setter for module
+    public int getModule() {
+        return module;
+    }
+
+    public void setModule(int module) {
+        this.module = module;
+    }
+
+    VeffCalibration(int module, double ldiff, double tdiff, int iCluster, int iBarHit){
+        this.module=module;
+        this.ldiff = ldiff;
+        this.tdiff = tdiff;
+        this.iCluster = iCluster;
+        this.iBarHit = iBarHit;
+    }
+
+    public static void main(String[] args) {
+    }
+}