Changeset c59be54 in git for converters/root2lhco.cpp

Timestamp:

Dec 19, 2014, 11:05:37 PM (10 years ago)

Author:

Pavel Demin <pavel.demin@…>

Branches:

ImprovedOutputFile, Timing, dual_readout, llp, master

Children:

b594101

Parents:

df04eb1

Message:

check if ROOT file contains all required branches (close #340)

File:

• converters/root2lhco.cpp (modified) (2 diffs)

converters/root2lhco.cpp

@@ -108 +108 @@
 
 LHCOWriter::LHCOWriter(ExRootTreeReader *treeReader, FILE *outputFile) :
-  fTriggerWord(0), fEventNumber(1), fTreeReader(0), fOutputFile(0)
+  fTriggerWord(0), fEventNumber(1), fTreeReader(0), fOutputFile(0),
+  fBranchEvent(0), fBranchTrack(0), fBranchTower(0), fBranchPhoton(0),
+  fBranchElectron(0), fBranchMuon(0), fBranchJet(0), fBranchMissingET(0)
 {
   fTreeReader = treeReader;
@@ -117 +119 @@
   // reconstructed tracks
   fBranchTrack = fTreeReader->UseBranch("Track");
-  fItTrack = fBranchTrack->MakeIterator();
   // calorimeter towers
   fBranchTower = fTreeReader->UseBranch("Tower");
-  fItTower = fBranchTower->MakeIterator();
   // reconstructed photons
   fBranchPhoton = fTreeReader->UseBranch("Photon");
-  fItPhoton = fBranchPhoton->MakeIterator();
   // reconstructed electrons
   fBranchElectron = fTreeReader->UseBranch("Electron");
-  fItElectron = fBranchElectron->MakeIterator();
   // reconstructed muons
   fBranchMuon = fTreeReader->UseBranch("Muon");
-  fItMuon = fBranchMuon->MakeIterator();
   // reconstructed jets
   fBranchJet = fTreeReader->UseBranch("Jet");
-  fItJet = fBranchJet->MakeIterator();
   // missing transverse energy
   fBranchMissingET = fTreeReader->UseBranch("MissingET");
+
+  if(!fBranchEvent || !fBranchTrack || !fBranchTower || !fBranchPhoton ||
+     !fBranchElectron || !fBranchMuon || !fBranchJet || !fBranchMissingET)
+  {
+    throw runtime_error("ROOT file doesn't contain all required branches");
+  }
+
+  fItTrack = fBranchTrack->MakeIterator();
+  fItTower = fBranchTower->MakeIterator();
+  fItPhoton = fBranchPhoton->MakeIterator();
+  fItElectron = fBranchElectron->MakeIterator();
+  fItMuon = fBranchMuon->MakeIterator();
+  fItJet = fBranchJet->MakeIterator();
 }