styles.py

Go to the documentation of this file.

########################### 
# generic drawing styles
import ROOT
truth_colours= [ROOT.kRed, ROOT.kMagenta+2,ROOT.kBlue+2,ROOT.kCyan+2, ROOT.kGreen+2, ROOT.kGray, ROOT.kOrange]
data_colours = [ROOT.kBlack, ROOT.kBlue, ROOT.kGreen, ROOT.kRed]
lines   = [0,1,2,3,0,1,2,3]
markers = [20,21,22,23,24,25,26]
########################### 
# parse drawing configuration options from the name of a histogram and a list of components
def parseDrawingConfig(plot_name, components, verbose=False):
    import naming_schema
    suffixes = naming_schema.standard_truth_components[1:]
    i_truth = 0
    # lazy man's object
    draw_config = type('draw_config', (object,), {})()
    draw_config.labels       = []
    draw_config.axis_labels  = parseAxisLabels(plot_name)
    draw_config.colours      = []
    draw_config.lines        = []
    draw_config.markers      = []
    draw_config.draw_options = []
    for i,component in enumerate(components):
        if hasattr(component, "label"):
            label = naming_schema.datasetToLabel(component.label)
        else:
            label = naming_schema.datasetToLabel(component.dataset)
        if ("-truth_components" in plot_name) and (label in draw_config.labels):
            label = suffixes[i_truth]
            i_truth +=1
            if i_truth >= len(suffixes): i_truth = 0
        draw_config.labels.append(label)
        draw_config.lines.append(lines[i])
        draw_config.markers.append(markers[i])

        if ("data" in draw_config.labels[-1]):
            if ("surface" in plot_name) or ("_vs_" in plot_name):
                if len(components) == 1:
                    draw_config.draw_options.append("COLZ")
                else:
                    draw_config.draw_options.append("BOX")
            elif ("contour" in plot_name):
                draw_config.draw_options.append("L")
            elif ("sensitivity_bestfit" in plot_name):
                draw_config.draw_options.append("P")
            else:
                draw_config.draw_options.append("EX09")
            draw_config.colours.append(data_colours[i])

        else:
            if ("surface" in plot_name) or ("_vs_" in plot_name):
                if len(components) == 1:
                    draw_config.draw_options.append("COLZ")
                else:
                    draw_config.draw_options.append("BOX")
            elif ("contour" in plot_name):
                draw_config.draw_options.append("L")
            elif ("sensitivity_bestfit" in plot_name):
                draw_config.draw_options.append("P")
            else:
                draw_config.draw_options.append("hist9")
            draw_config.colours.append(truth_colours[i])

    return draw_config
########################### 
# parse the axis labels from a plot name
def parseAxisLabels(name):
    import naming_schema
    import common_tools as tools
    # histograms should follow a logical naming convention:
    #   <category>/<observable>_<reconstruction level>_<truth level>
    #   category = reco, true, mixed
    if (not tools.isLogicalHistogramName(name)):
        print "style: Histogram name: %s does not follow expected naming conventions"%name
        return "x;y;"
    if "_vs_" in name:
        tokens = tools.getHistogramObservable(name).split("_vs_")
        labels = "%s;%s"%(naming_schema.observableName(tokens[0]),naming_schema.observableName(tokens[1]))
    else:
        labels = "%s;Entries"%naming_schema.observableName(tools.getHistogramObservable(name))
    return labels