Changeset cc8716b in git for modules/TreeWriter.cc

Timestamp:

Mar 7, 2022, 8:51:45 PM (3 years ago)

Author:

GitHub <noreply@…>

Branches:

master

Children:

7dbc149

Parents:

4e8e72b

git-author:

Christinaw97 <cwang5@…> (03/07/22 20:51:45)

git-committer:

GitHub <noreply@…> (03/07/22 20:51:45)

Message:

Update to handle CMS endcap muon detector showers for long-lived particles (#103)

• add CscCluster modules and classes
• add DelphesCscClusterFormula
• add example/ExampleCscCluster.py

Co-authored-by: christinaw97 <christina.wang@…>

File:

• modules/TreeWriter.cc (modified) (3 diffs)

modules/TreeWriter.cc

@@ -77 +77 @@
   fClassMap[Electron::Class()] = &TreeWriter::ProcessElectrons;
   fClassMap[Muon::Class()] = &TreeWriter::ProcessMuons;
+  fClassMap[CscCluster::Class()] = &TreeWriter::ProcessCscCluster;
   fClassMap[Jet::Class()] = &TreeWriter::ProcessJets;
   fClassMap[MissingET::Class()] = &TreeWriter::ProcessMissingET;
@@ -251 +252 @@
     entry->Z = position.Z();
     entry->T = position.T() * 1.0E-3 / c_light;
+
   }
 }
@@ -907 +909 @@
   }
 }
+//------------------------------------------------------------------------------
+
+void TreeWriter::ProcessCscCluster(ExRootTreeBranch *branch, TObjArray *array)
+{
+  TIter iterator(array);
+  Candidate *candidate = 0;
+  CscCluster *entry = 0;
+  Double_t pt, signPz, cosTheta, eta, rapidity;
+
+  const Double_t c_light = 2.99792458E8; // in unit of m/s
+
+  array->Sort();
+
+
+  // loop over all clusters
+  iterator.Reset();
+  while((candidate = static_cast<Candidate *>(iterator.Next())))
+  {
+    const TLorentzVector &momentum = candidate->Momentum;
+    const TLorentzVector &position = candidate->DecayPosition;
+
+    pt = momentum.Pt();
+    cosTheta = TMath::Abs(momentum.CosTheta());
+    signPz = (momentum.Pz() >= 0.0) ? 1.0 : -1.0;
+    eta = (cosTheta == 1.0 ? signPz * 999.9 : momentum.Eta());
+
+    entry = static_cast<CscCluster *>(branch->NewEntry());
+
+    entry->SetBit(kIsReferenced);
+    entry->SetUniqueID(candidate->GetUniqueID());
+
+    entry->Eta = eta;
+    entry->Phi = momentum.Phi();
+
+    entry->PT = momentum.Pt(); // pt of LLP
+    entry->Px = momentum.Px();// px of LLP
+    entry->Py = momentum.Py();// py of LLP
+    entry->Pz = momentum.Pz();// pz of LLP
+    entry->E = momentum.E(); // E of LLP
+    entry->pid = candidate->PID; // LLP pid
+    entry->Eem = candidate->Eem; // LLP Eem
+    entry->Ehad = candidate->Ehad; // LLP Ehad
+    Double_t beta = momentum.P()/momentum.E();
+    Double_t gamma = 1.0/sqrt(1-beta*beta);
+    Double_t decayDistance = sqrt(pow(position.X(),2)+pow(position.Y(),2)+pow(position.Z(),2)); // mm
+    entry->beta = beta; // LLP pid
+    entry->ctau = decayDistance/(beta * gamma); // LLP travel time in its rest frame
+    entry->T = decayDistance*(1./beta-1)* 1.0E-3/c_light*1e9; // ns
+    entry->X = position.X(); // LLP decay x
+    entry->Y = position.Y(); //  LLP decay y
+    entry->Z = position.Z(); //  LLP decay z
+  }
+}
 
 //------------------------------------------------------------------------------