Fix to fill detector and layer for Cosmics trajectory bank

JeffersonLab · Aug 28, 2023 · 2764a1b · 2764a1b
1 parent d5025a9
commit 2764a1b
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/reconstruction/cvt/src/main/java/org/jlab/rec/cvt/track/StraightTrack.java b/reconstruction/cvt/src/main/java/org/jlab/rec/cvt/track/StraightTrack.java
@@ -191,6 +191,7 @@ public void findTrajectory() {
                     stVec.setTrkThetaAtSurface(ThetaTrackIntersPlane);
                     stVec.setTrkToModuleAngle(trkToMPlnAngl);
                     stVec.setCalcCentroidStrip(CalcCentroidStrip);
+                    stVec.setSurfaceDetector(DetectorType.BST.getDetectorId()); 
                     if(stateVecs.size()>0 
                             && stateVecs.get(stateVecs.size()-1).x()==stVec.x()
                             && stateVecs.get(stateVecs.size()-1).y()==stVec.y()
@@ -230,7 +231,8 @@ public void findTrajectory() {
                     stVec.setTrkThetaAtSurface(ThetaTrackIntersSurf);
                     stVec.setTrkToModuleAngle(trkToMPlnAngl);
                     stVec.setCalcCentroidStrip(CalcCentroidStrip); 
-                     if(stateVecs.size()>0 
+                    stVec.setSurfaceDetector(DetectorType.BMT.getDetectorId()); 
+                    if(stateVecs.size()>0 
                             && stateVecs.get(stateVecs.size()-1).x()==stVec.x()
                             && stateVecs.get(stateVecs.size()-1).y()==stVec.y()
                             && stateVecs.get(stateVecs.size()-1).z()==stVec.z()) {

diff --git a/reconstruction/cvt/src/main/java/org/jlab/rec/cvt/trajectory/TrajectoryFinder.java b/reconstruction/cvt/src/main/java/org/jlab/rec/cvt/trajectory/TrajectoryFinder.java
@@ -120,6 +120,7 @@ public Trajectory findTrajectory(int id, Ray ray, ArrayList<Cross> candCrossList
                             continue;
                         }
 
+                        stVec.setSurfaceDetector(DetectorType.BST.getDetectorId());
                         Cluster clsOnTrk = null;
                         if (l % 2 == 0) {
                             clsOnTrk = c.getCluster1(); 
@@ -195,6 +196,7 @@ public Trajectory findTrajectory(int id, Ray ray, ArrayList<Cross> candCrossList
                             continue;
                         } 
 
+                        stVec.setSurfaceDetector(DetectorType.BMT.getDetectorId());
                         if (c.getType()==BMTType.C) { //C-detector measuring Z
                             //if(traj.isFinal) { // reset the cross only for final trajectory
 
@@ -226,9 +228,8 @@ public Trajectory findTrajectory(int id, Ray ray, ArrayList<Cross> candCrossList
         }
 
         //Collections.sort(stateVecs);
-
         stateVecs.sort(Comparator.comparing(StateVec::y));
-        for (int l = 0; l < stateVecs.size(); l++) {
+        for (int l = 0; l < stateVecs.size(); l++) { 
             stateVecs.get(l).setSurfaceDetector(DetectorType.CVT.getDetectorId());
             stateVecs.get(l).setPlaneIdx(l);
         }