From ebeff72ebf4b8f9e6e3b109c5118388f17a014e4 Mon Sep 17 00:00:00 2001
From: Fabrizio Grosa <fabrizio.grosa@cern.ch>
Date: Thu, 6 May 2021 15:15:16 +0200
Subject: [PATCH] Add possibility to parse multiple files

---
 femto/ProjectMassFromFemtoTask.py | 29 ++++++++++++++++++++++-------
 1 file changed, 22 insertions(+), 7 deletions(-)

diff --git a/femto/ProjectMassFromFemtoTask.py b/femto/ProjectMassFromFemtoTask.py
index 0dca1b22..b2cacc1e 100644
--- a/femto/ProjectMassFromFemtoTask.py
+++ b/femto/ProjectMassFromFemtoTask.py
@@ -22,6 +22,8 @@
                     help='suffix for directory inside task output file')
 parser.add_argument('--HFsuffix', metavar='text', default='',
                     help='HF suffix for directory inside task output file')
+parser.add_argument('--inFileNameAdd', action='append', default=None,
+                    help='additional root input file name')
 args = parser.parse_args()
 
 with open(args.cutSetFileName, 'r') as ymlCutSetFile:
@@ -30,15 +32,28 @@
 
 dirName = f'{args.prefix}_CharmFemto_{args.HFsuffix}DChargedQA{args.suffix}'
 listName = f'{dirName}/{args.prefix}_CharmFemto_{args.HFsuffix}DChargedQA{args.suffix}'
-print(f'Read input file: {args.inFileName}')
+
+inFileNames = [args.inFileName]
+if args.inFileNameAdd is not None:
+    for fileName in args.inFileNameAdd:
+        inFileNames.append(fileName)
+
+hDminusMassVsPt, hDplusMassVsPt = None, None
+for iFile, fileName in enumerate(inFileNames):
+    inFile = TFile.Open(fileName)
+    inList = inFile.Get(listName)
+    if iFile == 0:
+        hDminusMassVsPt = inList.FindObject('fHistDminusInvMassPt')
+        hDplusMassVsPt = inList.FindObject('fHistDplusInvMassPt')
+    else:
+        hDminusMassVsPt.Add(inList.FindObject('fHistDminusInvMassPt'))
+        hDplusMassVsPt.Add(inList.FindObject('fHistDplusInvMassPt'))
+    print(f'Read input file: {fileName}')
+print(f'           dir: {dirName}')
 print(f'           list: {listName}')
 
-inFile = TFile.Open(args.inFileName)
-inList = inFile.Get(listName)
-hDminusMassVsPt = inList.FindObject('fHistDminusInvMassPt')
-hDplusMassVsPt = inList.FindObject('fHistDplusInvMassPt')
-hMassVsPt = hDminusMassVsPt.Clone()
-if hDminusMassVsPt.GetYaxis().GetXmin() == hDplusMassVsPt.GetXaxis().GetXmin():
+hMassVsPt = hDminusMassVsPt.Clone('hMassVsPt')
+if hDminusMassVsPt.GetXaxis().GetXmin() == hDplusMassVsPt.GetXaxis().GetXmin():
     hMassVsPt.Add(hDplusMassVsPt)
 else:
     print('WARNING: using only D- because of different binning between D+ and D- histograms!')