create_input Namespace Reference

Functions
def	qsum_histograms (histo1, histo2)
def	copy_bin (histo, eta_range)
def	copy_dir (source, destination)
def	create_structured_file (path)
def	create_new_directories (output_file)
def	merge_histograms (old, new, merge_error=True)

Variables
string	old_filename = "/home/turra/simon/latest_simon.root"
	old_file = ROOT.TFile.Open(old_filename)
	old_scale_histogram = old_file.Get("Scales/es2012c/alphaZee_errStat")
	old_ct_histogram = old_file.Get("Resolution/es2012c/ctZee_errStat")
	old_scale_sys = old_file.Get("Scales/es2012c/alphaZee_errSyst")
	old_ct_sys = old_file.Get("Resolution/es2012c/ctZee_errSyst")
	file_christophe = ROOT.TFile("~/ElectronEnergyScaleFactor.root")
	scales_christophe = file_christophe.Get("alpha68")
	ct_christophe = file_christophe.Get("sigma24")
	scales_sys_christophe = file_christophe.Get("systAlpha")
	ct_sys_christophe = file_christophe.Get("systSigma")
def	histo_scale = merge_histograms(old_scale_histogram, scales_christophe)
def	histo_ct = merge_histograms(old_ct_histogram, ct_christophe)
	canvas1 = ROOT.TCanvas()
	canvas2 = ROOT.TCanvas()
	output_file = ROOT.TFile("xxx.root", "update")
def	new_scale_sys = qsum_histograms(scales_sys_christophe, old_scale_sys)
def	new_ct_sys = qsum_histograms(ct_sys_christophe, old_ct_sys)
	legend3 = ROOT.TLegend(0.6, 0.7, 0.9, 0.9)
	canvas3 = ROOT.TCanvas()
	legend4 = ROOT.TLegend(0.6, 0.7, 0.9, 0.9)
	canvas4 = ROOT.TCanvas()
	stefano_file = ROOT.TFile("egammaEnergyCorrectionDataMC15.root")
	f_ua2mev = ROOT.TFile.Open("~/uA2MeV.root")
	histo_ua2mev = f_ua2mev.Get("histo_uA2MeV_week12")

Function Documentation

copy_bin()

def create_input.copy_bin	(		histo,
			eta_range
	)

avoid bin with value == 0 

Definition at line 54 of file create_input.py.

def copy_bin(histo, eta_range):
    """ avoid bin with value == 0 """
    empty_bins_in_transition = set()
    for eta in eta_range:
        bin = histo.FindBin(eta)
        if (histo.GetBinContent(bin) == 0):
            empty_bins_in_transition.add(bin)
    assert len(empty_bins_in_transition) == 1
    empty_bin = list(empty_bins_in_transition)[0]
    if eta_range[-1] > 0:
        assert(histo_scale.GetBinContent(empty_bin - 1) != 0)
        histo.SetBinContent(empty_bin, histo.GetBinContent(empty_bin - 1))
        histo.SetBinError(empty_bin, histo.GetBinError(empty_bin - 1))
    else:
        assert(histo_scale.GetBinContent(empty_bin + 1) != 0)
        histo.SetBinContent(empty_bin, histo.GetBinContent(empty_bin + 1))
        histo.SetBinError(empty_bin, histo.GetBinError(empty_bin + 1))
 
 

copy_dir()

def create_input.copy_dir	(		source,
			destination
	)

Definition at line 73 of file create_input.py.

def copy_dir(source, destination):
    # equivalent of cp -r source/* destination
    savdir = ROOT.gDirectory
    destination.cd()
    # loop on all entries of this directory
    for key in source.GetListOfKeys():
        classname = key.GetClassName()
        cl = ROOT.gROOT.GetClass(classname)
        if (not cl):
            continue
        if (cl.InheritsFrom(ROOT.TDirectory.Class())):
            d = destination.mkdir(key.GetName())
            copy_dir(source.GetDirectory(key.GetName()), d)
            destination.cd()
        elif (cl.InheritsFrom(ROOT.TTree.Class())):
            T = source.Get(key.GetName())
            destination.cd()
            newT = T.CloneTree(-1, "fast")
            newT.Write()
        elif (cl.InheritsFrom(ROOT.TList.Class())):
            source.cd()
            obj = key.ReadObj()
            destination.cd()
            obj.Write(obj.GetName(), ROOT.TObject.kSingleKey)
            del obj
        else:
            source.cd()
            obj = key.ReadObj()
            destination.cd()
            obj.Write()
            del obj
 
    destination.SaveSelf(True)
    savdir.cd()
 
 

create_new_directories()

def create_input.create_new_directories

(

output_file

)

Definition at line 118 of file create_input.py.

def create_new_directories(output_file):
    output_file.Get("Scales").mkdir("es2015PRE")
    output_file.Get("Resolution").mkdir("es2015PRE")
    output_file.Get("PSRecalibration").mkdir("es2015PRE")
    output_file.Get("S1Recalibration").mkdir("es2015PRE")
 
 

create_structured_file()

def create_input.create_structured_file

(

path

)

Definition at line 109 of file create_input.py.

def create_structured_file(path):
    output_file = ROOT.TFile(path, "recreate")
    output_file.mkdir("Scales")
    output_file.mkdir("Resolution")
    output_file.mkdir("PSRecalibration")
    output_file.mkdir("S1Recalibration")
    return output_file
 
 

merge_histograms()

def create_input.merge_histograms	(		old,
			new,
			merge_error = True
	)

Definition at line 125 of file create_input.py.

def merge_histograms(old, new, merge_error=True):
    new_binning = []
    new_values = []
    new_errors = []
    UNDERFLOW = 0
    OVERFLOW = new.GetNbinsX() + 1
 
    for iold in range(1, old.GetNbinsX()):
        low = old.GetBinLowEdge(iold)
        r = low + old.GetBinWidth(iold)
 
        il_new = new.FindFixBin(low)
        ir_new = new.FindFixBin(r)
        remainer = None
 
        if il_new == UNDERFLOW and ir_new == UNDERFLOW:
            new_binning.append((low, r))
            new_values.append(old.GetBinContent(iold))
            new_errors.append(old.GetBinError(iold))
 
        elif il_new == UNDERFLOW and ir_new > UNDERFLOW and ir_new < OVERFLOW:
            if abs(new.GetBinLowEdge(1) - low) < 1E-100:
                continue
            new_binning.append((low, new.GetBinLowEdge(1)))
            new_values.append(old.GetBinContent(iold))
            new_errors.append(old.GetBinError(iold))
            if ir_new == OVERFLOW:
                remainer = iold
            break
    last_old = iold
 
    for inew in range(1, new.GetNbinsX() + 1):
        low = new.GetBinLowEdge(inew)
        r = low + new.GetBinWidth(inew)
        new_binning.append((low, r))
        new_values.append(new.GetBinContent(inew))
        new_errors.append(new.GetBinError(inew))
 
    if remainer is not None:
        new_binning.append((new.GetBinLowEdge(new.GetNbinsX()),
                            old.GetBinLowEdge(remainer)
                            + old.GetBinWidth(remainer)))
        new_values.append(old.GetBinContent(remainer))
        new_errors.append(old.GetBinError(remainer))
 
    for iold in range(last_old, old.GetNbinsX() + 1):
        low = old.GetBinLowEdge(iold)
        r = low + old.GetBinWidth(iold)
 
        il_new = new.FindFixBin(low)
        ir_new = new.FindFixBin(r)
 
        if il_new == OVERFLOW and ir_new == OVERFLOW:
            new_binning.append((low, r))
            new_values.append(old.GetBinContent(iold))
            new_errors.append(old.GetBinError(iold))
        elif il_new < OVERFLOW and ir_new == OVERFLOW:
            if abs(new.GetBinLowEdge(new.GetNbinsX() + 1) - r) < 1E-100:
                continue
            new_binning.append((new.GetBinLowEdge(new.GetNbinsX() + 1), r))
            new_values.append(old.GetBinContent(iold))
            new_errors.append(old.GetBinError(iold))
 
    new_edges = array('f', [x[0] for x in new_binning] + [new_binning[-1][1]])
    histo_type = type(new)
    result = histo_type(new.GetName(), new.GetTitle(),
                        len(new_edges) - 1, new_edges)
    for i, (v, e) in enumerate(zip(new_values, new_errors), 1):
        result.SetBinContent(i, v)
        if merge_error:
            result.SetBinError(i, e)
    return result
 
 

qsum_histograms()

def create_input.qsum_histograms	(		histo1,
			histo2
	)

Definition at line 41 of file create_input.py.

def qsum_histograms(histo1, histo2):
    new_histo = histo1.Clone()
    new_histo.Reset()
    for ibin in range(histo1.GetNbinsX() + 2):
        value1 = histo1.GetBinContent(ibin)
        central_value = histo1.GetBinCenter(ibin)
        ibin2 = histo2.FindBin(central_value)
        value2 = histo2.GetBinContent(ibin2)
        new_value = np.sqrt(value1 * value1 + value2 * value2)
        new_histo.SetBinContent(ibin, new_value)
    return new_histo
 
 

Variable Documentation

canvas1

create_input.canvas1 = ROOT.TCanvas()

Definition at line 217 of file create_input.py.

canvas2

create_input.canvas2 = ROOT.TCanvas()

Definition at line 222 of file create_input.py.

canvas3

create_input.canvas3 = ROOT.TCanvas()

Definition at line 258 of file create_input.py.

canvas4

create_input.canvas4 = ROOT.TCanvas()

Definition at line 273 of file create_input.py.

ct_christophe

create_input.ct_christophe = file_christophe.Get("sigma24")

Definition at line 30 of file create_input.py.

ct_sys_christophe

create_input.ct_sys_christophe = file_christophe.Get("systSigma")

Definition at line 33 of file create_input.py.

f_ua2mev

create_input.f_ua2mev = ROOT.TFile.Open("~/uA2MeV.root")

Definition at line 296 of file create_input.py.

file_christophe

create_input.file_christophe = ROOT.TFile("~/ElectronEnergyScaleFactor.root")

Definition at line 28 of file create_input.py.

histo_ct

def create_input.histo_ct = merge_histograms(old_ct_histogram, ct_christophe)

Definition at line 200 of file create_input.py.

histo_scale

def create_input.histo_scale = merge_histograms(old_scale_histogram, scales_christophe)

Definition at line 199 of file create_input.py.

histo_ua2mev

create_input.histo_ua2mev = f_ua2mev.Get("histo_uA2MeV_week12")

Definition at line 297 of file create_input.py.

legend3

create_input.legend3 = ROOT.TLegend(0.6, 0.7, 0.9, 0.9)

Definition at line 256 of file create_input.py.

legend4

create_input.legend4 = ROOT.TLegend(0.6, 0.7, 0.9, 0.9)

Definition at line 271 of file create_input.py.

new_ct_sys

def create_input.new_ct_sys = qsum_histograms(ct_sys_christophe, old_ct_sys)

Definition at line 245 of file create_input.py.

new_scale_sys

def create_input.new_scale_sys = qsum_histograms(scales_sys_christophe, old_scale_sys)

Definition at line 244 of file create_input.py.

old_ct_histogram

create_input.old_ct_histogram = old_file.Get("Resolution/es2012c/ctZee_errStat")

Definition at line 19 of file create_input.py.

old_ct_sys

create_input.old_ct_sys = old_file.Get("Resolution/es2012c/ctZee_errSyst")

Definition at line 21 of file create_input.py.

old_file

create_input.old_file = ROOT.TFile.Open(old_filename)

Definition at line 17 of file create_input.py.

old_filename

string create_input.old_filename = "/home/turra/simon/latest_simon.root"

Definition at line 16 of file create_input.py.

old_scale_histogram

create_input.old_scale_histogram = old_file.Get("Scales/es2012c/alphaZee_errStat")

Definition at line 18 of file create_input.py.

old_scale_sys

create_input.old_scale_sys = old_file.Get("Scales/es2012c/alphaZee_errSyst")

Definition at line 20 of file create_input.py.

output_file

create_input.output_file = ROOT.TFile("xxx.root", "update")

Definition at line 234 of file create_input.py.

scales_christophe

create_input.scales_christophe = file_christophe.Get("alpha68")

Definition at line 29 of file create_input.py.

scales_sys_christophe

create_input.scales_sys_christophe = file_christophe.Get("systAlpha")

Definition at line 32 of file create_input.py.

stefano_file

create_input.stefano_file = ROOT.TFile("egammaEnergyCorrectionDataMC15.root")

Definition at line 288 of file create_input.py.