Changeset 1361 in svn for trunk/modules/TreeWriter.cc

Timestamp:

Apr 8, 2014, 6:43:52 PM (11 years ago)

Author:

Pavel Demin

Message:

add HectorHit class and corresponding ROOT tree branch

File:

• trunk/modules/TreeWriter.cc (modified) (22 diffs)

trunk/modules/TreeWriter.cc

@@ -67 +67 @@
   fClassMap[Rho::Class()] = &TreeWriter::ProcessRho;
   fClassMap[Weight::Class()] = &TreeWriter::ProcessWeight;
+  fClassMap[HectorHit::Class()] = &TreeWriter::ProcessHectorHit;
 
   TBranchMap::iterator itBranchMap;
@@ -95 +96 @@
       continue;
     }
-    
+
     itClassMap = fClassMap.find(branchClass);
     if(itClassMap == fClassMap.end())
@@ -160 +161 @@
   GenParticle *entry = 0;
   Double_t pt, signPz, cosTheta, eta, rapidity;
-  
-  const Double_t c_light = 2.99792458E8;
-  
+
+  const Double_t c_light = 2.99792458E8;
+
   // loop over all particles
   iterator.Reset();
@@ -220 +221 @@
   Candidate *candidate = 0;
   Vertex *entry = 0;
-  
+
   const Double_t c_light = 2.99792458E8;
 
@@ -248 +249 @@
   Double_t pt, signz, cosTheta, eta, rapidity;
   const Double_t c_light = 2.99792458E8;
-  
+
   // loop over all tracks
   iterator.Reset();
@@ -310 +311 @@
   Double_t pt, signPz, cosTheta, eta, rapidity;
   const Double_t c_light = 2.99792458E8;
-  
+
   // loop over all towers
   iterator.Reset();
@@ -317 +318 @@
     const TLorentzVector &momentum = candidate->Momentum;
     const TLorentzVector &position = candidate->Position;
-    
+
     pt = momentum.Pt();
     cosTheta = TMath::Abs(momentum.CosTheta());
@@ -339 +340 @@
     entry->Edges[2] = candidate->Edges[2];
     entry->Edges[3] = candidate->Edges[3];
-    
+
     entry->T = position.T()*1.0E-3/c_light;
-    
+
     FillParticles(candidate, &entry->Particles);
   }
@@ -355 +356 @@
   Double_t pt, signPz, cosTheta, eta, rapidity;
   const Double_t c_light = 2.99792458E8;
-  
+
   array->Sort();
 
@@ -365 +366 @@
     const TLorentzVector &momentum = candidate->Momentum;
     const TLorentzVector &position = candidate->Position;
-    
+
 
     pt = momentum.Pt();
@@ -379 +380 @@
     entry->PT = pt;
     entry->E = momentum.E();
-    
+
     entry->T = position.T()*1.0E-3/c_light;
-   
+
     entry->EhadOverEem = candidate->Eem > 0.0 ? candidate->Ehad/candidate->Eem : 999.9;
 
@@ -397 +398 @@
   Double_t pt, signPz, cosTheta, eta, rapidity;
   const Double_t c_light = 2.99792458E8;
-  
+
   array->Sort();
 
@@ -406 +407 @@
     const TLorentzVector &momentum = candidate->Momentum;
     const TLorentzVector &position = candidate->Position;
-    
+
     pt = momentum.Pt();
     cosTheta = TMath::Abs(momentum.CosTheta());
@@ -418 +419 @@
     entry->Phi = momentum.Phi();
     entry->PT = pt;
-    
+
     entry->T = position.T()*1.0E-3/c_light;
-    
+
     entry->Charge = candidate->Charge;
 
@@ -437 +438 @@
   Muon *entry = 0;
   Double_t pt, signPz, cosTheta, eta, rapidity;
-  
-  const Double_t c_light = 2.99792458E8;
-  
+
+  const Double_t c_light = 2.99792458E8;
+
   array->Sort();
 
@@ -448 +449 @@
     const TLorentzVector &momentum = candidate->Momentum;
     const TLorentzVector &position = candidate->Position;
-    
+
 
     pt = momentum.Pt();
@@ -466 +467 @@
 
     entry->T = position.T()*1.0E-3/c_light;
-    
+
    // cout<<entry->PT<<","<<entry->T<<endl;
 
@@ -485 +486 @@
   Double_t ecalEnergy, hcalEnergy;
   const Double_t c_light = 2.99792458E8;
-  
+
   array->Sort();
 
@@ -493 +494 @@
   {
     TIter itConstituents(candidate->GetCandidates());
-    
-    const TLorentzVector &momentum = candidate->Momentum;
-    const TLorentzVector &position = candidate->Position;
-   
+
+    const TLorentzVector &momentum = candidate->Momentum;
+    const TLorentzVector &position = candidate->Position;
+
     pt = momentum.Pt();
     cosTheta = TMath::Abs(momentum.CosTheta());
@@ -510 +511 @@
 
     entry->T = position.T()*1.0E-3/c_light;
-    
+
     entry->Mass = momentum.M();
 
@@ -533 +534 @@
 
     entry->EhadOverEem = ecalEnergy > 0.0 ? hcalEnergy/ecalEnergy : 999.9;
-  
+
     //---   Pile-Up Jet ID variables ----
 
@@ -632 +633 @@
 //------------------------------------------------------------------------------
 
+void TreeWriter::ProcessHectorHit(ExRootTreeBranch *branch, TObjArray *array)
+{
+  TIter iterator(array);
+  Candidate *candidate = 0;
+  HectorHit *entry = 0;
+
+  // loop over all roman pot hits
+  iterator.Reset();
+  while((candidate = static_cast<Candidate*>(iterator.Next())))
+  {
+    const TLorentzVector &position = candidate->Position;
+    const TLorentzVector &momentum = candidate->Momentum;
+
+    entry = static_cast<HectorHit*>(branch->NewEntry());
+
+    entry->E = momentum.E();
+
+    entry->Tx = momentum.Px();
+    entry->Ty = momentum.Py();
+
+    entry->T = position.T();
+
+    entry->X = position.X();
+    entry->Y = position.Y();
+    entry->S = position.Z();
+  }
+}
+
+//------------------------------------------------------------------------------
+
 void TreeWriter::Process()
 {