Example #1
0
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)
Example #2
0
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?')
Example #3
0
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)
Example #4
0
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)
Example #5
0
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)
Example #6
0
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)