TreeName=name, branchblock=branchblock.getvalue(), setbranchesblock=setbranchesblock.getvalue(), getbranchesblock=getbranchesblock.getvalue() ) if __name__ == "__main__": parser = argparse.ArgumentParser() parser.add_argument('template_file', help='File path to .root file with template TTree') parser.add_argument('tree_path', help='Path in .root file to TTree') parser.add_argument('ClassName', help='Name of cython proxy class') args = parser.parse_args() file = ROOT.TFile(resolve_file(args.template_file), 'READ') tree = file.Get(args.tree_path) with open('%s.pyx' % args.ClassName, 'w') as pyx_file: pyx_file.write(make_pyx(args.ClassName, tree)) # Figure out the root include and lib paths incdir = subprocess.Popen( ['root-config', '--incdir'], stdout=subprocess.PIPE).communicate()[0].strip() libdir = subprocess.Popen( ['root-config', '--libdir'], stdout=subprocess.PIPE).communicate()[0].strip() with open('%s_setup.py' % args.ClassName, 'w') as setup_file: setup_file.write(_setup_template.format(
parser.add_argument('--debug', action='store_const', const=True, default=False, help='Print debug output') args = parser.parse_args() import ROOT files = [] if '.txt' in args.input: with open(args.input) as input_files: for input_file in input_files: input_file = input_file.strip() if input_file and '#' not in input_file: files.append(resolve_file(input_file)) else: files = [resolve_file(x.strip()) for x in args.input.split(',')] level = logging.INFO if args.debug: level = logging.DEBUG logging.basicConfig(stream=sys.stderr, level=level) log.info("Extracting meta info from %i files", len(files)) total_events = 0 run_lumis = {} for file in files: log.debug("OPEN file %s", file)
const=True, default=False, help='If true, include the run-lumi mask result') parser.add_argument('--debug', action='store_const', const=True, default=False, help='Print debug output') args = parser.parse_args() import ROOT files = [] if '.txt' in args.input: with open(args.input) as input_files: for input_file in input_files: input_file = input_file.strip() if input_file and '#' not in input_file: files.append(resolve_file(input_file)) else: files = [resolve_file(x.strip()) for x in args.input.split(',')] level = logging.INFO if args.debug: level = logging.DEBUG logging.basicConfig(stream=sys.stderr, level=level) log.info("Extracting meta info from %i files", len(files)) total_events = 0 run_lumis = {} for file in files: log.debug("OPEN file %s", file)