Changeset e999d62 in git

Timestamp:

Dec 5, 2017, 8:45:56 PM (7 years ago)

Author:

Michele Selvaggi <michele.selvaggi@…>

Branches:

ImprovedOutputFile, Timing, dual_readout, llp, master

Children:

8562e4e

Parents:

6d1fff5

Message:

subtract merged tracks from existing trk coll

Location:

modules

Files:

• DenseTrackFilter.cc (modified) (9 diffs)
• DenseTrackFilter.h (modified) (1 diff)

modules/DenseTrackFilter.cc

@@ -54 +54 @@
 
 DenseTrackFilter::DenseTrackFilter() :
-  fItTrackInputArray(0)
+  fItTrackInputArray(0),   fItDenseChargedInputArray(0)
 {
 }
@@ -113 +113 @@
   fEtaPhiRes = GetDouble("EtaPhiRes", 0.005);
 
-  fTrackInputArray = ImportArray(GetString("TrackInputArray", "ParticlePropagator/tracks"));
+  fTrackInputArray = ImportArray(GetString("TrackInputArray", "TrackMerger/tracks"));
   fItTrackInputArray = fTrackInputArray->MakeIterator();
+
+  fDenseChargedInputArray = ImportArray(GetString("DenseChargedInputArray", "DenseMergeTracks/tracks"));
+  fItDenseChargedInputArray = fDenseChargedInputArray->MakeIterator();
 
   fTrackOutputArray = ExportArray(GetString("TrackOutputArray", "tracks"));
@@ -125 +128 @@
   vector< vector< Double_t >* >::iterator itPhiBin;
   if(fItTrackInputArray) delete fItTrackInputArray;
+  if(fItDenseChargedInputArray) delete fItDenseChargedInputArray;
   for(itPhiBin = fPhiBins.begin(); itPhiBin != fPhiBins.end(); ++itPhiBin)
   {
@@ -135 +139 @@
 void DenseTrackFilter::Process()
 {
-  Candidate *candidate, *track, *bestTrack, *mother;
+  Candidate *candidate, *track, *bestTrack, *mother, *trackRef, *bestTrackRef;
   TLorentzVector position, momentum;
   Short_t etaBin, phiBin, flags;
@@ -142 +146 @@
 
   Double_t pt, ptmax, eta, phi;
+  Bool_t matched;
 
   vector< Double_t >::iterator itEtaBin;
@@ -152 +157 @@
 
   // loop over all tracks
-  fItTrackInputArray->Reset();
+  fItDenseChargedInputArray->Reset();
   number = -1;
-  while((track = static_cast<Candidate*>(fItTrackInputArray->Next())))
+  while((track = static_cast<Candidate*>(fItDenseChargedInputArray->Next())))
   {
     const TLorentzVector &trackPosition = track->Position;
@@ -205 +210 @@
       if(towerTrackHits > 0)
       {
-         mother = bestTrack;
-         candidate = static_cast<Candidate*>(bestTrack->Clone());
-         pt = candidate->Momentum.Pt();
-         eta = candidate->Momentum.Eta();
-         phi = candidate->Momentum.Phi();
-         eta = gRandom->Gaus(eta, fEtaPhiRes);
-         phi = gRandom->Gaus(phi, fEtaPhiRes);
-         candidate->Momentum.SetPtEtaPhiE(pt, eta, phi, pt*TMath::CosH(eta));
-         candidate->AddCandidate(mother);
-         fTrackOutputArray->Add(candidate);
+         bestTrackRef = static_cast<Candidate*>(bestTrack->GetCandidates()->At(0));
+         
+         // find corresponding track in properly propagated tracks
+         
+         
+         fItTrackInputArray->Reset();
+         matched = kFALSE;
+         int ntrack = 0;
+         while((track = static_cast<Candidate*>(fItTrackInputArray->Next())))
+         {
+           ntrack++;
+           trackRef = static_cast<Candidate*>(track->GetCandidates()->At(0));
+           if (trackRef->GetUniqueID() == bestTrackRef->GetUniqueID())
+           {
+             mother = track;
+             candidate = static_cast<Candidate*>(track->Clone());
+             pt = candidate->Momentum.Pt();
+             eta = candidate->Momentum.Eta();
+             phi = candidate->Momentum.Phi();
+             eta = gRandom->Gaus(eta, fEtaPhiRes);
+             phi = gRandom->Gaus(phi, fEtaPhiRes);
+             candidate->Momentum.SetPtEtaPhiE(pt, eta, phi, pt*TMath::CosH(eta));
+             candidate->AddCandidate(mother);
+             fTrackOutputArray->Add(candidate);
+             matched = kTRUE;
+           }
+           if (matched) break;
+         }
+         
+         // find track 
       }
 
@@ -226 +251 @@
     {
       ++towerTrackHits;
-      track = static_cast<Candidate*>(fTrackInputArray->At(number));
+      track = static_cast<Candidate*>(fDenseChargedInputArray->At(number));
       momentum = track->Momentum;
 
@@ -237 +262 @@
     }
   }
-}
+  
+}

modules/DenseTrackFilter.h

@@ -70 +70 @@
   TObjArray *fTrackOutputArray; //!
 
+
+  const TObjArray *fDenseChargedInputArray;
+  TIterator *fItDenseChargedInputArray;
+
   ClassDef(DenseTrackFilter, 1)
 };