release-05-02-19/doxygen/packers_8py_source.html

#!/usr/bin/env python3

# -*- coding: utf-8 -*-


import basf2 as b2

import simulation as sim

import reconstruction as rec

import rawdata as raw


from ROOT import Belle2


import sys

import os

import multiprocessing as mp


path_to_output = 'rawdata/tests/digits.root'

b2.set_random_seed("L1V0RN0")

b2.set_log_level(b2.LogLevel.WARNING)

# Disable tag replay, we want to test current packers

# independent of when the digits were created

b2.conditions.disable_globaltag_replay()


def create_digits():

    """Create the file 'digits.root' needed for testing the packers if it does not exist"""

    child_path = b2.create_path()

    child_path.add_module('EventInfoSetter',

                          evtNumList=[10])

    child_path.add_module('EvtGenInput')

    sim.add_simulation(path=child_path)

    child_path.add_module(

        'RootOutput',

        outputFileName='${BELLE2_LOCAL_DIR}/' +

        path_to_output,

        branchNames=list(rec.DIGITS_OBJECTS))

    child_path.add_module('Progress')

    b2.process(child_path)

    print(b2.statistics)


if Belle2.FileSystem.findFile(path_to_output, True) == '':

    # Execute create_digits() in a child process to avoid side effects

    child = mp.Process(target=create_digits)

    child.start()

    # Wait for simulation to finish

    child.join()

    # And exit if it had an error

    if child.exitcode != 0:

        sys.exit(child.exitcode)


# Here starts the main path

main_path = b2.create_path()

main_path.add_module('RootInput',

                     inputFileNames=Belle2.FileSystem.findFile(path_to_output))

raw.add_packers(path=main_path)

main_path.add_module('Progress')

b2.process(main_path)

print(b2.statistics)