FairRoot/PandaRoot
Public Member Functions | Static Public Attributes | Private Member Functions | Private Attributes | List of all members
PndTrackCombiner Class Reference

#include <PndTrackCombiner.h>

Public Member Functions

 PndTrackCombiner (TClonesArray *a, TClonesArray *b)
 
virtual ~PndTrackCombiner ()
 
void init (map< Int_t, vector< PndFtsExpandedTrackCand >> c)
 
vector< PndLineApproximation > combine (Int_t bigLayer)
 
vector< PndLineApproximation > combine (vector< PndFtsExpandedTrackCand > cands1, vector< PndFtsExpandedTrackCand > cands2, Int_t layer1, Int_t layer2)
 
vector< PndLineApproximation > combine (PndFtsExpandedTrackCand c1, PndFtsExpandedTrackCand c2, Int_t layer1, Int_t layer2)
 

Static Public Attributes

static Double_t compDist = 0.7
 
static Double_t compAngle = 2.5
 
static Double_t compQuali = 600
 

Private Member Functions

void addLine (vector< PndLineCombination > &lines, PndLineApproximation &l, Int_t max)
 
void addLineOld (vector< PndLineApproximation > &lines, PndLineApproximation &l, Int_t end)
 
void refitLines (vector< PndLineCombination > &lines)
 
vector< PndFtsExpandedTrackCand > compareAndRefit (Int_t layer1, Int_t layer2)
 
vector< PndLineApproximation > compareAndRefit (vector< PndLineApproximation > l1, vector< PndLineApproximation > l2)
 

Private Attributes

map< Int_t, vector
< PndFtsExpandedTrackCand > > 		fExpandedTrackCands
 
PndFtsLineComparator fComparator
 
TClonesArray * planes
 
TClonesArray * corrHits
 
Int_t firstRun = 0
 
Int_t trackNum
 
Double_t angleLimit
 
Double_t fts1ZPos = 295.4
 
Double_t fts2ZPos = 327.4
 
Double_t fts3ZPos = 394.5
 
Double_t fts4ZPos = 438.5
 
Double_t fts5ZPos = 607.5
 
Double_t fts6ZPos = 747.5
 
Double_t ftsPos [6]
 

Detailed Description

Definition at line 30 of file PndTrackCombiner.h.

Constructor & Destructor Documentation

PndTrackCombiner::PndTrackCombiner ( TClonesArray *  a,
TClonesArray *  b 
)
inline

Definition at line 36 of file PndTrackCombiner.h.

36 : fComparator(compDist,compAngle), planes(a), corrHits(b), firstRun(0), angleLimit(20){}
PndTrackCombiner::planes
TClonesArray * planes
Definition: PndTrackCombiner.h:60
PndTrackCombiner::corrHits
TClonesArray * corrHits
Definition: PndTrackCombiner.h:61
b
TTree * b
Definition: GammaSpectraAnalysis.C:24
PndTrackCombiner::compDist
static Double_t compDist
Definition: PndTrackCombiner.h:32
PndTrackCombiner::compAngle
static Double_t compAngle
Definition: PndTrackCombiner.h:33
a
Int_t a
Definition: anaLmdDigi.C:126
PndTrackCombiner::fComparator
PndFtsLineComparator fComparator
Definition: PndTrackCombiner.h:54
PndTrackCombiner::~PndTrackCombiner ( )
virtual

Definition at line 14 of file PndTrackCombiner.cxx.

14  {
15  // TODO Auto-generated destructor stubPndTrack*
16 }

Member Function Documentation

void PndTrackCombiner::addLine ( vector< PndLineCombination > &  lines,
PndLineApproximation &  l,
Int_t  max 
)
private

Definition at line 131 of file PndTrackCombiner.cxx.

References Bool_t, compQuali, Double_t, fComparator, PndLineApproximation::getLine(), PndFtsLineComparator::getQuality(), i, and max().

Referenced by combine().

131  {
132  PndLine &l = la.getLine();
133  Bool_t inUse = kFALSE;
134  for(int i=0;i<max;i++){
135  PndLine& l2=lines[i].getApprox1().getLine();
136  //compare with the line of other layer
137  Double_t quality = fComparator.getQuality(l,l2);
138  if(quality>compQuali)continue;
139  //is this line better as the current best line
140  if(lines[i].getQuality()>quality){
141  //better line found
142  lines[i].setApprox2(&la);
143  lines[i].setQuality(quality);
144  }
145  inUse = kTRUE;
146  }
147  if(!inUse){
148  PndLineCombination newCom(la);
149  lines.push_back(newCom);
150  }
151 }
i
Int_t i
Definition: run_full.C:25
max
friend F32vec4 max(const F32vec4 &a, const F32vec4 &b)
Definition: P4_F32vec4.h:26
PndFtsLineComparator::getQuality
Double_t getQuality(PndLine l1, PndLine l2)
Definition: PndFtsLineComparator.cxx:58
Double_t
Double_t
Definition: SimCompleteLinkDef.h:6
PndTrackCombiner::compQuali
static Double_t compQuali
Definition: PndTrackCombiner.h:34
PndTrackCombiner::fComparator
PndFtsLineComparator fComparator
Definition: PndTrackCombiner.h:54
void PndTrackCombiner::addLineOld ( vector< PndLineApproximation > &  lines,
PndLineApproximation &  l,
Int_t  end 
)
private

Definition at line 165 of file PndTrackCombiner.cxx.

References PndFtsLineComparator::areEqual2D(), fComparator, PndLineApproximation::getLine(), i, and PndLine::incrementRating().

165  {
166  PndLine &l = la.getLine();
167  for(int i=0;i<end;i++){
168  PndLine &l2 = lines[i].getLine();
169  if(fComparator.areEqual2D(l,l2)){
170  l.incrementRating();
171  l2.incrementRating();
172  }
173  }
174  lines.push_back(la);
175 }
i
Int_t i
Definition: run_full.C:25
PndLine::incrementRating
void incrementRating()
Definition: PndLine.h:43
PndFtsLineComparator::areEqual2D
Bool_t areEqual2D(PndLine l1, PndLine l2)
Definition: PndFtsLineComparator.cxx:29
PndTrackCombiner::fComparator
PndFtsLineComparator fComparator
Definition: PndTrackCombiner.h:54
vector< PndLineApproximation > PndTrackCombiner::combine ( Int_t  bigLayer)

Definition at line 18 of file PndTrackCombiner.cxx.

References a, addLine(), PndFtsLineComparator::areEqual2D(), b, Bool_t, combi, compareAndRefit(), fComparator, fExpandedTrackCands, ftsPos, PndLineCombination::getApprox1(), PndLineApproximation::getLine(), PndLine::getRating(), i, PndLineApproximation::newApproximation(), refitLines(), and PndFtsLineComparator::setZValue().

Referenced by PndForwardTrackFinderTask::Exec().

18  {
19  Int_t l1 = 4*bigLayer+0;
20  Int_t l2 = 4*bigLayer+1;
21  Int_t l3 = 4*bigLayer+2;
22  Int_t l4 = 4*bigLayer+3;
23 
24  vector<PndFtsExpandedTrackCand> layer1a4 = compareAndRefit(l1,l4);
25  vector<PndLineApproximation> arr[3];
26  arr[0] = combine(layer1a4,fExpandedTrackCands[l2],0,1);
27  arr[1] = combine(layer1a4,fExpandedTrackCands[l3],0,2);
28  arr[2] = combine(fExpandedTrackCands[l2],fExpandedTrackCands[l3],1,2);
29 
30 // cout << "firstAndLast=" << layer1a4.size() << endl;
31 // cout << "arr[0]=" << arr[0].size() << endl;
32 // cout << "arr[1]=" << arr[1].size() << endl;
33 // cout << "arr[2]=" << arr[2].size() << endl;
34 
35  //new implementation
36  vector<PndLineCombination> combindedLines;
37  //add the tracks of the first layer
38  for(size_t i=0;i<arr[0].size();i++){
39  PndLineApproximation newApprox(arr[0][i]);
40  combindedLines.push_back(newApprox);
41  }
42 
43  //firstRun = combindedLines.size();
44  for(int i=1;i<3;i++){
45  vector<PndLineApproximation> lines = arr[i];
46  fComparator.setZValue(ftsPos[bigLayer]+5*i);
47  Int_t maxVal = combindedLines.size(); //tracks in layers before -> no combination in same layer
48  for(size_t j=0;j<lines.size();j++){
49  addLine(combindedLines,lines[j],maxVal);
50  }
51  refitLines(combindedLines);
52  }
53  //get all lines with a high quality
54  vector<PndLineApproximation> result;
55  for(size_t i=0;i<combindedLines.size();i++){
56  PndLineCombination combi = combindedLines[i];
57  PndLineApproximation approx = combi.getApprox1();
58  if(approx.getLine().getRating()>2){
59  result.push_back(approx);
60  }
61  }
62  //combine similar tracks
63  vector<PndLineApproximation> result2;
64  fComparator.setZValue(ftsPos[bigLayer]);
65  Bool_t used[result.size()];
66  for(size_t i=0;i<result.size();i++)used[i]=kFALSE;
67  for(size_t i=0;i<result.size();i++){
68  if(used[i])continue;
69  PndLineApproximation a = result[i];
70  for(size_t j=i+1;j<result.size();j++){
71  if(used[j])continue;
72  PndLineApproximation b = result[j];
73  if(fComparator.areEqual2D(a.getLine(),b.getLine())){
74  a=a.newApproximation(b);
75  used[j]=kTRUE;
76  }
77  }
78  used[i]=kTRUE;
79  result2.push_back(a);
80  }
81  return result2;
82 }
PndTrackCombiner::ftsPos
Double_t ftsPos[6]
Definition: PndTrackCombiner.h:72
i
Int_t i
Definition: run_full.C:25
b
TTree * b
Definition: GammaSpectraAnalysis.C:24
PndLineApproximation::newApproximation
PndLineApproximation newApproximation(PndLineApproximation &approx2)
Definition: PndLineApproximation.cxx:54
PndTrackCombiner::addLine
void addLine(vector< PndLineCombination > &lines, PndLineApproximation &l, Int_t max)
Definition: PndTrackCombiner.cxx:131
PndTrackCombiner::compareAndRefit
vector< PndFtsExpandedTrackCand > compareAndRefit(Int_t layer1, Int_t layer2)
Definition: PndTrackCombiner.cxx:177
PndTrackCombiner::refitLines
void refitLines(vector< PndLineCombination > &lines)
Definition: PndTrackCombiner.cxx:153
PndFtsLineComparator::setZValue
void setZValue(Double_t z)
Definition: PndFtsLineComparator.h:28
a
Int_t a
Definition: anaLmdDigi.C:126
PndLineCombination::getApprox1
PndLineApproximation getApprox1()
Definition: PndLineCombination.h:21
combi
TGeoCombiTrans * combi
Definition: createROOTEnDrcGeo.C:66
PndLine::getRating
Int_t getRating()
Definition: PndLine.h:42
PndTrackCombiner::combine
vector< PndLineApproximation > combine(Int_t bigLayer)
Definition: PndTrackCombiner.cxx:18
PndFtsLineComparator::areEqual2D
Bool_t areEqual2D(PndLine l1, PndLine l2)
Definition: PndFtsLineComparator.cxx:29
PndTrackCombiner::fExpandedTrackCands
map< Int_t, vector< PndFtsExpandedTrackCand > > fExpandedTrackCands
Definition: PndTrackCombiner.h:53
PndTrackCombiner::fComparator
PndFtsLineComparator fComparator
Definition: PndTrackCombiner.h:54
vector< PndLineApproximation > PndTrackCombiner::combine ( vector< PndFtsExpandedTrackCand >  cands1,
vector< PndFtsExpandedTrackCand >  cands2,
Int_t  layer1,
Int_t  layer2 
)

Definition at line 85 of file PndTrackCombiner.cxx.

References combine(), PndLineApproximation::getLine(), PndFtsExpandedTrackCand::getLineApproximations(), i, PndLine::setRating(), and PndLineApproximation::setYInfo().

85  {
86  vector<PndLineApproximation> result;
87  for(size_t i=0;i<cands1.size();i++){
88  size_t resultSizeBefore = result.size();
89  PndFtsExpandedTrackCand cand1 = cands1[i];
90  for(size_t j=0;j<cands2.size();j++){
91  PndFtsExpandedTrackCand cand2 = cands2[j];
92  vector<PndLineApproximation> lines = combine(cand1,cand2,layer1,layer2);
93  result.insert(result.begin(),lines.begin(),lines.end());
94  }
95  //layer1 track not used?
96  if(layer1==0 && result.size()==resultSizeBefore){
97  PndLineApproximation la = cand1.getLineApproximations()[0];
98  la.getLine().setRating(3);
99  la.setYInfo(kFALSE);
100  result.push_back(la);
101  }
102  }
103  return result;
104 }
i
Int_t i
Definition: run_full.C:25
PndLine::setRating
void setRating(Int_t r)
Definition: PndLine.h:44
PndFtsExpandedTrackCand::getLineApproximations
vector< PndLineApproximation > getLineApproximations()
Definition: PndFtsExpandedTrackCand.h:25
PndTrackCombiner::combine
vector< PndLineApproximation > combine(Int_t bigLayer)
Definition: PndTrackCombiner.cxx:18
vector< PndLineApproximation > PndTrackCombiner::combine ( PndFtsExpandedTrackCand  c1,
PndFtsExpandedTrackCand  c2,
Int_t  layer1,
Int_t  layer2 
)

Definition at line 106 of file PndTrackCombiner.cxx.

References angle, angleLimit, Double_t, PndPlane::getIntersection(), PndFtsExpandedTrackCand::getLineApproximations(), PndLine::getRating(), PndLine::getZAngle(), i, CAMath::Max(), CAMath::Min(), p1, p2, and PndLine::setRating().

106  {
107  vector<PndLineApproximation> approx1 = c1.getLineApproximations();
108  vector<PndLineApproximation> approx2 = c2.getLineApproximations();
109  vector<PndLineApproximation> result;
110  for(size_t i=0;i<approx1.size();i++){
111  for(size_t j=0;j<approx2.size();j++){
112  PndLine line1 = approx1[i].getLine();
113  PndLine line2 = approx2[j].getLine();
114  PndPlane p1(line1,layer1);
115  PndPlane p2(line2,layer2);
116  PndLine l = p1.getIntersection(p2);
117 
118  Double_t angle = l.getZAngle();
119  if(TMath::Min(angle,180-angle)>angleLimit){
120  vector<PndFtsHit*> cHits1 = approx1[i].getHits();
121  vector<PndFtsHit*> cHits2 = approx2[j].getHits();
122  l.setRating(TMath::Max(line1.getRating(),line2.getRating()));
123  PndLineApproximation newApprox(l,cHits1,cHits2);
124  result.push_back(newApprox);
125  }
126  }
127  }
128  return result;
129 }
PndLine::getZAngle
Double_t getZAngle()
Definition: PndLine.h:39
i
Int_t i
Definition: run_full.C:25
CAMath::Min
static T Min(const T &x, const T &y)
Definition: PndCAMath.h:35
Double_t
Double_t
Definition: SimCompleteLinkDef.h:6
PndLine::setRating
void setRating(Int_t r)
Definition: PndLine.h:44
PndFtsExpandedTrackCand::getLineApproximations
vector< PndLineApproximation > getLineApproximations()
Definition: PndFtsExpandedTrackCand.h:25
PndLine::getRating
Int_t getRating()
Definition: PndLine.h:42
p2
TPad * p2
Definition: hist-t7.C:117
CAMath::Max
static T Max(const T &x, const T &y)
Definition: PndCAMath.h:36
p1
TPad * p1
Definition: hist-t7.C:116
angle
Double_t angle
Definition: createROOTEnDrcGeo.C:13
vector< PndFtsExpandedTrackCand > PndTrackCombiner::compareAndRefit ( Int_t  layer1,
Int_t  layer2 
)
private

Definition at line 177 of file PndTrackCombiner.cxx.

References Bool_t, fExpandedTrackCands, PndFtsExpandedTrackCand::getLineApproximations(), PndFtsExpandedTrackCand::getSourceHits(), and i.

Referenced by combine().

177  {
178  vector<PndFtsExpandedTrackCand> cands1 = fExpandedTrackCands[layer1];
179  vector<PndFtsExpandedTrackCand> cands2 = fExpandedTrackCands[layer2];
180  vector<PndFtsExpandedTrackCand> result;
181  Bool_t used2[cands2.size()]; //second cand is used
182  for(size_t j=0;j<cands2.size();j++)used2[j]=kFALSE;
183 
184  for(size_t i=0;i<cands1.size();i++){
185  Bool_t used = kFALSE;
186  for(size_t j=0;j<cands2.size();j++){
187  PndFtsExpandedTrackCand cand1 = cands1[i];
188  PndFtsExpandedTrackCand cand2 = cands2[j];
189  vector<PndLineApproximation> a1 = cand1.getLineApproximations();
190  vector<PndLineApproximation> a2 = cand2.getLineApproximations();
191  vector<PndLineApproximation> lines = compareAndRefit(a1,a2);
192  if(lines.size()==0)continue;
193  used = kTRUE;
194  used2[j]=kTRUE;
195  PndFtsExpandedTrackCand etc(cand1.getSourceHits(),cand2.getSourceHits(),lines);
196  result.push_back(etc);
197  }
198  if(!used)result.push_back(cands1[i]);
199  }
200  for(size_t j=0;j<cands2.size();j++){
201  if(!used2[j])result.push_back(cands2[j]);
202  }
203  return result;
204 }
i
Int_t i
Definition: run_full.C:25
PndFtsExpandedTrackCand::getSourceHits
vector< PndFtsHit * > getSourceHits()
Definition: PndFtsExpandedTrackCand.h:26
PndTrackCombiner::compareAndRefit
vector< PndFtsExpandedTrackCand > compareAndRefit(Int_t layer1, Int_t layer2)
Definition: PndTrackCombiner.cxx:177
PndFtsExpandedTrackCand::getLineApproximations
vector< PndLineApproximation > getLineApproximations()
Definition: PndFtsExpandedTrackCand.h:25
PndTrackCombiner::fExpandedTrackCands
map< Int_t, vector< PndFtsExpandedTrackCand > > fExpandedTrackCands
Definition: PndTrackCombiner.h:53
vector< PndLineApproximation > PndTrackCombiner::compareAndRefit ( vector< PndLineApproximation >  l1,
vector< PndLineApproximation >  l2 
)
private

Definition at line 207 of file PndTrackCombiner.cxx.

References PndLineApproximation::getLine(), PndLine::getP1(), i, and PndLineApproximation::newApproximation().

207  {
208  vector<PndLineApproximation> result;
209  for(size_t i=0;i<l1.size();i++){
210  PndLineApproximation approx1 = l1[i];
211  PndLine line1 = approx1.getLine();
212  fComparator.setZValue(line1.getP1()[2]);
213  for(size_t j=0;j<l2.size();j++){
214  PndLineApproximation approx2 = l2[j];
215  PndLine line2 = approx2.getLine();
216  if(fComparator.areEqual2D(line1,line2)){
217  PndLineApproximation newApprox = approx1.newApproximation(approx2);
218  result.push_back(newApprox);
219  return result;
220  }
221  }
222  }
223  return result;
224 }
i
Int_t i
Definition: run_full.C:25
PndLineApproximation::newApproximation
PndLineApproximation newApproximation(PndLineApproximation &approx2)
Definition: PndLineApproximation.cxx:54
PndFtsLineComparator::setZValue
void setZValue(Double_t z)
Definition: PndFtsLineComparator.h:28
PndFtsLineComparator::areEqual2D
Bool_t areEqual2D(PndLine l1, PndLine l2)
Definition: PndFtsLineComparator.cxx:29
PndLine::getP1
TVector3 getP1()
Definition: PndLine.h:32
PndTrackCombiner::fComparator
PndFtsLineComparator fComparator
Definition: PndTrackCombiner.h:54
void PndTrackCombiner::init ( map< Int_t, vector< PndFtsExpandedTrackCand >>  c)
inline

Definition at line 38 of file PndTrackCombiner.h.

References c.

Referenced by PndForwardTrackFinderTask::Exec().

38  {
39  fExpandedTrackCands=c;
40  trackNum=0;
41  firstRun=0;
42  ftsPos[0]=fts1ZPos;
43  ftsPos[1]=fts2ZPos;
44  ftsPos[2]=fts3ZPos;
45  ftsPos[3]=fts4ZPos;
46  ftsPos[4]=fts5ZPos;
47  ftsPos[5]=fts6ZPos;
48  }
PndTrackCombiner::ftsPos
Double_t ftsPos[6]
Definition: PndTrackCombiner.h:72
c
GeV c
Definition: ReadMesDecCorrRes_PiT.C:28
PndTrackCombiner::fExpandedTrackCands
map< Int_t, vector< PndFtsExpandedTrackCand > > fExpandedTrackCands
Definition: PndTrackCombiner.h:53
void PndTrackCombiner::refitLines ( vector< PndLineCombination > &  lines)
private

Definition at line 153 of file PndTrackCombiner.cxx.

References i, and PndLineApproximation::newApproximation().

Referenced by combine().

153  {
154  for(size_t i=0;i<lines.size();i++){
155  PndLineApproximation *a2 = lines[i].getApprox2();
156  if(a2!=0){
157  PndLineApproximation a1 = lines[i].getApprox1();
158  PndLineApproximation newApprox = a1.newApproximation(*a2);
159  lines[i].setApprox1(newApprox);
160  lines[i].setApprox2(0);
161  }
162  }
163 }
i
Int_t i
Definition: run_full.C:25
PndLineApproximation::newApproximation
PndLineApproximation newApproximation(PndLineApproximation &approx2)
Definition: PndLineApproximation.cxx:54

Member Data Documentation

Double_t PndTrackCombiner::angleLimit
private

Definition at line 64 of file PndTrackCombiner.h.

Referenced by combine().

Double_t PndTrackCombiner::compAngle = 2.5
static

Definition at line 33 of file PndTrackCombiner.h.

Referenced by PndForwardTrackFinderTask::PndForwardTrackFinderTask().

Double_t PndTrackCombiner::compDist = 0.7
static

Definition at line 32 of file PndTrackCombiner.h.

Referenced by PndForwardTrackFinderTask::PndForwardTrackFinderTask().

Double_t PndTrackCombiner::compQuali = 600
static

Definition at line 34 of file PndTrackCombiner.h.

Referenced by addLine(), and PndForwardTrackFinderTask::PndForwardTrackFinderTask().

TClonesArray* PndTrackCombiner::corrHits
private

Definition at line 61 of file PndTrackCombiner.h.

PndFtsLineComparator PndTrackCombiner::fComparator
private

Definition at line 54 of file PndTrackCombiner.h.

Referenced by addLine(), addLineOld(), and combine().

map<Int_t, vector<PndFtsExpandedTrackCand> > PndTrackCombiner::fExpandedTrackCands
private

Definition at line 53 of file PndTrackCombiner.h.

Referenced by combine(), and compareAndRefit().

Int_t PndTrackCombiner::firstRun = 0
private

Definition at line 62 of file PndTrackCombiner.h.

Double_t PndTrackCombiner::fts1ZPos = 295.4
private

Definition at line 66 of file PndTrackCombiner.h.

Double_t PndTrackCombiner::fts2ZPos = 327.4
private

Definition at line 67 of file PndTrackCombiner.h.

Double_t PndTrackCombiner::fts3ZPos = 394.5
private

Definition at line 68 of file PndTrackCombiner.h.

Double_t PndTrackCombiner::fts4ZPos = 438.5
private

Definition at line 69 of file PndTrackCombiner.h.

Double_t PndTrackCombiner::fts5ZPos = 607.5
private

Definition at line 70 of file PndTrackCombiner.h.

Double_t PndTrackCombiner::fts6ZPos = 747.5
private

Definition at line 71 of file PndTrackCombiner.h.

Double_t PndTrackCombiner::ftsPos[6]
private

Definition at line 72 of file PndTrackCombiner.h.

Referenced by combine().

TClonesArray* PndTrackCombiner::planes
private

Definition at line 60 of file PndTrackCombiner.h.

Int_t PndTrackCombiner::trackNum
private

Definition at line 63 of file PndTrackCombiner.h.

The documentation for this class was generated from the following files:
Generated on Sun Aug 18 2019 03:04:16 for FairRoot/PandaRoot by   doxygen 1.8.8