validationTestColorScheme.py

#!/usr/bin/env python3
 

 
"""
<header>
<output>validationTestColorScheme.root</output>
<contact>Kilian Lieret, Kilian.Lieret@campus.lmu.de</contact>
 
 
<description>
To test the color scheme.
Set generate_reference = True to generate the reference files.
</description>
</header>
"""
 
 
import basf2
from validationtest import add_properties
from ROOT import TFile, TH1F, TNamed
 
 
def generate_gaus(title, options, distort=0.0):
    """ Generate trivial gaus filled histogram.
    Use the distort parameter (0<=distort<=1) for distortion to get histograms
    with failing comparisons.
    """
    # fix seed, because we need to see failing and passing comparisons
 
    assert 0 <= distort <= 1
 
    basf2.set_random_seed(10)
 
    norm = 300
 
    name = title.lower().replace(" ", "_")
 
    gaus = TH1F(name, title, 100, -3, 3)
 
    gaus.FillRandom("gaus", int((1 - distort) * norm))
    # add a different histogram
    basf2.set_random_seed(100)
    distortion = TH1F(name, title, 100, -3, 3)
    distortion.FillRandom("gaus", int(distort * norm))
    gaus.Add(distortion)
 
    add_properties(gaus, options)
    gaus.Write()
 
 
if __name__ == "__main__":
    tfile = TFile("validationTestColorScheme.root", "RECREATE")
 
    TNamed(
        "Description",
        "These plots test the color scheme. Make sure to also check the 'expert' "
        "plot checkbox to see the color scheme for expert plots.",
    ).Write()
 
    mop_expert = {
        "Description": "Test color scheme",
        "Check": "Check color",
        "Contact": "Kilian Lieret, Kilian.Lieret@campus.lmu.de",
        "MetaOptions": "pvalue-error=0.6",
    }
    mop_shifter = mop_expert.copy()
    mop_shifter["MetaOptions"] += ", shifter"
 
    generate_reference = False
 
    if not generate_reference:
        # No reference files for the plots that have to be checked manually
        generate_gaus("manual", mop_shifter)
        generate_gaus("manual expert", mop_expert)
 
    if generate_reference:
        # The reference will contain the normal (undistorted) histogram
        # always.
        generate_gaus("pass", mop_shifter)
        generate_gaus("pass expert", mop_expert)
        generate_gaus("warn", mop_shifter)
        generate_gaus("fail", mop_shifter)
        generate_gaus("warn expert", mop_expert)
        generate_gaus("fail expert", mop_expert)
    else:
        generate_gaus("pass", mop_shifter, distort=0.1)
        generate_gaus("pass expert", mop_expert, distort=0.1)
        generate_gaus("warn", mop_shifter, distort=0.5)
        generate_gaus("fail", mop_shifter, distort=1)
        generate_gaus("warn expert", mop_expert, distort=0.5)
        generate_gaus("fail expert", mop_expert, distort=1)
 
    tfile.Close()