Changeset ebf40fd in git for modules/TrackCovariance.cc

Timestamp:

Nov 29, 2021, 3:18:22 PM (3 years ago)

Author:

Franco BEDESCHI <bed@…>

Branches:

master

Children:

bd376e3

Parents:

9a7ea36

Message:

Major update to handle highly displaced tracks

File:

• modules/TrackCovariance.cc (modified) (2 diffs)

modules/TrackCovariance.cc

@@ -103 +103 @@
   Double_t mass, p, pt, q, ct;
   Double_t dd0, ddz, dphi, dct, dp, dpt, dC;
+  //
+  // Get cylindrical box for fast track simulation
+  //
+  Double_t Rin = fGeometry->GetRmin();
+  Double_t ZinPos = fGeometry->GetZminPos();
+  Double_t ZinNeg = fGeometry->GetZminNeg();
 
   fItInputArray->Reset();
@@ -115 +121 @@
     const TLorentzVector &candidateMomentum = particle->Momentum;
 
-    if ( !fCovariance->IsAccepted(candidateMomentum.Vect()) ) continue;
+    Bool_t inside = TrkUtil::IsInside(candidatePosition.Vect(), Rin, ZinNeg, ZinPos); // Check if in inner box
+    Bool_t Accept = kTRUE;
+    if(inside) Accept = fCovariance->IsAccepted(candidateMomentum.Vect());
+    else       Accept = fCovariance->IsAccepted(candidatePosition.Vect(),candidateMomentum.Vect(), fGeometry);
+    if(!Accept) continue;
 
     mass = candidateMomentum.M();
 
-    ObsTrk track(candidatePosition.Vect(), candidateMomentum.Vect(), candidate->Charge, fBz, fCovariance);
+    ObsTrk track(candidatePosition.Vect(), candidateMomentum.Vect(), candidate->Charge, fCovariance, fGeometry);
 
     mother    = candidate;