Changeset a5af1df in git for modules/TreeWriter.cc

Timestamp:

Dec 9, 2021, 7:38:54 AM (3 years ago)

Author:

christinaw97 <christina.wang@…>

Children:

d612dec

Parents:

a5ff49f

Message:

added CscCluster modules and classes

File:

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

modules/TreeWriter.cc

@@ -75 +75 @@
   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;
@@ -182 +183 @@
     const TLorentzVector &momentum = candidate->Momentum;
     const TLorentzVector &position = candidate->Position;
+    const TLorentzVector &DecayPosition = candidate->DecayPosition;
 
     entry = static_cast<GenParticle *>(branch->NewEntry());
@@ -215 +217 @@
     entry->D0 = candidate->D0;
     entry->DZ = candidate->DZ;
-    entry->P = candidate->P;
+    entry->P = momentum.P();
     entry->PT = candidate->PT;
     entry->CtgTheta = candidate->CtgTheta;
@@ -230 +232 @@
     entry->Z = position.Z();
     entry->T = position.T() * 1.0E-3 / c_light;
+
+    entry->decayX = DecayPosition.X();
+    entry->decayY = DecayPosition.Y();
+    entry->decayZ = DecayPosition.Z();
+    entry->decayT = DecayPosition.T()* 1.0E-3 / c_light;
+    float beta = entry->P/momentum.E();
+    float gamma = 1./sqrt(1-beta*beta);
+    entry->ctau = sqrt(pow(entry->decayX-entry->X,2)+pow(entry->decayY-entry->Y,2)+pow(entry->decayZ-entry->Z,2))/(beta*gamma);// in millimeter
   }
 }
@@ -730 +740 @@
   }
 }
+//------------------------------------------------------------------------------
+
+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->Eta = position.Eta();
+    // entry->Phi = position.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 pid
+    entry->Ehad = candidate->Ehad; // LLP pid
+    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 pid
+
+    // entry->T = (position.T()-sqrt(pow(position.X(),2)+pow(position.Y(),2)+pow(position.Z(),2)))* 1.0E-3 / c_light; // LLP decay time-photon travel time
+
+    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
+    // entry->Size = 100;
+  }
+}
 
 //------------------------------------------------------------------------------