Changeset a26a130 in git

Timestamp:

Jun 1, 2021, 11:39:05 PM (4 years ago)

Author:

Pavel Demin <pavel.demin@…>

Branches:

master

Children:

90132e0

Parents:

90345e5

Message:

move arrays to FinalizeParticles in DelphesHepMC3Reader

Location:

classes

Files:

• DelphesHepMC3Reader.cc (modified) (6 diffs)
• DelphesHepMC3Reader.h (modified) (1 diff)

classes/DelphesHepMC3Reader.cc

@@ -326 +326 @@
     }
 
-    AnalyzeParticle(factory, allParticleOutputArray,
-      stableParticleOutputArray, partonOutputArray);
+    AnalyzeParticle(factory);
   }
 
   if(EventReady())
   {
-    FinalizeParticles(allParticleOutputArray);
+    FinalizeParticles(allParticleOutputArray, stableParticleOutputArray, partonOutputArray);
   }
 
@@ -374 +373 @@
 {
   Weight *element;
-  vector<double>::const_iterator itWeight;
-
-  for(itWeight = fWeights.begin(); itWeight != fWeights.end(); ++itWeight)
+  vector<double>::const_iterator itWeights;
+
+  for(itWeights = fWeights.begin(); itWeights != fWeights.end(); ++itWeights)
   {
     element = static_cast<Weight *>(branch->NewEntry());
 
-    element->Weight = *itWeight;
+    element->Weight = *itWeights;
   }
 }
@@ -406 +405 @@
     array = factory->NewArray();
     position->SetXYZT(0.0, 0.0, 0.0, 0.0);
-    array->Clear();
     fVertices.push_back(make_pair(position, array));
   }
@@ -432 +430 @@
 //---------------------------------------------------------------------------
 
-void DelphesHepMC3Reader::AnalyzeParticle(DelphesFactory *factory,
-  TObjArray *allParticleOutputArray,
+void DelphesHepMC3Reader::AnalyzeParticle(DelphesFactory *factory)
+{
+  Candidate *candidate;
+
+  candidate = factory->NewCandidate();
+
+  candidate->PID = fPID;
+
+  candidate->Status = fParticleStatus;
+
+  candidate->Mass = fMass;
+
+  candidate->Momentum.SetPxPyPzE(fPx, fPy, fPz, fE);
+
+  candidate->D1 = fParticleCode;
+
+  AnalyzeVertex(factory, fOutVertexCode, candidate);
+}
+
+//---------------------------------------------------------------------------
+
+void DelphesHepMC3Reader::FinalizeParticles(TObjArray *allParticleOutputArray,
   TObjArray *stableParticleOutputArray,
   TObjArray *partonOutputArray)
 {
+  TLorentzVector *position;
+  TObjArray *array;
   Candidate *candidate;
   TParticlePDG *pdgParticle;
   int pdgCode;
-
-  candidate = factory->NewCandidate();
-
-  candidate->PID = fPID;
-  pdgCode = TMath::Abs(candidate->PID);
-
-  candidate->Status = fParticleStatus;
-
-  pdgParticle = fPDG->GetParticle(fPID);
-  candidate->Charge = pdgParticle ? int(pdgParticle->Charge() / 3.0) : -999;
-  candidate->Mass = fMass;
-
-  candidate->Momentum.SetPxPyPzE(fPx, fPy, fPz, fE);
-  if(fMomentumCoefficient != 1.0)
-  {
-    candidate->Momentum *= fMomentumCoefficient;
-  }
-
-  candidate->D1 = fParticleCode;
-
-  AnalyzeVertex(factory, fOutVertexCode, candidate);
-
-  if(!pdgParticle) return;
-
-  if(fParticleStatus == 1)
-  {
-    stableParticleOutputArray->Add(candidate);
-  }
-  else if(pdgCode <= 5 || pdgCode == 21 || pdgCode == 15)
-  {
-    partonOutputArray->Add(candidate);
-  }
-}
-
-//---------------------------------------------------------------------------
-
-void DelphesHepMC3Reader::FinalizeParticles(TObjArray *allParticleOutputArray)
-{
-  TLorentzVector *position;
-  TObjArray *array;
-  Candidate *candidate;
   map<int, int >::iterator itVertexMap;
   map<int, pair<int, int> >::iterator itMotherMap;
@@ -500 +479 @@
       {
         candidate->Position *= fPositionCoefficient;
+      }
+
+      if(fMomentumCoefficient != 1.0)
+      {
+        candidate->Momentum *= fMomentumCoefficient;
       }
 
@@ -541 +525 @@
 
       ++counter;
+
+      pdgParticle = fPDG->GetParticle(candidate->PID);
+
+      candidate->Charge = pdgParticle ? int(pdgParticle->Charge() / 3.0) : -999;
+
+      if(!pdgParticle) continue;
+
+      pdgCode = TMath::Abs(candidate->PID);
+
+      if(candidate->Status == 1)
+      {
+        stableParticleOutputArray->Add(candidate);
+      }
+      else if(pdgCode <= 5 || pdgCode == 21 || pdgCode == 15)
+      {
+        partonOutputArray->Add(candidate);
+      }
     }
   }

classes/DelphesHepMC3Reader.h

@@ -65 +65 @@
   void AnalyzeVertex(DelphesFactory *factory, int code, Candidate *candidate = 0);
 
-  void AnalyzeParticle(DelphesFactory *factory,
-    TObjArray *allParticleOutputArray,
+  void AnalyzeParticle(DelphesFactory *factory);
+
+  void FinalizeParticles(TObjArray *allParticleOutputArray,
     TObjArray *stableParticleOutputArray,
     TObjArray *partonOutputArray);
-
-  void FinalizeParticles(TObjArray *allParticleOutputArray);
 
   FILE *fInputFile;