Changeset cc8716b in git for classes

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@…>

Location:

classes

Files:

• ClassesLinkDef.h (modified) (2 diffs)
• DelphesClasses.cc (modified) (5 diffs)
• DelphesClasses.h (modified) (3 diffs)
• DelphesCscClusterFormula.cc (added)
• DelphesCscClusterFormula.h (added)
• DelphesHepMC2Reader.cc (modified) (3 diffs)
• DelphesHepMC3Reader.cc (modified) (3 diffs)

classes/ClassesLinkDef.h

@@ -57 +57 @@
 #pragma link C++ class Electron+;
 #pragma link C++ class Muon+;
+#pragma link C++ class CscCluster+;
+
 #pragma link C++ class Jet+;
 #pragma link C++ class Track+;
@@ -66 +68 @@
 
 #endif
-

classes/DelphesClasses.cc

@@ -35 +35 @@
 CompBase *Electron::fgCompare = CompPT<Electron>::Instance();
 CompBase *Muon::fgCompare = CompPT<Muon>::Instance();
+
 CompBase *Jet::fgCompare = CompPT<Jet>::Instance();
 CompBase *Track::fgCompare = CompPT<Track>::Instance();
@@ -42 +43 @@
 CompBase *Vertex::fgCompare = CompSumPT2<Vertex>::Instance();
 CompBase *Candidate::fgCompare = CompMomentumPt<Candidate>::Instance();
+CompBase *CscCluster::fgCompare = CompE<CscCluster>::Instance();
 
 //------------------------------------------------------------------------------
@@ -218 +220 @@
   Position(0.0, 0.0, 0.0, 0.0),
   InitialPosition(0.0, 0.0, 0.0, 0.0),
+  DecayPosition(0.0, 0.0, 0.0, 0.0),
   PositionError(0.0, 0.0, 0.0, 0.0),
   Area(0.0, 0.0, 0.0, 0.0),
@@ -389 +392 @@
   object.Position = Position;
   object.InitialPosition = InitialPosition;
+  object.DecayPosition = DecayPosition;
   object.PositionError = PositionError;
   object.Area = Area;
@@ -527 +531 @@
   Position.SetXYZT(0.0, 0.0, 0.0, 0.0);
   InitialPosition.SetXYZT(0.0, 0.0, 0.0, 0.0);
+  DecayPosition.SetXYZT(0.0, 0.0, 0.0, 0.0);
   Area.SetXYZT(0.0, 0.0, 0.0, 0.0);
   TrackCovariance.Zero();

classes/DelphesClasses.h

@@ -158 +158 @@
   Float_t Z; // particle vertex position (z component) | hepevt.vhep[number][2]
 
+  Float_t decayX;
+  Float_t decayY;
+  Float_t decayZ;
+  Float_t decayT;
+
   static CompBase *fgCompare; //!
   const CompBase *GetCompare() const { return fgCompare; }
@@ -646 +651 @@
   ClassDef(HectorHit, 1)
 };
+//---------------------------------------------------------------------------
+
+class CscCluster: public SortableObject
+{
+public:
+  Float_t Eta; // eta of LLP
+  Float_t Phi; // phi of LLP
+  Float_t PT; // pt of LLP
+  Float_t Px;// px of LLP
+  Float_t Py;// py of LLP
+  Float_t Pz;// pz of LLP
+  Float_t E; // E of LLP
+  Float_t Ehad; // had energy of LLP
+  Float_t Eem; // em energy of LLP
+  Float_t pid; // LLP pid
+  Float_t T; // LLP decay time-photon travel time
+  Float_t X; // LLP decay x
+  Float_t Y; //  LLP decay y
+  Float_t Z; //  LLP decay z
+  Float_t R; //  LLP decay z
+  Float_t beta; // LLP beta
+  Float_t ctau; //LLP ctau
+
+
+  static CompBase *fgCompare; //!
+  const CompBase *GetCompare() const { return fgCompare; }
+
+  ClassDef(CscCluster, 5)
+};
 
 //---------------------------------------------------------------------------
@@ -690 +724 @@
   Float_t DeltaPhi;
 
-  TLorentzVector Momentum, Position, InitialPosition, PositionError, Area;
+  TLorentzVector Momentum, Position, InitialPosition, PositionError, DecayPosition, Area;
 
   Float_t L; // path length

classes/DelphesHepMC2Reader.cc

@@ -438 +438 @@
 {
   Candidate *candidate;
+  Candidate *candidateDaughter;
   map<int, pair<int, int> >::iterator itMotherMap;
   map<int, pair<int, int> >::iterator itDaughterMap;
@@ -446 +447 @@
     candidate = static_cast<Candidate *>(allParticleOutputArray->At(i));
 
+
     if(candidate->M1 > 0)
     {
@@ -477 +479 @@
         candidate->D1 = -1;
         candidate->D2 = -1;
-      }
+        const TLorentzVector &decayPosition = candidate->Position;
+        candidate->DecayPosition.SetXYZT(decayPosition.X(), decayPosition.Y(), decayPosition.Z(), decayPosition.T());// decay position
+     }
       else
       {
         candidate->D1 = itDaughterMap->second.first;
         candidate->D2 = itDaughterMap->second.second;
-      }
-    }
-  }
-}
-
-//---------------------------------------------------------------------------
+        candidateDaughter = static_cast<Candidate *>(allParticleOutputArray->At(candidate->D1));
+        const TLorentzVector &decayPosition = candidateDaughter->Position;
+        candidate->DecayPosition.SetXYZT(decayPosition.X(), decayPosition.Y(), decayPosition.Z(), decayPosition.T());// decay position
+        
+
+      }
+    }
+  }
+}
+
+//---------------------------------------------------------------------------

classes/DelphesHepMC3Reader.cc

@@ -458 +458 @@
   TObjArray *array;
   Candidate *candidate;
+  Candidate *candidateDaughter;
   TParticlePDG *pdgParticle;
   int pdgCode;
@@ -573 +574 @@
         candidate->D1 = -1;
         candidate->D2 = -1;
+        const TLorentzVector &decayPosition = candidate->Position;
+        candidate->DecayPosition.SetXYZT(decayPosition.X(), decayPosition.Y(), decayPosition.Z(), decayPosition.T());// decay position
       }
       else
@@ -578 +581 @@
         candidate->D1 = itDaughterMap->second.first;
         candidate->D2 = itDaughterMap->second.second;
-      }
-    }
-  }
-}
-
-//---------------------------------------------------------------------------
+        candidateDaughter = static_cast<Candidate *>(allParticleOutputArray->At(candidate->D1));
+        const TLorentzVector &decayPosition = candidateDaughter->Position;
+        candidate->DecayPosition.SetXYZT(decayPosition.X(), decayPosition.Y(), decayPosition.Z(), decayPosition.T());// decay position
+      }
+    }
+  }
+}
+
+//---------------------------------------------------------------------------