modify_crosssec_run3.py

# pylint: disable=missing-function-docstring
"""
file:  modify_input.py
brief: Perform adjustments on the input Run 3 cross section histogram.
usage: python3 modify_crosssec_run3.py file.root my_histo out_histo file_out.root
author: Maja Karwowska <mkarwowska@cern.ch>, Warsaw University of Technology
"""

import argparse

from ROOT import (  # pylint: disable=import-error,no-name-in-module
    TFile,
    gROOT,
)

# 2024 values for LHC22o
MLHEP_EV_SEL = 20430386.
NORM = 47092223769.611162532
BR = 0.0623

# 2025 values for LHC23_pass4_thin
MLHEP_EV_SEL = 258442910841. # 2 x 10^1
NORM = 3.0077675e+11

# 2025 values for multiplicity analysis
EV_SEL_MULT = 290860860000.
NORM_MB = 249371059919
NORM_2 = 37884927886
EV_FACTOR_2 = 0.85
NORM_3 = 50023302929
EV_FACTOR_3 = 0.91
NORM_4 = 49545723906
EV_FACTOR_4 = 0.96
NORM_5 = 49300695562
EV_FACTOR_5 = 0.98
NORM_6 = 22192632583
EV_FACTOR_6 = 0.99
NORM_7 = 2476292886
EV_FACTOR_7 = 1.0

def main():
    """
    Main function.
    """
    gROOT.SetBatch(True)

    parser = argparse.ArgumentParser(description="Arguments to pass")
    parser.add_argument("filename", help="input file with histogram")
    parser.add_argument("histname", help="histogram name")
    parser.add_argument("outhistname", help="outhistogram name")
    parser.add_argument("outname", help="output file for the new histogram")
    args = parser.parse_args()

    with TFile(args.filename) as fin, TFile(args.outname, "recreate") as fout:
        hist = fin.Get(args.histname)
        hist2 = hist.Clone(args.outhistname)
        hist2.SetDirectory(0)
        #hist2.Scale(0.000001 * MLHEP_EV_SEL / NORM)
        #hist2.Scale(0.000001)
        #hist.Scale(1./59400000000) # luminosity scaling, lumi in pb
        #hist.Scale(BR) # BR scaling back

        hist2.Scale(EV_SEL_MULT / NORM_7)
        hist2.Scale(EV_FACTOR_7)
        fout.cd()
        hist2.Write()


if __name__ == "__main__":
    main()