Changeset 4aec383 in git

Timestamp:

Jan 12, 2022, 9:07:49 PM (3 years ago)

Author:

christinaw97 <christina.wang@…>

Children:

2a1d95e

Parents:

157fe13

Message:

remove hard-coded portions of CscClusterId to card

Files:

• cards/delphes_card_CMS_CSCCluster.tcl (modified) (2 diffs)
• modules/CscClusterId.cc (modified) (6 diffs)
• modules/CscClusterId.h (modified) (1 diff)
• modules/LLPFilter.cc (modified) (1 diff)
• modules/LLPFilter.h (modified) (1 diff)

cards/delphes_card_CMS_CSCCluster.tcl

@@ -623 +623 @@
   set OutputArray cluster
 
+  set EtaCutMax 1.9
+
   # efficiency formula for Csc Cluster, as a function of LLP decay vertex in R, Z and hadronic and EM energy
   set EfficiencyFormula {
@@ -650 +652 @@
     (Ehad >= 200.0)*(0.2404))
   }
+  set EtaCutFormula {
+    (decayZ < 6320 || (decayZ < 7240 && decayR > 2750)) * (1.8) +
+    ( !(decayZ < 6320 || (decayZ < 7240 && decayR > 2750)) && decayZ < 9700) * (1.6) +
+    (decayZ >= 9700) * (1.8)
+  }
+
 }
 ######################

modules/CscClusterId.cc

@@ -21 +21 @@
   *  This module is specific to the CMS paper searching for neutral LLPs in the CMS endcap muon detectors: https://arxiv.org/abs/2107.04838
   *  It is implemented based on the cut_based_id.py function provided in the HEPData entry of the paper: https://www.hepdata.net/record/104408
+  *  to reproduce the cut-based ID efficiency of the CMS paper.
   *
   *  \author Christina Wang
@@ -48 +49 @@
 #include <sstream>
 #include <stdexcept>
-
+#include "assert.h"
 using namespace std;
 
@@ -54 +55 @@
 
 CscClusterId::CscClusterId() :
-  fFormula(0), fItInputArray(0)
+  fFormula(0), fEtaFormula(0), fItInputArray(0)
 {
   fFormula = new DelphesCscClusterFormula;
+  fEtaFormula = new DelphesCscClusterFormula;
 }
 
@@ -64 +66 @@
 {
   if(fFormula) delete fFormula;
+  if(fEtaFormula) delete fEtaFormula;
 }
 
@@ -73 +76 @@
 
   fFormula->Compile(GetString("EfficiencyFormula", "1.0"));
+  fEtaFormula->Compile(GetString("EtaCutFormula", "1.0"));
+  fEtaCutMax = GetDouble("EtaCutMax", 999.0);
 
   // import input array
@@ -114 +119 @@
     eta = (cosTheta == 1.0 ? signPz * 999.9 : momentum.Eta());
 
+    // calculate the NStation > 1 efficiency, implemented according to Additional Figure 8 in HEPData
     NStationEff = fFormula->Eval(decayR, decayZ, Ehad);
 
-    // assign average station for the cluster
-    if (decayZ < 6320) avgStation = 1;
-    else if (decayZ < 7240 && decayR > 2750)avgStation = 1;
-    else if (decayZ < 8500) avgStation = 2;
-    else if (decayZ < 9700) avgStation = 3;
-    else avgStation = 4;
-
-    // if NStation == 1, different eta cut is applied
-    if (avgStation == 1) eta_cut = 1.8;
-    else if (avgStation == 2) eta_cut = 1.6;
-    else if (avgStation == 3) eta_cut = 1.6;
-    else if (avgStation == 4) eta_cut = 1.8;
-    if(gRandom->Uniform() > NStationEff*(abs(eta)<1.9)+(1.0-NStationEff)*(abs(eta)<eta_cut)) continue;
+    // depending on the decay region (station Number), different eta cut is applied, implemented based on cut_based_id.py in HEPData
+    float eta_cut = fEtaFormula->Eval(decayR, decayZ);
+    if(gRandom->Uniform() > NStationEff*(abs(eta)<fEtaCutMax)+(1.0-NStationEff)*(abs(eta)<eta_cut)) continue;
 
     fOutputArray->Add(candidate);

modules/CscClusterId.h

@@ -46 +46 @@
 private:
   DelphesCscClusterFormula *fFormula; //!
+  DelphesCscClusterFormula *fEtaFormula; //!
+  Double_t fEtaCutMax;
 
   TIterator *fItInputArray; //!

modules/LLPFilter.cc

@@ -81 +81 @@
   fRequireDecayRegion = GetBool("RequireDecayRegion", 0);
 
-  fDecayRegionRMax = GetInt("DecayRegionRMax", 0); //mm
-  fDecayRegionRMin = GetInt("DecayRegionRMin", 0); //mm
-  fDecayRegionZMax = GetInt("DecayRegionZMax", 0); //mm
-  fDecayRegionZMin = GetInt("DecayRegionZMin", 0); //mm
-  fDecayRegionEtaMax = GetInt("DecayRegionEtaMax", 0); // requirement on abs(eta)
-  fDecayRegionEtaMin = GetInt("DecayRegionEtaMin", 0); //requirement on abs(eta)
+  fDecayRegionRMax = GetDouble("DecayRegionRMax", 0.0); //mm
+  fDecayRegionRMin = GetDouble("DecayRegionRMin", 0.0); //mm
+  fDecayRegionZMax = GetDouble("DecayRegionZMax", 0.0); //mm
+  fDecayRegionZMin = GetDouble("DecayRegionZMin", 0.0); //mm
+  fDecayRegionEtaMax = GetDouble("DecayRegionEtaMax", 0.0); // requirement on abs(eta)
+  fDecayRegionEtaMin = GetDouble("DecayRegionEtaMin", 0.0); //requirement on abs(eta)
 
 

modules/LLPFilter.h

@@ -52 +52 @@
   Double_t fPTMin; //!
   Bool_t fRequireDecayRegion;
-  Int_t fDecayRegionRMax;
-  Int_t fDecayRegionRMin;
-  Int_t fDecayRegionZMax;
-  Int_t fDecayRegionZMin;
-  Int_t fDecayRegionEtaMax;
-  Int_t fDecayRegionEtaMin;
+  Double_t fDecayRegionRMax;
+  Double_t fDecayRegionRMin;
+  Double_t fDecayRegionZMax;
+  Double_t fDecayRegionZMin;
+  Double_t fDecayRegionEtaMax;
+  Double_t fDecayRegionEtaMin;
   Int_t fDaughterNumber;
   Bool_t fInvert; //!