python.rootplot.tree2hists Namespace Reference

Classes
class	Plot
class	RootTree
	Define classes and generators #######################################. More...

Functions
def	duration_to_string (start, end)
def	join_cuts (list_of_cuts)
def	main ()
def	make_all_hists_all_files (list_of_RootTrees, list_of_Plots, cut_for_all_files, list_of_cutSets)
def	tree2hists_main (config_file)
	Define the main program #############################################. More...
def	write_default_T2H_config ()

Variables
	__license__

Detailed Description

Create ROOT Histograms from one or more ROOT TTrees or TNtuples.

Options are specified in the given configuration file.

Function Documentation

def python.rootplot.tree2hists.duration_to_string	(		start,
			end
	)

Definition at line 92 of file tree2hists.py.

Referenced by python.rootplot.tree2hists.tree2hists_main().

def duration_to_string(start, end):
    timeTaken = end - start
    hours, remainder = divmod(timeTaken.seconds, 3600)
    minutes, seconds = divmod(remainder, 60)
    if hours>0:
        return "%i hours, %i minutes" % (hours, minutes)
    elif minutes>0:
        return "%i minutes" % minutes
    return "%i seconds" % seconds                            

def python.rootplot.tree2hists.join_cuts

(

list_of_cuts

)

Joins list of cuts (strings) into something ROOT can handle.
Example:  given ('1<2','','5>4') returns '1<2&&5>4'

Definition at line 83 of file tree2hists.py.

References join().

Referenced by python.rootplot.tree2hists.make_all_hists_all_files().

def join_cuts(*list_of_cuts):
    """Joins list of cuts (strings) into something ROOT can handle.
    Example:  given ('1<2','','5>4') returns '1<2&&5>4'"""
    list_of_nonempty_cuts = []
    for cut in list_of_cuts:
        if cut:
            list_of_nonempty_cuts.append(cut)
    return '&&'.join(list_of_nonempty_cuts)

def python.rootplot.tree2hists.main

(

)

Definition at line 287 of file tree2hists.py.

References edm.print(), split, python.rootplot.tree2hists.tree2hists_main(), and python.rootplot.tree2hists.write_default_T2H_config().

def main():
    if len(sys.argv) > 1:
        if path.isfile(sys.argv[1]):
            config_file = sys.argv[1].split('.')[0]
            tree2hists_main(config_file)
        else:
            print("%s not found." % sys.argv[1])
            print("Create default config file by running tree2hists "
                  "with no argument.")
            sys.exit(1)
    else:
        if path.exists('t2h_config.py'):
            print("Run with specific config file, like:")
            print("  tree2hists t2h_config.py")
            sys.exit(1)
        write_default_T2H_config()
        sys.exit(0)    

def python.rootplot.tree2hists.make_all_hists_all_files	(		list_of_RootTrees,
			list_of_Plots,
			cut_for_all_files,
			list_of_cutSets
	)

Open root files one at a time, make plots, then close them.

Definition at line 152 of file tree2hists.py.

References join(), python.rootplot.tree2hists.join_cuts(), and edm.print().

Referenced by python.rootplot.tree2hists.tree2hists_main().

def make_all_hists_all_files(list_of_RootTrees, list_of_Plots, cut_for_all_files, list_of_cutSets):
    '''Open root files one at a time, make plots, then close them.'''
    list_of_plots_to_write = []

    # Create plots for each set of cuts
    for set_of_cuts in list_of_cutSets:
        histname_fix, title_fix, current_cut_set = set_of_cuts
        for plot in list_of_Plots:
            new_plot  = deepcopy(plot)
            new_title = ' '.join((plot.histogram.GetTitle(), title_fix))
            new_plot.histogram.SetTitle(new_title)
            list_of_plots_to_write.append((new_plot, set_of_cuts))
    
    for j, rootTree in enumerate(list_of_RootTrees):
        rootTree.tfile = TFile(rootTree.fileName, "read")           # Open TFile
        if rootTree.tfile.IsZombie():
            print("Error opening %s, exiting..." % rootTree.fileName)
            sys.exit(0)
        try:                                      # Try to get TTree from file.
            rootTree.tfile.GetObject(rootTree.treeName, rootTree.ttree)
        except:
            print("Error: %s not found in %s, exiting..." % (rootTree.treeName,
                                                             rootTree.fileName))
            sys.exit(1)

        print("\n%s: Opened %s  %i MB" % (datetime.now().strftime("%I:%M%p"),
                                          rootTree.fileName,
                                          path.getsize(rootTree.fileName)/1048576))
        print(" %s has %i entries, will plot with scale=%.2e, cuts='%s'" % (rootTree.treeName,
                                                                            rootTree.ttree.GetEntries(),
                                                                            rootTree.scale,
                                                                            rootTree.cuts))
        
        # Loop over plots
        print("   # entries                  var >> histogram")
        for i, (plot, set_of_cuts) in enumerate(list_of_plots_to_write):
            histname_fix, title_fix, current_cut_set = set_of_cuts
            tmp_hist = plot.histogram.Clone("temp")    # Create temp hist
            all_cuts = join_cuts(cut_for_all_files, rootTree.cuts,
                                 current_cut_set, plot.cuts) # Set cuts
            rootTree.ttree.Draw( "%s >> temp" % plot.treeVariable, all_cuts,
                                 "goff")               # Draw with graphics off
            tmp_hist.Scale(rootTree.scale)             # Scale temp
            entries_before = plot.histogram.GetEntries()
            plot.histogram.Add(tmp_hist)               # Add temp hist to total
            entries_after = plot.histogram.GetEntries()
            print(" %3i %7i %20s >> %s/%s" % (i, entries_after-entries_before,
                                              plot.treeVariable, histname_fix,
                                              plot.histogram.GetName()), end=' ')
            if plot.cuts:
                print("\textra cuts: %s" % plot.cuts, end=' ') # plot-specific cuts
            print()
            
        rootTree.tfile.Close()                                    # Close TFile
        print("%s: Closed %s" % (datetime.now().strftime("%I:%M%p"),
                                 rootTree.fileName))
    return list_of_plots_to_write

def python.rootplot.tree2hists.tree2hists_main

(

config_file

)

Define the main program #############################################.

Definition at line 212 of file tree2hists.py.

References python.rootplot.tree2hists.duration_to_string(), python.rootplot.tree2hists.make_all_hists_all_files(), and edm.print().

Referenced by python.rootplot.tree2hists.main().

def tree2hists_main(config_file):
    try:
        # Import only certain variables
        sys.path.insert(0, '')
        _temp = __import__(config_file, globals(), locals(),
                           ['list_of_files','output_filename',
                            'cut_for_all_files','cut_sets','list_of_plots'], -1)
        list_of_files     = _temp.list_of_files
        output_filename   = _temp.output_filename
        cut_for_all_files = _temp.cut_for_all_files
        cut_sets          = _temp.cut_sets
        list_of_plots     = _temp.list_of_plots
        for rootTree in list_of_files:
            if not path.isfile(rootTree.fileName):
                print("Error:\n  %s\nnot found for input." % rootTree.fileName)
                sys.exit(1)
        hist_names = [plot.name for plot in list_of_plots]
        if len(hist_names)>len(set(hist_names)):
            print(hist_names)
            print("Error: Each plot needs a unique name, exiting...")
            sys.exit(1)
        if path.isfile(output_filename):
            print("Warning: %s exists" % output_filename)
    except Exception as e:
        print(e)
        print("Error with %s" % config_file)
        sys.exit(1)

    if path.isfile('rootlogon.C'):
        print("Loading rootlogon.C")
        gROOT.Macro('rootlogon.C')    # Load functions from rootlogon script

    if cut_sets:
        print("\n%i defined cut sets:" % len(cut_sets))
        for cut in cut_sets:
            name, title_fix, current_cut_set = cut
            print("  %s\t: '%s'" % (name, current_cut_set))
        cut_names = [name for name,num,cut in cut_sets]
        if len(cut_names)>len(set(cut_names)):
            print("Error: Each cut set needs a unique name, exiting...")
            sys.exit(1)
    else:
        cut_sets = [("","","")] # Make all plots, no extra cuts

    print("\nCuts to apply to all files:\n\t'%s'" % cut_for_all_files)

    start_time = datetime.now()
    list_of_plots_to_write = make_all_hists_all_files(list_of_files,
                                                      list_of_plots,
                                                      cut_for_all_files,
                                                      cut_sets)
    end_time = datetime.now()
    print("Done drawing all plots after %s." % duration_to_string(start_time, end_time))

    #   Store and save/close files
    outputFile = TFile(output_filename, "recreate")
    if outputFile.IsZombie():
        print("Error opening %s for output exiting..." % output_filename)
        sys.exit(1)
    print("\nOpened output file. Saving histograms...")
    outputFile.cd()
    for set_of_cuts in cut_sets:
        outputFile.mkdir(set_of_cuts[0])
    print("   # entries  histogram")
    for i, (plot, cutset) in enumerate(list_of_plots_to_write):
        outputFile.cd(cutset[0])
        print(" %3i %7i  %s/%s" % (i, plot.histogram.GetEntries(),
                                   cutset[0],
                                   plot.histogram.GetName()))
        plot.histogram.Write()
    outputFile.Close()
    print("Done saving.")
    print("\nScaled & added histograms from %i TTrees saved in\n  %s" % (len(list_of_files), output_filename))

def python.rootplot.tree2hists.write_default_T2H_config

(

)

Writes configuration file for tree2hists

Definition at line 102 of file tree2hists.py.

References edm.print().

Referenced by python.rootplot.tree2hists.main().

def write_default_T2H_config():
    """Writes configuration file for tree2hists"""
    defaultConfig = '''# Configuration file for tree2hists
# Created %s.
try:
    ## the normal way to import from rootplot
    from rootplot.tree2hists import RootTree, Plot
except ImportError:
    ## special import for CMSSW installations of rootplot
    from PhysicsTools.PythonAnalysis.rootplot.tree2hists import RootTree, Plot
from array import array      # to allow making Float_t arrays for ROOT hists
from math import pi
from ROOT import TH1F, TH2F  # import other kinds of hists as neeeded

list_of_files = [RootTree("Treename", fileName="photons.root", scale=1.0, cuts=""),
                 RootTree("Treename", fileName="photons2.root", scale=1.0, cuts="")]

output_filename = "Hists_photons.root"

cut_for_all_files = "(!TTBit[36] && !TTBit[37] && !TTBit[38] && !TTBit[39] && !vtxIsFake && vtxNdof>4 && abs(vtxZ)<=15)"

# All plots are made for each "cut set".
# A "cut set" is 3 things: folder name to store hists in, string to add to hist titles, and cuts for these hists.
# Let cut_sets = [] to make all plots.
cut_sets = [
    ("barrel15to20", "(|#eta|<1.45, 15<E_{T}<20)", "et>15&&et<20&&abs(eta)<1.45"),
    ("barrel20to30", "(|#eta|<1.45, 20<E_{T}<30)", "et>20&&et<30&&abs(eta)<1.45"),
    ("endcap15to20", "(1.7<|#eta|<2.5, 15<E_{T}<20)", "et>15&&et<20&&abs(eta)>1.7&&abs(eta)<2.5"),
    ("endcap20to30", "(1.7<|#eta|<2.5, 20<E_{T}<30)", "et>20&&et<30&&abs(eta)>1.7&&abs(eta)<2.5")
    ]

# Define histograms to plot
bins_et     = array("f", [15.0, 20.0, 30.0, 50.0, 80.0, 120.0]) # example custom bins
list_of_plots = [
    Plot("et"           , TH1F("pho_et"           , "Lead #gamma: E_{T};E_{T} (GeV);entries/bin", 25, 0.0, 100.0)),
    Plot("eta"          , TH1F("pho_eta"          , "Lead #gamma: #eta;#eta;entries/bin"        , 25, -3.0, 3.0)),
    Plot("et"           , TH1F("pho_et_binned"    , "Lead #gamma: E_{T};E_{T} (GeV);entries/bin", len(bins_et)-1, bins_et)),
    Plot("sigmaIetaIeta", TH1F("pho_sigmaIEtaIEta", "Lead #gamma: #sigma_{i#etai#eta};#sigma_{i#etai#eta};entries/bin",20, 0, 0.06)),
    Plot("metEt/et"     , TH1F("metEt_over_phoEt" , "MET / E_{T}(#gamma);MET/E_{T}(sc);entries/bin"   , 20, 0.0, 3.0)),
    Plot("phi:eta"      , TH2F("phoPhi_vs_phoEta" , "Lead #gamma: #phi vs #eta;#eta;#phi"             , 50, -2.5, 2.5, 18, -pi, pi))
    ]
''' % datetime.now().strftime("%b %d, %Y")
    f = open('t2h_config.py', 'w')
    f.write(defaultConfig)
    f.close()
    print("Created default configuration file: t2h_config.py")
    print("Edit it, then run by typing:")
    print("  tree2hists t2h_config.py")

Variable Documentation

python.rootplot.tree2hists.__license__

private

Definition at line 14 of file tree2hists.py.