Skip to content

Commit b3d9fda

Browse files
committed
fix: some fixes
1 parent 493cef9 commit b3d9fda

File tree

1 file changed

+26
-20
lines changed

1 file changed

+26
-20
lines changed

reconstruction/cvt/src/main/java/org/jlab/rec/cvt/alignment/AlignmentBankReader.java

Lines changed: 26 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@
33
import java.util.ArrayList;
44
import java.util.List;
55
import java.util.Collections;
6+
import java.util.Map;
67

78
import org.jlab.detector.base.DetectorType;
89
import org.jlab.geom.prim.Line3D;
@@ -26,11 +27,16 @@
2627
*/
2728
public class AlignmentBankReader {
2829

30+
private Map<Integer,Cross> _SVTcrosses;
31+
private Map<Integer,Cluster> _SVTclusters;
32+
private Map<Integer,Cross> _BMTcrosses;
33+
private Map<Integer,Cluster> _BMTclusters;
34+
2935
public List<StraightTrack> getCosmics(DataEvent event) {
3036

3137

32-
var SVThits = RecoBankReader.readBSTHitBank(event, "BST::Hits");
33-
var BMThits = RecoBankReader.readBMTHitBank(event, "BMT::Hits");
38+
List<Hit> SVThits = RecoBankReader.readBSTHitBank(event, "BSTRec::Hits");
39+
List<Hit> BMThits = RecoBankReader.readBMTHitBank(event, "BMTRec::Hits");
3440
if(SVThits!= null) {
3541
Collections.sort(SVThits);
3642
}
@@ -41,20 +47,20 @@ public List<StraightTrack> getCosmics(DataEvent event) {
4147
Collections.sort(BMThits);
4248
}
4349

44-
var _SVTclusters = RecoBankReader.readBSTClusterBank(event, SVThits, "BSTRec::Clusters");
45-
var _BMTclusters = RecoBankReader.readBMTClusterBank(event, BMThits, "BMTRec::Clusters");
50+
_SVTclusters = RecoBankReader.readBSTClusterBank(event, SVThits, "BSTRec::Clusters");
51+
_BMTclusters = RecoBankReader.readBMTClusterBank(event, BMThits, "BMTRec::Clusters");
4652

4753

48-
var _SVTcrosses = RecoBankReader.readBSTCrossBank(event, _SVTclusters, "BSTRec::Crosses");
49-
var _BMTcrosses = RecoBankReader.readBMTCrossBank(event, _BMTclusters, "BMTRec::Crosses");
54+
_SVTcrosses = RecoBankReader.readBSTCrossBank(event, _SVTclusters, "BSTRec::Crosses");
55+
_BMTcrosses = RecoBankReader.readBMTCrossBank(event, _BMTclusters, "BMTRec::Crosses");
5056
if(_SVTcrosses!=null) {
51-
for(Cross cross : _SVTcrosses) {
57+
for(Cross cross : _SVTcrosses.values()) {
5258
cross.setCluster1(_SVTclusters.get(cross.getCluster1().getId()-1));
5359
cross.setCluster2(_SVTclusters.get(cross.getCluster2().getId()-1));
5460
}
5561
}
5662
if(_BMTcrosses!=null) {
57-
for(Cross cross : _BMTcrosses) {
63+
for(Cross cross : _BMTcrosses.values()) {
5864
cross.setCluster1(_BMTclusters.get(cross.getCluster1().getId()-1));
5965
}
6066
}
@@ -68,13 +74,13 @@ public List<StraightTrack> getCosmics(DataEvent event) {
6874
List<Cross> crosses = new ArrayList<>();
6975
for(Cross c : track) {
7076
if(_SVTcrosses!=null && c.getDetector()==DetectorType.BST) {
71-
for(Cross cross : _SVTcrosses) {
77+
for(Cross cross : _SVTcrosses.values()) {
7278
if(c.getId() == cross.getId())
7379
crosses.add(cross);
7480
}
7581
}
7682
if(_BMTcrosses!=null && c.getDetector()==DetectorType.BMT) {
77-
for(Cross cross : _BMTcrosses) {
83+
for(Cross cross : _BMTcrosses.values()) {
7884
if(c.getId() == cross.getId())
7985
crosses.add(cross);
8086
}
@@ -90,8 +96,8 @@ public List<StraightTrack> getCosmics(DataEvent event) {
9096
public List<Track> getTracks(DataEvent event) {
9197

9298

93-
var SVThits = RecoBankReader.readBSTHitBank(event, "BST::Hits");
94-
var BMThits = RecoBankReader.readBMTHitBank(event, "BMT::Hits");
99+
List<Hit> SVThits = RecoBankReader.readBSTHitBank(event, "BSTRec::Hits");
100+
List<Hit> BMThits = RecoBankReader.readBMTHitBank(event, "BMTRec::Hits");
95101
if(SVThits!= null) {
96102
Collections.sort(SVThits);
97103
}
@@ -102,20 +108,20 @@ public List<Track> getTracks(DataEvent event) {
102108
Collections.sort(BMThits);
103109
}
104110

105-
var _SVTclusters = RecoBankReader.readBSTClusterBank(event, SVThits, "BSTRec::Clusters");
106-
var _BMTclusters = RecoBankReader.readBMTClusterBank(event, BMThits, "BMT::Clusters");
111+
_SVTclusters = RecoBankReader.readBSTClusterBank(event, SVThits, "BSTRec::Clusters");
112+
_BMTclusters = RecoBankReader.readBMTClusterBank(event, BMThits, "BMT::Clusters");
107113

108114

109-
var _SVTcrosses = RecoBankReader.readBSTCrossBank(event, _SVTclusters, "BSTRec::Crosses");
110-
var _BMTcrosses = RecoBankReader.readBMTCrossBank(event, _BMTclusters, "BMTRec::Crosses");
115+
_SVTcrosses = RecoBankReader.readBSTCrossBank(event, _SVTclusters, "BSTRec::Crosses");
116+
_BMTcrosses = RecoBankReader.readBMTCrossBank(event, _BMTclusters, "BMTRec::Crosses");
111117
if(_SVTcrosses!=null) {
112-
for(Cross cross : _SVTcrosses) {
118+
for(Cross cross : _SVTcrosses.values()) {
113119
cross.setCluster1(_SVTclusters.get(cross.getCluster1().getId()-1));
114120
cross.setCluster2(_SVTclusters.get(cross.getCluster2().getId()-1));
115121
}
116122
}
117123
if(_BMTcrosses!=null) {
118-
for(Cross cross : _BMTcrosses) {
124+
for(Cross cross : _BMTcrosses.values()) {
119125
cross.setCluster1(_BMTclusters.get(cross.getCluster1().getId()-1));
120126
}
121127
}
@@ -132,13 +138,13 @@ public List<Track> getTracks(DataEvent event) {
132138
List<Cross> crosses = new ArrayList<>();
133139
for(Cross c : track) {
134140
if(_SVTcrosses!=null && c.getDetector()==DetectorType.BST) {
135-
for(Cross cross : _SVTcrosses) {
141+
for(Cross cross : _SVTcrosses.values()) {
136142
if(c.getId() == cross.getId())
137143
crosses.add(cross);
138144
}
139145
}
140146
if(_BMTcrosses!=null && c.getDetector()==DetectorType.BMT) {
141-
for(Cross cross : _BMTcrosses) {
147+
for(Cross cross : _BMTcrosses.values()) {
142148
if(c.getId() == cross.getId())
143149
crosses.add(cross);
144150
}

0 commit comments

Comments
 (0)