examplePython.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
 
"""
This file contains example for generating validation plots using python
"""
 
import basf2
import array
import numpy as np
from ROOT import TFile, TNtuple, TH1F, TF1, TH2F, TF2, TRandom3, gRandom, TNamed, TEfficiency, TGraphErrors
 
basf2.set_random_seed(1337)
 
tfile = TFile("examplePython.root", "RECREATE")
tntuple = TNtuple("ntuple_test", "ntuple test", "x:y:z:k")
 
prefix = "examplePython"
 
array_of_values = array.array('f', [23.4, 4.4, 5.12, -23.0])
tntuple.Fill(array_of_values)
 
tntuple.SetAlias('Description', "This is a description test. "
                 "Lorem ipsum sit dolor amet.  We also support $\LaTeX$! "
                 "\n <br> \n For example, here is the infamous "
                 "Einstein-Pythagoras-theorem: \n "
                 "$$a^2 + b^2 = \\frac{E}{m}$$ \n Of course, you can also"
                 "do other things, like $\\theta = 90^\circ$ or $D^- "
                 "\\rightarrow D^0 \pi^- \pi^+$. \n Sometimes it is "
                 "necessary to escape commands with a double backslash, "
                 "because e.g. \\theta will be interpreted as [tab]heta.")
tntuple.SetAlias('Check', "This is the check text.")
tntuple.SetAlias('Contact', "Name of the contact person.")
tntuple.SetAlias('MetaOptions', "some_meta_options")
 
# Overwrite the former TNtuple if one was there
tntuple.Write()
 
# Gauss Histogram
gausH = TH1F("gaus_histogram", prefix + " Gaus Histogram", 100, -3, 3)
gausH.FillRandom("gaus", 5000)
 
gausH.GetListOfFunctions().Add(TNamed('Description', "xlog"))
gausH.GetListOfFunctions().Add(TNamed('Check', "Gaus Histogram Check"))
gausH.GetListOfFunctions().Add(TNamed('Contact', "Gaus Histogram Contact"))
gausH.GetListOfFunctions().Add(TNamed('MetaOptions', "logx, nostats"))
 
gausH.Write()
 
gausH = TH1F("exp_histogram", prefix + " Exp Histogram", 100, 0, 10)
 
exp_fn = TF1("exp_fn", "exp(-x)", 0, 10)
gausH.FillRandom("exp_fn", 5000)
 
gausH.GetListOfFunctions().Add(TNamed('Description', "Expert Validation Plot"))
gausH.GetListOfFunctions().Add(TNamed('Check', "Exp Histogram Check"))
gausH.GetListOfFunctions().Add(TNamed('Contact', "Exp Histogram Contact"))
gausH.GetListOfFunctions().Add(TNamed(
    'MetaOptions', "logy, nostats, C"
))
 
gausH.Write()
 
# Example for a 2D histogram
hist2d = TH2F("example_2d_histogram", "example_2d_title",
              60, -3, 3, 60, -3, 3)
 
mean = (0, 0)
cov = [[1, 0], [0, 1]]
for i in range(10000):
    x, y = np.random.multivariate_normal(mean, cov)
    hist2d.Fill(x, y)
 
hist2d.GetListOfFunctions().Add(TNamed('Description', "Some 2D Histogram"))
hist2d.GetListOfFunctions().Add(TNamed('Check', "Check For Something"))
hist2d.GetListOfFunctions().Add(TNamed('Contact', "Contact Someone"))
hist2d.GetListOfFunctions().Add(TNamed('MetaOptions', "contz"))
 
hist2d.Write()
 
# add TEfficiency plot
gaus_passed = TH1F("gaus_passed", "gaus_passed", 50, -0.5, 49.5)
gaus_total = TH1F("gaus_total", "gaus_total", 50, -0.5, 49.5)
 
for i in range(5000):
    # shifted to more passed at larger positions
    ratio = float(i) / 5000.0
    passed = gRandom.Uniform(ratio * 0.45, 1.0)
    pos = gRandom.Uniform(ratio * 30.0, 49.5)
    gaus_total.Fill(pos)
# fixme
if passed > 0.5:
    gaus_passed.Fill(pos)
 
teff = TEfficiency(gaus_passed, gaus_total)
teff.SetName("TEfficiency")
teff.GetListOfFunctions().Add(TNamed('Description', "Efficiency plot of something"))
teff.GetListOfFunctions().Add(TNamed('Check', "Check For Something"))
teff.GetListOfFunctions().Add(TNamed('Contact', "Contact Someone"))
teff.GetListOfFunctions().Add(TNamed('MetaOptions', ""))
 
teff.Write()
 
# add TGraphErrors plot
graph_err = TGraphErrors()
 
graph_err.Set(50)
for i in range(50):
    # shifted to more passed at larger positions
    ratio = float(i) / 50.0
    passed = gRandom.Uniform(ratio * 0.45, 1.0)
 
# fixme
graph_err.SetPoint(i, i + 1.0, passed)
 
graph_err.SetName("TGraphErrors")
graph_err.GetListOfFunctions().Add(TNamed('Description', "TGraphErrors plot of something"))
graph_err.GetListOfFunctions().Add(TNamed('Check', "Check For Something"))
graph_err.GetListOfFunctions().Add(TNamed('Contact', "Contact Someone"))
graph_err.GetListOfFunctions().Add(TNamed('MetaOptions', ""))
 
graph_err.Write()
 
tfile.Close()