Changeset 1349 in svn for trunk

Timestamp:

Jan 17, 2014, 2:29:22 PM (11 years ago)

Author:

Michele Selvaggi

Message:

JetId v2

Location:

trunk

Files:

• examples/delphes_card_CMS_PileUp.tcl (modified) (4 diffs)
• modules/PileUpJetID.cc (modified) (6 diffs)
• modules/PileUpJetID.h (modified) (1 diff)
• modules/TrackPileUpSubtractor.cc (modified) (2 diffs)

trunk/examples/delphes_card_CMS_PileUp.tcl

@@ -25 +25 @@
   Rho
   FastJetFinder
+  PileUpJetID
   JetPileUpSubtractor
 
@@ -61 +62 @@
 
   # pre-generated minbias input file
-  set PileUpFile MinBias.pileup
+  set PileUpFile ../../Delphes/MinBias.pileup
 
   # average expected pile up
@@ -325 +326 @@
   add InputArray ElectronEnergySmearing/electrons electrons
   add InputArray MuonMomentumSmearing/muons muons
-
+  
+  set VertexInputArray PileUpMerger/vertices
   # assume perfect pile-up subtraction for tracks with |z| > fZVertexResolution
   # Z vertex resolution in m
@@ -404 +406 @@
 
 ###########################
+# Jet Pile-Up ID
+###########################
+
+module PileUpJetID PileUpJetID {
+  set JetInputArray FastJetFinder/jets
+  set TrackInputArray Calorimeter/eflowTracks
+  set NeutralInputArray Calorimeter/eflowTowers
+
+  set VertexInputArray PileUpMerger/vertices
+  # assume perfect pile-up subtraction for tracks with |z| > fZVertexResolution
+  # Z vertex resolution in m
+  set ZVertexResolution 0.0001
+  
+  set OutputArray jets
+
+  set UseConstituents 0
+  set ParameterR 0.5
+
+  set JetPTMin 20.0
+}
+
+###########################
 # Jet Pile-Up Subtraction
 ###########################
 
 module JetPileUpSubtractor JetPileUpSubtractor {
-  set JetInputArray FastJetFinder/jets
+  set JetInputArray PileUpJetID/jets
   set RhoInputArray Rho/rho
 

trunk/modules/PileUpJetID.cc

@@ -36 +36 @@
 
 PileUpJetID::PileUpJetID() :
-  fItJetInputArray(0),fTrackInputArray(0),fNeutralInputArray(0)
+  fItJetInputArray(0),fTrackInputArray(0),fNeutralInputArray(0),fItVertexInputArray(0) 
 {
 
@@ -68 +68 @@
   fNeutralInputArray = ImportArray(GetString("NeutralInputArray", "Calorimeter/eflowTowers"));
   fItNeutralInputArray = fNeutralInputArray->MakeIterator();
-
-  // create output array(s)
+  
+  fVertexInputArray = ImportArray(GetString("VertexInputArray", "PileUpMerger/vertices"));
+  fItVertexInputArray = fVertexInputArray->MakeIterator();
+  
+  fZVertexResolution  = GetDouble("ZVertexResolution", 0.005)*1.0E3;
+// create output array(s)
 
   fOutputArray = ExportArray(GetString("OutputArray", "jets"));
@@ -83 +87 @@
   if(fItTrackInputArray) delete fItTrackInputArray;
   if(fItNeutralInputArray) delete fItNeutralInputArray;
+  if(fItVertexInputArray) delete fItVertexInputArray;
 
 }
@@ -92 +97 @@
   Candidate *candidate, *constituent;
   TLorentzVector momentum, area;
+  Double_t zvtx=0;
 
   Candidate *trk;
+
+ // find z position of primary vertex
+   
+  fItVertexInputArray->Reset();
+  while((candidate = static_cast<Candidate*>(fItVertexInputArray->Next())))
+  {
+    if(!candidate->IsPU)
+    {
+    zvtx = candidate->Position.Z();
+    break;
+    }
+  }
 
   // loop over all input candidates
@@ -129 +147 @@
         } else {
           // charged
-          if (constituent->IsPU) {
+          if (constituent->IsPU && TMath::Abs(constituent->Position.Z()-zvtx) > fZVertexResolution) {
             sumptchpu += pt;
           } else {
@@ -146 +164 @@
       // Not using constituents, using dr
       fItTrackInputArray->Reset();
-      while ((trk = static_cast<Candidate*>(fItTrackInputArray->Next()))) {
+       while ((trk = static_cast<Candidate*>(fItTrackInputArray->Next()))) {
         if (trk->Momentum.DeltaR(candidate->Momentum) < fParameterR) {
           float pt = trk->Momentum.Pt();
           sumpt += pt;
           sumptch += pt;
-          if (trk->IsPU) {
+          if (trk->IsPU && TMath::Abs(trk->Position.Z()-zvtx) > fZVertexResolution) {
             sumptchpu += pt;
           } else {

trunk/modules/PileUpJetID.h

@@ -53 +53 @@
 
   TObjArray *fOutputArray; //!
+  
+  TIterator *fItVertexInputArray; //!
+  const TObjArray *fVertexInputArray; //!
 
+  Double_t fZVertexResolution;
 
   ClassDef(PileUpJetID, 1)

trunk/modules/TrackPileUpSubtractor.cc

@@ -110 +110 @@
   while((candidate = static_cast<Candidate*>(fItVertexInputArray->Next())))
   {
-    if(candidate->IsPU == 0)
+    if(!candidate->IsPU)
     {
     zvtx = candidate->Position.Z();
-    break;
+    // break;
     }
   }
-
 
   // loop over all input arrays
@@ -130 +129 @@
       particle = static_cast<Candidate*>(candidate->GetCandidates()->At(0));
       z = particle->Position.Z();
-
+      
       // apply pile-up subtraction
       // assume perfect pile-up subtraction for tracks outside fZVertexResolution
+      
       if(candidate->IsPU && TMath::Abs(z-zvtx) > fZVertexResolution) continue;