Public Member Functions
	__init__ (self, input_file='../V0ValidationHarvested.root', output_file='V0Validation.root')

	collect_histograms (self)

	plot (self)

Static Public Member Functions
	efficiency_plot (found, all, title, x_variable, x_unit, description='', check='', contact='', meta_options='')

	histogram_plot (hist, title, x_variable, x_unit=None, description='', check='', contact='', meta_options='')

	histogram_2d_plot (hist, title, x_variable, y_variable, x_unit=None, y_unit=None, description='', check='', contact='', meta_options='')

Public Attributes
	input_file = input_file
	Input file.

	output_file = output_file
	Output file.

	hist_r = ROOT.TH1F("", "True R", 20, 0, 20)
	MCTruth R.

	hist_theta = ROOT.TH1F("", "True Theta", 26, 20, 150)
	MCTruth Theta.

	hist_phi = ROOT.TH1F("", "True Phi", 36, -180, 180)
	MCTruth Phi.

	hist_p = ROOT.TH1F("", "True P", 25, 0.0, 1.0)
	MCTruth Momentum.

	hist_r_found = ROOT.TH1F("", "Found R", 20, 0, 20)
	Reconstructed R.

	hist_theta_found = ROOT.TH1F("", "Found Theta", 26, 20, 150)
	Reconstructed Theta.

	hist_phi_found = ROOT.TH1F("", "Found Phi", 36, -180, 180)
	Reconstructed Phi.

	hist_p_found = ROOT.TH1F("", "Found P", 25, 0.0, 1.0)
	Reconstructed Momentum.

	hist_invariant_mass = ROOT.TH1F("", "", 60, 0.470, 0.530)
	Invariant mass histogram.

	hist_invariant_mass_res = ROOT.TH1F("", "", 40, -0.02, 0.02)
	Invariant mass residual histogram.

	hist_chi2 = ROOT.TH1F("", "", 50, 0, 50)
	Chi2 of all vertex fits.

	hist_chi2_inside = ROOT.TH1F("", "", 50, 0, 50)
	Chi2 of vertex fit inside beampipe.

	hist_chi2_outside = ROOT.TH1F("", "", 50, 0, 50)
	Chi2 of vertex fit outside beampipe.

	hist_mass_vs_mc_mass = ROOT.TH2F("", "", 80, 0, 0.8, 80, 0, 0.8)
	2D histogram; invariant mass vs reconstructed mass

	hist_invariant_mass_residuum = ROOT.TH1F("", "", 60, -0.05, 0.05)
	Invariant mass residuum histogram.

	hist_r_residuum = ROOT.TH1F("", "", 60, -0.1, 0.1)
	R residuum histogram.

	hist_theta_residuum = ROOT.TH1F("", "", 60, -0.1, 0.1)
	Theta residuum histogram.

	hist_phi_residuum = ROOT.TH1F("", "", 60, -0.1, 0.1)
	Phi residuum histogram.

	hist_p_residuum = ROOT.TH1F("", "", 60, -0.05, 0.05)
	Momentum residuum histogram.

Detailed Description

Reads the output created by the V0Harvester and creates plots from it.

Definition at line 27 of file v0ValidationCreatePlots.py.

Constructor & Destructor Documentation

◆ init()

__init__	(	self,
		input_file = '../V0ValidationHarvested.root',
		output_file = 'V0Validation.root' )

Reads the output created by the V0Harvester defines histograms which will be filled later.

:param input_file: Output of V0ValidationHarvester.
:param output_file: Plots displayed in the V0Validation.

Definition at line 31 of file v0ValidationCreatePlots.py.

    def __init__(self, input_file='../V0ValidationHarvested.root', output_file='V0Validation.root'):
        """Reads the output created by the V0Harvester defines histograms which will be filled later.
 
        :param input_file: Output of V0ValidationHarvester.
        :param output_file: Plots displayed in the V0Validation.
        """
        
        self.input_file = input_file
        
        self.output_file = output_file
 
        self.hist_r = ROOT.TH1F("", "True R", 20, 0, 20)
        
        self.hist_theta = ROOT.TH1F("", "True Theta", 26, 20, 150)
        
        self.hist_phi = ROOT.TH1F("", "True Phi", 36, -180, 180)
        
        self.hist_p = ROOT.TH1F("", "True P", 25, 0.0, 1.0)
 
        self.hist_r_found = ROOT.TH1F("", "Found R", 20, 0, 20)
        
        self.hist_theta_found = ROOT.TH1F("", "Found Theta", 26, 20, 150)
        
        self.hist_phi_found = ROOT.TH1F("", "Found Phi", 36, -180, 180)
        
        self.hist_p_found = ROOT.TH1F("", "Found P", 25, 0.0, 1.0)
 
        self.hist_invariant_mass = ROOT.TH1F("", "", 60, 0.470, 0.530)
        
        self.hist_invariant_mass_res = ROOT.TH1F("", "", 40, -0.02, 0.02)
 
        self.hist_chi2 = ROOT.TH1F("", "", 50, 0, 50)
        
        self.hist_chi2_inside = ROOT.TH1F("", "", 50, 0, 50)
        
        self.hist_chi2_outside = ROOT.TH1F("", "", 50, 0, 50)
 
        self.hist_mass_vs_mc_mass = ROOT.TH2F("", "", 80, 0, 0.8, 80, 0, 0.8)
 
        self.hist_invariant_mass_residuum = ROOT.TH1F("", "", 60, -0.05, 0.05)
        
        self.hist_r_residuum = ROOT.TH1F("", "", 60, -0.1, 0.1)
        
        self.hist_theta_residuum = ROOT.TH1F("", "", 60, -0.1, 0.1)
        
        self.hist_phi_residuum = ROOT.TH1F("", "", 60, -0.1, 0.1)
        
        self.hist_p_residuum = ROOT.TH1F("", "", 60, -0.05, 0.05)

Member Function Documentation

◆ collect_histograms()

collect_histograms ( self )

Fills the defined histograms with the V0Harvester data.

:return: self

Definition at line 86 of file v0ValidationCreatePlots.py.

    def collect_histograms(self):
        """Fills the defined histograms with the V0Harvester data.
 
        :return: self
        """
        input_root_file = ROOT.TFile.Open(self.input_file, "READ")
 
        for event in input_root_file.V0Harvester_tree:
            self.hist_r.Fill(event.R_MC)
            self.hist_theta.Fill(numpy.rad2deg(event.THETA_MC))
            self.hist_phi.Fill(numpy.rad2deg(event.PHI_MC))
            self.hist_p.Fill(event.P_MC)
 
            if event.FOUND:
                self.hist_r_found.Fill(event.R_MC)
                self.hist_theta_found.Fill(numpy.rad2deg(event.THETA_MC))
                self.hist_phi_found.Fill(numpy.rad2deg(event.PHI_MC))
                self.hist_p_found.Fill(event.P_MC)
 
                self.hist_invariant_mass.Fill(event.M)
 
                # Residuum histograms
                self.hist_invariant_mass_residuum.Fill(event.M - event.M_MC)
                self.hist_r_residuum.Fill(event.R - event.R_MC)
                self.hist_theta_residuum.Fill(event.THETA - event.THETA_MC)
                self.hist_phi_residuum.Fill(event.PHI - event.PHI_MC)
                self.hist_p_residuum.Fill(event.P - event.P_MC)
 
                self.hist_chi2.Fill(event.CHI2)
                if event.R_MC > 1.0:
                    self.hist_chi2_outside.Fill(event.CHI2)
                else:
                    assert event.R_MC <= 1.0
                    self.hist_chi2_inside.Fill(event.CHI2)
 
                self.hist_mass_vs_mc_mass.Fill(event.M, event.M_MC)
 
        input_root_file.Close()
        return self
 

◆ efficiency_plot()

efficiency_plot	(	found,
		all,
		title,
		x_variable,
		x_unit,
		description = '',
		check = '',
		contact = '',
		meta_options = '' )

static

Create an efficiency plot.

:param found: Histogram with all found entries (i.e. reconstructed).
:param all: Histogram with all entries (i.e. MCTruth).
:param title: Title of the histogram.
:param x_variable: x variable.
:param x_unit: x unit.
:param description: Description text shown on the validation page.
:param check: Check text shown on the validation page.
:param contact: Contact text shown on the validation page.
:param meta_options: Meta options for the validation page.
:return: ROOT.TEfficiency

Definition at line 127 of file v0ValidationCreatePlots.py.

    def efficiency_plot(found, all, title, x_variable, x_unit, description='', check='', contact='', meta_options=''):
        """Create an efficiency plot.
 
        :param found: Histogram with all found entries (i.e. reconstructed).
        :param all: Histogram with all entries (i.e. MCTruth).
        :param title: Title of the histogram.
        :param x_variable: x variable.
        :param x_unit: x unit.
        :param description: Description text shown on the validation page.
        :param check: Check text shown on the validation page.
        :param contact: Contact text shown on the validation page.
        :param meta_options: Meta options for the validation page.
        :return: ROOT.TEfficiency
        """
        efficiency = ROOT.TEfficiency(found, all)
        efficiency.SetName("".join(title.split()))
        ylabel = f'Efficiency / ({(found.GetXaxis().GetXmax() - found.GetXaxis().GetXmin()) / found.GetNbinsX()} {x_unit})'
        efficiency.SetTitle(f"{title};{x_variable} / ({x_unit});{ylabel}")
        efficiency.GetListOfFunctions().Add(ROOT.TNamed('Description', description))
        efficiency.GetListOfFunctions().Add(ROOT.TNamed('Check', check))
        efficiency.GetListOfFunctions().Add(ROOT.TNamed('Contact', contact))
        efficiency.GetListOfFunctions().Add(ROOT.TNamed('MetaOptions', meta_options))
        return efficiency
 

◆ histogram_2d_plot()

histogram_2d_plot	(	hist,
		title,
		x_variable,
		y_variable,
		x_unit = None,
		y_unit = None,
		description = '',
		check = '',
		contact = '',
		meta_options = '' )

static

Create a 2d hisogram plot.

:param hist: TH2F
:param title: Title of the histogram.
:param x_variable: x variable
:param y_variable: y variable
:param x_unit: x unit
:param y_unit: y unit
:param description: Description text shown on the validation page.
:param check: Check text shown on the validation page.
:param contact: Contact text shown on the validation page.
:param meta_options: Meta options for the validation page.
:return: ROOT.TEfficiency
:return:

Definition at line 178 of file v0ValidationCreatePlots.py.

                          description='', check='', contact='', meta_options=''):
        """Create a 2d hisogram plot.
 
        :param hist: TH2F
        :param title: Title of the histogram.
        :param x_variable: x variable
        :param y_variable: y variable
        :param x_unit: x unit
        :param y_unit: y unit
        :param description: Description text shown on the validation page.
        :param check: Check text shown on the validation page.
        :param contact: Contact text shown on the validation page.
        :param meta_options: Meta options for the validation page.
        :return: ROOT.TEfficiency
        :return:
        """
        hist.SetName("".join(title.split()))
        xlabel = f'{x_variable} / ({x_unit})' if x_unit is not None else f'{x_variable}'
        ylabel = f'{y_variable} / ({y_unit})' if y_unit is not None else f'{y_variable}'
        hist.SetTitle(f"{title};{xlabel};{ylabel}")
        hist.GetListOfFunctions().Add(ROOT.TNamed('Description', description))
        hist.GetListOfFunctions().Add(ROOT.TNamed('Check', check))
        hist.GetListOfFunctions().Add(ROOT.TNamed('Contact', contact))
        hist.GetListOfFunctions().Add(ROOT.TNamed('MetaOptions', meta_options))
        return hist
 

◆ histogram_plot()

histogram_plot	(	hist,
		title,
		x_variable,
		x_unit = None,
		description = '',
		check = '',
		contact = '',
		meta_options = '' )

static

Create (annotate) an histogram plot.

:param hist: TH1F
:param title: Title of the histogram.
:param x_variable: x variable.
:param x_unit: x unit.
:param description: Description text shown on the validation page.
:param check: Check text shown on the validation page.
:param contact: Contact text shown on the validation page.
:param meta_options: Meta options for the validation page.
:return: modified hist

Definition at line 152 of file v0ValidationCreatePlots.py.

    def histogram_plot(hist, title, x_variable, x_unit=None, description='', check='', contact='', meta_options=''):
        """Create (annotate) an histogram plot.
 
        :param hist: TH1F
        :param title: Title of the histogram.
        :param x_variable: x variable.
        :param x_unit: x unit.
        :param description: Description text shown on the validation page.
        :param check: Check text shown on the validation page.
        :param contact: Contact text shown on the validation page.
        :param meta_options: Meta options for the validation page.
        :return: modified hist
        """
        hist.SetName("".join(title.split()))
        xlabel = f'{x_variable} / ({x_unit})' if x_unit is not None else f'{x_variable}'
        ylabel = f'Entries / ({(hist.GetXaxis().GetXmax() - hist.GetXaxis().GetXmin()) / hist.GetNbinsX()} {x_unit})' \
            if x_unit is not None \
            else f'Entries / ({(hist.GetXaxis().GetXmax() - hist.GetXaxis().GetXmin()) / hist.GetNbinsX()})'
        hist.SetTitle(f"{title};{xlabel};{ylabel}")
        hist.GetListOfFunctions().Add(ROOT.TNamed('Description', description))
        hist.GetListOfFunctions().Add(ROOT.TNamed('Check', check))
        hist.GetListOfFunctions().Add(ROOT.TNamed('Contact', contact))
        hist.GetListOfFunctions().Add(ROOT.TNamed('MetaOptions', meta_options))
        return hist
 

◆ plot()

plot ( self )

Create plots with the data filled with 'collect_histograms'.

:return: self

Definition at line 205 of file v0ValidationCreatePlots.py.

    def plot(self):
        """Create plots with the data filled with 'collect_histograms'.
 
        :return: self
        """
        output_root_file = ROOT.TFile.Open(self.output_file, "RECREATE")
 
        V0ValidationPlots.efficiency_plot(
            self.hist_r_found,
            self.hist_r,
            'Efficiency vs R',
            'r',
            'cm',
            description='Reconstruction Efficiency vs. r (perpendicular)',
            check='Check that efficiencies are close to the reference. Report significant deviations.',
            contact='software-tracking@belle2.org',
            meta_options='shifter').Write()
 
        V0ValidationPlots.efficiency_plot(self.hist_theta_found, self.hist_theta, 'Efficiency vs Theta', 'Theta', 'deg',
                                          description='Reconstruction Efficiency vs. theta',
                                          check='',
                                          contact='software-tracking@belle2.org',
                                          meta_options='').Write()
 
        V0ValidationPlots.efficiency_plot(self.hist_phi_found, self.hist_phi, 'Efficiency vs Phi', 'Phi', 'deg',
                                          description='Reconstruction Efficiency vs phi',
                                          check='',
                                          contact='software-tracking@belle2.org',
                                          meta_options='').Write()
 
        V0ValidationPlots.efficiency_plot(self.hist_p_found, self.hist_p, 'Efficiency vs P', 'P', 'GeV',
                                          description='Reconstruction Efficiency vs momentum',
                                          check='',
                                          contact='software-tracking@belle2.org',
                                          meta_options='').Write()
 
        V0ValidationPlots.histogram_plot(self.hist_invariant_mass, "KShort Invariant Mass", "m", "GeV",
                                         description='Reconstructed invariant mass of KShorts with standard reconstruction',
                                         check='Invariant mass peak around KShort nominal mass 497.61 MeV.',
                                         contact='software-tracking@belle2.org',
                                         meta_options='shifter').Write()
 
        V0ValidationPlots.histogram_plot(self.hist_invariant_mass_residuum, "KShort Invariant Mass Residuum", "Rec - MC", "GeV",
                                         description='Invariant mass residuum of KShorts with standard reconstruction',
                                         check='',
                                         contact='software-tracking@belle2.org',
                                         meta_options='').Write()
 
        V0ValidationPlots.histogram_plot(self.hist_chi2, "Chi2 of Vertex Fits.", "Chi2", None,
                                         description='Chi2 distributions of the vertex fits.',
                                         check='Check if distribution looks like a Chi2 distribution with 1 dof',
                                         contact='software-tracking@belle2.org',
                                         meta_options='').Write()
 
        V0ValidationPlots.histogram_plot(self.hist_chi2_inside, "Chi2 of Vertex Fits Inside Beampipe.", "Chi2", None,
                                         description='Chi2 distributions of the vertex fits inside the beampipe.',
                                         check='Check if distribution looks like a Chi2 distribution with 1 dof',
                                         contact='software-tracking@belle2.org',
                                         meta_options='').Write()
 
        V0ValidationPlots.histogram_plot(self.hist_chi2_outside, "Chi2 of Vertex Fits Outside Beampipe.", "Chi2", None,
                                         description='Chi2 distributions of the vertex fits outside the beampipe.',
                                         check='Check if distribution looks like a Chi2 distribution with 1 dof',
                                         contact='software-tracking@belle2.org',
                                         meta_options='').Write()
 
        V0ValidationPlots.histogram_2d_plot(self.hist_mass_vs_mc_mass, "Reconstructed vs MC Mass.",
                                            "Reconstructed Mass", "GeV", "MC Mass", "GeV",
                                            description="Reconstructed mass vs invariant Mass.",
                                            check="",
                                            contact="software-tracking@belle2.org",
                                            meta_options='').Write()
 
        V0ValidationPlots.histogram_plot(self.hist_r_residuum, "KShort R Residuum", "Rec - MC", "cm",
                                         description='R residuum of KShorts with standard reconstruction',
                                         check='',
                                         contact='software-tracking@belle2.org',
                                         meta_options='').Write()
        V0ValidationPlots.histogram_plot(self.hist_theta_residuum, "KShort Theta Residuum", "Rec - MC", "rad",
                                         description='Theta residuum of KShorts with standard reconstruction',
                                         check='',
                                         contact='software-tracking@belle2.org',
                                         meta_options='').Write()
        V0ValidationPlots.histogram_plot(self.hist_phi_residuum, "KShort Phi Residuum", "Rec - MC", "rad",
                                         description='Phi residuum of KShorts with standard reconstruction',
                                         check='',
                                         contact='software-tracking@belle2.org',
                                         meta_options='').Write()
        V0ValidationPlots.histogram_plot(self.hist_p_residuum, "KShort Momentum Residuum", "Rec - MC", "GeV",
                                         description='Momentum residuum of KShorts with standard reconstruction',
                                         check='',
                                         contact='software-tracking@belle2.org',
                                         meta_options='').Write()
 
        output_root_file.Write()
        output_root_file.Close()
        return self
 
 

Member Data Documentation

◆ hist_chi2

hist_chi2 = ROOT.TH1F("", "", 50, 0, 50)

Chi2 of all vertex fits.

Definition at line 66 of file v0ValidationCreatePlots.py.

◆ hist_chi2_inside

hist_chi2_inside = ROOT.TH1F("", "", 50, 0, 50)

Chi2 of vertex fit inside beampipe.

Definition at line 68 of file v0ValidationCreatePlots.py.

◆ hist_chi2_outside

hist_chi2_outside = ROOT.TH1F("", "", 50, 0, 50)

Chi2 of vertex fit outside beampipe.

Definition at line 70 of file v0ValidationCreatePlots.py.

◆ hist_invariant_mass

hist_invariant_mass = ROOT.TH1F("", "", 60, 0.470, 0.530)

Invariant mass histogram.

Definition at line 61 of file v0ValidationCreatePlots.py.

◆ hist_invariant_mass_res

hist_invariant_mass_res = ROOT.TH1F("", "", 40, -0.02, 0.02)

Invariant mass residual histogram.

Definition at line 63 of file v0ValidationCreatePlots.py.

◆ hist_invariant_mass_residuum

hist_invariant_mass_residuum = ROOT.TH1F("", "", 60, -0.05, 0.05)

Invariant mass residuum histogram.

Definition at line 76 of file v0ValidationCreatePlots.py.

◆ hist_mass_vs_mc_mass

hist_mass_vs_mc_mass = ROOT.TH2F("", "", 80, 0, 0.8, 80, 0, 0.8)

2D histogram; invariant mass vs reconstructed mass

Definition at line 73 of file v0ValidationCreatePlots.py.

◆ hist_p

hist_p = ROOT.TH1F("", "True P", 25, 0.0, 1.0)

MCTruth Momentum.

Definition at line 49 of file v0ValidationCreatePlots.py.

◆ hist_p_found

hist_p_found = ROOT.TH1F("", "Found P", 25, 0.0, 1.0)

Reconstructed Momentum.

Definition at line 58 of file v0ValidationCreatePlots.py.

◆ hist_p_residuum

hist_p_residuum = ROOT.TH1F("", "", 60, -0.05, 0.05)

Momentum residuum histogram.

Definition at line 84 of file v0ValidationCreatePlots.py.

◆ hist_phi

hist_phi = ROOT.TH1F("", "True Phi", 36, -180, 180)

MCTruth Phi.

Definition at line 47 of file v0ValidationCreatePlots.py.

◆ hist_phi_found

hist_phi_found = ROOT.TH1F("", "Found Phi", 36, -180, 180)

Reconstructed Phi.

Definition at line 56 of file v0ValidationCreatePlots.py.

◆ hist_phi_residuum

hist_phi_residuum = ROOT.TH1F("", "", 60, -0.1, 0.1)

Phi residuum histogram.

Definition at line 82 of file v0ValidationCreatePlots.py.

◆ hist_r

hist_r = ROOT.TH1F("", "True R", 20, 0, 20)

MCTruth R.

Definition at line 43 of file v0ValidationCreatePlots.py.

◆ hist_r_found

hist_r_found = ROOT.TH1F("", "Found R", 20, 0, 20)

Reconstructed R.

Definition at line 52 of file v0ValidationCreatePlots.py.

◆ hist_r_residuum

hist_r_residuum = ROOT.TH1F("", "", 60, -0.1, 0.1)

R residuum histogram.

Definition at line 78 of file v0ValidationCreatePlots.py.

◆ hist_theta

hist_theta = ROOT.TH1F("", "True Theta", 26, 20, 150)

MCTruth Theta.

Definition at line 45 of file v0ValidationCreatePlots.py.

◆ hist_theta_found

hist_theta_found = ROOT.TH1F("", "Found Theta", 26, 20, 150)

Reconstructed Theta.

Definition at line 54 of file v0ValidationCreatePlots.py.

◆ hist_theta_residuum

hist_theta_residuum = ROOT.TH1F("", "", 60, -0.1, 0.1)

Theta residuum histogram.

Definition at line 80 of file v0ValidationCreatePlots.py.

◆ input_file

input_file = input_file

Input file.

Definition at line 38 of file v0ValidationCreatePlots.py.

◆ output_file

output_file = output_file

Output file.

Definition at line 40 of file v0ValidationCreatePlots.py.

The documentation for this class was generated from the following file:

tracking/validation/v0ValidationCreatePlots.py

Public Member Functions

Static Public Member Functions

Public Attributes

Detailed Description

Constructor & Destructor Documentation

◆ __init__()

Member Function Documentation

◆ collect_histograms()

◆ efficiency_plot()

◆ histogram_2d_plot()

◆ histogram_plot()

◆ plot()

Member Data Documentation

◆ hist_chi2

◆ hist_chi2_inside

◆ hist_chi2_outside

◆ hist_invariant_mass

◆ hist_invariant_mass_res

◆ hist_invariant_mass_residuum

◆ hist_mass_vs_mc_mass

◆ hist_p

◆ hist_p_found

◆ hist_p_residuum

◆ hist_phi

◆ hist_phi_found

◆ hist_phi_residuum

◆ hist_r

◆ hist_r_found

◆ hist_r_residuum

◆ hist_theta

◆ hist_theta_found

◆ hist_theta_residuum

◆ input_file

◆ output_file

◆ init()