def main(): utils.load_packages() utils.generate_dicts() utils.import_SUSYDefs_enums() parser = optparse.OptionParser(usage=usage) parser.add_option('-f', '--fill', action='store_true', default=False, help='force filling histos') parser.add_option('-l', '--labels', help='labels for plots (comma separated)') parser.add_option('-n', '--normalize', action='store_true', help='normalize histograms to unity') parser.add_option('-N', '--num-entries', help='loop only over N entries') parser.add_option('-s', '--sample', help='sample name') parser.add_option('-v', '--verbose', action='store_true', default=False) (opts, args) = parser.parse_args() print 'labels ',opts.labels if not args_are_valid(args) : parser.error('invalid arguments') verbose = opts.verbose input1 = args[0] input2 = args[1] out_dir = args[2] histo_cache_filename = os.path.join(out_dir, 'plot_comparison.root') fill_histos = opts.fill or not os.path.exists(histo_cache_filename) allOpts = [x.dest for x in parser._get_all_options()[1:]] if opts.verbose : print '\nUsing the following opts:\n' print '\n'.join("%s : %s"%(o, str(getattr(opts, o))) for o in allOpts) print '\n'.join(["%s : %s"%(o, str(eval(o))) for o in ['histo_cache_filename', 'fill_histos']]) if fill_histos : run_fill(input1, input2, histo_cache_filename, opts) run_plot(histo_cache_filename, out_dir, opts)
def main(): utils.load_packages() utils.generate_dicts() utils.import_SUSYDefs_enums() input1 = sys.argv[1] input_file = r.TFile.Open(input1) tree = input_file.Get('susyNt') tree.StartViewer() answ = '' while answ != 'y': answ = raw_input('Quit?')
def main(): utils.load_packages() utils.generate_dicts() utils.import_SUSYDefs_enums() sample_name = 'Sherpa_CT10_lllnu_WZ' input_dir = '/var/tmp/susynt_dev/data/ntup_susy/' #input_dir = '/var/tmp/susynt_dev/data/ntup_common/' input_files = glob.glob(os.path.join(input_dir, '*.root*')) chain = r.TChain('susyNt') for f in input_files : chain.Add(f) num_entries = chain.GetEntries() num_entries_to_process = num_entries if num_entries<1e4 else int(1e4) print "About to loop on %d entries"%num_entries_to_process run_with_chain(chain, num_entries_to_process)
def main(): utils.load_packages() utils.generate_dicts() utils.import_SUSYDefs_enums() sample_name = 'mc14_13TeV.110401.PowhegPythia_P2012_ttbar_nonallhad' # input_dir = '/var/tmp/susynt_dev/data/ntup_susy/' # #input_dir = '/var/tmp/susynt_dev/data/ntup_common/' # input_files = glob.glob(os.path.join(input_dir, '*.root*')) input_files = ['./susyNt.root'] chain = r.TChain('susyNt') for f in input_files : chain.Add(f) num_entries = chain.GetEntries() num_entries_to_process = num_entries if num_entries<1e4 else int(1e4) print "About to loop on %d entries"%num_entries_to_process run_with_chain(chain, num_entries_to_process)
def main(): utils.load_packages() utils.generate_dicts() utils.import_SUSYDefs_enums() sample_name = 'mc14_13TeV.110401.PowhegPythia_P2012_ttbar_nonallhad' # input_dir = '/var/tmp/susynt_dev/data/ntup_susy/' # #input_dir = '/var/tmp/susynt_dev/data/ntup_common/' # input_files = glob.glob(os.path.join(input_dir, '*.root*')) input_files = ['./susyNt.root'] chain = r.TChain('susyNt') for f in input_files: chain.Add(f) num_entries = chain.GetEntries() num_entries_to_process = num_entries if num_entries < 1e4 else int(1e4) print "About to loop on %d entries" % num_entries_to_process run_with_chain(chain, num_entries_to_process)
def main(): utils.load_packages() utils.generate_dicts() utils.import_SUSYDefs_enums() sample_name = "Sherpa_CT10_lllnu_WZ" input_dir = "/var/tmp/susynt_dev/data/ntup_susy/" # input_dir = '/var/tmp/susynt_dev/data/ntup_common/' input_files = glob.glob(os.path.join(input_dir, "*.root*")) chain = r.TChain("susyNt") for f in input_files: chain.Add(f) num_entries = chain.GetEntries() num_entries_to_process = num_entries if num_entries < 1e4 else int(1e4) print "About to loop on %d entries" % num_entries_to_process run_with_chain(chain, num_entries_to_process)
def main(): utils.load_packages() utils.generate_dicts() utils.import_SUSYDefs_enums() parser = optparse.OptionParser(usage=usage) parser.add_option('-f', '--fill', action='store_true', default=False, help='force filling histos') parser.add_option('-l', '--labels', help='labels for plots (comma separated)') parser.add_option('-n', '--normalize', action='store_true', help='normalize histograms to unity') parser.add_option('-N', '--num-entries', help='loop only over N entries') parser.add_option('-s', '--sample', help='sample name') parser.add_option('-v', '--verbose', action='store_true', default=False) (opts, args) = parser.parse_args() print 'labels ', opts.labels if not args_are_valid(args): parser.error('invalid arguments') verbose = opts.verbose input1 = args[0] input2 = args[1] out_dir = args[2] histo_cache_filename = os.path.join(out_dir, 'plot_comparison.root') fill_histos = opts.fill or not os.path.exists(histo_cache_filename) allOpts = [x.dest for x in parser._get_all_options()[1:]] if opts.verbose: print '\nUsing the following opts:\n' print '\n'.join("%s : %s" % (o, str(getattr(opts, o))) for o in allOpts) print '\n'.join([ "%s : %s" % (o, str(eval(o))) for o in ['histo_cache_filename', 'fill_histos'] ]) if fill_histos: run_fill(input1, input2, histo_cache_filename, opts) run_plot(histo_cache_filename, out_dir, opts)