modularAnalysis.py

#!/usr/bin/env python3
 
 
 
"""
This module defines wrapper functions around the analysis modules.
"""
 
import b2bii
from basf2 import register_module, create_path
from basf2 import B2INFO, B2WARNING, B2ERROR, B2FATAL
import basf2
import subprocess
 
 
def setAnalysisConfigParams(configParametersAndValues, path):
    """
    Sets analysis configuration parameters.
 
    These are:
 
    - 'tupleStyle': 'Default' (default) or 'Laconic'
 
      - defines the style of the branch name in the ntuple
 
    - 'mcMatchingVersion': Specifies what version of mc matching algorithm is going to be used:
 
      - 'Belle' - analysis of Belle MC
      - 'BelleII' (default) - all other cases
 
    @param configParametersAndValues dictionary of parameters and their values of the form {param1: value, param2: value, ...)
    @param modules are added to this path
    """
 
    conf = register_module('AnalysisConfiguration')
 
    allParameters = ['tupleStyle', 'mcMatchingVersion']
 
    keys = configParametersAndValues.keys()
    for key in keys:
        if key not in allParameters:
            allParametersString = ', '.join(allParameters)
            B2ERROR('Invalid analysis configuration parameter: ' + key + '.\n'
                    'Please use one of the following: ' + allParametersString)
 
    for param in allParameters:
        if param in configParametersAndValues:
            conf.param(param, configParametersAndValues.get(param))
 
    path.add_module(conf)
 
 
def inputMdst(filename, path, environmentType='default', skipNEvents=0, entrySequence=None, *, parentLevel=0, **kwargs):
    """
    Loads the specified :ref:`mDST <mdst>` (or :ref:`uDST <analysis_udstoutput>`) file with the RootInput module.
 
    The correct environment (e.g. magnetic field settings) is determined from
    ``environmentType``.  Options are either: 'default' (for Belle II MC and
    data: falls back to database), 'Belle': for analysis of converted Belle 1
    data and MC.
 
    Parameters:
        filename (str): the name of the file to be loaded
        path (basf2.Path): modules are added to this path
        environmentType (str): type of the environment to be loaded (either 'default' or 'Belle')
        skipNEvents (int): N events of the input file are skipped
        entrySequence (str): The number sequences (e.g. 23:42,101) defining the entries which are processed.
        parentLevel (int): Number of generations of parent files (files used as input when creating a file) to be read
    """
 
    # FIXME remove this check of "filename" at release-07
    if filename == 'default':
        B2FATAL("""
We have simplified the arguments to inputMdst! If you are running on Belle II
data or MC, you don't have to use "default" any more.
Please replace:
   inputMdst("default", "/your/input/file.root", path=mypath)
With:
   inputMdst("/your/input/file.root", path=mypath)
                """)
    elif filename == "Belle":
        B2FATAL("""
We have reordered the arguments to inputMdst! If you are running on Belle 1
data or MC, you need to specify the 'environmentType'.
Please replace:
   inputMdst("Belle", "/your/input/file.root", path=mypath)
With:
   inputMdst("/your/input/file.root", path=mypath, environmentType='Belle')
                """)
    elif filename in [f"MC{i}" for i in range(5, 10)]:
        B2FATAL(f"We no longer support the MC version {filename}. Sorry.")
 
    if entrySequence is not None:
        entrySequence = [entrySequence]
 
    inputMdstList([filename], path, environmentType, skipNEvents, entrySequence, parentLevel=parentLevel, **kwargs)
 
 
def inputMdstList(
        filelist,
        path,
        environmentType='default',
        skipNEvents=0,
        entrySequences=None,
        *,
        parentLevel=0,
        useB2BIIDBCache=True):
    """
    Loads the specified list of :ref:`mDST <mdst>` (or :ref:`uDST <analysis_udstoutput>`) files with the RootInput module.
 
    The correct environment (e.g. magnetic field settings) is determined from
    ``environmentType``.  Options are either: 'default' (for Belle II MC and
    data: falls back to database), 'Belle': for analysis of converted Belle 1
    data and MC.
 
    Parameters:
        filelist (list(str)): the filename list of files to be loaded
        path (basf2.Path): modules are added to this path
        environmentType (str): type of the environment to be loaded (either 'default' or 'Belle')
        skipNEvents (int): N events of the input files are skipped
        entrySequences (list(str)): The number sequences (e.g. 23:42,101) defining
            the entries which are processed for each inputFileName.
        parentLevel (int): Number of generations of parent files (files used as input when creating a file) to be read
        useB2BIIDBCache (bool): Loading of local KEKCC database (only to be deactivated in very special cases)
    """
 
    # FIXME remove this check of "filename" at release-07
    if filelist == 'default':
        B2FATAL("""
We have simplified the arguments to inputMdstList! If you are running on
Belle II data or MC, you don't have to use "default" any more.
Please replace:
   inputMdstList("default", list_of_your_files, path=mypath)
With:
   inputMdstList(list_of_your_files, path=mypath)
                """)
    elif filelist == "Belle":
        B2FATAL("""
We have reordered the arguments to inputMdstList! If you are running on
Belle 1 data or MC, you need to specify the 'environmentType'.
Please replace:
   inputMdstList("Belle", list_of_your_files, path=mypath)
With:
   inputMdstList(list_of_your_files, path=mypath, environmentType='Belle')
                """)
    elif filelist in [f"MC{i}" for i in range(5, 10)]:
        B2FATAL(f"We no longer support the MC version {filelist}. Sorry.")
 
    roinput = register_module('RootInput')
    roinput.param('inputFileNames', filelist)
    roinput.param('skipNEvents', skipNEvents)
    if entrySequences is not None:
        roinput.param('entrySequences', entrySequences)
    roinput.param('parentLevel', parentLevel)
 
    path.add_module(roinput)
    path.add_module('ProgressBar')
 
    if environmentType == 'Belle':
        # Belle 1 constant magnetic field
        # -------------------------------
        # n.b. slightly unfortunate syntax: the MagneticField is a member of the
        # Belle2 namespace but will be set to the Belle 1 values
        from ROOT import Belle2  # reduced scope of potentially-misbehaving import
        from ROOT.Math import XYZVector
        belle1_field = Belle2.MagneticField()
        belle1_field.addComponent(Belle2.MagneticFieldComponentConstant(XYZVector(0, 0, 1.5 * Belle2.Unit.T)))
        Belle2.DBStore.Instance().addConstantOverride("MagneticField", belle1_field, False)
        # also set the MC matching for Belle 1
        setAnalysisConfigParams({'mcMatchingVersion': 'Belle'}, path)
        b2bii.setB2BII()
        if useB2BIIDBCache:
            basf2.conditions.metadata_providers = ["/sw/belle/b2bii/database/conditions/b2bii.sqlite"]
            basf2.conditions.payload_locations = ["/sw/belle/b2bii/database/conditions/"]
 
 
def outputMdst(filename, path):
    """
    Saves mDST (mini-Data Summary Tables) to the output root file.
 
    .. warning::
 
        This function is kept for backward-compatibility.
        Better to use `mdst.add_mdst_output` directly.
 
    """
 
    import mdst
    mdst.add_mdst_output(path, mc=True, filename=filename)
 
 
def outputUdst(filename, particleLists=None, includeArrays=None, path=None, dataDescription=None):
    """
    Save uDST (user-defined Data Summary Tables) = MDST + Particles + ParticleLists
    The charge-conjugate lists of those given in particleLists are also stored.
    Additional Store Arrays and Relations to be stored can be specified via includeArrays
    list argument.
 
    Note:
        This does not reduce the amount of Particle objects saved,
        see `udst.add_skimmed_udst_output` for a function that does.
 
    """
 
    import udst
    udst.add_udst_output(
        path=path, filename=filename, particleLists=particleLists,
        additionalBranches=includeArrays, dataDescription=dataDescription)
 
 
def outputIndex(filename, path, includeArrays=None, keepParents=False, mc=True):
    """
    Write out all particle lists as an index file to be reprocessed using parentLevel flag.
    Additional branches necessary for file to be read are automatically included.
    Additional Store Arrays and Relations to be stored can be specified via includeArrays
    list argument.
 
    @param str filename the name of the output index file
    @param str path modules are added to this path
    @param list(str) includeArrays: datastore arrays/objects to write to the output
        file in addition to particle lists and related information
    @param bool keepParents whether the parents of the input event will be saved as the parents of the same event
        in the output index file. Useful if you are only adding more information to another index file
    @param bool mc whether the input data is MC or not
    """
 
    if includeArrays is None:
        includeArrays = []
 
    # Module to mark all branches to not be saved except particle lists
    onlyPLists = register_module('OnlyWriteOutParticleLists')
    path.add_module(onlyPLists)
 
    # Set up list of all other branches we need to make index file complete
    partBranches = [
        'Particles',
        'ParticlesToMCParticles',
        'ParticlesToPIDLikelihoods',
        'ParticleExtraInfoMap',
        'EventExtraInfo'
    ]
    branches = ['EventMetaData']
    persistentBranches = ['FileMetaData']
    if mc:
        branches += []
        # persistentBranches += ['BackgroundInfos']
    branches += partBranches
    branches += includeArrays
 
    r1 = register_module('RootOutput')
    r1.param('outputFileName', filename)
    r1.param('additionalBranchNames', branches)
    r1.param('branchNamesPersistent', persistentBranches)
    r1.param('keepParents', keepParents)
    path.add_module(r1)
 
 
def setupEventInfo(noEvents, path):
    """
    Prepare to generate events. This function sets up the EventInfoSetter.
    You should call this before adding a generator from generators.
    The experiment and run numbers are set to 0 (run independent generic MC in phase 3).
    https://xwiki.desy.de/xwiki/rest/p/59192
 
    Parameters:
        noEvents (int): number of events to be generated
        path (basf2.Path): modules are added to this path
    """
 
    evtnumbers = register_module('EventInfoSetter')
    evtnumbers.param('evtNumList', [noEvents])
    evtnumbers.param('runList', [0])
    evtnumbers.param('expList', [0])
    path.add_module(evtnumbers)
 
 
def loadGearbox(path, silence_warning=False):
    """
    Loads Gearbox module to the path.
 
    Warning:
        Should be used in a job with *cosmic event generation only*
 
    Needed for scripts which only generate cosmic events in order to
    load the geometry.
 
    @param path modules are added to this path
    @param silence_warning stops a verbose warning message if you know you want to use this function
    """
 
    if not silence_warning:
        B2WARNING("""You are overwriting the geometry from the database with Gearbox.
          This is fine if you're generating cosmic events. But in most other cases you probably don't want this.
 
          If you're really sure you know what you're doing you can suppress this message with:
 
          >>> loadGearbox(silence_warning=True)
 
                  """)
 
    paramloader = register_module('Gearbox')
    path.add_module(paramloader)
 
 
def printPrimaryMCParticles(path, **kwargs):
    """
    Prints all primary MCParticles, that is particles from
    the physics generator and not particles created by the simulation
 
    This is equivalent to `printMCParticles(onlyPrimaries=True, path=path) <printMCParticles>` and additional
    keyword arguments are just forwarded to that function
    """
 
    return printMCParticles(onlyPrimaries=True, path=path, **kwargs)
 
 
def printMCParticles(onlyPrimaries=False, maxLevel=-1, path=None, *,
                     showProperties=False, showMomenta=False, showVertices=False, showStatus=False, suppressPrint=False):
    """
    Prints all MCParticles or just primary MCParticles up to specified level. -1 means no limit.
 
    By default this will print a tree of just the particle names and their pdg
    codes in the event, for example ::
 
        [INFO] Content of MCParticle list
        ├── e- (11)
        ├── e+ (-11)
        ╰── Upsilon(4S) (300553)
            ├── B+ (521)
            │   ├── anti-D_0*0 (-10421)
            │   │   ├── D- (-411)
            │   │   │   ├── K*- (-323)
            │   │   │   │   ├── anti-K0 (-311)
            │   │   │   │   │   ╰── K_S0 (310)
            │   │   │   │   │       ├── pi+ (211)
            │   │   │   │   │       │   ╰╶╶ p+ (2212)
            │   │   │   │   │       ╰── pi- (-211)
            │   │   │   │   │           ├╶╶ e- (11)
            │   │   │   │   │           ├╶╶ n0 (2112)
            │   │   │   │   │           ├╶╶ n0 (2112)
            │   │   │   │   │           ╰╶╶ n0 (2112)
            │   │   │   │   ╰── pi- (-211)
            │   │   │   │       ├╶╶ anti-nu_mu (-14)
            │   │   │   │       ╰╶╶ mu- (13)
            │   │   │   │           ├╶╶ nu_mu (14)
            │   │   │   │           ├╶╶ anti-nu_e (-12)
            │   │   │   │           ╰╶╶ e- (11)
            │   │   │   ╰── K_S0 (310)
            │   │   │       ├── pi0 (111)
            │   │   │       │   ├── gamma (22)
            │   │   │       │   ╰── gamma (22)
            │   │   │       ╰── pi0 (111)
            │   │   │           ├── gamma (22)
            │   │   │           ╰── gamma (22)
            │   │   ╰── pi+ (211)
            │   ├── mu+ (-13)
            │   │   ├╶╶ anti-nu_mu (-14)
            │   │   ├╶╶ nu_e (12)
            │   │   ╰╶╶ e+ (-11)
            │   ├── nu_mu (14)
            │   ╰── gamma (22)
            ...
 
 
    There's a distinction between primary and secondary particles. Primary    particles are the ones created by the physics generator while secondary
    particles are ones generated by the simulation of the detector interaction.
 
    Secondaries are indicated with a dashed line leading to the particle name
    and if the output is to the terminal they will be printed in red. If
    ``onlyPrimaries`` is True they will not be included in the tree.
 
    On demand, extra information on all the particles can be displayed by
    enabling any of the ``showProperties``, ``showMomenta``, ``showVertices``
    and ``showStatus`` flags. Enabling all of them will look like
    this::
 
        ...
        ╰── pi- (-211)
            │ mass=0.14 energy=0.445 charge=-1 lifetime=6.36
            │ p=(0.257, -0.335, 0.0238) |p|=0.423
            │ production vertex=(0.113, -0.0531, 0.0156), time=0.00589
            │ status flags=PrimaryParticle, StableInGenerator, StoppedInDetector
            │ list index=48
            │
            ╰╶╶ n0 (2112)
                mass=0.94 energy=0.94 charge=0 lifetime=5.28e+03
                p=(-0.000238, -0.0127, 0.0116) |p|=0.0172
                production vertex=(144, 21.9, -1.29), time=39
                status flags=StoppedInDetector
                creation process=HadronInelastic
                list index=66
 
    The first line of extra information is enabled by ``showProperties``, the
    second line by ``showMomenta``, the third line by ``showVertices`` and the
    last two lines by ``showStatus``. Note that all values are given in Belle II
    standard units, that is GeV, centimeter and nanoseconds.
 
    The depth of the tree can be limited with the ``maxLevel`` argument: If it's
    bigger than zero it will limit the tree to the given number of generations.
    A visual indicator will be added after each particle which would have
    additional daughters that are skipped due to this limit. An example event
    with ``maxLevel=3`` is given below. In this case only the tau neutrino and
    the pion don't have additional daughters. ::
 
        [INFO] Content of MCParticle list
        ├── e- (11)
        ├── e+ (-11)
        ╰── Upsilon(4S) (300553)
            ├── B+ (521)
            │   ├── anti-D*0 (-423) → …
            │   ├── tau+ (-15) → …
            │   ╰── nu_tau (16)
            ╰── B- (-521)
                ├── D*0 (423) → …
                ├── K*- (-323) → …
                ├── K*+ (323) → …
                ╰── pi- (-211)
 
    The same information will be stored in the branch ``__MCDecayString__`` of
    TTree created by `VariablesToNtuple` or `VariablesToEventBasedTree` module.
    This branch is automatically created when `PrintMCParticles` modules is called.
    Printing the information on the log message can be suppressed if ``suppressPrint``
    is True, while the branch ``__MCDecayString__``. This option helps to reduce the
    size of the log message.
 
    Parameters:
        onlyPrimaries (bool): If True show only primary particles, that is particles coming from
            the generator and not created by the simulation.
        maxLevel (int): If 0 or less print the whole tree, otherwise stop after n generations
        showProperties (bool): If True show mass, energy and charge of the particles
        showMomenta (bool): if True show the momenta of the particles
        showVertices (bool): if True show production vertex and production time of all particles
        showStatus (bool): if True show some status information on the particles.
            For secondary particles this includes creation process.
        suppressPrint (bool): if True printing the information on the log message is suppressed.
            Even if True, the branch ``__MCDecayString__`` is created.
    """
 
    return path.add_module(
        "PrintMCParticles",
        onlyPrimaries=onlyPrimaries,
        maxLevel=maxLevel,
        showProperties=showProperties,
        showMomenta=showMomenta,
        showVertices=showVertices,
        showStatus=showStatus,
        suppressPrint=suppressPrint,
    )
 
 
def correctBrems(outputList,
                 inputList,
                 gammaList,
                 maximumAcceptance=3.0,
                 multiplePhotons=False,
                 usePhotonOnlyOnce=True,
                 writeOut=False,
                 path=None):
    """
    For each particle in the given ``inputList``, copies it to the ``outputList`` and adds the
    4-vector of the photon(s) in the ``gammaList`` which has(have) a weighted named relation to
    the particle's track, set by the ``ECLTrackBremFinder`` module during reconstruction.
 
    Warning:
        This can only work if the mdst file contains the *Bremsstrahlung* named relation. Official MC samples
        up to and including MC12 and proc9 **do not** contain this. Newer production campaigns (from proc10 and MC13) do.
        However, studies by the tau WG revealed that the cuts applied by the ``ECLTrackBremFinder`` module are too tight.
        These will be loosened but this will only have effect with proc13 and MC15.
        If your analysis is very sensitive to the Bremsstrahlung corrections, it is advised to use `correctBremsBelle`.
 
    Information:
        A detailed description of how the weights are set can be found directly at the documentation of the
        `BremsFinder` module.
 
        Please note that a new particle is always generated, with the old particle and -if found- one or more
        photons as daughters.
 
        The ``inputList`` should contain particles with associated tracks. Otherwise, the module will exit with an error.
 
        The ``gammaList`` should contain photons. Otherwise, the module will exit with an error.
 
    @param outputList   The output particle list name containing the corrected particles
    @param inputList    The initial particle list name containing the particles to correct. *It should already exist.*
    @param gammaList    The photon list containing possibly bremsstrahlung photons; *It should already exist.*
    @param maximumAcceptance Maximum value of the relation weight. Should be a number between [0,3)
    @param multiplePhotons Whether to use only one photon (the one with the smallest acceptance) or as many as possible
    @param usePhotonOnlyOnce If true, each brems candidate is used to correct only the track with the smallest relation weight
    @param writeOut      Whether `RootOutput` module should save the created ``outputList``
    @param path          The module is added to this path
    """
 
    import b2bii
    if b2bii.isB2BII():
        B2ERROR("The BremsFinder can only be run over Belle II data.")
 
    bremscorrector = register_module('BremsFinder')
    bremscorrector.set_name('bremsCorrector_' + outputList)
    bremscorrector.param('inputList', inputList)
    bremscorrector.param('outputList', outputList)
    bremscorrector.param('gammaList', gammaList)
    bremscorrector.param('maximumAcceptance', maximumAcceptance)
    bremscorrector.param('multiplePhotons', multiplePhotons)
    bremscorrector.param('usePhotonOnlyOnce', usePhotonOnlyOnce)
    bremscorrector.param('writeOut', writeOut)
    path.add_module(bremscorrector)
 
 
def copyList(outputListName, inputListName, writeOut=False, path=None):
    """
    Copy all Particle indices from input ParticleList to the output ParticleList.
    Note that the Particles themselves are not copied. The original and copied
    ParticleLists will point to the same Particles.
 
    @param ouputListName copied ParticleList
    @param inputListName original ParticleList to be copied
    @param writeOut      whether RootOutput module should save the created ParticleList
    @param path          modules are added to this path
    """
 
    copyLists(outputListName, [inputListName], writeOut, path)
 
 
def correctBremsBelle(outputListName,
                      inputListName,
                      gammaListName,
                      multiplePhotons=True,
                      angleThreshold=0.05,
                      usePhotonOnlyOnce=False,
                      writeOut=False,
                      path=None):
    """
    Run the Belle - like brems finding on the ``inputListName`` of charged particles.
    Adds all photons in ``gammaListName`` to a copy of the charged particle that are within
    ``angleThreshold``.
 
    Tip:
        Studies by the tau WG show that using a rather wide opening angle (up to
        0.2 rad) and rather low energetic photons results in good correction.
        However, this should only serve as a starting point for your own studies
        because the optimal criteria are likely mode-dependent
 
    Parameters:
       outputListName (str): The output charged particle list containing the corrected charged particles
       inputListName (str): The initial charged particle list containing the charged particles to correct.
       gammaListName (str): The gammas list containing possibly radiative gammas, should already exist.
       multiplePhotons (bool): How many photons should be added to the charged particle? nearest one -> False,
             add all the photons within the cone -> True
       angleThreshold (float): The maximum angle in radians between the charged particle and the (radiative)
              gamma to be accepted.
       writeOut (bool): whether RootOutput module should save the created ParticleList
       usePhotonOnlyOnce (bool): If true, a photon is used for correction of the closest charged particle in the inputList.
                                 If false, a photon is allowed to be used for correction multiple times (Default).
 
       Warning:
           One cannot use a photon twice to reconstruct a composite particle. Thus, for example, if ``e+`` and ``e-`` are corrected
           with a ``gamma``, the pair of ``e+`` and ``e-`` cannot form a ``J/psi -> e+ e-`` candidate.
 
       path (basf2.Path): modules are added to this path
    """
 
    fsrcorrector = register_module('BelleBremRecovery')
    fsrcorrector.set_name('BelleFSRCorrection_' + outputListName)
    fsrcorrector.param('inputListName', inputListName)
    fsrcorrector.param('outputListName', outputListName)
    fsrcorrector.param('gammaListName', gammaListName)
    fsrcorrector.param('multiplePhotons', multiplePhotons)
    fsrcorrector.param('angleThreshold', angleThreshold)
    fsrcorrector.param('usePhotonOnlyOnce', usePhotonOnlyOnce)
    fsrcorrector.param('writeOut', writeOut)
    path.add_module(fsrcorrector)
 
 
def copyLists(outputListName, inputListNames, writeOut=False, path=None):
    """
    Copy all Particle indices from all input ParticleLists to the
    single output ParticleList.
    Note that the Particles themselves are not copied.
    The original and copied ParticleLists will point to the same Particles.
 
    Duplicates are removed based on the first-come, first-served principle.
    Therefore, the order of the input ParticleLists matters.
 
    .. seealso::
        If you want to select the best duplicate based on another criterion, have
        a look at the function `mergeListsWithBestDuplicate`.
 
    .. note::
        Two particles that differ only by the order of their daughters are
        considered duplicates and one of them will be removed.
 
    @param ouputListName copied ParticleList
    @param inputListName vector of original ParticleLists to be copied
    @param writeOut      whether RootOutput module should save the created ParticleList
    @param path          modules are added to this path
    """
 
    pmanipulate = register_module('ParticleListManipulator')
    pmanipulate.set_name('PListCopy_' + outputListName)
    pmanipulate.param('outputListName', outputListName)
    pmanipulate.param('inputListNames', inputListNames)
    pmanipulate.param('writeOut', writeOut)
    path.add_module(pmanipulate)
 
 
def copyParticles(outputListName, inputListName, writeOut=False, path=None):
    """
    Create copies of Particles given in the input ParticleList and add them to the output ParticleList.
 
    The existing relations of the original Particle (or it's (grand-)^n-daughters)
    are copied as well. Note that only the relation is copied and that the related
    object is not. Copied particles are therefore related to the *same* object as
    the original ones.
 
    @param ouputListName new ParticleList filled with copied Particles
    @param inputListName input ParticleList with original Particles
    @param writeOut      whether RootOutput module should save the created ParticleList
    @param path          modules are added to this path
    """
 
    # first copy original particles to the new ParticleList
    pmanipulate = register_module('ParticleListManipulator')
    pmanipulate.set_name('PListCopy_' + outputListName)
    pmanipulate.param('outputListName', outputListName)
    pmanipulate.param('inputListNames', [inputListName])
    pmanipulate.param('writeOut', writeOut)
    path.add_module(pmanipulate)
 
    # now replace original particles with their copies
    pcopy = register_module('ParticleCopier')
    pcopy.param('inputListNames', [outputListName])
    path.add_module(pcopy)
 
 
def cutAndCopyLists(outputListName, inputListNames, cut, writeOut=False, path=None):
    """
    Copy candidates from all lists in ``inputListNames`` to
    ``outputListName`` if they pass ``cut`` (given selection criteria).
 
    Note:
        Note that the Particles themselves are not copied.
        The original and copied ParticleLists will point to the same Particles.
 
    Example:
        Require energetic pions safely inside the cdc
 
        .. code-block:: python
 
            cutAndCopyLists("pi+:energeticPions", ["pi+:good", "pi+:loose"], "[E > 2] and thetaInCDCAcceptance", path=mypath)
 
    Warning:
        You must use square braces ``[`` and ``]`` for conditional statements.
 
    Parameters:
        outputListName (str): the new ParticleList name
        inputListName (list(str)): list of input ParticleList names
        cut (str): Candidates that do not pass these selection criteria are removed from the ParticleList
        writeOut (bool): whether RootOutput module should save the created ParticleList
        path (basf2.Path): modules are added to this path
    """
 
    pmanipulate = register_module('ParticleListManipulator')
    pmanipulate.set_name('PListCutAndCopy_' + outputListName)
    pmanipulate.param('outputListName', outputListName)
    pmanipulate.param('inputListNames', inputListNames)
    pmanipulate.param('cut', cut)
    pmanipulate.param('writeOut', writeOut)
    path.add_module(pmanipulate)
 
 
def cutAndCopyList(outputListName, inputListName, cut, writeOut=False, path=None):
    """
    Copy candidates from ``inputListName`` to ``outputListName`` if they pass
    ``cut`` (given selection criteria).
 
    Note:
        Note the Particles themselves are not copied.
        The original and copied ParticleLists will point to the same Particles.
 
    Example:
        require energetic pions safely inside the cdc
 
        .. code-block:: python
 
            cutAndCopyList("pi+:energeticPions", "pi+:loose", "[E > 2] and thetaInCDCAcceptance", path=mypath)
 
    Warning:
        You must use square braces ``[`` and ``]`` for conditional statements.
 
    Parameters:
        outputListName (str): the new ParticleList name
        inputListName (str): input ParticleList name
        cut (str): Candidates that do not pass these selection criteria are removed from the ParticleList
        writeOut (bool): whether RootOutput module should save the created ParticleList
        path (basf2.Path): modules are added to this path
    """
 
    cutAndCopyLists(outputListName, [inputListName], cut, writeOut, path)
 
 
def removeTracksForTrackingEfficiencyCalculation(inputListNames, fraction, path=None):
    """
    Randomly remove tracks from the provided particle lists to estimate the tracking efficiency.
    Takes care of the duplicates, if any.
 
    Parameters:
        inputListNames (list(str)): input particle list names
        fraction (float): fraction of particles to be removed randomly
        path (basf2.Path): module is added to this path
    """
 
    trackingefficiency = register_module('TrackingEfficiency')
    trackingefficiency.param('particleLists', inputListNames)
    trackingefficiency.param('frac', fraction)
    path.add_module(trackingefficiency)
 
 
def scaleTrackMomenta(inputListNames, scale=float('nan'), payloadName="", scalingFactorName="SF", path=None):
    """
    Scale momenta of the particles according to a scaling factor scale.
    This scaling factor can either be given as constant number or as the name of the payload which contains
    the variable scale factors.
    If the particle list contains composite particles, the momenta of the track-based daughters are scaled.
    Subsequently, the momentum of the mother particle is updated as well.
 
    Parameters:
        inputListNames (list(str)): input particle list names
        scale (float): scaling factor (1.0 -- no scaling)
        payloadName (str): name of the payload which contains the phase-space dependent scaling factors
        scalingFactorName (str): name of scaling factor variable in the payload.
        path (basf2.Path): module is added to this path
    """
 
    import b2bii
    if b2bii.isB2BII():
        B2ERROR("The tracking momentum scaler can only be run over Belle II data.")
 
    TrackingMomentumScaleFactors = register_module('TrackingMomentumScaleFactors')
    TrackingMomentumScaleFactors.param('particleLists', inputListNames)
    TrackingMomentumScaleFactors.param('scale', scale)
    TrackingMomentumScaleFactors.param('payloadName', payloadName)
    TrackingMomentumScaleFactors.param('scalingFactorName', scalingFactorName)
 
    path.add_module(TrackingMomentumScaleFactors)
 
 
def correctTrackEnergy(inputListNames, correction=float('nan'), payloadName="", correctionName="SF", path=None):
    """
    Correct the energy loss of tracks according to a 'correction' value.
    This correction can either be given as constant number or as the name of the payload which contains
    the variable corrections.
    If the particle list contains composite particles, the momenta of the track-based daughters are corrected.
    Subsequently, the momentum of the mother particle is updated as well.
 
    Parameters:
        inputListNames (list(str)): input particle list names
        correction (float): correction value to be subtracted to the particle energy (0.0 -- no correction)
        payloadName (str): name of the payload which contains the phase-space dependent scaling factors
        correctionName (str): name of correction variable in the payload.
        path (basf2.Path): module is added to this path
    """
 
    import b2bii
    if b2bii.isB2BII():
        B2ERROR("The tracking energy correction can only be run over Belle II data.")
 
    TrackingEnergyLossCorrection = register_module('TrackingEnergyLossCorrection')
    TrackingEnergyLossCorrection.param('particleLists', inputListNames)
    TrackingEnergyLossCorrection.param('correction', correction)
    TrackingEnergyLossCorrection.param('payloadName', payloadName)
    TrackingEnergyLossCorrection.param('correctionName', correctionName)
 
    path.add_module(TrackingEnergyLossCorrection)
 
 
def smearTrackMomenta(inputListNames, payloadName="", smearingFactorName="smear", path=None):
    """
    Smear the momenta of the particles according the values read from the given payload.
    If the particle list contains composite particles, the momenta of the track-based daughters are smeared.
    Subsequently, the momentum of the mother particle is updated as well.
 
    Parameters:
        inputListNames (list(str)): input particle list names
        payloadName (str): name of the payload which contains the smearing values
        smearingFactorName (str): name of smearing factor variable in the payload.
        path (basf2.Path): module is added to this path
    """
 
    TrackingMomentumScaleFactors = register_module('TrackingMomentumScaleFactors')
    TrackingMomentumScaleFactors.param('particleLists', inputListNames)
    TrackingMomentumScaleFactors.param('payloadName', payloadName)
    TrackingMomentumScaleFactors.param('smearingFactorName',  smearingFactorName)
 
    path.add_module(TrackingMomentumScaleFactors)
 
 
def mergeListsWithBestDuplicate(outputListName,
                                inputListNames,
                                variable,
                                preferLowest=True,
                                writeOut=False,
                                ignoreMotherFlavor=False,
                                path=None):
    """
    Merge input ParticleLists into one output ParticleList. Only the best
    among duplicates is kept. The lowest or highest value (configurable via
    preferLowest) of the provided variable determines which duplicate is the
    best.
 
    @param ouputListName name of merged ParticleList
    @param inputListName vector of original ParticleLists to be merged
    @param variable      variable to determine best duplicate
    @param preferLowest  whether lowest or highest value of variable should be preferred
    @param writeOut      whether RootOutput module should save the created ParticleList
    @param ignoreMotherFlavor whether the flavor of the mother particle is ignored when trying to find duplicates
    @param path          modules are added to this path
    """
 
    pmanipulate = register_module('ParticleListManipulator')
    pmanipulate.set_name('PListMerger_' + outputListName)
    pmanipulate.param('outputListName', outputListName)
    pmanipulate.param('inputListNames', inputListNames)
    pmanipulate.param('variable', variable)
    pmanipulate.param('preferLowest', preferLowest)
    pmanipulate.param('writeOut', writeOut)
    pmanipulate.param('ignoreMotherFlavor', ignoreMotherFlavor)
    path.add_module(pmanipulate)
 
 
def fillSignalSideParticleList(outputListName, decayString, path):
    """
    This function should only be used in the ROE path, that is a path
    that is executed for each ROE object in the DataStore.
 
    Example: fillSignalSideParticleList('gamma:sig','B0 -> K*0 ^gamma', roe_path)
 
    Function will create a ParticleList with name 'gamma:sig' which will be filled
    with the existing photon Particle, being the second daughter of the B0 candidate
    to which the ROE object has to be related.
 
    @param ouputListName name of the created ParticleList
    @param decayString specify Particle to be added to the ParticleList
    """
 
    pload = register_module('SignalSideParticleListCreator')
    pload.set_name('SSParticleList_' + outputListName)
    pload.param('particleListName', outputListName)
    pload.param('decayString', decayString)
    path.add_module(pload)
 
 
def fillParticleLists(decayStringsWithCuts, writeOut=False, path=None, enforceFitHypothesis=False,
                      loadPhotonsFromKLM=False):
    """
    Creates Particles of the desired types from the corresponding ``mdst`` dataobjects,
    loads them to the ``StoreArray<Particle>`` and fills the ParticleLists.
 
    The multiple ParticleLists with their own selection criteria are specified
    via list tuples (decayString, cut), for example
 
    .. code-block:: python
 
        kaons = ('K+:mykaons', 'kaonID>0.1')
        pions = ('pi+:mypions','pionID>0.1')
        fillParticleLists([kaons, pions], path=mypath)
 
    If you are unsure what selection you want, you might like to see the
    :doc:`StandardParticles` functions.
 
    The type of the particles to be loaded is specified via the decayString module parameter.
    The type of the ``mdst`` dataobject that is used as an input is determined from the type of
    the particle. The following types of the particles can be loaded:
 
    * charged final state particles (input ``mdst`` type = Tracks)
        - e+, mu+, pi+, K+, p, deuteron (and charge conjugated particles)
 
    * neutral final state particles
        - "gamma"           (input ``mdst`` type = ECLCluster)
        - "K_S0", "Lambda0" (input ``mdst`` type = V0)
        - "K_L0"            (input ``mdst`` type = KLMCluster or ECLCluster)
 
    Note:
        For "K_S0" and "Lambda0" you must specify the daughter ordering.
 
    For example, to load V0s as :math:`\\Lambda^0\\to p^+\\pi^-` decays from V0s:
 
    .. code-block:: python
 
        v0lambdas = ('Lambda0 -> p+ pi-', '0.9 < M < 1.3')
        fillParticleLists([kaons, pions, v0lambdas], path=mypath)
 
    Tip:
        Gammas can also be loaded from KLMClusters by explicitly setting the
        parameter ``loadPhotonsFromKLM`` to True. However, this should only be
        done in selected use-cases and the effect should be studied carefully.
 
    Tip:
        For "K_L0" it is now possible to load from ECLClusters, to revert to
        the old (Belle) behavior, you can require ``'isFromKLM > 0'``.
 
    .. code-block:: python
 
        klongs = ('K_L0', 'isFromKLM > 0')
        fillParticleLists([kaons, pions, klongs], path=mypath)
 
    * Charged kinks final state particles (input ``mdst`` type = Kink)
 
    Note:
        To reconstruct charged particle kink you must specify the daughter.
 
    For example, to load Kinks as :math:`K^- \\to \\pi^-\\pi^0` decays from Kinks:
 
    .. code-block:: python
 
        kinkKaons = ('K- -> pi-', yourCut)
        fillParticleLists([kaons, pions, v0lambdas, kinkKaons], path=mypath)
 
 
    Parameters:
        decayStringsWithCuts (list): A list of python ntuples of (decayString, cut).
                                     The decay string determines the type of Particle
                                     and the name of the ParticleList.
                                     If the input MDST type is V0 the whole
                                     decay chain needs to be specified, so that
                                     the user decides and controls the daughters
                                     ' order (e.g. ``K_S0 -> pi+ pi-``).
                                     If the input MDST type is Kink the decay chain needs to be specified
                                     with only one daughter (e.g. ``K- -> pi-``).
                                     The cut is the selection criteria
                                     to be added to the ParticleList. It can be an empty string.
        writeOut (bool):             whether RootOutput module should save the created ParticleList
        path (basf2.Path):           modules are added to this path
        enforceFitHypothesis (bool): If true, Particles will be created only for the tracks which have been fitted
                                     using a mass hypothesis of the exact type passed to fillParticleLists().
                                     If enforceFitHypothesis is False (the default) the next closest fit hypothesis
                                     in terms of mass difference will be used if the fit using exact particle
                                     type is not available.
        loadPhotonsFromKLM (bool):   If true, photon candidates will be created from KLMClusters as well.
    """
 
    pload = register_module('ParticleLoader')
    pload.set_name('ParticleLoader_' + 'PLists')
    pload.param('decayStrings', [decayString for decayString, cut in decayStringsWithCuts])
    pload.param('writeOut', writeOut)
    pload.param("enforceFitHypothesis", enforceFitHypothesis)
    path.add_module(pload)
 
    from ROOT import Belle2
    decayDescriptor = Belle2.DecayDescriptor()
    for decayString, cut in decayStringsWithCuts:
        if not decayDescriptor.init(decayString):
            raise ValueError("Invalid decay string")
        # need to check some logic to unpack possible scenarios
        if decayDescriptor.getNDaughters() > 0:
            # ... then we have an actual decay in the decay string which must be a V0 (if more than 1 daughter)
            # or a kink (if 1 daughter)
            # the particle loader automatically calls this "V0" or "kink", respectively, so we have to copy over
            # the list to name/format that user wants
            if (decayDescriptor.getNDaughters() == 1) & (decayDescriptor.getMother().getLabel() != 'kink'):
                copyList(decayDescriptor.getMother().getFullName(), decayDescriptor.getMother().getName() + ':kink',
                         writeOut, path)
            if (decayDescriptor.getNDaughters() > 1) & (decayDescriptor.getMother().getLabel() != 'V0'):
                copyList(decayDescriptor.getMother().getFullName(), decayDescriptor.getMother().getName() + ':V0', writeOut, path)
        elif decayDescriptor.getMother().getLabel() != 'all':
            # then we have a non-V0/kink particle which the particle loader automatically calls "all"
            # as with the special V0 and kink cases we have to copy over the list to the name/format requested
            copyList(decayString, decayDescriptor.getMother().getName() + ':all', writeOut, path)
 
        # optionally apply a cut
        if cut != "":
            applyCuts(decayDescriptor.getMother().getFullName(), cut, path)
 
        if decayString.startswith("gamma"):
            # keep KLM-source photons as a experts-only for now: they are loaded by the particle loader,
            # but the user has to explicitly request them.
            if not loadPhotonsFromKLM:
                applyCuts(decayString, 'isFromECL', path)
 
 
def fillParticleList(decayString, cut, writeOut=False, path=None, enforceFitHypothesis=False,
                     loadPhotonsFromKLM=False):
    """
    Creates Particles of the desired type from the corresponding ``mdst`` dataobjects,
    loads them to the StoreArray<Particle> and fills the ParticleList.
 
    See also:
        the :doc:`StandardParticles` functions.
 
    The type of the particles to be loaded is specified via the decayString module parameter.
    The type of the ``mdst`` dataobject that is used as an input is determined from the type of
    the particle. The following types of the particles can be loaded:
 
    * charged final state particles (input ``mdst`` type = Tracks)
        - e+, mu+, pi+, K+, p, deuteron (and charge conjugated particles)
 
    * neutral final state particles
        - "gamma"           (input ``mdst`` type = ECLCluster)
        - "K_S0", "Lambda0" (input ``mdst`` type = V0)
        - "K_L0"            (input ``mdst`` type = KLMCluster or ECLCluster)
 
    Note:
        For "K_S0" and "Lambda0" you must specify the daughter ordering.
 
    For example, to load V0s as :math:`\\Lambda^0\\to p^+\\pi^-` decays from V0s:
 
    .. code-block:: python
 
        fillParticleList('Lambda0 -> p+ pi-', '0.9 < M < 1.3', path=mypath)
 
    Tip:
        Gammas can also be loaded from KLMClusters by explicitly setting the
        parameter ``loadPhotonsFromKLM`` to True. However, this should only be
        done in selected use-cases and the effect should be studied carefully.
 
    Tip:
        For "K_L0" it is now possible to load from ECLClusters, to revert to
        the old (Belle) behavior, you can require ``'isFromKLM > 0'``.
 
    .. code-block:: python
 
        fillParticleList('K_L0', 'isFromKLM > 0', path=mypath)
 
    * Charged kinks final state particles (input ``mdst`` type = Kink)
 
    .. note::
        To reconstruct charged particle kink you must specify the daughter.
 
    For example, to load Kinks as :math:`K^- \\to \\pi^-\\pi^0` decays from Kinks:
 
    .. code-block:: python
 
        fillParticleList('K- -> pi-', yourCut, path=mypath)
 
 
    Parameters:
        decayString (str):           Type of Particle and determines the name of the ParticleList.
                                     If the input MDST type is V0 the whole decay chain needs to be specified, so that
                                     the user decides and controls the daughters' order (e.g. ``K_S0 -> pi+ pi-``).
                                     If the input MDST type is Kink the decay chain needs to be specified
                                     with only one daughter (e.g. ``K- -> pi-``).
        cut (str):                   Particles need to pass these selection criteria to be added to the ParticleList
        writeOut (bool):             whether RootOutput module should save the created ParticleList
        path (basf2.Path):           modules are added to this path
        enforceFitHypothesis (bool): If true, Particles will be created only for the tracks which have been fitted
                                     using a mass hypothesis of the exact type passed to fillParticleLists().
                                     If enforceFitHypothesis is False (the default) the next closest fit hypothesis
                                     in terms of mass difference will be used if the fit using exact particle
                                     type is not available.
        loadPhotonsFromKLM (bool):   If true, photon candidates will be created from KLMClusters as well.
    """
 
    pload = register_module('ParticleLoader')
    pload.set_name('ParticleLoader_' + decayString)
    pload.param('decayStrings', [decayString])
    pload.param('writeOut', writeOut)
    pload.param("enforceFitHypothesis", enforceFitHypothesis)
    path.add_module(pload)
 
    # need to check some logic to unpack possible scenarios
    from ROOT import Belle2
    decayDescriptor = Belle2.DecayDescriptor()
    if not decayDescriptor.init(decayString):
        raise ValueError("Invalid decay string")
    if decayDescriptor.getNDaughters() > 0:
        # ... then we have an actual decay in the decay string which must be a V0 (if more than 1 daughter)
        # or a kink (if 1 daughter)
        # the particle loader automatically calls this "V0" or "kink", respectively, so we have to copy over
        # the list to name/format that user wants
        if (decayDescriptor.getNDaughters() == 1) & (decayDescriptor.getMother().getLabel() != 'kink'):
            copyList(decayDescriptor.getMother().getFullName(), decayDescriptor.getMother().getName() + ':kink',
                     writeOut, path)
        if (decayDescriptor.getNDaughters() > 1) & (decayDescriptor.getMother().getLabel() != 'V0'):
            copyList(decayDescriptor.getMother().getFullName(), decayDescriptor.getMother().getName() + ':V0', writeOut,
                     path)
    elif decayDescriptor.getMother().getLabel() != 'all':
        # then we have a non-V0/kink particle which the particle loader automatically calls "all"
        # as with the special V0 and kink cases we have to copy over the list to the name/format requested
        copyList(decayString, decayDescriptor.getMother().getName() + ':all', writeOut, path)
 
    # optionally apply a cut
    if cut != "":
        applyCuts(decayDescriptor.getMother().getFullName(), cut, path)
 
    if decayString.startswith("gamma"):
        # keep KLM-source photons as a experts-only for now: they are loaded by the particle loader,
        # but the user has to explicitly request them.
        if not loadPhotonsFromKLM:
            applyCuts(decayString, 'isFromECL', path)
 
 
def fillParticleListWithTrackHypothesis(decayString,
                                        cut,
                                        hypothesis,
                                        writeOut=False,
                                        enforceFitHypothesis=False,
                                        path=None):
    """
    As fillParticleList, but if used for a charged FSP, loads the particle with the requested hypothesis if available
 
    @param decayString   specifies type of Particles and determines the name of the ParticleList
    @param cut           Particles need to pass these selection criteria to be added to the ParticleList
    @param hypothesis    the PDG code of the desired track hypothesis
    @param writeOut      whether RootOutput module should save the created ParticleList
    @param enforceFitHypothesis If true, Particles will be created only for the tracks which have been fitted
                                using a mass hypothesis of the exact type passed to fillParticleLists().
                                If enforceFitHypothesis is False (the default) the next closest fit hypothesis
                                in terms of mass difference will be used if the fit using exact particle
                                type is not available.
    @param path          modules are added to this path
    """
 
    pload = register_module('ParticleLoader')
    pload.set_name('ParticleLoader_' + decayString)
    pload.param('decayStrings', [decayString])
    pload.param('trackHypothesis', hypothesis)
    pload.param('writeOut', writeOut)
    pload.param("enforceFitHypothesis", enforceFitHypothesis)
    path.add_module(pload)
 
    from ROOT import Belle2
    decayDescriptor = Belle2.DecayDescriptor()
    if not decayDescriptor.init(decayString):
        raise ValueError("Invalid decay string")
    if decayDescriptor.getMother().getLabel() != 'all':
        # the particle loader automatically calls particle lists of charged FSPs "all"
        # so we have to copy over the list to the name/format requested
        copyList(decayString, decayDescriptor.getMother().getName() + ':all', writeOut, path)
 
    # apply a cut if a non-empty cut string is provided
    if cut != "":
        applyCuts(decayString, cut, path)
 
 
def fillConvertedPhotonsList(decayString, cut, writeOut=False, path=None):
    """
    Creates photon Particle object for each e+e- combination in the V0 StoreArray.
 
    Note:
        You must specify the daughter ordering.
 
    .. code-block:: python
 
        fillConvertedPhotonsList('gamma:converted -> e+ e-', '', path=mypath)
 
    Parameters:
        decayString (str): Must be gamma to an e+e- pair. You must specify the daughter ordering.
                           Will also determine the name of the particleList.
        cut (str):         Particles need to pass these selection criteria to be added to the ParticleList
        writeOut (bool):   whether RootOutput module should save the created ParticleList
        path (basf2.Path): modules are added to this path
 
    """
 
    import b2bii
    if b2bii.isB2BII():
        B2ERROR('For Belle converted photons are available in the pre-defined list "gamma:v0mdst".')
 
    pload = register_module('ParticleLoader')
    pload.set_name('ParticleLoader_' + decayString)
    pload.param('decayStrings', [decayString])
    pload.param('addDaughters', True)
    pload.param('writeOut', writeOut)
    path.add_module(pload)
 
    from ROOT import Belle2
    decayDescriptor = Belle2.DecayDescriptor()
    if not decayDescriptor.init(decayString):
        raise ValueError("Invalid decay string")
    if decayDescriptor.getMother().getLabel() != 'V0':
        # the particle loader automatically calls converted photons "V0" so we have to copy over
        # the list to name/format that user wants
        copyList(decayDescriptor.getMother().getFullName(), decayDescriptor.getMother().getName() + ':V0', writeOut, path)
 
    # apply a cut if a non-empty cut string is provided
    if cut != "":
        applyCuts(decayDescriptor.getMother().getFullName(), cut, path)
 
 
def fillParticleListFromROE(decayString,
                            cut,
                            maskName='all',
                            sourceParticleListName='',
                            useMissing=False,
                            writeOut=False,
                            path=None):
    """
    Creates Particle object for each ROE of the desired type found in the
    StoreArray<RestOfEvent>, loads them to the StoreArray<Particle>
    and fills the ParticleList. If useMissing is True, then the missing
    momentum is used instead of ROE.
 
    The type of the particles to be loaded is specified via the decayString module parameter.
 
    @param decayString             specifies type of Particles and determines the name of the ParticleList.
                                   Source ROEs can be taken as a daughter list, for example:
                                   'B0:tagFromROE -> B0:signal'
    @param cut                     Particles need to pass these selection criteria to be added to the ParticleList
    @param maskName                Name of the ROE mask to use
    @param sourceParticleListName  Use related ROEs to this particle list as a source
    @param useMissing              Use missing momentum instead of ROE momentum
    @param writeOut                whether RootOutput module should save the created ParticleList
    @param path                    modules are added to this path
    """
 
    pload = register_module('ParticleLoader')
    pload.set_name('ParticleLoader_' + decayString)
    pload.param('decayStrings', [decayString])
    pload.param('writeOut', writeOut)
    pload.param('roeMaskName', maskName)
    pload.param('useMissing', useMissing)
    pload.param('sourceParticleListName', sourceParticleListName)
    pload.param('useROEs', True)
    path.add_module(pload)
 
    from ROOT import Belle2
    decayDescriptor = Belle2.DecayDescriptor()
    if not decayDescriptor.init(decayString):
        raise ValueError("Invalid decay string")
 
    # apply a cut if a non-empty cut string is provided
    if cut != "":
        applyCuts(decayDescriptor.getMother().getFullName(), cut, path)
 
 
def fillParticleListFromDummy(decayString,
                              mdstIndex=0,
                              covMatrix=10000.,
                              treatAsInvisible=True,
                              writeOut=False,
                              path=None):
    """
    Creates a ParticleList and fills it with dummy Particles. For self-conjugated Particles one dummy
    Particle is created, for Particles that are not self-conjugated one Particle and one anti-Particle is
    created. The four-momentum is set to zero.
 
    The type of the particles to be loaded is specified via the decayString module parameter.
 
    @param decayString             specifies type of Particles and determines the name of the ParticleList
    @param mdstIndex               sets the mdst index of Particles
    @param covMatrix               sets the value of the diagonal covariance matrix of Particles
    @param treatAsInvisible        whether treeFitter should treat the Particles as invisible
    @param writeOut                whether RootOutput module should save the created ParticleList
    @param path                    modules are added to this path
    """
 
    pload = register_module('ParticleLoader')
    pload.set_name('ParticleLoader_' + decayString)
    pload.param('decayStrings', [decayString])
    pload.param('useDummy', True)
    pload.param('dummyMDSTIndex', mdstIndex)
    pload.param('dummyCovMatrix', covMatrix)
    pload.param('dummyTreatAsInvisible', treatAsInvisible)
    pload.param('writeOut', writeOut)
    path.add_module(pload)
 
 
def fillParticleListFromMC(decayString,
                           cut,
                           addDaughters=False,
                           skipNonPrimaryDaughters=False,
                           writeOut=False,
                           path=None,
                           skipNonPrimary=False,
                           skipInitial=True):
    """
    Creates Particle object for each MCParticle of the desired type found in the StoreArray<MCParticle>,
    loads them to the StoreArray<Particle> and fills the ParticleList.
 
    The type of the particles to be loaded is specified via the decayString module parameter.
 
    @param decayString             specifies type of Particles and determines the name of the ParticleList
    @param cut                     Particles need to pass these selection criteria to be added to the ParticleList
    @param addDaughters            adds the bottom part of the decay chain of the particle to the datastore and
                                   sets mother-daughter relations
    @param skipNonPrimaryDaughters if true, skip non primary daughters, useful to study final state daughter particles
    @param writeOut                whether RootOutput module should save the created ParticleList
    @param path                    modules are added to this path
    @param skipNonPrimary          if true, skip non primary particle
    @param skipInitial             if true, skip initial particles
    """
 
    pload = register_module('ParticleLoader')
    pload.set_name('ParticleLoader_' + decayString)
    pload.param('decayStrings', [decayString])
    pload.param('addDaughters', addDaughters)
    pload.param('skipNonPrimaryDaughters', skipNonPrimaryDaughters)
    pload.param('writeOut', writeOut)
    pload.param('useMCParticles', True)
    pload.param('skipNonPrimary', skipNonPrimary)
    pload.param('skipInitial', skipInitial)
    path.add_module(pload)
 
    from ROOT import Belle2
    decayDescriptor = Belle2.DecayDescriptor()
    if not decayDescriptor.init(decayString):
        raise ValueError("Invalid decay string")
 
    # apply a cut if a non-empty cut string is provided
    if cut != "":
        applyCuts(decayString, cut, path)
 
 
def fillParticleListsFromMC(decayStringsWithCuts,
                            addDaughters=False,
                            skipNonPrimaryDaughters=False,
                            writeOut=False,
                            path=None,
                            skipNonPrimary=False,
                            skipInitial=True):
    """
    Creates Particle object for each MCParticle of the desired type found in the StoreArray<MCParticle>,
    loads them to the StoreArray<Particle> and fills the ParticleLists.
 
    The types of the particles to be loaded are specified via the (decayString, cut) tuples given in a list.
    For example:
 
    .. code-block:: python
 
        kaons = ('K+:gen', '')
        pions = ('pi+:gen', 'pionID>0.1')
        fillParticleListsFromMC([kaons, pions], path=mypath)
 
    .. tip::
        Daughters of ``Lambda0`` are not primary, but ``Lambda0`` is not final state particle.
        Thus, when one reconstructs a particle from ``Lambda0``, that is created with
        ``addDaughters=True`` and ``skipNonPrimaryDaughters=True``, the particle always has ``isSignal==0``.
        Please set options for ``Lambda0`` to use MC-matching variables properly as follows,
        ``addDaughters=True`` and ``skipNonPrimaryDaughters=False``.
 
    @param decayString             specifies type of Particles and determines the name of the ParticleList
    @param cut                     Particles need to pass these selection criteria to be added to the ParticleList
    @param addDaughters            adds the bottom part of the decay chain of the particle to the datastore and
                                   sets mother-daughter relations
    @param skipNonPrimaryDaughters if true, skip non primary daughters, useful to study final state daughter particles
    @param writeOut                whether RootOutput module should save the created ParticleList
    @param path                    modules are added to this path
    @param skipNonPrimary          if true, skip non primary particle
    @param skipInitial             if true, skip initial particles
    """
 
    pload = register_module('ParticleLoader')
    pload.set_name('ParticleLoader_' + 'PLists')
    pload.param('decayStrings', [decayString for decayString, cut in decayStringsWithCuts])
    pload.param('addDaughters', addDaughters)
    pload.param('skipNonPrimaryDaughters', skipNonPrimaryDaughters)
    pload.param('writeOut', writeOut)
    pload.param('useMCParticles', True)
    pload.param('skipNonPrimary', skipNonPrimary)
    pload.param('skipInitial', skipInitial)
    path.add_module(pload)
 
    from ROOT import Belle2
    decayDescriptor = Belle2.DecayDescriptor()
    for decayString, cut in decayStringsWithCuts:
        if not decayDescriptor.init(decayString):
            raise ValueError("Invalid decay string")
 
        # apply a cut if a non-empty cut string is provided
        if cut != "":
            applyCuts(decayString, cut, path)
 
 
def fillParticleListFromChargedCluster(outputParticleList,
                                       inputParticleList,
                                       cut,
                                       useOnlyMostEnergeticECLCluster=True,
                                       writeOut=False,
                                       path=None):
    """
    Creates the Particle object from ECLCluster and KLMCluster that are being matched with the Track of inputParticleList.
 
    @param outputParticleList       The output ParticleList. Only neutral final state particles are supported.
    @param inputParticleList        The input ParticleList that is required to have the relation to the Track object.
    @param cut                      Particles need to pass these selection criteria to be added to the ParticleList
    @param useOnlyMostEnergeticECLCluster If True, only the most energetic ECLCluster among ones that are matched with the Track is
                                          used. If False, all matched ECLClusters are loaded. The default is True. Regardless of
                                          this option, the KLMCluster is loaded.
    @param writeOut                whether RootOutput module should save the created ParticleList
    @param path                    modules are added to this path
    """
 
    pload = register_module('ParticleLoader')
    pload.set_name('ParticleLoader_' + outputParticleList)
 
    pload.param('decayStrings', [outputParticleList])
    pload.param('sourceParticleListName', inputParticleList)
    pload.param('writeOut', writeOut)
    pload.param('loadChargedCluster', True)
    pload.param('useOnlyMostEnergeticECLCluster', useOnlyMostEnergeticECLCluster)
    path.add_module(pload)
 
    # apply a cut if a non-empty cut string is provided
    if cut != "":
        applyCuts(outputParticleList, cut, path)
 
 
def extractParticlesFromROE(particleLists,
                            signalSideParticleList=None,
                            maskName='all',
                            writeOut=False,
                            path=None):
    """
    Extract Particle objects that belong to the Rest-Of-Events and fill them into the ParticleLists.
    The types of the particles other than those specified by ``particleLists`` are not stored.
    If one creates a ROE with ``fillWithMostLikely=True`` via `buildRestOfEvent`, for example,
    one should create particleLists for not only ``pi+``, ``gamma``, ``K_L0`` but also other charged final state particles.
 
    When one calls the function in the main path, one has to set the argument ``signalSideParticleList`` and the signal side
    ParticleList must have only one candidate.
 
    .. code-block:: python
 
        buildRestOfEvent('B0:sig', fillWithMostLikely=True, path=mypath)
 
        roe_path = create_path()
        deadEndPath = create_path()
        signalSideParticleFilter('B0:sig', '', roe_path, deadEndPath)
 
        plists = ['%s:in_roe' % ptype for ptype in ['pi+', 'gamma', 'K_L0', 'K+', 'p+', 'e+', 'mu+']]
        extractParticlesFromROE(plists, maskName='all', path=roe_path)
 
        # one can analyze these ParticleLists in the roe_path
 
        mypath.for_each('RestOfEvent', 'RestOfEvents', roe_path)
 
        rankByLowest('B0:sig', 'deltaE', numBest=1, path=mypath)
        extractParticlesFromROE(plists, signalSideParticleList='B0:sig', maskName='all', path=mypath)
 
        # one can analyze these ParticleLists in the main path
 
 
    @param particleLists (str or list(str)) Name of output ParticleLists
    @param signalSideParticleList (str)     Name of signal side ParticleList
    @param maskName (str)                   Name of the ROE mask to be applied on Particles
    @param writeOut (bool)                  whether RootOutput module should save the created ParticleList
    @param path (basf2.Path)                modules are added to this path
    """
 
    if isinstance(particleLists, str):
        particleLists = [particleLists]
 
    pext = register_module('ParticleExtractorFromROE')
    pext.set_name('ParticleExtractorFromROE_' + '_'.join(particleLists))
    pext.param('outputListNames', particleLists)
    if signalSideParticleList is not None:
        pext.param('signalSideParticleListName', signalSideParticleList)
    pext.param('maskName', maskName)
    pext.param('writeOut', writeOut)
    path.add_module(pext)
 
 
def applyCuts(list_name, cut, path):
    """
    Removes particle candidates from ``list_name`` that do not pass ``cut``
    (given selection criteria).
 
    Example:
        require energetic pions safely inside the cdc
 
        .. code-block:: python
 
            applyCuts("pi+:mypions", "[E > 2] and thetaInCDCAcceptance", path=mypath)
 
    Warning:
        You must use square braces ``[`` and ``]`` for conditional statements.
 
    Parameters:
        list_name (str): input ParticleList name
        cut (str): Candidates that do not pass these selection criteria are removed from the ParticleList
        path (basf2.Path): modules are added to this path
    """
 
    pselect = register_module('ParticleSelector')
    pselect.set_name('ParticleSelector_applyCuts_' + list_name)
    pselect.param('decayString', list_name)
    pselect.param('cut', cut)
    path.add_module(pselect)
 
 
def applyEventCuts(cut, path, metavariables=None):
    """
    Removes events that do not pass the ``cut`` (given selection criteria).
 
    Example:
        continuum events (in mc only) with more than 5 tracks
 
        .. code-block:: python
 
            applyEventCuts("[nTracks > 5] and [isContinuumEvent], path=mypath)
 
    .. warning::
      Only event-based variables are allowed in this function
      and only square brackets ``[`` and ``]`` for conditional statements.
 
    Parameters:
        cut (str): Events that do not pass these selection criteria are skipped
        path (basf2.Path): modules are added to this path
        metavariables (list(str)): List of meta variables to be considered in decomposition of cut
    """
 
    import b2parser
    from variables import variables
 
    def find_vars(t: tuple, var_list: list, meta_list: list) -> None:
        """ Recursive helper function to find variable names """
        if not isinstance(t, tuple):
            return
        if t[0] == b2parser.B2ExpressionParser.node_types['IdentifierNode']:
            var_list += [t[1]]
            return
        if t[0] == b2parser.B2ExpressionParser.node_types['FunctionNode']:
            meta_list.append(list(t[1:]))
            return
        for i in t:
            if isinstance(i, tuple):
                find_vars(i, var_list, meta_list)
 
    def check_variable(var_list: list, metavar_ids: list) -> None:
        for var_string in var_list:
            # Check if the var_string is alias
            orig_name = variables.resolveAlias(var_string)
            if orig_name != var_string:
                var_list_temp = []
                meta_list_temp = []
                find_vars(b2parser.parse(orig_name), var_list_temp, meta_list_temp)
 
                check_variable(var_list_temp, metavar_ids)
                check_meta(meta_list_temp, metavar_ids)
            else:
                # Get the variable
                var = variables.getVariable(var_string)
                if event_var_id not in var.description:
                    B2ERROR(f'Variable {var_string} is not an event-based variable! "\
                    "Please check your inputs to the applyEventCuts method!')
 
    def check_meta(meta_list: list, metavar_ids: list) -> None:
        for meta_string_list in meta_list:
            var_list_temp = []
            while meta_string_list[0] in metavar_ids:
                # remove special meta variable
                meta_string_list.pop(0)
                for meta_string in meta_string_list[0].split(","):
                    find_vars(b2parser.parse(meta_string), var_list_temp, meta_string_list)
                if len(meta_string_list) > 0:
                    meta_string_list.pop(0)
                if len(meta_string_list) == 0:
                    break
                if len(meta_string_list) > 1:
                    meta_list += meta_string_list[1:]
                if isinstance(meta_string_list[0], list):
                    meta_string_list = [element for element in meta_string_list[0]]
 
            check_variable(var_list_temp, metavar_ids)
 
            if len(meta_string_list) == 0:
                continue
            elif len(meta_string_list) == 1:
                var = variables.getVariable(meta_string_list[0])
            else:
                var = variables.getVariable(meta_string_list[0], meta_string_list[1].split(","))
            # Check if the variable's description contains event-based marker
            if event_var_id in var.description:
                continue
            # Throw an error message if non event-based variable is used
            B2ERROR(f'Variable {var.name} is not an event-based variable! Please check your inputs to the applyEventCuts method!')
 
    event_var_id = '[Eventbased]'
    metavar_ids = ['formula', 'abs',
                   'cos', 'acos',
                   'tan', 'atan',
                   'sin', 'asin',
                   'exp', 'log', 'log10',
                   'min', 'max',
                   'isNAN']
    if metavariables:
        metavar_ids += metavariables
 
    var_list = []
    meta_list = []
    find_vars(b2parser.parse(cut), var_list=var_list, meta_list=meta_list)
 
    if len(var_list) == 0 and len(meta_list) == 0:
        B2WARNING(f'Cut string "{cut}" has no variables for applyEventCuts helper function!')
 
    check_variable(var_list, metavar_ids)
    check_meta(meta_list, metavar_ids)
 
    eselect = register_module('VariableToReturnValue')
    eselect.param('variable', 'passesEventCut(' + cut + ')')
    path.add_module(eselect)
    empty_path = create_path()
    eselect.if_value('<1', empty_path)
 
 
def reconstructDecay(decayString,
                     cut,
                     dmID=0,
                     writeOut=False,
                     path=None,
                     candidate_limit=None,
                     ignoreIfTooManyCandidates=True,
                     chargeConjugation=True,
                     allowChargeViolation=False):
    r"""
    Creates new Particles by making combinations of existing Particles - it reconstructs unstable particles via their specified
    decay mode, e.g. in form of a :ref:`DecayString`: :code:`D0 -> K- pi+` or :code:`B+ -> anti-D0 pi+`, ... All possible
    combinations are created (particles are used only once per candidate) and combinations that pass the specified selection
    criteria are saved to a newly created (mother) ParticleList. By default the charge conjugated decay is reconstructed as well
    (meaning that the charge conjugated mother list is created as well) but this can be deactivated.
 
    One can use an ``@``-sign to mark a particle as unspecified for inclusive analyses,
    e.g. in a DecayString: :code:`'@Xsd -> K+ pi-'`.
 
    .. seealso:: :ref:`Marker_of_unspecified_particle`
 
    .. warning::
        The input ParticleLists are typically ordered according to the upstream reconstruction algorithm.
        Therefore, if you combine two or more identical particles in the decay chain you should not expect to see the same
        distribution for the daughter kinematics as they may be sorted by geometry, momentum etc.
 
        For example, in the decay :code:`D0 -> pi0 pi0` the momentum distributions of the two ``pi0`` s are not identical.
        This can be solved by manually randomising the lists before combining.
 
    See Also:
 
        * `Particle combiner how does it work? <https://questions.belle2.org/question/4318/particle-combiner-how-does-it-work/>`_
        * `Identical particles in decay chain <https://questions.belle2.org/question/5724/identical-particles-in-decay-chain/>`_
 
    @param decayString :ref:`DecayString` specifying what kind of the decay should be reconstructed
                       (from the DecayString the mother and daughter ParticleLists are determined)
    @param cut         created (mother) Particles are added to the mother ParticleList if they
                       pass give cuts (in VariableManager style) and rejected otherwise
    @param dmID        user specified decay mode identifier
    @param writeOut    whether RootOutput module should save the created ParticleList
    @param path        modules are added to this path
    @param candidate_limit Maximum amount of candidates to be reconstructed. If
                       the number of candidates is exceeded a Warning will be
                       printed.
                       By default, all these candidates will be removed and event will be ignored.
                       This behaviour can be changed by \'ignoreIfTooManyCandidates\' flag.
                       If no value is given the amount is limited to a sensible
                       default. A value <=0 will disable this limit and can
                       cause huge memory amounts so be careful.
    @param ignoreIfTooManyCandidates whether event should be ignored or not if number of reconstructed
                       candidates reaches limit. If event is ignored, no candidates are reconstructed,
                       otherwise, number of candidates in candidate_limit is reconstructed.
    @param chargeConjugation boolean to decide whether charge conjugated mode should be reconstructed as well (on by default)
    @param allowChargeViolation whether the decay string needs to conserve the electric charge
    """
 
    pmake = register_module('ParticleCombiner')
    pmake.set_name('ParticleCombiner_' + decayString)
    pmake.param('decayString', decayString)
    pmake.param('cut', cut)
    pmake.param('decayMode', dmID)
    pmake.param('writeOut', writeOut)
    if candidate_limit is not None:
        pmake.param("maximumNumberOfCandidates", candidate_limit)
    pmake.param("ignoreIfTooManyCandidates", ignoreIfTooManyCandidates)
    pmake.param('chargeConjugation', chargeConjugation)
    pmake.param("allowChargeViolation", allowChargeViolation)
    path.add_module(pmake)
 
 
def combineAllParticles(inputParticleLists, outputList, cut='', writeOut=False, path=None):
    """
    Creates a new Particle as the combination of all Particles from all
    provided inputParticleLists. However, each particle is used only once
    (even if duplicates are provided) and the combination has to pass the
    specified selection criteria to be saved in the newly created (mother)
    ParticleList.
 
    @param inputParticleLists List of input particle lists which are combined to the new Particle
    @param outputList         Name of the particle combination created with this module
    @param cut                created (mother) Particle is added to the mother ParticleList if it passes
                              these given cuts (in VariableManager style) and is rejected otherwise
    @param writeOut           whether RootOutput module should save the created ParticleList
    @param path               module is added to this path
    """
 
    pmake = register_module('AllParticleCombiner')
    pmake.set_name('AllParticleCombiner_' + outputList)
    pmake.param('inputListNames', inputParticleLists)
    pmake.param('outputListName', outputList)
    pmake.param('cut', cut)
    pmake.param('writeOut', writeOut)
    path.add_module(pmake)
 
 
def reconstructMissingKlongDecayExpert(decayString,
                                       cut,
                                       dmID=0,
                                       writeOut=False,
                                       path=None,
                                       recoList="_reco"):
    """
    Creates a list of K_L0's and of B -> K_L0 + X, with X being a fully-reconstructed state.
    The K_L0 momentum is determined from kinematic constraints of the two-body B decay into K_L0 and X
 
    @param decayString DecayString specifying what kind of the decay should be reconstructed
                       (from the DecayString the mother and daughter ParticleLists are determined)
    @param cut         Particles are added to the K_L0 and B ParticleList if the B candidates
                       pass the given cuts (in VariableManager style) and rejected otherwise
    @param dmID        user specified decay mode identifier
    @param writeOut    whether RootOutput module should save the created ParticleList
    @param path        modules are added to this path
    @param recoList    suffix appended to original K_L0 and B ParticleList that identify the newly created K_L0 and B lists
    """
 
    pcalc = register_module('KlongMomentumCalculatorExpert')
    pcalc.set_name('KlongMomentumCalculatorExpert_' + decayString)
    pcalc.param('decayString', decayString)
    pcalc.param('writeOut', writeOut)
    pcalc.param('recoList', recoList)
    path.add_module(pcalc)
 
    rmake = register_module('KlongDecayReconstructorExpert')
    rmake.set_name('KlongDecayReconstructorExpert_' + decayString)
    rmake.param('decayString', decayString)
    rmake.param('cut', cut)
    rmake.param('decayMode', dmID)
    rmake.param('writeOut', writeOut)
    rmake.param('recoList', recoList)
    path.add_module(rmake)
 
 
def setBeamConstrainedMomentum(particleList, decayStringTarget, decayStringDaughters, path=None):
    """
    Replace the four-momentum of the target Particle by p(beam) - p(selected daughters).
    The momentum of the mother Particle will not be changed.
 
    @param particleList         mother Particlelist
    @param decayStringTarget    DecayString specifying the target particle whose momentum
                                will be updated
    @param decayStringDaughters DecayString specifying the daughter particles used to replace
                                the momentum of the target particle by p(beam)-p(daughters)
    """
 
    mod = register_module('ParticleMomentumUpdater')
    mod.set_name('ParticleMomentumUpdater' + particleList)
    mod.param('particleList', particleList)
    mod.param('decayStringTarget', decayStringTarget)
    mod.param('decayStringDaughters', decayStringDaughters)
    path.add_module(mod)
 
 
def updateKlongKinematicsExpert(particleList,
                                writeOut=False,
                                path=None):
    """
    Calculates and updates the kinematics of B->K_L0 + something else with same method as
    `reconstructMissingKlongDecayExpert`. This helps to revert the kinematics after the vertex fitting.
 
    @param particleList input ParticleList of B meson that decays to K_L0 + X
    @param writeOut     whether RootOutput module should save the ParticleList
    @param path         modules are added to this path
    """
 
    mod = register_module('KlongMomentumUpdaterExpert')
    mod.set_name('KlongMomentumUpdaterExpert_' + particleList)
    mod.param('listName', particleList)
    mod.param('writeOut', writeOut)
    path.add_module(mod)
 
 
def replaceMass(replacerName, particleLists=None, pdgCode=22, path=None):
    """
    replaces the mass of the particles inside the given particleLists
    with the invariant mass of the particle corresponding to the given pdgCode.
 
    @param particleLists new ParticleList filled with copied Particles
    @param pdgCode PDG   code for mass reference
    @param path          modules are added to this path
    """
 
    if particleLists is None:
        particleLists = []
 
    # first copy original particles to the new ParticleList
    pmassupdater = register_module('ParticleMassUpdater')
    pmassupdater.set_name('ParticleMassUpdater_' + replacerName)
    pmassupdater.param('particleLists', particleLists)
    pmassupdater.param('pdgCode', pdgCode)
    path.add_module(pmassupdater)
 
 
def reconstructRecoil(decayString,
                      cut,
                      dmID=0,
                      writeOut=False,
                      path=None,
                      candidate_limit=None,
                      allowChargeViolation=False):
    """
    Creates new Particles that recoil against the input particles.
 
    For example the decay string M -> D1 D2 D3 will:
     - create mother Particle M for each unique combination of D1, D2, D3 Particles
     - Particles D1, D2, D3 will be appended as daughters to M
     - the 4-momentum of the mother Particle M is given by
         p(M) = p(HER) + p(LER) - Sum_i p(Di)
 
    @param decayString DecayString specifying what kind of the decay should be reconstructed
                       (from the DecayString the mother and daughter ParticleLists are determined)
    @param cut         created (mother) Particles are added to the mother ParticleList if they
                       pass give cuts (in VariableManager style) and rejected otherwise
    @param dmID        user specified decay mode identifier
    @param writeOut    whether RootOutput module should save the created ParticleList
    @param path        modules are added to this path
    @param candidate_limit Maximum amount of candidates to be reconstructed. If
                       the number of candidates is exceeded no candidate will be
                       reconstructed for that event and a Warning will be
                       printed.
                       If no value is given the amount is limited to a sensible
                       default. A value <=0 will disable this limit and can
                       cause huge memory amounts so be careful.
    @param allowChargeViolation whether the decay string needs to conserve the electric charge
    """
 
    pmake = register_module('ParticleCombiner')
    pmake.set_name('ParticleCombiner_' + decayString)
    pmake.param('decayString', decayString)
    pmake.param('cut', cut)
    pmake.param('decayMode', dmID)
    pmake.param('writeOut', writeOut)
    pmake.param('recoilParticleType', 1)
    if candidate_limit is not None:
        pmake.param("maximumNumberOfCandidates", candidate_limit)
    pmake.param('allowChargeViolation', allowChargeViolation)
    path.add_module(pmake)
 
 
def reconstructRecoilDaughter(decayString,
                              cut,
                              dmID=0,
                              writeOut=False,
                              path=None,
                              candidate_limit=None,
                              allowChargeViolation=False):
    """
    Creates new Particles that are daughters of the particle reconstructed in the recoil (always assumed to be the first daughter).
 
    For example the decay string M -> D1 D2 D3 will:
     - create mother Particle M for each unique combination of D1, D2, D3 Particles
     - Particles D1, D2, D3 will be appended as daughters to M
     - the 4-momentum of the mother Particle M is given by
         p(M) = p(D1) - Sum_i p(Di), where i>1
 
    @param decayString DecayString specifying what kind of the decay should be reconstructed
                       (from the DecayString the mother and daughter ParticleLists are determined)
    @param cut         created (mother) Particles are added to the mother ParticleList if they
                       pass give cuts (in VariableManager style) and rejected otherwise
    @param dmID        user specified decay mode identifier
    @param writeOut    whether RootOutput module should save the created ParticleList
    @param path        modules are added to this path
    @param candidate_limit Maximum amount of candidates to be reconstructed. If
                       the number of candidates is exceeded no candidate will be
                       reconstructed for that event and a Warning will be
                       printed.
                       If no value is given the amount is limited to a sensible
                       default. A value <=0 will disable this limit and can
                       cause huge memory amounts so be careful.
    @param allowChargeViolation whether the decay string needs to conserve the electric charge taking into account that the first
                       daughter is actually the mother
    """
 
    pmake = register_module('ParticleCombiner')
    pmake.set_name('ParticleCombiner_' + decayString)
    pmake.param('decayString', decayString)
    pmake.param('cut', cut)
    pmake.param('decayMode', dmID)
    pmake.param('writeOut', writeOut)
    pmake.param('recoilParticleType', 2)
    if candidate_limit is not None:
        pmake.param("maximumNumberOfCandidates", candidate_limit)
    pmake.param('allowChargeViolation', allowChargeViolation)
    path.add_module(pmake)
 
 
def rankByHighest(particleList,
                  variable,
                  numBest=0,
                  outputVariable='',
                  allowMultiRank=False,
                  cut='',
                  overwriteRank=False,
                  path=None):
    """
    Ranks particles in the input list by the given variable (highest to lowest), and stores an integer rank for each Particle
    in an :b2:var:`extraInfo` field ``${variable}_rank`` starting at 1 (best).
    The list is also sorted from best to worst candidate
    (each charge, e.g. B+/B-, separately).
    This can be used to perform a best candidate selection by cutting on the corresponding rank value, or by specifying
    a non-zero value for 'numBest'.
 
    .. tip::
        Extra-info fields can be accessed by the :b2:var:`extraInfo` metavariable.
        These variable names can become clunky, so it's probably a good idea to set an alias.
        For example if you rank your B candidates by momentum,
 
        .. code:: python
 
            rankByHighest("B0:myCandidates", "p", path=mypath)
            vm.addAlias("momentumRank", "extraInfo(p_rank)")
 
 
    @param particleList     The input ParticleList
    @param variable         Variable to order Particles by.
    @param numBest          If not zero, only the $numBest Particles in particleList with rank <= numBest are kept.
    @param outputVariable   Name for the variable that will be created which contains the rank, Default is '${variable}_rank'.
    @param allowMultiRank   If true, candidates with the same value will get the same rank.
    @param cut              Only candidates passing the cut will be ranked. The others will have rank -1
    @param overwriteRank    If true, the extraInfo of rank is overwritten when the particle has already the extraInfo.
    @param path             modules are added to this path
    """
 
    bcs = register_module('BestCandidateSelection')
    bcs.set_name('BestCandidateSelection_' + particleList + '_' + variable)
    bcs.param('particleList', particleList)
    bcs.param('variable', variable)
    bcs.param('numBest', numBest)
    bcs.param('outputVariable', outputVariable)
    bcs.param('allowMultiRank', allowMultiRank)
    bcs.param('cut', cut)
    bcs.param('overwriteRank', overwriteRank)
    path.add_module(bcs)
 
 
def rankByLowest(particleList,
                 variable,
                 numBest=0,
                 outputVariable='',
                 allowMultiRank=False,
                 cut='',
                 overwriteRank=False,
                 path=None):
    """
    Ranks particles in the input list by the given variable (lowest to highest), and stores an integer rank for each Particle
    in an :b2:var:`extraInfo` field ``${variable}_rank`` starting at 1 (best).
    The list is also sorted from best to worst candidate
    (each charge, e.g. B+/B-, separately).
    This can be used to perform a best candidate selection by cutting on the corresponding rank value, or by specifying
    a non-zero value for 'numBest'.
 
    .. tip::
        Extra-info fields can be accessed by the :b2:var:`extraInfo` metavariable.
        These variable names can become clunky, so it's probably a good idea to set an alias.
        For example if you rank your B candidates by :b2:var:`dM`,
 
        .. code:: python
 
            rankByLowest("B0:myCandidates", "dM", path=mypath)
            vm.addAlias("massDifferenceRank", "extraInfo(dM_rank)")
 
 
    @param particleList     The input ParticleList
    @param variable         Variable to order Particles by.
    @param numBest          If not zero, only the $numBest Particles in particleList with rank <= numBest are kept.
    @param outputVariable   Name for the variable that will be created which contains the rank, Default is '${variable}_rank'.
    @param allowMultiRank   If true, candidates with the same value will get the same rank.
    @param cut              Only candidates passing the cut will be ranked. The others will have rank -1
    @param overwriteRank    If true, the extraInfo of rank is overwritten when the particle has already the extraInfo.
    @param path             modules are added to this path
    """
 
    bcs = register_module('BestCandidateSelection')
    bcs.set_name('BestCandidateSelection_' + particleList + '_' + variable)
    bcs.param('particleList', particleList)
    bcs.param('variable', variable)
    bcs.param('numBest', numBest)
    bcs.param('selectLowest', True)
    bcs.param('allowMultiRank', allowMultiRank)
    bcs.param('outputVariable', outputVariable)
    bcs.param('cut', cut)
    bcs.param('overwriteRank', overwriteRank)
    path.add_module(bcs)
 
 
def applyRandomCandidateSelection(particleList, path=None):
    """
    If there are multiple candidates in the provided particleList, all but one of them are removed randomly.
    This is done on a event-by-event basis.
 
    @param particleList     ParticleList for which the random candidate selection should be applied
    @param path             module is added to this path
    """
 
    rcs = register_module('BestCandidateSelection')
    rcs.set_name('RandomCandidateSelection_' + particleList)
    rcs.param('particleList', particleList)
    rcs.param('variable', 'random')
    rcs.param('selectLowest', False)
    rcs.param('allowMultiRank', False)
    rcs.param('numBest', 1)
    rcs.param('cut', '')
    rcs.param('outputVariable', '')
    path.add_module(rcs)
 
 
def printDataStore(eventNumber=-1, path=None):
    """
    Prints the contents of DataStore in the first event (or a specific event number or all events).
    Will list all objects and arrays (including size).
 
    See also:
        The command line tool: ``b2file-size``.
 
    Parameters:
        eventNumber (int): Print the datastore only for this event. The default
            (-1) prints only the first event, 0 means print for all events (can produce large output)
        path (basf2.Path): the PrintCollections module is added to this path
 
    Warning:
        This will print a lot of output if you print it for all events and process many events.
 
    """
 
    printDS = register_module('PrintCollections')
    printDS.param('printForEvent', eventNumber)
    path.add_module(printDS)
 
 
def printVariableValues(list_name, var_names, path):
    """
    Prints out values of specified variables of all Particles included in given ParticleList. For debugging purposes.
 
    @param list_name input ParticleList name
    @param var_names vector of variable names to be printed
    @param path         modules are added to this path
    """
 
    prlist = register_module('ParticlePrinter')
    prlist.set_name('ParticlePrinter_' + list_name)
    prlist.param('listName', list_name)
    prlist.param('fullPrint', False)
    prlist.param('variables', var_names)
    path.add_module(prlist)
 
 
def printList(list_name, full, path):
    """
    Prints the size and executes Particle->print() (if full=True)
    method for all Particles in given ParticleList. For debugging purposes.
 
    @param list_name input ParticleList name
    @param full      execute Particle->print() method for all Particles
    @param path      modules are added to this path
    """
 
    prlist = register_module('ParticlePrinter')
    prlist.set_name('ParticlePrinter_' + list_name)
    prlist.param('listName', list_name)
    prlist.param('fullPrint', full)
    path.add_module(prlist)
 
 
def variablesToNtuple(decayString, variables, treename='variables', filename='ntuple.root', path=None, basketsize=1600,
                      signalSideParticleList="", filenameSuffix="", useFloat=False, storeEventType=True,
                      ignoreCommandLineOverride=False):
    """
    Creates and fills a flat ntuple with the specified variables from the VariableManager.
    If a decayString is provided, then there will be one entry per candidate (for particle in list of candidates).
    If an empty decayString is provided, there will be one entry per event (useful for trigger studies, etc).
 
    Parameters:
        decayString (str): specifies type of Particles and determines the name of the ParticleList
        variables (list(str)): the list of variables (which must be registered in the VariableManager)
        treename (str): name of the ntuple tree
        filename (str): which is used to store the variables
        path (basf2.Path): the basf2 path where the analysis is processed
        basketsize (int): size of baskets in the output ntuple in bytes
        signalSideParticleList (str): The name of the signal-side ParticleList.
                                      Only valid if the module is called in a for_each loop over the RestOfEvent.
        filenameSuffix (str): suffix to be appended to the filename before ``.root``.
        useFloat (bool): Use single precision (float) instead of double precision (double)
                         for floating-point numbers.
        storeEventType (bool) : if true, the branch __eventType__ is added for the MC event type information.
                                The information is available from MC16 on.
        ignoreCommandLineOverride (bool) : if true, ignore override of file name via command line argument ``-o``.
 
    .. tip:: The output filename can be overridden using the ``-o`` argument of basf2.
    """
 
    output = register_module('VariablesToNtuple')
    output.set_name('VariablesToNtuple_' + decayString)
    output.param('particleList', decayString)
    output.param('variables', variables)
    output.param('fileName', filename)
    output.param('treeName', treename)
    output.param('basketSize', basketsize)
    output.param('signalSideParticleList', signalSideParticleList)
    output.param('fileNameSuffix', filenameSuffix)
    output.param('useFloat', useFloat)
    output.param('storeEventType', storeEventType)
    output.param('ignoreCommandLineOverride', ignoreCommandLineOverride)
    path.add_module(output)
 
 
def variablesToHistogram(decayString,
                         variables,
                         variables_2d=None,
                         filename='ntuple.root',
                         path=None, *,
                         directory=None,
                         prefixDecayString=False,
                         filenameSuffix="",
                         ignoreCommandLineOverride=False):
    """
    Creates and fills a flat ntuple with the specified variables from the VariableManager
 
    Parameters:
        decayString (str): specifies type of Particles and determines the name of the ParticleList
        variables (list(tuple))): variables + binning which must be registered in the VariableManager
        variables_2d (list(tuple)): pair of variables + binning for each which must be registered in the VariableManager
        filename (str): which is used to store the variables
        path (basf2.Path): the basf2 path where the analysis is processed
        directory (str): directory inside the output file where the histograms should be saved.
            Useful if you want to have different histograms in the same file to separate them.
        prefixDecayString (bool): If True the decayString will be prepended to the directory name to allow for more
            programmatic naming of the structure in the file.
        filenameSuffix (str): suffix to be appended to the filename before ``.root``.
        ignoreCommandLineOverride (bool) : if true, ignore override of file name via command line argument ``-o``.
 
    .. tip:: The output filename can be overridden using the ``-o`` argument of basf2.
    """
 
    if variables_2d is None:
        variables_2d = []
    output = register_module('VariablesToHistogram')
    output.set_name('VariablesToHistogram_' + decayString)
    output.param('particleList', decayString)
    output.param('variables', variables)
    output.param('variables_2d', variables_2d)
    output.param('fileName', filename)
    output.param('fileNameSuffix', filenameSuffix)
    output.param('ignoreCommandLineOverride', ignoreCommandLineOverride)
    if directory is not None or prefixDecayString:
        if directory is None:
            directory = ""
        if prefixDecayString:
            directory = decayString + "_" + directory
        output.param("directory", directory)
    path.add_module(output)
 
 
def variablesToExtraInfo(particleList, variables, option=0, path=None):
    """
    For each particle in the input list the selected variables are saved in an extra-info field with the given name.
    Can be used when wanting to save variables before modifying them, e.g. when performing vertex fits.
 
    Parameters:
        particleList (str):         The input ParticleList
        variables (dict[str,str]):  Dictionary of Variables (key) and extraInfo names (value).
        option (int):               Option to overwrite an existing extraInfo. Choose among -1, 0, 1, 2.
                                    An existing extra info with the same name will be overwritten if the new
                                    value is lower / will never be overwritten / will be overwritten if the
                                    new value is higher / will always be overwritten (option = -1/0/1/2).
        path (basf2.Path):          modules are added to this path
    """
 
    mod = register_module('VariablesToExtraInfo')
    mod.set_name('VariablesToExtraInfo_' + particleList)
    mod.param('particleList', particleList)
    mod.param('variables', variables)
    mod.param('overwrite', option)
    path.add_module(mod)
 
 
def variablesToDaughterExtraInfo(particleList, decayString, variables, option=0, path=None):
    """
    For each daughter particle specified via decay string the selected variables (estimated for the mother particle)
    are saved in an extra-info field with the given name. In other words, the property of mother is saved as extra-info
    to specified daughter particle.
 
    Parameters:
        particleList (str):         The input ParticleList
        decayString (str):          Decay string that specifies to which daughter the extra info should be appended
        variables (dict[str,str]):  Dictionary of Variables (key) and extraInfo names (value).
        option (int):               Option to overwrite an existing extraInfo. Choose among -1, 0, 1, 2.
                                    An existing extra info with the same name will be overwritten if the new
                                    value is lower / will never be overwritten / will be overwritten if the
                                    new value is higher / will always be overwritten (option = -1/0/1/2).
        path (basf2.Path):          modules are added to this path
    """
 
    mod = register_module('VariablesToExtraInfo')
    mod.set_name('VariablesToDaughterExtraInfo_' + particleList)
    mod.param('particleList', particleList)
    mod.param('decayString', decayString)
    mod.param('variables', variables)
    mod.param('overwrite', option)
    path.add_module(mod)
 
 
def variablesToEventExtraInfo(particleList, variables, option=0, path=None):
    """
    For each particle in the input list the selected variables are saved in an event-extra-info field with the given name,
    Can be used to save MC truth information, for example, in a ntuple of reconstructed particles.
 
    .. tip::
        When the function is called first time not in the main path but in a sub-path e.g. ``roe_path``,
        the eventExtraInfo cannot be accessed from the main path because of the shorter lifetime of the event-extra-info field.
        If one wants to call the function in a sub-path, one has to call the function in the main path beforehand.
 
    Parameters:
        particleList (str):         The input ParticleList
        variables (dict[str,str]):  Dictionary of Variables (key) and extraInfo names (value).
        option (int):               Option to overwrite an existing extraInfo. Choose among -1, 0, 1, 2.
                                    An existing extra info with the same name will be overwritten if the new
                                    value is lower / will never be overwritten / will be overwritten if the
                                    new value is higher / will always be overwritten (option = -1/0/1/2).
        path (basf2.Path):          modules are added to this path
    """
 
    mod = register_module('VariablesToEventExtraInfo')
    mod.set_name('VariablesToEventExtraInfo_' + particleList)
    mod.param('particleList', particleList)
    mod.param('variables', variables)
    mod.param('overwrite', option)
    path.add_module(mod)
 
 
def variableToSignalSideExtraInfo(particleList, varToExtraInfo, path):
    """
    Write the value of specified variables estimated for the single particle in the input list (has to contain exactly 1
    particle) as an extra info to the particle related to current ROE.
    Should be used only in the for_each roe path.
 
    Parameters:
        particleList (str):              The input ParticleList
        varToExtraInfo (dict[str,str]):  Dictionary of Variables (key) and extraInfo names (value).
        path (basf2.Path):               modules are added to this path
    """
 
    mod = register_module('SignalSideVariablesToExtraInfo')
    mod.set_name('SigSideVarToExtraInfo_' + particleList)
    mod.param('particleListName', particleList)
    mod.param('variableToExtraInfo', varToExtraInfo)
    path.add_module(mod)
 
 
def signalRegion(particleList, cut, path=None, name="isSignalRegion", blind_data=True):
    """
    Define and blind a signal region.
    Per default, the defined signal region is cut out if ran on data.
    This function will provide a new variable 'isSignalRegion' as default, which is either 0 or 1 depending on the cut
    provided.
 
    Example:
        .. code-block:: python
 
            ma.reconstructDecay("B+:sig -> D+ pi0", "Mbc>5.2", path=path)
            ma.signalRegion("B+:sig",
                             "Mbc>5.27 and abs(deltaE)<0.2",
                             blind_data=True,
                             path=path)
            ma.variablesToNtuples("B+:sig", ["isSignalRegion"], path=path)
 
    Parameters:
        particleList (str):     The input ParticleList
        cut (str):              Cut string describing the signal region
        path (basf2.Path)::     Modules are added to this path
        name (str):             Name of the Signal region in the variable manager
        blind_data (bool):      Automatically exclude signal region from data
 
    """
 
    from variables import variables
    mod = register_module('VariablesToExtraInfo')
    mod.set_name(f'{name}_' + particleList)
    mod.param('particleList', particleList)
    mod.param('variables', {f"passesCut({cut})": name})
    variables.addAlias(name, f"extraInfo({name})")
    path.add_module(mod)
 
    # Check if we run on Data
    if blind_data:
        applyCuts(particleList, f"{name}==0 or isMC==1", path=path)
 
 
def removeExtraInfo(particleLists=None, removeEventExtraInfo=False, path=None):
    """
    Removes the ExtraInfo of the given particleLists. If specified (removeEventExtraInfo = True) also the EventExtraInfo is removed.
    """
 
    if particleLists is None:
        particleLists = []
    mod = register_module('ExtraInfoRemover')
    mod.param('particleLists', particleLists)
    mod.param('removeEventExtraInfo', removeEventExtraInfo)
    path.add_module(mod)
 
 
def signalSideParticleFilter(particleList, selection, roe_path, deadEndPath):
    """
    Checks if the current ROE object in the for_each roe path (argument roe_path) is related
    to the particle from the input ParticleList. Additional selection criteria can be applied.
    If ROE is not related to any of the Particles from ParticleList or the Particle doesn't
    meet the selection criteria the execution of deadEndPath is started. This path, as the name
    suggests should be empty and its purpose is to end the execution of for_each roe path for
    the current ROE object.
 
    @param particleList  The input ParticleList
    @param selection Selection criteria that Particle needs meet in order for for_each ROE path to continue
    @param for_each roe path in which this filter is executed
    @param deadEndPath empty path that ends execution of or_each roe path for the current ROE object.
    """
 
    mod = register_module('SignalSideParticleFilter')
    mod.set_name('SigSideParticleFilter_' + particleList)
    mod.param('particleLists', [particleList])
    mod.param('selection', selection)
    roe_path.add_module(mod)
    mod.if_false(deadEndPath)
 
 
def signalSideParticleListsFilter(particleLists, selection, roe_path, deadEndPath):
    """
    Checks if the current ROE object in the for_each roe path (argument roe_path) is related
    to the particle from the input ParticleList. Additional selection criteria can be applied.
    If ROE is not related to any of the Particles from ParticleList or the Particle doesn't
    meet the selection criteria the execution of deadEndPath is started. This path, as the name
    suggests should be empty and its purpose is to end the execution of for_each roe path for
    the current ROE object.
 
    @param particleLists  The input ParticleLists
    @param selection Selection criteria that Particle needs meet in order for for_each ROE path to continue
    @param for_each roe path in which this filter is executed
    @param deadEndPath empty path that ends execution of or_each roe path for the current ROE object.
    """
 
    mod = register_module('SignalSideParticleFilter')
    mod.set_name('SigSideParticleFilter_' + particleLists[0])
    mod.param('particleLists', particleLists)
    mod.param('selection', selection)
    roe_path.add_module(mod)
    mod.if_false(deadEndPath)
 
 
def reconstructMCDecay(
    decayString,
    cut,
    dmID=0,
    writeOut=False,
    path=None,
    chargeConjugation=True,
):
    r"""
    Finds and creates a ``ParticleList`` from given decay string.
    ``ParticleList`` of daughters with sub-decay is created.
 
    Only the particles made from MCParticle, which can be loaded by `fillParticleListFromMC`, are accepted as daughters.
 
    Only signal particle, which means :b2:var:`isSignal` is equal to 1, is stored. One can use the decay string grammar
    to change the behavior of :b2:var:`isSignal`. One can find detailed information in :ref:`DecayString`.
 
    .. tip::
        If one uses same sub-decay twice, same particles are registered to a ``ParticleList``. For example,
        ``K_S0:pi0pi0 =direct=> [pi0:gg =direct=> gamma:MC gamma:MC] [pi0:gg =direct=> gamma:MC gamma:MC]``.
        One can skip the second sub-decay, ``K_S0:pi0pi0 =direct=> [pi0:gg =direct=> gamma:MC gamma:MC] pi0:gg``.
 
    .. tip::
        It is recommended to use only primary particles as daughter particles unless you want to explicitly study the secondary
        particles. The behavior of MC-matching for secondary particles from a stable particle decay is not guaranteed.
        Please consider to use `fillParticleListFromMC` with ``skipNonPrimary=True`` to load daughter particles.
        Moreover, it is recommended to load ``K_S0`` and ``Lambda0`` directly from MCParticle by `fillParticleListFromMC` rather
        than reconstructing from two pions or a proton-pion pair, because their direct daughters can be the secondary particle.
 
 
    @param decayString :ref:`DecayString` specifying what kind of the decay should be reconstructed
                       (from the DecayString the mother and daughter ParticleLists are determined)
    @param cut         created (mother) Particles are added to the mother ParticleList if they
                       pass given cuts (in VariableManager style) and rejected otherwise
                       isSignal==1 is always required by default.
    @param dmID        user specified decay mode identifier
    @param writeOut    whether RootOutput module should save the created ParticleList
    @param path        modules are added to this path
    @param chargeConjugation boolean to decide whether charge conjugated mode should be reconstructed as well (on by default)
    """
 
    pmake = register_module('ParticleCombinerFromMC')
    pmake.set_name('ParticleCombinerFromMC_' + decayString)
    pmake.param('decayString', decayString)
    pmake.param('cut', cut)
    pmake.param('decayMode', dmID)
    pmake.param('writeOut', writeOut)
    pmake.param('chargeConjugation', chargeConjugation)
    path.add_module(pmake)
 
 
def findMCDecay(
    list_name,
    decay,
    writeOut=False,
    appendAllDaughters=False,
    skipNonPrimaryDaughters=True,
    path=None,
):
    """
    Finds and creates a ``ParticleList`` for all ``MCParticle`` decays matching a given :ref:`DecayString`.
    The decay string is required to describe correctly what you want.
    In the case of inclusive decays, you can use :ref:`Grammar_for_custom_MCMatching`
 
    The output particles has only the daughter particles written in the given decay string, if
    ``appendAllDaughters=False`` (default). If ``appendAllDaughters=True``, all daughters of the matched MCParticle are
    appended in the order defined at the MCParticle level. For example,
 
    .. code-block:: python
 
        findMCDecay('B0:Xee', 'B0 -> e+ e- ... ?gamma', appendAllDaughters=False, path=mypath)
 
    The output ParticleList ``B0:Xee`` will match the inclusive ``B0 -> e+ e-`` decays (but neutrinos are not included),
    in both cases of ``appendAllDaughters`` is false and true.
    If the ``appendAllDaughters=False`` as above example, the ``B0:Xee`` has only two electrons as daughters.
    While, if ``appendAllDaughters=True``, all daughters of the matched MCParticles are appended. When the truth decay mode of
    the MCParticle is ``B0 -> [K*0 -> K+ pi-] [J/psi -> e+ e-]``, the first daughter of ``B0:Xee`` is ``K*0`` and ``e+``
    will be the first daughter of second daughter of ``B0:Xee``.
 
    The option ``skipNonPrimaryDaughters`` only has an effect if ``appendAllDaughters=True``. If ``skipNonPrimaryDaughters=True``,
    all primary daughters are appended but the secondary particles are not.
 
    .. tip::
        Daughters of ``Lambda0`` are not primary, but ``Lambda0`` is not a final state particle.
        In order for the MCMatching to work properly, the daughters of ``Lambda0`` are appended to
        ``Lambda0`` regardless of the value of the option ``skipNonPrimaryDaughters``.
 
 
    @param list_name The output particle list name
    @param decay     The decay string which you want
    @param writeOut  Whether `RootOutput` module should save the created ``outputList``
    @param skipNonPrimaryDaughters if true, skip non primary daughters, useful to study final state daughter particles
    @param appendAllDaughters if true, not only the daughters described in the decay string but all daughters are appended
    @param path      modules are added to this path
    """
 
    decayfinder = register_module('MCDecayFinder')
    decayfinder.set_name('MCDecayFinder_' + list_name)
    decayfinder.param('listName', list_name)
    decayfinder.param('decayString', decay)
    decayfinder.param('appendAllDaughters', appendAllDaughters)
    decayfinder.param('skipNonPrimaryDaughters', skipNonPrimaryDaughters)
    decayfinder.param('writeOut', writeOut)
    path.add_module(decayfinder)
 
 
def summaryOfLists(particleLists, outputFile=None, path=None):
    """
    Prints out Particle statistics at the end of the job: number of events with at
    least one candidate, average number of candidates per event, etc.
    If an output file name is provided the statistics is also dumped into a json file with that name.
 
    @param particleLists list of input ParticleLists
    @param outputFile output file name (not created by default)
    """
 
    particleStats = register_module('ParticleStats')
    particleStats.param('particleLists', particleLists)
    if outputFile is not None:
        particleStats.param('outputFile', outputFile)
    path.add_module(particleStats)
 
 
def matchMCTruth(list_name, path):
    """
    Performs MC matching (sets relation Particle->MCParticle) for
    all particles (and its (grand)^N-daughter particles) in the specified
    ParticleList.
 
    @param list_name name of the input ParticleList
    @param path      modules are added to this path
    """
 
    mcMatch = register_module('MCMatcherParticles')
    mcMatch.set_name('MCMatch_' + list_name)
    mcMatch.param('listName', list_name)
    path.add_module(mcMatch)
 
 
def looseMCTruth(list_name, path):
    """
    Performs loose MC matching for all particles in the specified
    ParticleList.
    The difference between loose and normal mc matching algorithm is that
    the loose algorithm will find the common mother of the majority of daughter
    particles while the normal algorithm finds the common mother of all daughters.
    The results of loose mc matching algorithm are stored to the following extraInfo
    items:
 
    - looseMCMotherPDG: PDG code of most common mother
    - looseMCMotherIndex: 1-based StoreArray<MCParticle> index of most common mother
    - looseMCWrongDaughterN: number of daughters that don't originate from the most common mother
    - looseMCWrongDaughterPDG: PDG code of the daughter that doesn't originate from the most common mother (only if
      looseMCWrongDaughterN = 1)
    - looseMCWrongDaughterBiB: 1 if the wrong daughter is Beam Induced Background Particle
 
    @param list_name name of the input ParticleList
    @param path      modules are added to this path
    """
 
    mcMatch = register_module('MCMatcherParticles')
    mcMatch.set_name('LooseMCMatch_' + list_name)
    mcMatch.param('listName', list_name)
    mcMatch.param('looseMCMatching', True)
    path.add_module(mcMatch)
 
 
def buildRestOfEvent(target_list_name, inputParticlelists=None,
                     fillWithMostLikely=True,
                     chargedPIDPriors=None, path=None):
    """
    Creates for each Particle in the given ParticleList a RestOfEvent
    dataobject and makes basf2 relation between them. User can provide additional
    particle lists with a different particle hypothesis like ['K+:good, e+:good'], etc.
 
    @param target_list_name   name of the input ParticleList
    @param inputParticlelists list of user-defined input particle list names, which serve
                              as source of particles to build the ROE, the FSP particles from
                              target_list_name are automatically excluded from the ROE object
    @param fillWithMostLikely By default the module uses the most likely particle mass hypothesis for charged particles
                              based on the PID likelihood. Turn this behavior off if you want to configure your own
                              input particle lists.
    @param chargedPIDPriors   The prior PID fractions, that are used to regulate the
                              amount of certain charged particle species, should be a list of
                              six floats if not None. The order of particle types is
                              the following: [e-, mu-, pi-, K-, p+, d+]
    @param path               modules are added to this path
    """
 
    if inputParticlelists is None:
        inputParticlelists = []
    fillParticleList('pi+:all', '', path=path)
    if fillWithMostLikely:
        from stdCharged import stdMostLikely
        stdMostLikely(chargedPIDPriors, '_roe', path=path)
        inputParticlelists = [f'{ptype}:mostlikely_roe' for ptype in ['K+', 'p+', 'e+', 'mu+']]
    import b2bii
    if not b2bii.isB2BII():
        fillParticleList('gamma:all', '', path=path)
        fillParticleList('K_L0:roe_default', 'isFromKLM > 0', path=path)
        inputParticlelists += ['pi+:all', 'gamma:all', 'K_L0:roe_default']
    else:
        inputParticlelists += ['pi+:all', 'gamma:mdst']
    roeBuilder = register_module('RestOfEventBuilder')
    roeBuilder.set_name('ROEBuilder_' + target_list_name)
    roeBuilder.param('particleList', target_list_name)
    roeBuilder.param('particleListsInput', inputParticlelists)
    roeBuilder.param('mostLikely', fillWithMostLikely)
    path.add_module(roeBuilder)
 
 
def buildNestedRestOfEvent(target_list_name, maskName='all', path=None):
    """
    Creates for each Particle in the given ParticleList a RestOfEvent
    @param target_list_name      name of the input ParticleList
    @param mask_name             name of the ROEMask to be used
    @param path                  modules are added to this path
    """
 
    roeBuilder = register_module('RestOfEventBuilder')
    roeBuilder.set_name('NestedROEBuilder_' + target_list_name)
    roeBuilder.param('particleList', target_list_name)
    roeBuilder.param('nestedROEMask', maskName)
    roeBuilder.param('createNestedROE', True)
    path.add_module(roeBuilder)
 
 
def buildRestOfEventFromMC(target_list_name, inputParticlelists=None, path=None):
    """
    Creates for each Particle in the given ParticleList a RestOfEvent
    @param target_list_name   name of the input ParticleList
    @param inputParticlelists list of input particle list names, which serve
                              as a source of particles to build ROE, the FSP particles from
                              target_list_name are excluded from ROE object
    @param path               modules are added to this path
    """
 
    if inputParticlelists is None:
        inputParticlelists = []
    if (len(inputParticlelists) == 0):
        # Type of particles to use for ROEBuilder
        # K_S0 and Lambda0 are added here because some of them have interacted
        # with the detector material
        types = ['gamma', 'e+', 'mu+', 'pi+', 'K+', 'p+', 'K_L0',
                 'n0', 'nu_e', 'nu_mu', 'nu_tau',
                 'K_S0', 'Lambda0']
        for t in types:
            fillParticleListFromMC(f"{t}:roe_default_gen", 'mcPrimary > 0 and nDaughters == 0',
                                   True, True, path=path)
            inputParticlelists += [f"{t}:roe_default_gen"]
    roeBuilder = register_module('RestOfEventBuilder')
    roeBuilder.set_name('MCROEBuilder_' + target_list_name)
    roeBuilder.param('particleList', target_list_name)
    roeBuilder.param('particleListsInput', inputParticlelists)
    roeBuilder.param('fromMC', True)
    path.add_module(roeBuilder)
 
 
def appendROEMask(list_name,
                  mask_name,
                  trackSelection,
                  eclClusterSelection,
                  klmClusterSelection='',
                  path=None):
    """
    Loads the ROE object of a particle and creates a ROE mask with a specific name. It applies
    selection criteria for tracks and eclClusters which will be used by variables in ROEVariables.cc.
 
    - append a ROE mask with all tracks in ROE coming from the IP region
 
    .. code-block:: python
 
        appendROEMask('B+:sig', 'IPtracks', '[dr < 2] and [abs(dz) < 5]', path=mypath)
 
    - append a ROE mask with only ECL-based particles that pass as good photon candidates
 
    .. code-block:: python
 
        goodPhotons = 'inCDCAcceptance and clusterErrorTiming < 1e6 and [clusterE1E9 > 0.4 or E > 0.075]'
        appendROEMask('B+:sig', 'goodROEGamma', '', goodPhotons, path=mypath)
 
 
    @param list_name             name of the input ParticleList
    @param mask_name             name of the appended ROEMask
    @param trackSelection        decay string for the track-based particles in ROE
    @param eclClusterSelection   decay string for the ECL-based particles in ROE
    @param klmClusterSelection   decay string for the KLM-based particles in ROE
    @param path                  modules are added to this path
    """
 
    roeMask = register_module('RestOfEventInterpreter')
    roeMask.set_name('RestOfEventInterpreter_' + list_name + '_' + mask_name)
    roeMask.param('particleList', list_name)
    roeMask.param('ROEMasks', [(mask_name, trackSelection, eclClusterSelection, klmClusterSelection)])
    path.add_module(roeMask)
 
 
def appendROEMasks(list_name, mask_tuples, path=None):
    """
    Loads the ROE object of a particle and creates a ROE mask with a specific name. It applies
    selection criteria for track-, ECL- and KLM-based particles which will be used by ROE variables.
 
    The multiple ROE masks with their own selection criteria are specified
    via list of tuples (mask_name, trackParticleSelection, eclParticleSelection, klmParticleSelection) or
    (mask_name, trackSelection, eclClusterSelection) in case with fractions.
 
    - Example for two tuples, one with and one without fractions
 
    .. code-block:: python
 
        ipTracks     = ('IPtracks', '[dr < 2] and [abs(dz) < 5]', '', '')
        goodPhotons = 'inCDCAcceptance and [clusterErrorTiming < 1e6] and [clusterE1E9 > 0.4 or E > 0.075]'
        goodROEGamma = ('ROESel', '[dr < 2] and [abs(dz) < 5]', goodPhotons, '')
        goodROEKLM     = ('IPtracks', '[dr < 2] and [abs(dz) < 5]', '', 'nKLMClusterTrackMatches == 0')
        appendROEMasks('B+:sig', [ipTracks, goodROEGamma, goodROEKLM], path=mypath)
 
    @param list_name             name of the input ParticleList
    @param mask_tuples           array of ROEMask list tuples to be appended
    @param path                  modules are added to this path
    """
 
    compatible_masks = []
    for mask in mask_tuples:
        # add empty KLM-based selection if it's absent:
        if len(mask) == 3:
            compatible_masks += [(*mask, '')]
        else:
            compatible_masks += [mask]
    roeMask = register_module('RestOfEventInterpreter')
    roeMask.set_name('RestOfEventInterpreter_' + list_name + '_' + 'MaskList')
    roeMask.param('particleList', list_name)
    roeMask.param('ROEMasks', compatible_masks)
    path.add_module(roeMask)
 
 
def updateROEMask(list_name,
                  mask_name,
                  trackSelection,
                  eclClusterSelection='',
                  klmClusterSelection='',
                  path=None):
    """
    Update an existing ROE mask by applying additional selection cuts for
    tracks and/or clusters.
 
    See function `appendROEMask`!
 
    @param list_name             name of the input ParticleList
    @param mask_name             name of the ROEMask to update
    @param trackSelection        decay string for the track-based particles in ROE
    @param eclClusterSelection   decay string for the ECL-based particles in ROE
    @param klmClusterSelection   decay string for the KLM-based particles in ROE
    @param path                  modules are added to this path
    """
 
    roeMask = register_module('RestOfEventInterpreter')
    roeMask.set_name('RestOfEventInterpreter_' + list_name + '_' + mask_name)
    roeMask.param('particleList', list_name)
    roeMask.param('ROEMasks', [(mask_name, trackSelection, eclClusterSelection, klmClusterSelection)])
    roeMask.param('update', True)
    path.add_module(roeMask)
 
 
def updateROEMasks(list_name, mask_tuples, path):
    """
    Update existing ROE masks by applying additional selection cuts for tracks
    and/or clusters.
 
    The multiple ROE masks with their own selection criteria are specified
    via list tuples (mask_name, trackSelection, eclClusterSelection, klmClusterSelection)
 
    See function `appendROEMasks`!
 
    @param list_name             name of the input ParticleList
    @param mask_tuples           array of ROEMask list tuples to be appended
    @param path                  modules are added to this path
    """
 
    compatible_masks = []
    for mask in mask_tuples:
        # add empty KLM-based selection if it's absent:
        if len(mask) == 3:
            compatible_masks += [(*mask, '')]
        else:
            compatible_masks += [mask]
 
    roeMask = register_module('RestOfEventInterpreter')
    roeMask.set_name('RestOfEventInterpreter_' + list_name + '_' + 'MaskList')
    roeMask.param('particleList', list_name)
    roeMask.param('ROEMasks', compatible_masks)
    roeMask.param('update', True)
    path.add_module(roeMask)
 
 
def keepInROEMasks(list_name, mask_names, cut_string, path=None):
    """
    This function is used to apply particle list specific cuts on one or more ROE masks (track or eclCluster).
    With this function one can KEEP the tracks/eclclusters used in particles from provided particle list.
    This function should be executed only in the for_each roe path for the current ROE object.
 
    To avoid unnecessary computation, the input particle list should only contain particles from ROE
    (use cut 'isInRestOfEvent == 1'). To update the ECLCluster masks, the input particle list should be a photon
    particle list (e.g. 'gamma:someLabel'). To update the Track masks, the input particle list should be a charged
    pion particle list (e.g. 'pi+:someLabel').
 
    Updating a non-existing mask will create a new one.
 
    - keep only those tracks that were used in provided particle list
 
    .. code-block:: python
 
        keepInROEMasks('pi+:goodTracks', 'mask', '', path=mypath)
 
    - keep only those clusters that were used in provided particle list and pass a cut, apply to several masks
 
    .. code-block:: python
 
        keepInROEMasks('gamma:goodClusters', ['mask1', 'mask2'], 'E > 0.1', path=mypath)
 
 
    @param list_name    name of the input ParticleList
    @param mask_names   array of ROEMasks to be updated
    @param cut_string   decay string with which the mask will be updated
    @param path         modules are added to this path
    """
 
    updateMask = register_module('RestOfEventUpdater')
    updateMask.set_name('RestOfEventUpdater_' + list_name + '_masks')
    updateMask.param('particleList', list_name)
    updateMask.param('updateMasks', mask_names)
    updateMask.param('cutString', cut_string)
    updateMask.param('discard', False)
    path.add_module(updateMask)
 
 
def discardFromROEMasks(list_name, mask_names, cut_string, path=None):
    """
    This function is used to apply particle list specific cuts on one or more ROE masks (track or eclCluster).
    With this function one can DISCARD the tracks/eclclusters used in particles from provided particle list.
    This function should be executed only in the for_each roe path for the current ROE object.
 
    To avoid unnecessary computation, the input particle list should only contain particles from ROE
    (use cut 'isInRestOfEvent == 1'). To update the ECLCluster masks, the input particle list should be a photon
    particle list (e.g. 'gamma:someLabel'). To update the Track masks, the input particle list should be a charged
    pion particle list (e.g. 'pi+:someLabel').
 
    Updating a non-existing mask will create a new one.
 
    - discard tracks that were used in provided particle list
 
    .. code-block:: python
 
        discardFromROEMasks('pi+:badTracks', 'mask', '', path=mypath)
 
    - discard clusters that were used in provided particle list and pass a cut, apply to several masks
 
    .. code-block:: python
 
        discardFromROEMasks('gamma:badClusters', ['mask1', 'mask2'], 'E < 0.1', path=mypath)
 
 
    @param list_name    name of the input ParticleList
    @param mask_names   array of ROEMasks to be updated
    @param cut_string   decay string with which the mask will be updated
    @param path         modules are added to this path
    """
 
    updateMask = register_module('RestOfEventUpdater')
    updateMask.set_name('RestOfEventUpdater_' + list_name + '_masks')
    updateMask.param('particleList', list_name)
    updateMask.param('updateMasks', mask_names)
    updateMask.param('cutString', cut_string)
    updateMask.param('discard', True)
    path.add_module(updateMask)
 
 
def optimizeROEWithV0(list_name, mask_names, cut_string, path=None):
    """
    This function is used to apply particle list specific cuts on one or more ROE masks for Tracks.
    It is possible to optimize the ROE selection by treating tracks from V0's separately, meaning,
    taking V0's 4-momentum into account instead of 4-momenta of tracks. A cut for only specific V0's
    passing it can be applied.
 
    The input particle list should be a V0 particle list: K_S0 ('K_S0:someLabel', ''),
    Lambda ('Lambda:someLabel', '') or converted photons ('gamma:someLabel').
 
    Updating a non-existing mask will create a new one.
 
    - treat tracks from K_S0 inside mass window separately, replace track momenta with K_S0 momentum
 
    .. code-block:: python
 
        optimizeROEWithV0('K_S0:opt', 'mask', '0.450 < M < 0.550', path=mypath)
 
    @param list_name    name of the input ParticleList
    @param mask_names   array of ROEMasks to be updated
    @param cut_string   decay string with which the mask will be updated
    @param path         modules are added to this path
    """
 
    updateMask = register_module('RestOfEventUpdater')
    updateMask.set_name('RestOfEventUpdater_' + list_name + '_masks')
    updateMask.param('particleList', list_name)
    updateMask.param('updateMasks', mask_names)
    updateMask.param('cutString', cut_string)
    path.add_module(updateMask)
 
 
def updateROEUsingV0Lists(target_particle_list, mask_names, default_cleanup=True, selection_cuts=None,
                          apply_mass_fit=False, fitter='treefit', path=None):
    """
    This function creates V0 particle lists (photons, :math:`K^0_S` and :math:`\\Lambda^0`)
    and it uses V0 candidates to update the Rest Of Event, which is associated to the target particle list.
    It is possible to apply a standard or customized selection and mass fit to the V0 candidates.
 
 
    @param target_particle_list  name of the input ParticleList
    @param mask_names            array of ROE masks to be applied
    @param default_cleanup       if True, predefined cuts will be applied on the V0 lists
    @param selection_cuts        a single string of selection cuts or tuple of three strings (photon_cuts, K_S0_cuts, Lambda0_cuts),
                                 which will be applied to the V0 lists. These cuts will have a priority over the default ones.
    @param apply_mass_fit        if True, a mass fit will be applied to the V0 particles
    @param fitter                string, that represent a fitter choice: "treefit" for TreeFitter and "kfit" for KFit
    @param path                  modules are added to this path
    """
 
    roe_path = create_path()
    deadEndPath = create_path()
    signalSideParticleFilter(target_particle_list, '', roe_path, deadEndPath)
 
    if (default_cleanup and selection_cuts is None):
        B2INFO("Using default cleanup in updateROEUsingV0Lists.")
        selection_cuts = 'abs(dM) < 0.1 '
        selection_cuts += 'and daughter(0,particleID) > 0.2 and daughter(1,particleID) > 0.2 '
        selection_cuts += 'and daughter(0,thetaInCDCAcceptance) and daughter(1,thetaInCDCAcceptance)'
    if (selection_cuts is None or selection_cuts == ''):
        B2INFO("No cleanup in updateROEUsingV0Lists.")
        selection_cuts = ('True', 'True', 'True')
    if (isinstance(selection_cuts, str)):
        selection_cuts = (selection_cuts, selection_cuts, selection_cuts)
    # The isInRestOfEvent variable will be applied on FSPs of composite particles automatically:
    roe_cuts = 'isInRestOfEvent > 0'
    fillConvertedPhotonsList('gamma:v0_roe -> e+ e-', f'{selection_cuts[0]} and {roe_cuts}',
                             path=roe_path)
    fillParticleList('K_S0:v0_roe -> pi+ pi-', f'{selection_cuts[1]} and {roe_cuts}',
                     path=roe_path)
    fillParticleList('Lambda0:v0_roe -> p+ pi-', f'{selection_cuts[2]} and {roe_cuts}',
                     path=roe_path)
    fitter = fitter.lower()
    if (fitter != 'treefit' and fitter != 'kfit'):
        B2WARNING('Argument "fitter" in updateROEUsingV0Lists has only "treefit" and "kfit" options, '
                  f'but "{fitter}" was provided! TreeFitter will be used instead.')
        fitter = 'treefit'
    from vertex import kFit, treeFit
    for v0 in ['gamma:v0_roe', 'K_S0:v0_roe', 'Lambda0:v0_roe']:
        if (apply_mass_fit and fitter == 'kfit'):
            kFit(v0, conf_level=0.0, fit_type='massvertex', path=roe_path)
        if (apply_mass_fit and fitter == 'treefit'):
            treeFit(v0, conf_level=0.0, massConstraint=[v0.split(':')[0]], path=roe_path)
        optimizeROEWithV0(v0, mask_names, '', path=roe_path)
    path.for_each('RestOfEvent', 'RestOfEvents', roe_path)
 
 
def printROEInfo(mask_names=None, full_print=False,
                 unpackComposites=True, path=None):
    """
    This function prints out the information for the current ROE, so it should only be used in the for_each path.
    It prints out basic ROE object info.
 
    If mask names are provided, specific information for those masks will be printed out.
 
    It is also possible to print out all particles in a given mask if the
    'full_print' is set to True.
 
    @param mask_names         array of ROEMask names for printing out info
    @param unpackComposites   if true, replace composite particles by their daughters
    @param full_print         print out particles in mask
    @param path               modules are added to this path
    """
 
    if mask_names is None:
        mask_names = []
    printMask = register_module('RestOfEventPrinter')
    printMask.set_name('RestOfEventPrinter')
    printMask.param('maskNames', mask_names)
    printMask.param('fullPrint', full_print)
    printMask.param('unpackComposites', unpackComposites)
    path.add_module(printMask)
 
 
def buildContinuumSuppression(list_name, roe_mask, path):
    """
    Creates for each Particle in the given ParticleList a ContinuumSuppression
    dataobject and makes basf2 relation between them.
 
    :param list_name: name of the input ParticleList
    :param roe_mask: name of the ROE mask
    :param path: modules are added to this path
    """
 
    qqBuilder = register_module('ContinuumSuppressionBuilder')
    qqBuilder.set_name('QQBuilder_' + list_name)
    qqBuilder.param('particleList', list_name)
    qqBuilder.param('ROEMask', roe_mask)
    path.add_module(qqBuilder)
 
 
def removeParticlesNotInLists(lists_to_keep, path):
    """
    Removes all Particles that are not in a given list of ParticleLists (or daughters of those).
    All relations from/to Particles, daughter indices, and other ParticleLists are fixed.
 
    @param lists_to_keep Keep the Particles and their daughters in these ParticleLists.
    @param path      modules are added to this path
    """
 
    mod = register_module('RemoveParticlesNotInLists')
    mod.param('particleLists', lists_to_keep)
    path.add_module(mod)
 
 
def inclusiveBtagReconstruction(upsilon_list_name, bsig_list_name, btag_list_name, input_lists_names, path):
    """
    Reconstructs Btag from particles in given ParticleLists which do not share any final state particles (mdstSource) with Bsig.
 
    @param upsilon_list_name Name of the ParticleList to be filled with 'Upsilon(4S) -> B:sig anti-B:tag'
    @param bsig_list_name Name of the Bsig ParticleList
    @param btag_list_name Name of the Bsig ParticleList
    @param input_lists_names List of names of the ParticleLists which are used to reconstruct Btag from
    """
 
    btag = register_module('InclusiveBtagReconstruction')
    btag.set_name('InclusiveBtagReconstruction_' + bsig_list_name)
    btag.param('upsilonListName', upsilon_list_name)
    btag.param('bsigListName', bsig_list_name)
    btag.param('btagListName', btag_list_name)
    btag.param('inputListsNames', input_lists_names)
    path.add_module(btag)
 
 
def selectDaughters(particle_list_name, decay_string, path):
    """
    Redefine the Daughters of a particle: select from decayString
 
    @param particle_list_name input particle list
    @param decay_string  for selecting the Daughters to be preserved
    """
 
    seld = register_module('SelectDaughters')
    seld.set_name('SelectDaughters_' + particle_list_name)
    seld.param('listName', particle_list_name)
    seld.param('decayString', decay_string)
    path.add_module(seld)
 
 
def markDuplicate(particleList, prioritiseV0, path):
    """
    Call DuplicateVertexMarker to find duplicate particles in a list and
    flag the ones that should be kept
 
    @param particleList input particle list
    @param prioritiseV0 if true, give V0s a higher priority
    """
 
    markdup = register_module('DuplicateVertexMarker')
    markdup.param('particleList', particleList)
    markdup.param('prioritiseV0', prioritiseV0)
    path.add_module(markdup)
 
 
PI0ETAVETO_COUNTER = 0
 
 
def oldwritePi0EtaVeto(
    particleList,
    decayString,
    workingDirectory='.',
    pi0vetoname='Pi0_Prob',
    etavetoname='Eta_Prob',
    downloadFlag=True,
    selection='',
    path=None
):
    """
    Give pi0/eta probability for hard photon.
 
    In the default weight files a value of 1.4 GeV is set as the lower limit for the hard photon energy in the CMS frame.
 
    The current default weight files are optimised using MC9.
    The input variables are as below. Aliases are set to some variables during training.
 
    * M: pi0/eta candidates Invariant mass
    * lowE: soft photon energy in lab frame
    * cTheta: soft photon ECL cluster's polar angle
    * Zmva: soft photon output of MVA using Zernike moments of the cluster
    * minC2Hdist: soft photon distance from eclCluster to nearest point on nearest Helix at the ECL cylindrical radius
 
    If you don't have weight files in your workingDirectory,
    these files are downloaded from database to your workingDirectory automatically.
    Please refer to analysis/examples/tutorials/B2A306-B02RhoGamma-withPi0EtaVeto.py
    about how to use this function.
 
    NOTE:
      Please don't use following ParticleList names elsewhere:
 
      ``gamma:HARDPHOTON``, ``pi0:PI0VETO``, ``eta:ETAVETO``,
      ``gamma:PI0SOFT + str(PI0ETAVETO_COUNTER)``, ``gamma:ETASOFT + str(PI0ETAVETO_COUNTER)``
 
      Please don't use ``lowE``, ``cTheta``, ``Zmva``, ``minC2Hdist`` as alias elsewhere.
 
    @param particleList     The input ParticleList
    @param decayString specify Particle to be added to the ParticleList
    @param workingDirectory The weight file directory
    @param downloadFlag whether download default weight files or not
    @param pi0vetoname extraInfo name of pi0 probability
    @param etavetoname extraInfo name of eta probability
    @param selection Selection criteria that Particle needs meet in order for for_each ROE path to continue
    @param path       modules are added to this path
    """
 
    import b2bii
    if b2bii.isB2BII():
        B2ERROR("The old pi0 / eta veto is not suitable for Belle analyses.")
 
    import os
    import basf2_mva
 
    global PI0ETAVETO_COUNTER
 
    if PI0ETAVETO_COUNTER == 0:
        from variables import variables
        variables.addAlias('lowE', 'daughter(1,E)')
        variables.addAlias('cTheta', 'daughter(1,clusterTheta)')
        variables.addAlias('Zmva', 'daughter(1,clusterZernikeMVA)')
        variables.addAlias('minC2Tdist', 'daughter(1,minC2TDist)')
        variables.addAlias('cluNHits', 'daughter(1,clusterNHits)')
        variables.addAlias('E9E21', 'daughter(1,clusterE9E21)')
 
    PI0ETAVETO_COUNTER = PI0ETAVETO_COUNTER + 1
 
    roe_path = create_path()
 
    deadEndPath = create_path()
 
    signalSideParticleFilter(particleList, selection, roe_path, deadEndPath)
 
    fillSignalSideParticleList('gamma:HARDPHOTON', decayString, path=roe_path)
 
    pi0softname = 'gamma:PI0SOFT'
    etasoftname = 'gamma:ETASOFT'
    softphoton1 = pi0softname + str(PI0ETAVETO_COUNTER)
    softphoton2 = etasoftname + str(PI0ETAVETO_COUNTER)
 
    fillParticleList(
        softphoton1,
        '[clusterReg==1 and E>0.025] or [clusterReg==2 and E>0.02] or [clusterReg==3 and E>0.02]',
        path=roe_path)
    applyCuts(softphoton1, 'abs(clusterTiming)<120', path=roe_path)
    fillParticleList(
        softphoton2,
        '[clusterReg==1 and E>0.035] or [clusterReg==2 and E>0.03] or [clusterReg==3 and E>0.03]',
        path=roe_path)
    applyCuts(softphoton2, 'abs(clusterTiming)<120', path=roe_path)
 
    reconstructDecay('pi0:PI0VETO -> gamma:HARDPHOTON ' + softphoton1, '', path=roe_path)
    reconstructDecay('eta:ETAVETO -> gamma:HARDPHOTON ' + softphoton2, '', path=roe_path)
 
    if not os.path.isdir(workingDirectory):
        os.mkdir(workingDirectory)
        B2INFO('oldwritePi0EtaVeto: ' + workingDirectory + ' has been created as workingDirectory.')
 
    if not os.path.isfile(workingDirectory + '/pi0veto.root'):
        if downloadFlag:
            basf2_mva.download('Pi0VetoIdentifier', workingDirectory + '/pi0veto.root')
            B2INFO('oldwritePi0EtaVeto: pi0veto.root has been downloaded from database to workingDirectory.')
 
    if not os.path.isfile(workingDirectory + '/etaveto.root'):
        if downloadFlag:
            basf2_mva.download('EtaVetoIdentifier', workingDirectory + '/etaveto.root')
            B2INFO('oldwritePi0EtaVeto: etaveto.root has been downloaded from database to workingDirectory.')
 
    roe_path.add_module('MVAExpert', listNames=['pi0:PI0VETO'], extraInfoName='Pi0Veto',
                        identifier=workingDirectory + '/pi0veto.root')
    roe_path.add_module('MVAExpert', listNames=['eta:ETAVETO'], extraInfoName='EtaVeto',
                        identifier=workingDirectory + '/etaveto.root')
 
    rankByHighest('pi0:PI0VETO', 'extraInfo(Pi0Veto)', numBest=1, path=roe_path)
    rankByHighest('eta:ETAVETO', 'extraInfo(EtaVeto)', numBest=1, path=roe_path)
 
    variableToSignalSideExtraInfo('pi0:PI0VETO', {'extraInfo(Pi0Veto)': pi0vetoname}, path=roe_path)
    variableToSignalSideExtraInfo('eta:ETAVETO', {'extraInfo(EtaVeto)': etavetoname}, path=roe_path)
 
    path.for_each('RestOfEvent', 'RestOfEvents', roe_path)
 
 
def writePi0EtaVeto(
    particleList,
    decayString,
    mode='standardMC15rd',
    selection='',
    path=None,
    suffix='',
    hardParticle='gamma',
    pi0PayloadNameOverride=None,
    pi0SoftPhotonCutOverride=None,
    etaPayloadNameOverride=None,
    etaSoftPhotonCutOverride=None,
    requireSoftPhotonIsInROE=False,
    pi0Selection='[0.03 < M < 0.23]',
    etaSelection='[0.25 < M < 0.75]'
):
    """
    Give pi0/eta probability for hard photon.
 
    In the default weight files a value of 1.4 GeV is set as the lower limit for the hard photon energy in the CMS frame.
    For MC15rd weight files, the BtoXGamma skim is applied during the MVA training.
 
    The current default weight files are optimised using MC15rd. The weight files for MC12 (last version) are still available.
 
    The input variables of the mva training for pi0 veto using MC15rd are:
 
    * M: Invariant mass of pi0 candidates
    * cosHelicityAngleMomentum: Cosine of angle between momentum difference of the photons in the pi0 rest frame
      and momentum of pi0 in lab frame
    * daughter(1,E): soft photon energy in lab frame
    * daughter(1,clusterTheta): soft photon ECL cluster's polar angle
    * daughter(1,clusterLAT): soft photon lateral energy distribution
 
    The input variables of the mva training for eta veto using MC15rd are:
 
    * M: Invariant mass of eta candidates
    * cosHelicityAngleMomentum: Cosine of angle between momentum difference of the photons in the eta rest frame
      and momentum of eta in lab frame
    * daughter(1,E): soft photon energy in lab frame
    * daughter(1,clusterTheta): soft photon ECL cluster's polar angle
    * daughter(1,clusterLAT): soft photon lateral energy distribution
    * daughter(1,clusterNHits): soft photon total crystal weights sum(w_i) with w_i<=1
    * daughter(1,clusterE1E9): soft photon ratio between energies of central crystal and inner 3x3 crystals
    * daughter(1,clusterE9E21): soft photon ratio of energies in inner 3x3 crystals and 5x5 crystals without corners
    * daughter(1,clusterSecondMoment): soft photon second moment
    * daughter(1,clusterAbsZernikeMoment40): soft photon Zernike moment 40
    * daughter(1,clusterAbsZernikeMoment51): soft photon Zernike moment 51
 
    The input variables of the mva training using MC12 are:
 
    * M: Invariant mass of pi0/eta candidates
    * daughter(1,E): soft photon energy in lab frame
    * daughter(1,clusterTheta): soft photon ECL cluster's polar angle
    * daughter(1,minC2TDist): soft photon distance from eclCluster to nearest point on nearest Helix at the ECL cylindrical radius
    * daughter(1,clusterZernikeMVA): soft photon output of MVA using Zernike moments of the cluster
    * daughter(1,clusterNHits): soft photon total crystal weights sum(w_i) with w_i<=1
    * daughter(1,clusterE9E21): soft photon ratio of energies in inner 3x3 crystals and 5x5 crystals without corners
    * cosHelicityAngleMomentum: Cosine of angle between momentum difference of the photons in the pi0/eta rest frame
      and momentum of pi0/eta in lab frame
 
    The following strings are available for mode:
 
    * standard: loose energy cut and no clusterNHits cut are applied to soft photon
    * tight: tight energy cut and no clusterNHits cut are applied to soft photon
    * cluster: loose energy cut and clusterNHits cut are applied to soft photon
    * both: tight energy cut and clusterNHits cut are applied to soft photon
    * standardMC15rd: loose energy cut is applied to soft photon and the weight files are trained using MC15rd
    * tightMC15rd: tight energy cut is applied to soft photon and the weight files are trained using MC15rd
 
    The final probability of the pi0/eta veto is stored as an extraInfo. If no suffix is set it can be obtained from the variables
    `pi0Prob`/`etaProb`. Otherwise, it is available as '{Pi0, Eta}ProbOrigin', '{Pi0, Eta}ProbTightEnergyThreshold', '{Pi0,
    Eta}ProbLargeClusterSize', '{Pi0, Eta}ProbTightEnergyThresholdAndLargeClusterSize', '{Pi0, Eta}ProbOriginMC15rd', or
    '{Pi0, Eta}ProbTightEnergyThresholdMC15rd' for the six modes described above, with the chosen suffix appended. If one would
    like to call this veto twice in one script, add suffix in the second time!
    The second highest probability of the pi0/eta veto also is stored as an extraInfo, with a prefix of 'second' to the previous
    ones, e.g. secondPi0ProbOrigin{suffix}. This can be used to do validation/systematics study.
 
    NOTE:
      Please don't use following ParticleList names elsewhere:
 
      ``gamma:HardPhoton``,
      ``gamma:Pi0Soft + ListName + '_' + particleList.replace(':', '_')``,
      ``gamma:EtaSoft + ListName + '_' + particleList.replace(':', '_')``,
      ``pi0:EtaVeto + ListName``,
      ``eta:EtaVeto + ListName``
 
    @param particleList     the input ParticleList
    @param decayString    specify Particle to be added to the ParticleList
    @param mode       choose one mode out of 'standardMC15rd', 'tightMC15rd', 'standard', 'tight', 'cluster' and 'both'
    @param selection    selection criteria that Particle needs meet in order for for_each ROE path to continue
    @param path           modules are added to this path
    @param suffix           optional suffix to be appended to the usual extraInfo name
    @param hardParticle           particle name which is used to calculate the pi0/eta probability (default is gamma)
    @param pi0PayloadNameOverride  specify the payload name of pi0 veto only if one wants to use non-default one. (default is None)
    @param pi0SoftPhotonCutOverride specify the soft photon selection criteria of pi0 veto only if one wants to use non-default one.
                                    (default is None)
    @param etaPayloadNameOverride  specify the payload name of eta veto only if one wants to use non-default one. (default is None)
    @param etaSoftPhotonCutOverride specify the soft photon selection criteria of eta veto only if one wants to use non-default one.
                                    (default is None)
    @param requireSoftPhotonIsInROE specify if the soft photons used to build pi0 and eta candidates have to be in the current ROE
                                    or not. Default is False, i.e. all soft photons in the event are used.
    @param pi0Selection     Selection for the pi0 reconstruction. Default is '(0.03 < M < 0.23)'.
    @param etaSelection     Selection for the eta reconstruction. Default is '(0.25 < M < 0.75)'.
    """
 
    import b2bii
    if b2bii.isB2BII():
        B2ERROR("The pi0 / eta veto is not suitable for Belle analyses.")
 
    if (requireSoftPhotonIsInROE):
        B2WARNING("Requiring the soft photon to being in the ROE was not done for the MVA training. "
                  "Please check the results carefully.")
    if (mode == 'standardMC15rd' or mode == 'tightMC15rd'):
        if (pi0Selection != '[0.03 < M < 0.23]' or etaSelection != '[0.25 < M < 0.75]'):
            B2WARNING(
                "Personal selection criteria for the pi0 or the eta during reconstructDecay were not used during the MVA training. "
                "Please check the results carefully.")
    else:
        if (pi0Selection != '' or etaSelection != ''):
            B2WARNING(
                "Personal selection criteria for the pi0 or the eta during reconstructDecay were not used during the MVA training. "
                "Please check the results carefully.")
 
    renameSuffix = False
 
    for module in path.modules():
        if module.type() == "SubEvent" and not renameSuffix:
            for subpath in [p.values for p in module.available_params() if p.name == "path"]:
                if renameSuffix:
                    break
                for submodule in subpath.modules():
                    if f'{hardParticle}:HardPhoton{suffix}' in submodule.name():
                        suffix += '_0'
                        B2WARNING("Same extension already used in writePi0EtaVeto, append '_0'")
                        renameSuffix = True
                        break
 
    roe_path = create_path()
    deadEndPath = create_path()
    signalSideParticleFilter(particleList, selection, roe_path, deadEndPath)
    fillSignalSideParticleList(f'{hardParticle}:HardPhoton{suffix}', decayString, path=roe_path)
 
    dictListName = {'standard': 'Origin',
                    'tight': 'TightEnergyThreshold',
                    'cluster': 'LargeClusterSize',
                    'both': 'TightEnrgyThresholdAndLargeClusterSize',
                    'standardMC15rd': 'OriginMC15rd',
                    'tightMC15rd': 'TightEnergyThresholdMC15rd'}
 
    dictPi0EnergyCut = {
        'standard': '[[clusterReg==1 and E>0.025] or [clusterReg==2 and E>0.02] or [clusterReg==3 and E>0.02]]',
        'tight': '[[clusterReg==1 and E>0.03] or [clusterReg==2 and E>0.03] or [clusterReg==3 and E>0.04]]',
        'cluster': '[[clusterReg==1 and E>0.025] or [clusterReg==2 and E>0.02] or [clusterReg==3 and E>0.02]]',
        'both': '[[clusterReg==1 and E>0.03] or [clusterReg==2 and E>0.03] or [clusterReg==3 and E>0.04]]',
        'standardMC15rd': '[[clusterReg==1 and E>0.0225] or [clusterReg==2 and E>0.02] or [clusterReg==3 and E>0.02]]',
        'tightMC15rd': '[[clusterReg==1 and E>0.03] or [clusterReg==2 and E>0.03] or [clusterReg==3 and E>0.04]]'}
 
    dictEtaEnergyCut = {
        'standard': '[[clusterReg==1 and E>0.035] or [clusterReg==2 and E>0.03] or [clusterReg==3 and E>0.03]]',
        'tight': '[[clusterReg==1 and E>0.06] or [clusterReg==2 and E>0.06] or [clusterReg==3 and E>0.06]]',
        'cluster': '[[clusterReg==1 and E>0.035] or [clusterReg==2 and E>0.03] or [clusterReg==3 and E>0.03]]',
        'both': '[[clusterReg==1 and E>0.06] or [clusterReg==2 and E>0.06] or [clusterReg==3 and E>0.06]]',
        'standardMC15rd': '[[clusterReg==1 and E>0.0225] or [clusterReg==2 and E>0.02] or [clusterReg==3 and E>0.02]]',
        'tightMC15rd': '[[clusterReg==1 and E>0.03] or [clusterReg==2 and E>0.03] or [clusterReg==3 and E>0.04]]'}
 
    dictNHitsTimingCut = {'standard': 'clusterNHits >= 0 and abs(clusterTiming)<clusterErrorTiming',
                          'tight': 'clusterNHits >= 0 and abs(clusterTiming)<clusterErrorTiming',
                          'cluster': 'clusterNHits >= 2 and abs(clusterTiming)<clusterErrorTiming',
                          'both': 'clusterNHits >= 2 and abs(clusterTiming)<clusterErrorTiming',
                          'standardMC15rd': 'clusterNHits > 1.5 and abs(clusterTiming) < 200',
                          'tightMC15rd': 'clusterNHits > 1.5 and abs(clusterTiming) < 200'}
 
    dictPi0PayloadName = {'standard': 'Pi0VetoIdentifierStandard',
                          'tight': 'Pi0VetoIdentifierWithHigherEnergyThreshold',
                          'cluster': 'Pi0VetoIdentifierWithLargerClusterSize',
                          'both': 'Pi0VetoIdentifierWithHigherEnergyThresholdAndLargerClusterSize',
                          'standardMC15rd': 'Pi0VetoIdentifierStandardMC15rd',
                          'tightMC15rd': 'Pi0VetoIdentifierWithHigherEnergyThresholdMC15rd'}
 
    dictEtaPayloadName = {'standard': 'EtaVetoIdentifierStandard',
                          'tight': 'EtaVetoIdentifierWithHigherEnergyThreshold',
                          'cluster': 'EtaVetoIdentifierWithLargerClusterSize',
                          'both': 'EtaVetoIdentifierWithHigherEnergyThresholdAndLargerClusterSize',
                          'standardMC15rd': 'EtaVetoIdentifierStandardMC15rd',
                          'tightMC15rd': 'EtaVetoIdentifierWithHigherEnergyThresholdMC15rd'}
 
    dictPi0ExtraInfoName = {'standard': 'Pi0ProbOrigin',
                            'tight': 'Pi0ProbTightEnergyThreshold',
                            'cluster': 'Pi0ProbLargeClusterSize',
                            'both': 'Pi0ProbTightEnergyThresholdAndLargeClusterSize',
                            'standardMC15rd': 'Pi0ProbOriginMC15rd',
                            'tightMC15rd': 'Pi0ProbTightEnergyThresholdMC15rd'}
 
    dictEtaExtraInfoName = {'standard': 'EtaProbOrigin',
                            'tight': 'EtaProbTightEnergyThreshold',
                            'cluster': 'EtaProbLargeClusterSize',
                            'both': 'EtaProbTightEnergyThresholdAndLargeClusterSize',
                            'standardMC15rd': 'EtaProbOriginMC15rd',
                            'tightMC15rd': 'EtaProbTightEnergyThresholdMC15rd'}
 
    ListName = dictListName[mode]
    Pi0EnergyCut = dictPi0EnergyCut[mode]
    EtaEnergyCut = dictEtaEnergyCut[mode]
    NHitsTimingCut = dictNHitsTimingCut[mode]
    Pi0PayloadName = dictPi0PayloadName[mode]
    EtaPayloadName = dictEtaPayloadName[mode]
    Pi0ExtraInfoName = dictPi0ExtraInfoName[mode]
    EtaExtraInfoName = dictEtaExtraInfoName[mode]
 
    # pi0 veto
    if pi0PayloadNameOverride is not None:
        Pi0PayloadName = pi0PayloadNameOverride
        B2WARNING("You're using personal weight files, be careful. ")
    if pi0SoftPhotonCutOverride is None:
        Pi0SoftPhotonCut = Pi0EnergyCut + ' and ' + NHitsTimingCut
    else:
        Pi0SoftPhotonCut = pi0SoftPhotonCutOverride
        B2WARNING("You're applying personal cuts on the soft photon candidates, be careful. ")
 
    if requireSoftPhotonIsInROE:
        Pi0SoftPhotonCut += ' and isInRestOfEvent==1'
 
    # define the particleList name for soft photon
    pi0soft = f'gamma:Pi0Soft{suffix}' + ListName + '_' + particleList.replace(':', '_')
    # fill the particleList for soft photon with energy, timing and clusterNHits cuts
    fillParticleList(pi0soft, Pi0SoftPhotonCut, path=roe_path)
    # reconstruct pi0
    reconstructDecay('pi0:Pi0Veto' + ListName + suffix + f' -> {hardParticle}:HardPhoton{suffix} ' + pi0soft, pi0Selection,
                     allowChargeViolation=True, path=roe_path)
    # MVA training is conducted.
    roe_path.add_module('MVAExpert', listNames=['pi0:Pi0Veto' + ListName + suffix],
                        extraInfoName=Pi0ExtraInfoName, identifier=Pi0PayloadName)
    # Pick up the pi0/eta candidate with the highest pi0/eta probability.
    rankByHighest(
        'pi0:Pi0Veto' + ListName + suffix,
        'extraInfo(' + Pi0ExtraInfoName + ')',
        numBest=2,
        outputVariable="Pi0VetoRank",
        path=roe_path)
    cutAndCopyList(outputListName='pi0:Pi0VetoFirst' + ListName + suffix,
                   inputListName='pi0:Pi0Veto' + ListName + suffix,
                   cut='extraInfo(Pi0VetoRank)==1',
                   path=roe_path)
    variableToSignalSideExtraInfo('pi0:Pi0VetoFirst' + ListName + suffix,
                                  {'extraInfo(' + Pi0ExtraInfoName + ')': Pi0ExtraInfoName + suffix}, path=roe_path)
    # Pick up the pi0/eta candidate with the second highest pi0/eta probability.
    cutAndCopyList(outputListName='pi0:Pi0VetoSecond' + ListName + suffix,
                   inputListName='pi0:Pi0Veto' + ListName + suffix,
                   cut='extraInfo(Pi0VetoRank)==2',
                   path=roe_path)
    variableToSignalSideExtraInfo('pi0:Pi0VetoSecond' + ListName + suffix,
                                  {'extraInfo(' + Pi0ExtraInfoName + ')': 'second' + Pi0ExtraInfoName + suffix}, path=roe_path)
 
    # eta veto
    if etaPayloadNameOverride is not None:
        EtaPayloadName = etaPayloadNameOverride
        B2WARNING("You're using personal weight files, be careful. ")
    if etaSoftPhotonCutOverride is None:
        EtaSoftPhotonCut = EtaEnergyCut + ' and ' + NHitsTimingCut
    else:
        EtaSoftPhotonCut = etaSoftPhotonCutOverride
        B2WARNING("You're applying personal cuts on the soft photon candidates, be careful. ")
 
    if requireSoftPhotonIsInROE:
        EtaSoftPhotonCut += ' and isInRestOfEvent==1'
 
    etasoft = f'gamma:EtaSoft{suffix}' + ListName + '_' + particleList.replace(':', '_')
    fillParticleList(etasoft, EtaSoftPhotonCut, path=roe_path)
    reconstructDecay('eta:EtaVeto' + ListName + suffix + f' -> {hardParticle}:HardPhoton{suffix} ' + etasoft, etaSelection,
                     allowChargeViolation=True, path=roe_path)
    roe_path.add_module('MVAExpert', listNames=['eta:EtaVeto' + ListName + suffix],
                        extraInfoName=EtaExtraInfoName, identifier=EtaPayloadName)
    rankByHighest(
        'eta:EtaVeto' + ListName + suffix,
        'extraInfo(' + EtaExtraInfoName + ')',
        numBest=2,
        outputVariable="EtaVetoRank",
        path=roe_path)
    cutAndCopyList(outputListName='eta:EtaVetoFirst' + ListName + suffix,
                   inputListName='eta:EtaVeto' + ListName + suffix,
                   cut='extraInfo(EtaVetoRank)==1',
                   path=roe_path)
    variableToSignalSideExtraInfo('eta:EtaVetoFirst' + ListName + suffix,
                                  {'extraInfo(' + EtaExtraInfoName + ')': EtaExtraInfoName + suffix}, path=roe_path)
    cutAndCopyList(outputListName='eta:EtaVetoSecond' + ListName + suffix,
                   inputListName='eta:EtaVeto' + ListName + suffix,
                   cut='extraInfo(EtaVetoRank)==2',
                   path=roe_path)
    variableToSignalSideExtraInfo('eta:EtaVetoSecond' + ListName + suffix,
                                  {'extraInfo(' + EtaExtraInfoName + ')': 'second' + EtaExtraInfoName + suffix}, path=roe_path)
 
    path.for_each('RestOfEvent', 'RestOfEvents', roe_path)
 
 
def lowEnergyPi0Identification(pi0List, gammaList, payloadNameSuffix,
                               path=None):
    """
    Calculate low-energy pi0 identification.
    The result is stored as ExtraInfo ``lowEnergyPi0Identification`` for
    the list pi0List.
 
    Parameters:
        pi0List (str): Pi0 list.
 
        gammaList (str): Gamma list. First, an energy cut E > 0.2 is applied to the photons from this list.
                         Then, all possible combinations with a pi0 daughter photon are formed except the one
                         corresponding to the reconstructed pi0.
                         The maximum low-energy pi0 veto value is calculated for such photon pairs
                         and used as one of the input variables for the identification classifier.
 
        payloadNameSuffix (str): Payload name suffix. The weight payloads are stored in the analysis global
                                 tag and have the following names:\n
                                 * ``'LowEnergyPi0Veto' + payloadNameSuffix``
                                 * ``'LowEnergyPi0Identification' + payloadNameSuffix``\n
                                 The possible suffixes are:\n
                                 * ``'Belle1'`` for Belle data.
                                 * ``'Belle2Release5'`` for Belle II release 5 data (MC14, proc12, buckets 16 - 25).
                                 * ``'Belle2Release6'`` for Belle II release 6 data (MC15, proc13, buckets 26 - 36).
 
        path (basf2.Path): Module path.
    """
 
    # Select photons with higher energy for formation of veto combinations.
    gammaListVeto = f'{gammaList}_pi0veto'
    cutAndCopyList(gammaListVeto, gammaList, 'E > 0.2', path=path)
    import b2bii
    payload_name = 'LowEnergyPi0Veto' + payloadNameSuffix
    path.add_module('LowEnergyPi0VetoExpert', identifier=payload_name,
                    VetoPi0Daughters=True, GammaListName=gammaListVeto,
                    Pi0ListName=pi0List, Belle1=b2bii.isB2BII())
    payload_name = 'LowEnergyPi0Identification' + payloadNameSuffix
    path.add_module('LowEnergyPi0IdentificationExpert',
                    identifier=payload_name, Pi0ListName=pi0List,
                    Belle1=b2bii.isB2BII())
 
 
def getNeutralHadronGeomMatches(
        particleLists,
        addKL=True,
        addNeutrons=False,
        efficiencyCorrectionKl=0.83,
        efficiencyCorrectionNeutrons=1.0,
        path=None):
    """
    For an ECL-based list, assign the mcdistanceKL and mcdistanceNeutron variables that correspond
    to the distance to the closest MC KL and neutron, respectively.
    @param particleLists the input ParticleLists, must be ECL-based lists (e.g. photons)
    @param addKL (default True) add distance to MC KL
    @param addNeutrons (default False) add distance to MC neutrons
    @param efficiencyCorrectionKl (default 0.83) apply overall efficiency correction
    @param efficiencyCorrectionNeutrons (default 1.0) apply overall efficiency correction
    @param path    modules are added to this path
    """
    from ROOT import Belle2
    Const = Belle2.Const
 
    if addKL:
        path.add_module(
            "NeutralHadronMatcher",
            particleLists=particleLists,
            mcPDGcode=Const.Klong.getPDGCode(),
            efficiencyCorrection=efficiencyCorrectionKl)
    if addNeutrons:
        path.add_module(
            "NeutralHadronMatcher",
            particleLists=particleLists,
            mcPDGcode=Const.neutron.getPDGCode(),
            efficiencyCorrection=efficiencyCorrectionNeutrons)
 
 
def getBeamBackgroundProbability(particleList, weight, path=None):
    """
    Assign a probability to each ECL cluster as being signal like (1) compared to beam background like (0)
    @param particleList    the input ParticleList, must be a photon list
    @param weight    type of weight file to use
    @param path    modules are added to this path
    """
 
    import b2bii
    if b2bii.isB2BII() and weight != "Belle":
        B2WARNING("weight type must be 'Belle' for b2bii.")
 
    path.add_module('MVAExpert',
                    listNames=particleList,
                    extraInfoName='beamBackgroundSuppression',
                    identifier=f'BeamBackgroundMVA_{weight}')
 
 
def getFakePhotonProbability(particleList, weight, path=None):
    """
    Assign a probability to each ECL cluster as being signal like (1) compared to fake photon like (0)
    @param particleList    the input ParticleList, must be a photon list
    @param weight    type of weight file to use
    @param path    modules are added to this path
    """
 
    import b2bii
    if b2bii.isB2BII() and weight != "Belle":
        B2WARNING("weight type must be 'Belle' for b2bii.")
 
    path.add_module('MVAExpert',
                    listNames=particleList,
                    extraInfoName='fakePhotonSuppression',
                    identifier=f'FakePhotonMVA_{weight}')
 
 
def buildEventKinematics(inputListNames=None, default_cleanup=True, custom_cuts=None,
                         chargedPIDPriors=None, fillWithMostLikely=False, path=None):
    """
    Calculates the global kinematics of the event (visible energy, missing momentum, missing mass...)
    using ParticleLists provided. If no ParticleList is provided, default ParticleLists are used
    (all track and all hits in ECL without associated track).
 
    The visible energy missing values are
    stored in a EventKinematics dataobject.
 
    @param inputListNames     list of ParticleLists used to calculate the global event kinematics.
                              If the list is empty, default ParticleLists pi+:evtkin and gamma:evtkin are filled.
    @param fillWithMostLikely if True, the module uses the most likely particle mass hypothesis for charged particles
                              according to the PID likelihood and the option inputListNames will be ignored.
    @param chargedPIDPriors   The prior PID fractions, that are used to regulate
                              amount of certain charged particle species, should be a list of
                              six floats if not None. The order of particle types is
                              the following: [e-, mu-, pi-, K-, p+, d+]
    @param default_cleanup    if True and either inputListNames empty or fillWithMostLikely True, default clean up cuts are applied
    @param custom_cuts        tuple of selection cut strings of form (trackCuts, photonCuts), default is None,
                              which would result in a standard predefined selection cuts
    @param path               modules are added to this path
    """
 
    if inputListNames is None:
        inputListNames = []
    trackCuts = 'pt > 0.1'
    trackCuts += ' and thetaInCDCAcceptance'
    trackCuts += ' and abs(dz) < 3'
    trackCuts += ' and dr < 0.5'
 
    gammaCuts = 'E > 0.05'
    gammaCuts += ' and thetaInCDCAcceptance'
    if not b2bii.isB2BII():
        gammaCuts += ' and abs(clusterTiming) < 200'
    if (custom_cuts is not None):
        trackCuts, gammaCuts = custom_cuts
 
    if fillWithMostLikely:
        from stdCharged import stdMostLikely
        stdMostLikely(chargedPIDPriors, '_evtkin', path=path)
        inputListNames = [f'{ptype}:mostlikely_evtkin' for ptype in ['K+', 'p+', 'e+', 'mu+', 'pi+']]
        if b2bii.isB2BII():
            copyList('gamma:evtkin', 'gamma:mdst', path=path)
        else:
            fillParticleList('gamma:evtkin', '', path=path)
        inputListNames += ['gamma:evtkin']
        if default_cleanup:
            B2INFO("Using default cleanup in EventKinematics module.")
            for ptype in ['K+', 'p+', 'e+', 'mu+', 'pi+']:
                applyCuts(f'{ptype}:mostlikely_evtkin', trackCuts, path=path)
            applyCuts('gamma:evtkin', gammaCuts, path=path)
        else:
            B2INFO("No cleanup in EventKinematics module.")
    if not inputListNames:
        B2INFO("Creating particle lists pi+:evtkin and gamma:evtkin to get the global kinematics of the event.")
        fillParticleList('pi+:evtkin', '', path=path)
        if b2bii.isB2BII():
            copyList('gamma:evtkin', 'gamma:mdst', path=path)
        else:
            fillParticleList('gamma:evtkin', '', path=path)
        particleLists = ['pi+:evtkin', 'gamma:evtkin']
        if default_cleanup:
            if (custom_cuts is not None):
                B2INFO("Using default cleanup in EventKinematics module.")
            applyCuts('pi+:evtkin', trackCuts, path=path)
            applyCuts('gamma:evtkin', gammaCuts, path=path)
        else:
            B2INFO("No cleanup in EventKinematics module.")
    else:
        particleLists = inputListNames
 
    eventKinematicsModule = register_module('EventKinematics')
    eventKinematicsModule.set_name('EventKinematics_reco')
    eventKinematicsModule.param('particleLists', particleLists)
    path.add_module(eventKinematicsModule)
 
 
def buildEventKinematicsFromMC(inputListNames=None, selectionCut='', path=None):
    """
    Calculates the global kinematics of the event (visible energy, missing momentum, missing mass...)
    using generated particles. If no ParticleList is provided, default generated ParticleLists are used.
 
    @param inputListNames     list of ParticleLists used to calculate the global event kinematics.
                              If the list is empty, default ParticleLists are filled.
    @param selectionCut       optional selection cuts
    @param path               Path to append the eventKinematics module to.
    """
 
    if inputListNames is None:
        inputListNames = []
    if (len(inputListNames) == 0):
        # Type of particles to use for EventKinematics
        # K_S0 and Lambda0 are added here because some of them have interacted
        # with the detector material
        types = ['gamma', 'e+', 'mu+', 'pi+', 'K+', 'p+',
                 'K_S0', 'Lambda0']
        for t in types:
            fillParticleListFromMC(f"{t}:evtkin_default_gen", 'mcPrimary > 0 and nDaughters == 0',
                                   True, True, path=path)
            if (selectionCut != ''):
                applyCuts(f"{t}:evtkin_default_gen", selectionCut, path=path)
            inputListNames += [f"{t}:evtkin_default_gen"]
 
    eventKinematicsModule = register_module('EventKinematics')
    eventKinematicsModule.set_name('EventKinematics_gen')
    eventKinematicsModule.param('particleLists', inputListNames)
    eventKinematicsModule.param('usingMC', True)
    path.add_module(eventKinematicsModule)
 
 
def buildEventShape(inputListNames=None,
                    default_cleanup=True,
                    custom_cuts=None,
                    allMoments=False,
                    cleoCones=True,
                    collisionAxis=True,
                    foxWolfram=True,
                    harmonicMoments=True,
                    jets=True,
                    sphericity=True,
                    thrust=True,
                    checkForDuplicates=False,
                    path=None):
    """
    Calculates the event-level shape quantities (thrust, sphericity, Fox-Wolfram moments...)
    using the particles in the lists provided by the user. If no particle list is provided,
    the function will internally create a list of good tracks and a list of good photons
    with (optionally) minimal quality cuts.
 
 
    The results of the calculation are then stored into the EventShapeContainer dataobject,
    and are accessible using the variables of the EventShape group.
 
    The user can switch the calculation of certain quantities on or off to save computing
    time. By default the calculation of the high-order moments (5-8) is turned off.
    Switching off an option will make the corresponding variables not available.
 
    Info:
       The user can provide as many particle lists as needed, using also composite particles.
       In these cases, it is recommended to activate the checkForDuplicates flag since it
       will eliminate duplicates, e.g., if the same track is provided multiple times
       (either with different mass hypothesis or once as an independent particle and once
       as daughter of a composite particle). The first occurrence will be used in the
       calculations so the order in which the particle lists are given as well as within
       the particle lists matters.
 
    @param inputListNames     List of ParticleLists used to calculate the
                              event shape variables. If the list is empty the default
                              particleLists pi+:evtshape and gamma:evtshape are filled.
    @param default_cleanup    If True, applies standard cuts on pt and cosTheta when
                              defining the internal lists. This option is ignored if the
                              particleLists are provided by the user.
    @param custom_cuts        tuple of selection cut strings of form (trackCuts, photonCuts), default is None,
                              which would result in a standard predefined selection cuts
    @param path               Path to append the eventShape modules to.
    @param thrust             Enables the calculation of thrust-related quantities (CLEO
                              cones, Harmonic moments, jets).
    @param collisionAxis      Enables the calculation of the  quantities related to the
                              collision axis .
    @param foxWolfram         Enables the calculation of the Fox-Wolfram moments.
    @param harmonicMoments    Enables the calculation of the Harmonic moments with respect
                              to both the thrust axis and, if collisionAxis = True, the collision axis.
    @param allMoments         If True, calculates also the  FW and harmonic moments from order
                              5 to 8 instead of the low-order ones only.
    @param cleoCones          Enables the calculation of the CLEO cones with respect to both the thrust
                              axis and, if collisionAxis = True, the collision axis.
    @param jets               Enables the calculation of the hemisphere momenta and masses.
                              Requires thrust = True.
    @param sphericity         Enables the calculation of the sphericity-related quantities.
    @param checkForDuplicates Perform a check for duplicate particles before adding them. Regardless of the value of this option,
                              it is recommended to consider sanitizing the lists you are passing to the function since this will
                              speed up the processing.
 
    """
 
    if inputListNames is None:
        inputListNames = []
    trackCuts = 'pt > 0.1'
    trackCuts += ' and thetaInCDCAcceptance'
    trackCuts += ' and abs(dz) < 3.0'
    trackCuts += ' and dr < 0.5'
 
    gammaCuts = 'E > 0.05'
    gammaCuts += ' and thetaInCDCAcceptance'
    if not b2bii.isB2BII():
        gammaCuts += ' and abs(clusterTiming) < 200'
    if (custom_cuts is not None):
        trackCuts, gammaCuts = custom_cuts
 
    if not inputListNames:
        B2INFO("Creating particle lists pi+:evtshape and gamma:evtshape to get the event shape variables.")
        fillParticleList('pi+:evtshape', '', path=path)
        if b2bii.isB2BII():
            copyList('gamma:evtshape', 'gamma:mdst', path=path)
        else:
            fillParticleList(
                'gamma:evtshape',
                '',
                path=path)
        particleLists = ['pi+:evtshape', 'gamma:evtshape']
 
        if default_cleanup:
            if (custom_cuts is not None):
                B2INFO("Applying standard cuts")
            applyCuts('pi+:evtshape', trackCuts, path=path)
 
            applyCuts('gamma:evtshape', gammaCuts, path=path)
        else:
            B2WARNING("Creating the default lists with no cleanup.")
    else:
        particleLists = inputListNames
 
    eventShapeModule = register_module('EventShapeCalculator')
    eventShapeModule.set_name('EventShape')
    eventShapeModule.param('particleListNames', particleLists)
    eventShapeModule.param('enableAllMoments', allMoments)
    eventShapeModule.param('enableCleoCones', cleoCones)
    eventShapeModule.param('enableCollisionAxis', collisionAxis)
    eventShapeModule.param('enableFoxWolfram', foxWolfram)
    eventShapeModule.param('enableJets', jets)
    eventShapeModule.param('enableHarmonicMoments', harmonicMoments)
    eventShapeModule.param('enableSphericity', sphericity)
    eventShapeModule.param('enableThrust', thrust)
    eventShapeModule.param('checkForDuplicates', checkForDuplicates)
 
    path.add_module(eventShapeModule)
 
 
def labelTauPairMC(printDecayInfo=False, path=None, TauolaBelle=False, mapping_minus=None, mapping_plus=None):
    """
    Search tau leptons into the MC information of the event. If confirms it's a generated tau pair decay,
    labels the decay generated of the positive and negative leptons using the ID of KKMC tau decay table.
 
    @param printDecayInfo:  If true, prints ID and prong of each tau lepton in the event.
    @param path:        module is added to this path
    @param TauolaBelle: if False, TauDecayMode is set. If True, TauDecayMarker is set.
    @param mapping_minus: if None, the map is the default one, else the path for the map is given by the user for tau-
    @param mapping_plus: if None, the map is the default one, else the path for the map is given by the user for tau+
    """
 
    from basf2 import find_file
    if not TauolaBelle:
 
        if printDecayInfo:
            m_printmode = 'all'
        else:
            m_printmode = 'default'
 
        if mapping_minus is None:
            mp_file_minus = find_file('data/analysis/modules/TauDecayMode/map_tauminus.txt')
        else:
            mp_file_minus = mapping_minus
 
        if mapping_plus is None:
            mp_file_plus = find_file('data/analysis/modules/TauDecayMode/map_tauplus.txt')
        else:
            mp_file_plus = mapping_plus
 
        path.add_module('TauDecayMode', printmode=m_printmode, file_minus=mp_file_minus, file_plus=mp_file_plus)
 
    else:
        tauDecayMarker = register_module('TauDecayMarker')
        tauDecayMarker.set_name('TauDecayMarker_')
 
        path.add_module(tauDecayMarker, printDecayInfo=printDecayInfo)
 
 
def tagCurlTracks(particleLists,
                  mcTruth=False,
                  responseCut=-1.0,
                  selectorType='cut',
                  ptCut=0.5,
                  expert_train=False,
                  expert_filename="",
                  path=None):
    """
    Warning:
        The cut selector is not calibrated with Belle II data and should not be used without extensive study.
 
    Identifies curl tracks and tags them with extraInfo(isCurl=1) for later removal.
    For Belle data with a `b2bii` analysis the available cut based selection is described in `BN1079`_.
 
      .. _BN1079: https://belle.kek.jp/secured/belle_note/gn1079/bn1079.pdf
 
 
    The module loops over all particles in a given list with a transverse momentum below the pre-selection **ptCut**
    and assigns them to bundles based on the response of the chosen **selector** and the required minimum response set by the
    **responseCut**. Once all particles are assigned they are ranked by 25dr^2+dz^2. All but the lowest are tagged
    with extraInfo(isCurl=1) to allow for later removal by cutting the list or removing these from ROE as
    applicable.
 
 
    @param particleLists: list of particle lists to check for curls.
    @param mcTruth:       bool flag to additionally assign particles with extraInfo(isTruthCurl) and
                          extraInfo(truthBundleSize). To calculate these particles are assigned to bundles by their
                          genParticleIndex then ranked and tagged as normal.
    @param responseCut:   float min classifier response that considers two tracks to come from the same particle.
                          If set to ``-1`` a cut value optimised to maximise the accuracy on a BBbar sample is used.
                          Note 'cut' selector is binary 0/1.
    @param selectorType:  string name of selector to use. The available options are 'cut' and 'mva'.
                          It is strongly recommended to used the 'mva' selection. The 'cut' selection
                          is based on BN1079 and is only calibrated for Belle data.
 
    @param ptCut:         Pre-selection cut on transverse momentum. Only tracks below that are considered as curler candidates.
 
    @param expert_train:  flag to set training mode if selector has a training mode (mva).
    @param expert_filename: set file name of produced training ntuple (mva).
    @param path:          module is added to this path.
    """
 
    import b2bii
    belle = b2bii.isB2BII()
 
    if (not isinstance(particleLists, list)):
        particleLists = [particleLists]  # in case user inputs a particle list as string
 
    curlTagger = register_module('CurlTagger')
    curlTagger.set_name('CurlTagger_')
    curlTagger.param('particleLists', particleLists)
    curlTagger.param('belle', belle)
    curlTagger.param('mcTruth', mcTruth)
    curlTagger.param('responseCut', responseCut)
    if abs(responseCut + 1) < 1e-9:
        curlTagger.param('usePayloadCut', True)
    else:
        curlTagger.param('usePayloadCut', False)
 
    curlTagger.param('selectorType', selectorType)
    curlTagger.param('ptCut', ptCut)
    curlTagger.param('train', expert_train)
    curlTagger.param('trainFilename', expert_filename)
 
    path.add_module(curlTagger)
 
 
def applyChargedPidMVA(particleLists, path, trainingMode, chargeIndependent=False, binaryHypoPDGCodes=(0, 0)):
    """
    Use an MVA to perform particle identification for charged stable particles, using the `ChargedPidMVA` module.
 
    The module decorates Particle objects in the input ParticleList(s) with variables
    containing the appropriate MVA score, which can be used to select candidates by placing a cut on it.
 
    Note:
        The MVA algorithm used is a gradient boosted decision tree (**TMVA 4.3.0**, **ROOT 6.20/04**).
 
    The module can perform either 'binary' PID between input S, B particle mass hypotheses according to the following scheme:
 
    * e (11) vs. pi (211)
    * mu (13) vs. pi (211)
    * pi (211) vs. K (321)
    * K (321) vs. pi (211)
 
    , or 'global' PID, namely "one-vs-others" separation. The latter exploits an MVA algorithm trained in multi-class mode,
    and it's the default behaviour. Currently, the multi-class training separates the following standard charged hypotheses:
 
    - e (11), mu (13), pi (211), K (321)
 
    Warning:
        In order to run the `ChargedPidMVA` and ensure the most up-to-date MVA training weights are applied,
        it is necessary to append the latest analysis global tag (GT) to the steering script.
 
    Parameters:
        particleLists (list(str)): the input list of DecayStrings, where each selected (^) daughter should correspond to a
                                   standard charged ParticleList, e.g. ``['Lambda0:sig -> ^p+ ^pi-', 'J/psi:sig -> ^mu+ ^mu-']``.
                                   One can also directly pass a list of standard charged ParticleLists,
                                   e.g. ``['e+:my_electrons', 'pi+:my_pions']``.
                                   Note that charge-conjugated ParticleLists will automatically be included.
        path (basf2.Path): the module is added to this path.
        trainingMode (``Belle2.ChargedPidMVAWeights.ChargedPidMVATrainingMode``): enum identifier of the training mode.
          Needed to pick up the correct payload from the DB. Available choices:
 
          * c_Classification=0
          * c_Multiclass=1
          * c_ECL_Classification=2
          * c_ECL_Multiclass=3
          * c_PSD_Classification=4
          * c_PSD_Multiclass=5
          * c_ECL_PSD_Classification=6
          * c_ECL_PSD_Multiclass=7
 
        chargeIndependent (bool, ``optional``): use a BDT trained on a sample of inclusively charged particles.
        binaryHypoPDGCodes (tuple(int, int), ``optional``): the pdgIds of the signal, background mass hypothesis.
          Required only for binary PID mode.
    """
 
    import b2bii
    if b2bii.isB2BII():
        B2ERROR("Charged PID via MVA is not available for Belle data.")
 
    from ROOT import Belle2
 
    TrainingMode = Belle2.ChargedPidMVAWeights.ChargedPidMVATrainingMode
    Const = Belle2.Const
 
    plSet = set(particleLists)
 
    # Map the training mode enum value to the actual name of the payload in the GT.
    payloadNames = {
        TrainingMode.c_Classification:
        {"mode": "Classification", "detector": "ALL"},
        TrainingMode.c_Multiclass:
        {"mode": "Multiclass", "detector": "ALL"},
        TrainingMode.c_ECL_Classification:
        {"mode": "ECL_Classification", "detector": "ECL"},
        TrainingMode.c_ECL_Multiclass:
        {"mode": "ECL_Multiclass", "detector": "ECL"},
        TrainingMode.c_PSD_Classification:
        {"mode": "PSD_Classification", "detector": "ALL"},
        TrainingMode.c_PSD_Multiclass:
        {"mode": "PSD_Multiclass", "detector": "ALL"},
        TrainingMode.c_ECL_PSD_Classification:
        {"mode": "ECL_PSD_Classification", "detector": "ECL"},
        TrainingMode.c_ECL_PSD_Multiclass:
        {"mode": "ECL_PSD_Multiclass", "detector": "ECL"},
    }
 
    if payloadNames.get(trainingMode) is None:
        B2FATAL("The chosen training mode integer identifier:\n", trainingMode,
                "\nis not supported. Please choose among the following:\n",
                "\n".join(f"{key}:{val.get('mode')}" for key, val in sorted(payloadNames.items())))
 
    mode = payloadNames.get(trainingMode).get("mode")
    detector = payloadNames.get(trainingMode).get("detector")
 
    payloadName = f"ChargedPidMVAWeights_{mode}"
 
    # Map pdgIds of std charged particles to name identifiers,
    # and binary bkg identifiers.
    stdChargedMap = {
        Const.electron.getPDGCode():
        {"pName": "e", "pFullName": "electron", "pNameBkg": "pi", "pdgIdBkg": Const.pion.getPDGCode()},
        Const.muon.getPDGCode():
        {"pName": "mu", "pFullName": "muon", "pNameBkg": "pi", "pdgIdBkg": Const.pion.getPDGCode()},
        Const.pion.getPDGCode():
        {"pName": "pi", "pFullName": "pion", "pNameBkg": "K", "pdgIdBkg": Const.kaon.getPDGCode()},
        Const.kaon.getPDGCode():
        {"pName": "K", "pFullName": "kaon", "pNameBkg": "pi", "pdgIdBkg": Const.pion.getPDGCode()},
        Const.proton.getPDGCode():
        {"pName": "p", "pFullName": "proton", "pNameBkg": "pi", "pdgIdBkg": Const.pion.getPDGCode()},
        Const.deuteron.getPDGCode():
        {"pName": "d", "pFullName": "deuteron", "pNameBkg": "pi", "pdgIdBkg": Const.pion.getPDGCode()},
    }
 
    if binaryHypoPDGCodes == (0, 0):
 
        # MULTI-CLASS training mode.
        chargedpid = register_module("ChargedPidMVAMulticlass")
        chargedpid.set_name(f"ChargedPidMVAMulticlass_{mode}")
 
    else:
 
        # BINARY training mode.
        # In binary mode, enforce check on input S, B hypotheses compatibility.
 
        binaryOpts = [(pdgIdSig, info["pdgIdBkg"]) for pdgIdSig, info in stdChargedMap.items()]
 
        if binaryHypoPDGCodes not in binaryOpts:
            B2FATAL("No charged pid MVA was trained to separate ", binaryHypoPDGCodes[0], " vs. ", binaryHypoPDGCodes[1],
                    ". Please choose among the following pairs:\n",
                    "\n".join(f"{opt[0]} vs. {opt[1]}" for opt in binaryOpts))
 
        decayDescriptor = Belle2.DecayDescriptor()
        for name in plSet:
            if not decayDescriptor.init(name):
                raise ValueError(f"Invalid particle list {name} in applyChargedPidMVA!")
            msg = f"Input ParticleList: {name}"
            pdgs = [abs(decayDescriptor.getMother().getPDGCode())]
            daughter_pdgs = decayDescriptor.getSelectionPDGCodes()
            if len(daughter_pdgs) > 0:
                pdgs = daughter_pdgs
            for idaughter, pdg in enumerate(pdgs):
                if abs(pdg) not in binaryHypoPDGCodes:
                    if daughter_pdgs:
                        msg = f"Selected daughter {idaughter} in ParticleList: {name}"
                    B2WARNING(
                        f"{msg} (PDG={pdg}) is neither signal ({binaryHypoPDGCodes[0]}) nor background ({binaryHypoPDGCodes[1]}).")
 
        chargedpid = register_module("ChargedPidMVA")
        chargedpid.set_name(f"ChargedPidMVA_{binaryHypoPDGCodes[0]}_vs_{binaryHypoPDGCodes[1]}_{mode}")
        chargedpid.param("sigHypoPDGCode", binaryHypoPDGCodes[0])
        chargedpid.param("bkgHypoPDGCode", binaryHypoPDGCodes[1])
 
    chargedpid.param("particleLists", list(plSet))
    chargedpid.param("payloadName", payloadName)
    chargedpid.param("chargeIndependent", chargeIndependent)
 
    # Ensure the module knows whether we are using ECL-only training mode.
    if detector == "ECL":
        chargedpid.param("useECLOnlyTraining", True)
 
    path.add_module(chargedpid)
 
 
def calculateTrackIsolation(
        decay_string,
        path,
        *detectors,
        reference_list_name=None,
        vars_for_nearest_part=[],
        highest_prob_mass_for_ext=True,
        exclude_pid_det_weights=False):
    """
    Given an input decay string, compute variables that quantify track helix-based isolation of the charged
    stable particles in the input decay chain.
 
    Note:
        An "isolation score" can be defined using the distance
        of each particle to its closest neighbour, defined as the segment connecting the two
        extrapolated track helices intersection points on a given cylindrical surface.
        The distance variables defined in the `VariableManager` is named `minET2ETDist`,
        the isolation scores are named `minET2ETIsoScore`, `minET2ETIsoScoreAsWeightedAvg`.
 
    The definition of distance and the number of distances that are calculated per sub-detector is based on
    the following recipe:
 
    * **CDC**: as the segmentation is very coarse along :math:`z`,
      the distance is defined as the cord length on the :math:`(\\rho=R, \\phi)` plane.
      A total of 9 distances are calculated: the cylindrical surfaces are defined at radiuses
      that correspond to the positions of the 9 CDC wire superlayers: :math:`R_{i}^{\\mathrm{CDC}}~(i \\in \\{0,...,8\\})`.
 
    * **TOP**: as there is no segmentation along :math:`z`,
      the distance is defined as the cord length on the :math:`(\\rho=R, \\phi)` plane.
      Only one distance at the TOP entry radius :math:`R_{0}^{\\mathrm{TOP}}` is calculated.
 
    * **ARICH**: as there is no segmentation along :math:`z`,
      the distance is defined as the distance on the :math:`(\\rho=R, \\phi)` plane at fixed :math:`z=Z`.
      Only one distance at the ARICH photon detector entry coordinate :math:`Z_{0}^{\\mathrm{ARICH}}` is calculated.
 
    * **ECL**: the distance is defined on the :math:`(\\rho=R, \\phi, z)` surface in the barrel,
      on the :math:`(\\rho, \\phi, z=Z)` surface in the endcaps.
      Two distances are calculated: one at the ECL entry surface :math:`R_{0}^{\\mathrm{ECL}}` (barrel),
      :math:`Z_{0}^{\\mathrm{ECL}}` (endcaps), and one at :math:`R_{1}^{\\mathrm{ECL}}` (barrel),
      :math:`Z_{1}^{\\mathrm{ECL}}` (endcaps), corresponding roughly to the mid-point
      of the longitudinal size of the crystals.
 
    * **KLM**: the distance is defined on the :math:`(\\rho=R, \\phi, z)` surface in the barrel,
      on the :math:`(\\rho, \\phi, z=Z)` surface in the endcaps.
      Only one distance at the KLM first strip entry surface :math:`R_{0}^{\\mathrm{KLM}}` (barrel),
      :math:`Z_{0}^{\\mathrm{KLM}}` (endcaps) is calculated.
 
    Parameters:
        decay_string (str): name of the input decay string with selected charged stable daughters,
                            for example: ``Lambda0:merged -> ^p+ ^pi-``.
                            Alternatively, it can be a particle list for charged stable particles
                            as defined in ``Const::chargedStableSet``, for example: ``mu+:all``.
                            The charge-conjugate particle list will be also processed automatically.
        path (basf2.Path): path to which module(s) will be added.
        *detectors: detectors for which track isolation variables will be calculated.
                    Choose among: ``{'CDC', 'TOP', 'ARICH', 'ECL', 'KLM'}``.
        reference_list_name (Optional[str]): name of the input charged stable particle list for the reference tracks.
                                             By default, the ``:all`` ParticleList of the same type
                                             of the selected particle in ``decay_string`` is used.
                                             The charge-conjugate particle list will be also processed automatically.
        vars_for_nearest_part (Optional[list(str)]): a list of variables to calculate for the nearest particle in the reference
                                                     list at each detector surface. It uses the metavariable `minET2ETDistVar`.
                                                     If unset, only the distances to the nearest neighbour
                                                     per detector are calculated.
        highest_prob_mass_for_hex (Optional[bool]): if this option is set to True (default), the helix extrapolation
                                                    for the particles will use the track fit result for the most
                                                    probable mass hypothesis, namely, the one that gives the highest
                                                    chi2Prob of the fit. Otherwise, it uses the mass hypothesis that
                                                    corresponds to the particle lists PDG.
        exclude_pid_det_weights (Optional[bool]): if this option is set to False (default), the isolation score
                                                  calculation will take into account the weight that each detector has on the PID
                                                  for the particle species of interest.
 
    Returns:
        dict(int, list(str)): a dictionary mapping the PDG of each reference particle list to its isolation variables.
 
    """
 
    import pdg
    from ROOT import Belle2, TDatabasePDG
 
    decayDescriptor = Belle2.DecayDescriptor()
    if not decayDescriptor.init(decay_string):
        B2FATAL(f"Invalid particle list {decay_string} in calculateTrackIsolation!")
    no_reference_list_name = not reference_list_name
 
    det_and_layers = {
        "CDC": list(range(9)),
        "TOP": [0],
        "ARICH": [0],
        "ECL": [0, 1],
        "KLM": [0],
    }
    if any(d not in det_and_layers for d in detectors):
        B2FATAL(
            "Your input detector list: ",
            detectors,
            " contains an invalid choice. Please select among: ",
            list(
                det_and_layers.keys()))
 
    # The module allows only one daughter to be selected at a time,
    # that's why here we preprocess the input decay string.
    select_symbol = '^'
    processed_decay_strings = []
    if select_symbol in decay_string:
        splitted_ds = decay_string.split(select_symbol)
        for i in range(decay_string.count(select_symbol)):
            tmp = list(splitted_ds)
            tmp.insert(i+1, select_symbol)
            processed_decay_strings += [''.join(tmp)]
    else:
        processed_decay_strings += [decay_string]
 
    reference_lists_to_vars = {}
 
    for processed_dec in processed_decay_strings:
        if no_reference_list_name:
            decayDescriptor.init(processed_dec)
            selected_daughter_pdgs = decayDescriptor.getSelectionPDGCodes()
            if len(selected_daughter_pdgs) > 0:
                reference_list_name = f'{TDatabasePDG.Instance().GetParticle(abs(selected_daughter_pdgs[-1])).GetName()}:all'
            else:
                reference_list_name = f'{processed_dec.split(":")[0]}:all'
 
        ref_pdg = pdg.from_name(reference_list_name.split(":")[0])
 
        trackiso = path.add_module("TrackIsoCalculator",
                                   decayString=processed_dec,
                                   detectorNames=list(detectors),
                                   particleListReference=reference_list_name,
                                   useHighestProbMassForExt=highest_prob_mass_for_ext,
                                   excludePIDDetWeights=exclude_pid_det_weights)
        trackiso.set_name(f"TrackIsoCalculator_{'_'.join(detectors)}_{processed_dec}_VS_{reference_list_name}")
 
        # Metavariables for the distances to the closest reference tracks at each detector surface.
        # Always calculate them.
        # Ensure the flag for the mass hypothesis of the fit is set.
        trackiso_vars = [
            f"minET2ETDist({d}, {d_layer}, {reference_list_name}, {int(highest_prob_mass_for_ext)})"
            for d in detectors for d_layer in det_and_layers[d]]
        # Track isolation score.
        trackiso_vars += [
            f"minET2ETIsoScore({reference_list_name}, {int(highest_prob_mass_for_ext)}, {', '.join(detectors)})",
            f"minET2ETIsoScoreAsWeightedAvg({reference_list_name}, {int(highest_prob_mass_for_ext)}, {', '.join(detectors)})",
        ]
        # Optionally, calculate the input variables for the nearest neighbour in the reference list.
        if vars_for_nearest_part:
            trackiso_vars.extend(
                [
                    f"minET2ETDistVar({d}, {d_layer}, {reference_list_name}, {v})"
                    for d in detectors for d_layer in det_and_layers[d] for v in vars_for_nearest_part
                ])
        trackiso_vars.sort()
 
        reference_lists_to_vars[ref_pdg] = trackiso_vars
 
    return reference_lists_to_vars
 
 
def calculateDistance(list_name, decay_string, mode='vertextrack', path=None):
    """
    Calculates distance between two vertices, distance of closest approach between a vertex and a track,\
    distance of closest approach between a vertex and btube. For track, this calculation ignores track curvature,\
    it's negligible for small distances.The user should use extraInfo(CalculatedDistance)\
    to get it. A full example steering file is at analysis/tests/test_DistanceCalculator.py
 
    Example:
        .. code-block:: python
 
            from modularAnalysis import calculateDistance
            calculateDistance('list_name', 'decay_string', "mode", path=user_path)
 
    @param list_name              name of the input ParticleList
    @param decay_string           select particles between the distance of closest approach will be calculated
    @param mode                   Specifies how the distance is calculated
                                  vertextrack: calculate the distance of closest approach between a track and a\
                                   vertex, taking the first candidate as vertex, default
                                  trackvertex: calculate the distance of closest approach between a track and a\
                                   vertex, taking the first candidate as track
                                  2tracks: calculates the distance of closest approach between two tracks
                                  2vertices: calculates the distance between two vertices
                                  vertexbtube: calculates the distance of closest approach between a vertex and btube
                                  trackbtube: calculates the distance of closest approach between a track and btube
    @param path                   modules are added to this path
 
    """
 
    dist_mod = register_module('DistanceCalculator')
 
    dist_mod.set_name('DistanceCalculator_' + list_name)
    dist_mod.param('listName', list_name)
    dist_mod.param('decayString', decay_string)
    dist_mod.param('mode', mode)
    path.add_module(dist_mod)
 
 
def addInclusiveDstarReconstruction(decayString, slowPionCut, DstarCut, path):
    """
    Adds the InclusiveDstarReconstruction module to the given path.
    This module creates a D* particle list by estimating the D* four momenta
    from slow pions, specified by a given cut. The D* energy is approximated
    as  E(D*) = m(D*)/(m(D*) - m(D)) * E(pi). The absolute value of the D*
    momentum is calculated using the D* PDG mass and the direction is collinear
    to the slow pion direction. The charge of the given pion list has to be consistent
    with the D* charge
 
    @param decayString Decay string, must be of form ``D* -> pi``
    @param slowPionCut Cut applied to the input pion list to identify slow pions
    @param DstarCut Cut applied to the output D* list
    @param path the module is added to this path
    """
 
    incl_dstar = register_module("InclusiveDstarReconstruction")
    incl_dstar.param("decayString", decayString)
    incl_dstar.param("slowPionCut", slowPionCut)
    incl_dstar.param("DstarCut", DstarCut)
    path.add_module(incl_dstar)
 
 
def scaleError(outputListName, inputListName,
               scaleFactors=[1.149631, 1.085547, 1.151704, 1.096434, 1.086659],
               scaleFactorsNoPXD=[1.149631, 1.085547, 1.151704, 1.096434, 1.086659],
               d0Resolution=[0.00115328, 0.00134704],
               z0Resolution=[0.00124327, 0.0013272],
               d0MomThr=0.500000,
               z0MomThr=0.500000,
               path=None):
    """
    This module creates a new charged particle list.
    The helix errors of the new particles are scaled by constant factors.
    Two sets of five scale factors are defined for tracks with and without a PXD hit.
    The scale factors are in order of (d0, phi0, omega, z0, tanlambda).
    For tracks with a PXD hit, in order to avoid severe underestimation of d0 and z0 errors,
    lower limits (best resolution) can be set in a momentum-dependent form.
    This module is supposed to be used only for TDCPV analysis and for low-momentum (0-3 GeV/c) tracks in BBbar events.
    Details will be documented in a Belle II note, BELLE2-NOTE-PH-2021-038.
 
    @param inputListName Name of input charged particle list to be scaled
    @param outputListName Name of output charged particle list with scaled error
    @param scaleFactors List of five constants to be multiplied to each of helix errors (for tracks with a PXD hit)
    @param scaleFactorsNoPXD List of five constants to be multiplied to each of helix errors (for tracks without a PXD hit)
    @param d0Resolution List of two parameters, (a [cm], b [cm/(GeV/c)]),
                        defining d0 best resolution as sqrt{ a**2 + (b / (p*beta*sinTheta**1.5))**2 }
    @param z0Resolution List of two parameters, (a [cm], b [cm/(GeV/c)]),
                        defining z0 best resolution as sqrt{ a**2 + (b / (p*beta*sinTheta**2.5))**2 }
    @param d0MomThr d0 best resolution is kept constant below this momentum
    @param z0MomThr z0 best resolution is kept constant below this momentum
 
    """
 
    scale_error = register_module("HelixErrorScaler")
    scale_error.set_name('ScaleError_' + inputListName)
    scale_error.param('inputListName', inputListName)
    scale_error.param('outputListName', outputListName)
    scale_error.param('scaleFactors_PXD', scaleFactors)
    scale_error.param('scaleFactors_noPXD', scaleFactorsNoPXD)
    scale_error.param('d0ResolutionParameters', d0Resolution)
    scale_error.param('z0ResolutionParameters', z0Resolution)
    scale_error.param('d0MomentumThreshold', d0MomThr)
    scale_error.param('z0MomentumThreshold', z0MomThr)
    path.add_module(scale_error)
 
 
def estimateAndAttachTrackFitResult(inputListName, path=None):
    """
    Create a TrackFitResult from the momentum of the Particle assuming it originates from the IP and make a relation between them.
    The covariance, detector hit information, and fit-related information (pValue, NDF) are assigned meaningless values. The input
    Particles must not have already Track or TrackFitResult and thus are supposed to be composite particles, recoil, dummy
    particles, and so on.
 
 
    .. warning:: Since the source type is not overwritten as Track, not all track-related variables are guaranteed to be available.
 
 
    @param inputListName Name of input ParticleList
    """
 
    estimator = register_module("TrackFitResultEstimator")
    estimator.set_name("trackFitResultEstimator_" + inputListName)
    estimator.param("inputListName", inputListName)
    path.add_module(estimator)
 
 
def correctEnergyBias(inputListNames, tableName, path=None):
    """
    Scale energy of the particles according to the scaling factor.
    If the particle list contains composite particles, the energy of the daughters are scaled.
    Subsequently, the energy of the mother particle is updated as well.
 
    Parameters:
        inputListNames (list(str)): input particle list names
        tableName : stored in localdb and created using ParticleWeightingLookUpCreator
        path (basf2.Path): module is added to this path
    """
 
    import b2bii
    if b2bii.isB2BII():
        B2ERROR("The energy bias cannot be corrected with this tool for Belle data.")
 
    correctenergybias = register_module('EnergyBiasCorrection')
    correctenergybias.param('particleLists', inputListNames)
    correctenergybias.param('tableName', tableName)
    path.add_module(correctenergybias)
 
 
def twoBodyISRPhotonCorrector(outputListName, inputListName, massiveParticle, path=None):
    """
    Sets photon kinematics to corrected values in two body decays with an ISR photon
    and a massive particle. The original photon kinematics are kept in the input
    particleList and can be accessed using the originalParticle() metavariable on the
    new list.
 
    @param ouputListName    new ParticleList filled with copied Particles
    @param inputListName    input ParticleList with original Particles
    @param massiveParticle  name or PDG code of massive particle participating in the two
                            body decay with the ISR photon
    @param path             modules are added to this path
    """
 
    # set the corrected energy of the photon in a new list
    photon_energy_correction = register_module('TwoBodyISRPhotonCorrector')
    photon_energy_correction.set_name('TwoBodyISRPhotonCorrector_' + outputListName)
    photon_energy_correction.param('outputGammaList', outputListName)
    photon_energy_correction.param('inputGammaList', inputListName)
 
    # prepare PDG code of massive particle
    if isinstance(massiveParticle, int):
        photon_energy_correction.param('massiveParticlePDGCode', massiveParticle)
    else:
        from ROOT import Belle2
        decayDescriptor = Belle2.DecayDescriptor()
        if not decayDescriptor.init(massiveParticle):
            raise ValueError("TwoBodyISRPhotonCorrector: value of massiveParticle must be" +
                             " an int or valid decay string.")
        pdgCode = decayDescriptor.getMother().getPDGCode()
        photon_energy_correction.param('massiveParticlePDGCode', pdgCode)
 
    path.add_module(photon_energy_correction)
 
 
def addPhotonEfficiencyRatioVariables(inputListNames, tableName, path=None):
    """
    Add photon Data/MC detection efficiency ratio weights to the specified particle list
 
    Parameters:
        inputListNames (list(str)): input particle list names
        tableName : taken from database with appropriate name
        path (basf2.Path): module is added to this path
    """
 
    import b2bii
    if b2bii.isB2BII():
        B2ERROR("For Belle data the photon data/MC detection efficiency ratio is not available with this tool.")
 
    photon_efficiency_correction = register_module('PhotonEfficiencySystematics')
    photon_efficiency_correction.param('particleLists', inputListNames)
    photon_efficiency_correction.param('tableName', tableName)
    path.add_module(photon_efficiency_correction)
 
 
def addPi0VetoEfficiencySystematics(particleList, decayString, tableName, threshold, mode='standard', suffix='', path=None):
    """
    Add pi0 veto Data/MC efficiency ratio weights to the specified particle list
 
    @param particleList   the input ParticleList
    @param decayString    specify hard photon to be performed pi0 veto (e.g. 'B+:sig -> rho+:sig ^gamma:hard')
    @param tableName      table name corresponding to payload version (e.g. 'Pi0VetoEfficiencySystematics_Mar2022')
    @param threshold      pi0 veto threshold (0.10, 0.11, ..., 0.99)
    @param mode           choose one mode (same as writePi0EtaVeto) out of 'standard', 'tight', 'cluster' and 'both'
    @param suffix         optional suffix to be appended to the usual extraInfo name
    @param path           the module is added to this path
 
    The following extraInfo are available related with the given particleList:
 
    * Pi0VetoEfficiencySystematics_{mode}{suffix}_data_MC_ratio             : weight of Data/MC for the veto efficiency
    * Pi0VetoEfficiencySystematics_{mode}{suffix}_data_MC_uncertainty_stat  : the statistical uncertainty of the weight
    * Pi0VetoEfficiencySystematics_{mode}{suffix}_data_MC_uncertainty_sys   : the systematic uncertainty of the weight
    * Pi0VetoEfficiencySystematics_{mode}{suffix}_data_MC_uncertainty_total : the total uncertainty of the weight
    * Pi0VetoEfficiencySystematics_{mode}{suffix}_threshold                 : threshold of the pi0 veto
    """
 
    import b2bii
    if b2bii.isB2BII():
        B2ERROR("For Belle data the pi0 veto data/MC efficiency ratio weights are not available via this tool.")
 
    pi0veto_efficiency_correction = register_module('Pi0VetoEfficiencySystematics')
    pi0veto_efficiency_correction.param('particleLists', particleList)
    pi0veto_efficiency_correction.param('decayString', decayString)
    pi0veto_efficiency_correction.param('tableName', tableName)
    pi0veto_efficiency_correction.param('threshold', threshold)
    pi0veto_efficiency_correction.param('mode', mode)
    pi0veto_efficiency_correction.param('suffix', suffix)
    path.add_module(pi0veto_efficiency_correction)
 
 
def getAnalysisGlobaltag(timeout=180) -> str:
    """
    Returns a string containing the name of the latest and recommended analysis globaltag.
 
    Parameters:
        timeout: Seconds to wait for b2conditionsdb-recommend
    """
 
    import b2bii
    if b2bii.isB2BII():
        B2ERROR("The getAnalysisGlobaltag() function cannot be used for Belle data.")
 
    # b2conditionsdb-recommend relies on a different repository, so it's better to protect
    # this function against potential failures of check_output.
    try:
        tags = subprocess.check_output(
            ['b2conditionsdb-recommend', '--oneline'],
            timeout=timeout
        ).decode('UTF-8').rstrip().split(' ')
        analysis_tag = ''
        for tag in tags:
            if tag.startswith('analysis_tools'):
                analysis_tag = tag
        return analysis_tag
    # In case of issues with git, b2conditionsdb-recommend may take too much time.
    except subprocess.TimeoutExpired as te:
        B2FATAL(f'A {te} exception was raised during the call of getAnalysisGlobaltag(). '
                'The function took too much time to retrieve the requested information '
                'from the versioning repository.\n'
                'Please try to re-run your job. In case of persistent failures, there may '
                'be issues with the DESY collaborative services, so please contact the experts.')
    except subprocess.CalledProcessError as ce:
        B2FATAL(f'A {ce} exception was raised during the call of getAnalysisGlobaltag(). '
                'Please try to re-run your job. In case of persistent failures, please contact '
                'the experts.')
 
 
def getAnalysisGlobaltagB2BII() -> str:
    """
    Get recommended global tag for B2BII analysis.
    """
 
    import b2bii
    if not b2bii.isB2BII():
        B2ERROR('The getAnalysisGlobaltagB2BII() function cannot be used for Belle II data.')
    from versioning import recommended_b2bii_analysis_global_tag
    return recommended_b2bii_analysis_global_tag()
 
 
def getNbarIDMVA(particleList: str, path=None):
    """
    This function can give a score to predict if it is a anti-n0.
    It is not used to predict n0.
    Currently, this can be used only for ECL cluster.
    output will be stored in extraInfo(nbarID); -1 means MVA invalid
 
    @param particleList     The input ParticleList name or a decay string which contains a full mother particle list name.
                            Only one selected daughter is supported.
    @param path             modules are added to this path
    """
    import b2bii
    from ROOT import Belle2
 
    if b2bii.isB2BII():
        B2ERROR("The MVA-based anti-neutron PID is only available for Belle II data.")
 
    from variables import variables
 
    variables.addAlias('V1', 'clusterHasPulseShapeDiscrimination')
    variables.addAlias('V2', 'clusterE')
    variables.addAlias('V3', 'clusterLAT')
    variables.addAlias('V4', 'clusterE1E9')
    variables.addAlias('V5', 'clusterE9E21')
    variables.addAlias('V6', 'clusterZernikeMVA')
    variables.addAlias('V7', 'clusterAbsZernikeMoment40')
    variables.addAlias('V8', 'clusterAbsZernikeMoment51')
 
    variables.addAlias(
        'nbarIDValid',
        'passesCut(V1 == 1 and V2 >= 0 and V3 >= 0 and V4 >= 0 and V5 >= 0 and V6 >= 0 and V7 >= 0 and V8 >= 0)')
    variables.addAlias('nbarIDmod', 'conditionalVariableSelector(nbarIDValid == 1, extraInfo(nbarIDFromMVA), constant(-1.0))')
 
    path.add_module('MVAExpert', listNames=particleList, extraInfoName='nbarIDFromMVA', identifier='db_nbarIDECL')
    decayDescriptor = Belle2.DecayDescriptor()
    if not decayDescriptor.init(particleList):
        raise ValueError(f"Provided decay string is invalid: {particleList}")
    if decayDescriptor.getNDaughters() == 0:
        variablesToExtraInfo(particleList, {'nbarIDmod': 'nbarID'}, option=2, path=path)
    else:
        listname = decayDescriptor.getMother().getFullName()
        variablesToDaughterExtraInfo(listname, particleList, {'nbarIDmod': 'nbarID'}, option=2, path=path)
 
 
def reconstructDecayWithNeutralHadron(decayString, cut, allowGamma=False, allowAnyParticleSource=False, path=None, **kwargs):
    r"""
    Reconstructs decay with a long-lived neutral hadron e.g.
    :math:`B^0 \to J/\psi K_L^0`,
    :math:`B^0 \to p \bar{n} D^*(2010)^-`.
 
    The calculation is done with IP constraint and mother mass constraint.
 
    The decay string passed in must satisfy the following rules:
 
    - The neutral hadron must be **selected** in the decay string with the
      caret (``^``) e.g. ``B0:sig -> J/psi:sig ^K_L0:sig``. (Note the caret
      next to the neutral hadron.)
    - There can only be **one neutral hadron in a decay**.
    - The neutral hadron has to be a direct daughter of its mother.
 
    .. note:: This function forwards its arguments to `reconstructDecay`,
       so please check the documentation of `reconstructDecay` for all
       possible arguments.
 
    @param decayString A decay string following the mentioned rules
    @param cut Cut to apply to the particle list
    @param allowGamma Whether allow the selected particle to be ``gamma``
    @param allowAnyParticleSource Whether allow the selected particle to be from any source.
                                  Should only be used when studying control sample.
    @param path The path to put in the module
    """
 
    reconstructDecay(decayString, cut, path=path, **kwargs)
    module = register_module('NeutralHadron4MomentumCalculator')
    module.set_name('NeutralHadron4MomentumCalculator_' + decayString)
    module.param('decayString', decayString)
    module.param('allowGamma', allowGamma)
    module.param('allowAnyParticleSource', allowAnyParticleSource)
    path.add_module(module)
 
 
def updateMassHypothesis(particleList, pdg, writeOut=False, path=None):
    """
    Module to update the mass hypothesis of a given input particle list with the chosen PDG.
    A new particle list is created with updated mass hypothesis.
    The allowed mass hypotheses for both input and output are electrons, muons, pions, kaons and protons.
 
    .. note:
        The new particle list is named after the input one, with the additional suffix ``_converted_from_OLDHYPOTHESIS``,
        e.g. ``e+:all`` converted to muons becomes ``mu+:all_converted_from_e``.
 
    @param particleList The input particle list name
    @param pdg          The PDG code for the new mass hypothesis, in [11, 13, 211, 321, 2212]
    @param writeOut     Whether `RootOutput` module should save the new particle list
    @param path         Modules are added to this path
    """
    mass_updater = register_module("ParticleMassHypothesesUpdater")
    mass_updater.set_name("ParticleMassHypothesesUpdater_" + particleList + "_to_" + str(pdg))
    mass_updater.param("particleList", particleList)
    mass_updater.param("writeOut", writeOut)
    mass_updater.param("pdgCode", pdg)
    path.add_module(mass_updater)
 
 
func_requiring_analysisGT = [
    correctTrackEnergy, scaleTrackMomenta, smearTrackMomenta, oldwritePi0EtaVeto, writePi0EtaVeto, lowEnergyPi0Identification,
    getBeamBackgroundProbability, getFakePhotonProbability, tagCurlTracks, applyChargedPidMVA, correctEnergyBias,
    addPhotonEfficiencyRatioVariables, addPi0VetoEfficiencySystematics, getNbarIDMVA]
for _ in func_requiring_analysisGT:
    _.__doc__ += "\n    .. note:: This function (optionally) requires a payload stored in the analysis GlobalTag. "\
                    "Please append or prepend the latest one from `getAnalysisGlobaltag` or `getAnalysisGlobaltagB2BII`.\n"
 
 
if __name__ == '__main__':
    from basf2.utils import pretty_print_module
    pretty_print_module(__name__, "modularAnalysis")