Change branch name in utils.py

benchmarks/imaging_ecal/scripts/utils.py

@@ -110,7 +110,7 @@ def get_all_mcp(path, evnums=None, branch='mcparticles2'):
     return pd.DataFrame(data=dbuf[:idb], columns=['event', 'px', 'py', 'pz', 'pid', 'status', 'g4Parent', 'vex', 'vey', 'vez'])
 
 # read hits data from root file
-def get_hits_data(path, evnums=None, branch='RecoEcalBarrelHits'):
+def get_hits_data(path, evnums=None, branch='RecoEcalBarreImaginglHits'):
     f = ROOT.TFile(path)
     events = f.events
     if evnums is None:
@@ -136,7 +136,7 @@ def get_hits_data(path, evnums=None, branch='RecoEcalBarrelHits'):
 
 
 # read layers data from root file
-def get_layers_data(path, evnums=None, branch="EcalBarrelClustersLayers"):
+def get_layers_data(path, evnums=None, branch="EcalBarrelImagingClustersLayers"):
     f = ROOT.TFile(path)
     events = f.events
     if evnums is None:
@@ -163,7 +163,7 @@ def get_layers_data(path, evnums=None, branch="EcalBarrelClustersLayers"):
 
 
 # read clusters data from root file
-def get_clusters_data(path, evnums=None, branch='EcalBarrelClustersReco'):
+def get_clusters_data(path, evnums=None, branch='EcalBarrelImagingClustersReco'):
     f = ROOT.TFile(path)
     events = f.events
     if evnums is None: