Source code for iov_conditional

#!/usr/bin/env python3

##########################################################################
# basf2 (Belle II Analysis Software Framework)                           #
# Author: The Belle II Collaboration                                     #
#                                                                        #
# See git log for contributors and copyright holders.                    #
# This file is licensed under LGPL-3.0, see LICENSE.md.                  #
##########################################################################
"""
iov_conditional - Functions to Execute Paths Depending on Experiment Phases
===========================================================================

This module contains some convenience functions to execute different paths
depending on the experiment run number. This allows to write one common steering
file for phases 2 and 3.

Warning:
    All Modules will be initialized on startup of the framework. This might
    cause problems if the modules initialize some common global state. For
    example, adding ``Geometry`` modules with different parameters will not
    work as expected.
"""
import basf2

#: Definition of Phase 2 in a) real data and b) run-independent MC/development globaltags
PHASE2_IOVS = [(1, 0, 4, -1), (1002, 0, 1002, -1)]


[docs] def make_conditional_at(path, iov_list, path_when_in_iov, path_when_not_in_iov=None): """ Branch the current path execution based on the given iov list. If the current exp/run is met by one of the IoV conditions, execute the path given in ``path_when_in_iov`` and continue the normal path afterwards. Otherwise execute the ``path_when_not_in_iov`` when given. Basically, this function is just a wrapper around adding the :b2:mod:`IoVDependentCondition` module to the path. See Also: If you just need to distinguish between phase 2/3 please use `phase_2_conditional()` Example: >>> make_conditional_at(path, iov_list=[(0, 0, 0, -1)], ... path_when_in_iov=exp_0_path, ... path_when_not_in_iov=not_exp_0_path) will branch the path for all events with experiment number 0 to the modules in ``exp_0_path`` and into ``not_exp_0_path`` in all other cases. Parameters: path (basf2.Path): Branch the execution of the given path. iov_list (list(tuple)): Under which IoV conditions should the ``path_when_in_iov`` be executed. It should be a list in the form ``[(min exp, min run, max exp, max run), ...]`` path_when_in_iov (basf2.Path): Which branch to execute, if one of the IoV conditions is met. path_when_not_in_iov (basf2.Path): If given, execute this path in all cases, none IoV condition is met. """ condition_module = path.add_module("IoVDependentCondition", iovList=iov_list) condition_module.if_true(path_when_in_iov, basf2.AfterConditionPath.CONTINUE) if path_when_not_in_iov: condition_module.if_false(path_when_not_in_iov, basf2.AfterConditionPath.CONTINUE)
[docs] def phase_2_conditional(path, phase2_path, phase3_path=None): """ Handy shortcut for phase 2/3 conditional module execution. See Also: `make_conditional_at()` Parameters: path (basf2.Path): Branch the execution of the given path. phase2_path (basf2.Path): Call this path only when in phase 2. phase3_path (basf2.Path): When given, call this path only when in phase 3. """ make_conditional_at(path=path, iov_list=PHASE2_IOVS, path_when_in_iov=phase2_path, path_when_not_in_iov=phase3_path)