Changeset c7e90d8 in git

Timestamp:

Jan 26, 2020, 3:26:13 PM (5 years ago)

Author:

michele <michele.selvaggi@…>

Branches:

Timing

Children:

9e3f2fb

Parents:

62764fb

Message:

fixed PileUpSubtractor4D

Location:

modules

Files:

• ModulesLinkDef.h (modified) (2 diffs)
• PileUpSubtractor4D.cc (modified) (8 diffs)
• PileUpSubtractor4D.h (modified) (2 diffs)

modules/ModulesLinkDef.h

@@ -56 +56 @@
 #include "modules/PileUpMerger.h"
 #include "modules/JetPileUpSubtractor.h"
-#include "modules/TrackTimingPileUpSubtractor.h"
+#include "modules/TrackPileUpSubtractor.h"
 #include "modules/TaggingParticlesSkimmer.h"
 #include "modules/PileUpJetID.h"
+#include "modules/PileUpSubtractor4D.h"
 #include "modules/PhotonID.h"
 #include "modules/ConstituentFilter.h"
@@ -113 +114 @@
 #pragma link C++ class PileUpMerger+;
 #pragma link C++ class JetPileUpSubtractor+;
-#pragma link C++ class TrackTimingPileUpSubtractor+;
+#pragma link C++ class TrackPileUpSubtractor+;
 #pragma link C++ class TaggingParticlesSkimmer+;
 #pragma link C++ class PileUpJetID+;
+#pragma link C++ class PileUpSubtractor4D+;
 #pragma link C++ class PhotonID+;
 #pragma link C++ class ConstituentFilter+;

modules/PileUpSubtractor4D.cc

@@ -17 +17 @@
  */
 
-/** \class TrackPileUpSubtractor
+/** \class PileUpSubtractor4D
  *
- *  Subtract pile-up contribution from tracks.
+ *  Subtract pile-up contribution from tracks using both spatial and timing
+ *  information
  *
- *  \author P. Demin - UCL, Louvain-la-Neuve
+ *  \author M. Selvaggi - CERN
  *
  */
@@ -52 +53 @@
 //------------------------------------------------------------------------------
 
-PileUpSubtractor4D::PileUpSubtractor4D() :
-  fFormula(0)
+PileUpSubtractor4D::PileUpSubtractor4D()
 {
-  fFormula = new DelphesFormula;
 }
 
@@ -62 +61 @@
 PileUpSubtractor4D::~PileUpSubtractor4D()
 {
-  if(fFormula) delete fFormula;
 }
 
@@ -107 +105 @@
   {
     iterator = itInputMap->first;
-
     if(iterator) delete iterator;
   }
@@ -118 +115 @@
 void PileUpSubtractor4D::Process()
 {
-  Candidate *candidate, *particle;
+  Candidate *candidate;
   map<TIterator *, TObjArray *>::iterator itInputMap;
   TIterator *iterator;
   TObjArray *array;
   Double_t z, zvtx = 0;
-  Double_t z_err, zvtx_err = 0;
+  Double_t zvtx_err = 0;
   Double_t t, tvtx = 0;
-  Double_t t_err, tvtx_err = 0;
+  Double_t tvtx_err = 0;
   Double_t sumPTSquare = 0;
   Double_t tempPTSquare = 0;
-  Double_t pt, eta, phi, e;
   Double_t distanceCharged, distanceNeutral = 0;
 
@@ -144 +140 @@
       tvtx = candidate->Position.T();
       tvtx_err = candidate->PositionError.T();
-    } 
+    }
   }
 
@@ -157 +153 @@
     while((candidate = static_cast<Candidate *>(iterator->Next())))
     {
-      particle = static_cast<Candidate *>(candidate->GetCandidates()->At(0));
-      const TLorentzVector &candidateMomentum = particle->Momentum;
 
-      eta = candidateMomentum.Eta();
-      pt = candidateMomentum.Pt();
-      phi = candidateMomentum.Phi();
-      e = candidateMomentum.E();
-
-      z = particle->Position.Z();
-      z_err = particle->PositionError.Z();
-      t = particle->InitialPosition.T();
-      t_err = particle->PositionError.T();
+      z = candidate->Position.Z();
+      t = candidate->InitialPosition.T();
 
       distanceCharged = TMath::Sqrt(pow((zvtx - z),2)/pow((zvtx_err),2) + pow((tvtx - t),2)/pow((tvtx_err),2));
@@ -176 +163 @@
       {
         candidate->IsRecoPU = 0;
+        if(candidate->Momentum.Pt() > fPTMin) array->Add(candidate);
       }
       else if(candidate->Charge == 0 && distanceNeutral < fNeutralMinSignificance)
       {
         candidate->IsRecoPU = 0;
-      }  
+        if(candidate->Momentum.Pt() > fPTMin) array->Add(candidate);
+      }
       else
       {
         candidate->IsRecoPU = 1;
-        if(candidate->Momentum.Pt() > fPTMin) array->Add(candidate);
       }
     }

modules/PileUpSubtractor4D.h

@@ -34 +34 @@
 class TIterator;
 class TObjArray;
-class DelphesFormula;
 
 class PileUpSubtractor4D: public DelphesModule
@@ -47 +46 @@
 
 private:
-  DelphesFormula *fFormula; //!
 
   Double_t fChargedMinSignificance;
   Double_t fNeutralMinSignificance;
-
   Double_t fPTMin;
 
   std::map<TIterator *, TObjArray *> fInputMap; //!
 
+  TIterator *fItVertexInputArray; //!
+
+  const TObjArray *fVertexInputArray; //!
+
   ClassDef(PileUpSubtractor4D, 1)
 
-    TIterator *fItVertexInputArray; //!
-
-  const TObjArray *fVertexInputArray; //!
 };