validationTestPlots.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
 
"""
<header>
<output>validationTestPlots.root, validationTestPlotsExpertOnly.root</output>
<contact>Kilian Lieret, Kilian.Lieret@campus.lmu.de</contact>
</header>
"""
 
"""
This file will generate various output into a root-file to have
a fast turn-around during development
"""
 
from validationtest import add_properties
 
import basf2
import array
import numpy as np
from datetime import datetime
import ROOT
 
# make sure we are able to always create the same plots
basf2.set_random_seed(1337)
 
tfile = ROOT.TFile("validationTestPlots.root", "RECREATE")
 
 
# NTuples
# ======================================
 
tntuple = ROOT.TNtuple("ntuple_test", "ntuple test", "x:y:z:k")
 
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', "shifter, some_meta_options")
 
# Overwrite the former TNtuple if one was there
tntuple.Write()
 
# Gauss Histogram
# ======================================
 
gaus_h = ROOT.TH1F("gaus_histogram", " Gaus Histogram", 100, -3, 3)
gaus_h.FillRandom("gaus", 500)
 
add_properties(
    gaus_h,
    {
        'Description': "xlog",
        'Check': "Gaus Histogram Check",
        'Contact': "Gaus Histogram Contact",
        'MetaOptions': "shifter, logx, nostats"
    }
)
 
gaus_h.Write()
 
# Warnings
# ======================================
 
warnings_h = ROOT.TH1F(
    "warnings_histogram",
    "Histogram with missing check and description",
    100, -3, 3
)
warnings_h.FillRandom("gaus", 500)
 
add_properties(
    warnings_h,
    {
        'Description': "",
        'Check': "",
        'Contact': "Kilian Lieret, Kilian.Lieret@campus.lmu.de",
        'MetaOptions': "shifter, logx, nostats"
    }
)
 
warnings_h.Write()
 
# Exp histograms
# ======================================
 
exp_h = ROOT.TH1F("exp_histogram", " Expert Histogram", 100, 0, 10)
 
exp_fn = ROOT.TF1("exp_fn", "exp(-x)", 0, 10)
exp_h.FillRandom("exp_fn", 500)
 
add_properties(
    exp_h,
    {
        'Description': "Expert Validation Plot",
        'Check': "Expert Histogram Check. Should only appear if expert button"
                 "was checked.",
        'Contact': "Expert Histogram Contact",
        'MetaOptions': "logy, nostats, C"
    }
)
 
exp_h.Write()
 
 
# Gaus changing histogram
# ======================================
 
# set new random seed, to have changed a changed diagram for every
# execution
basf2.set_random_seed(datetime.now().microsecond)
 
gaus_changing = ROOT.TH1F(
    "gaus_changing_histogram",
    "Gaus Changing Histogram",
    100, -5, 5
)
mean = ROOT.gRandom.Uniform(-1.0, 1.0)
 
for i in range(500):
    v = ROOT.gRandom.Gaus(mean, 1.0)
    gaus_changing.Fill(v)
 
add_properties(
    gaus_changing,
    {
        'Description': "xlog ylog with stats. I can haz $\LaTeX$?",
        'Check': "Gaus Changing Histogram Check",
        'Contact': "Gaus Changing Histogram Contact",
        'MetaOptions': "shifter, logx, logy"
    }
)
 
gaus_changing.Write()
 
# 2D histogram
# ======================================
hist2d = ROOT.TH2F("example_2d_histogram", "example_2d_title",
                   60, -3, 3, 60, -3, 3)
 
mean = (0, 0)
cov = [[1, 0], [0, 1]]
for _ in range(500):
    x, y = np.random.multivariate_normal(mean, cov)
    hist2d.Fill(x, y)
 
 
add_properties(
    hist2d,
    {
        'Description': "Some 2D Histogram",
        'Check': "Check For Something",
        'Contact': "Contact Someone",
        'MetaOptions': "shifter, contz"
    }
)
 
hist2d.Write()
 
# TEfficiency plot
# ======================================
 
gaus_passed = ROOT.TH1F("gaus_passed", "gaus_passed", 50, -0.5, 49.5)
gaus_total = ROOT.TH1F("gaus_total", "gaus_total", 50, -0.5, 49.5)
 
for i in range(500):
    # shifted to more passed at larger positions
    ratio = float(i) / 500.0
    passed = ROOT.gRandom.Uniform(ratio * 0.45, 1.0)
    pos = ROOT.gRandom.Uniform(ratio * 30.0, 49.5)
 
    gaus_total.Fill(pos)
    if passed > 0.5:
        gaus_passed.Fill(pos)
 
teff = ROOT.TEfficiency(gaus_passed, gaus_total)
teff.SetName("TEfficiency")
teff.SetTitle("Tefficiency")
 
add_properties(
    teff,
    {
        'Description': "Efficiency plot of something",
        'Check': "Check For Something",
        'Contact': "Contact Someone",
        'MetaOptions': "shifter"
    }
)
 
teff.Write()
 
# TGraphErrors
# ======================================
 
graph_err = ROOT.TGraphErrors()
 
graph_err.Set(50)
for i in range(50):
    # shifted to more passed at larger positions
    ratio = float(i) / 50.0
    passed = ROOT.gRandom.Uniform(ratio * 0.45, 1.0)
 
    graph_err.SetPoint(i, i + 1.0, passed)
 
graph_err.SetName("TGraphErrors")
graph_err.SetTitle("TGraphErrors")
add_properties(
    graph_err,
    {
        'Description': "TGraphErrors plot of something",
        'Check': "Check For Something",
        'Contact': "Contact Someone",
        'MetaOptions': "shifter"
    }
)
 
graph_err.Write()
 
# Html content
# ======================================
 
# generate some user-defined HTML
html_content = ROOT.TNamed("This is a bold HTML tag", "<p><b>THIS IS USER'S HTML</b></p>")
 
html_content.Write()
 
tfile.Close()
 
# Expert only
# ======================================
 
tfile = ROOT.TFile("validationTestPlotsExpertOnly.root", "RECREATE")
gaus_h = ROOT.TH1F("gaus_histogram_exprt", " Gaus Histogram Expert", 100, -3, 3)
gaus_h.FillRandom("gaus", 500)
gaus_h.Write()
tfile.Close()