4 Airflow script to perform BeamSpot calibration.
7 from prompt
import CalibrationSettings
10 settings = CalibrationSettings(
11 name=
"BeamSpot Calibrations",
12 expert_username=
"zlebcr",
14 input_data_formats=[
"cdst"],
15 input_data_names=[
"mumutight_calib"],
17 "outerLoss":
"pow(rawTime - 2.0, 2) + 10 * pow(maxGap, 2)",
18 "innerLoss":
"pow(rawTime - 0.5, 2) + 10 * pow(maxGap, 2)"},
24 def get_calibrations(input_data, **kwargs):
27 input_data (dict): Should contain every name from the 'input_data_names' variable as a key.
28 Each value is a dictionary with {"/path/to/file_e1_r5.root": IoV(1,5,1,5), ...}. Useful for
29 assigning to calibration.files_to_iov
31 **kwargs: Configuration options to be sent in. Since this may change we use kwargs as a way to help prevent
32 backwards compatibility problems. But you could use the correct arguments in b2caf-prompt-run for this
33 release explicitly if you want to.
35 Currently only kwargs["output_iov"] is used. This is the output IoV range that your payloads should
36 correspond to. Generally your highest ExpRun payload should be open ended e.g. IoV(3,4,-1,-1)
39 list(caf.framework.Calibration): All of the calibration objects we want to assign to the CAF process
46 file_to_iov_physics = input_data[
"mumutight_calib"]
51 max_files_per_run = 1000000
57 reduced_file_to_iov_physics = filter_by_max_files_per_run(file_to_iov_physics, max_files_per_run)
58 input_files_physics = list(reduced_file_to_iov_physics.keys())
59 basf2.B2INFO(f
"Total number of files actually used as input = {len(input_files_physics)}")
63 requested_iov = kwargs.get(
"requested_iov",
None)
65 from caf.utils
import IoV
67 output_iov = IoV(requested_iov.exp_low, requested_iov.run_low, -1, -1)
72 from ROOT.Belle2
import BeamSpotAlgorithm
73 from basf2
import create_path, register_module
74 import modularAnalysis
as ana
79 from caf.framework
import Calibration
80 from caf.strategies
import SingleIOV
83 rec_path_1 = create_path()
85 muSelection =
'[p>1.0]'
86 muSelection +=
' and abs(dz)<2.0 and abs(dr)<0.5'
87 muSelection +=
' and nPXDHits >=1 and nSVDHits >= 8 and nCDCHits >= 20'
88 ana.fillParticleList(
'mu+:BS', muSelection, path=rec_path_1)
89 ana.reconstructDecay(
'Upsilon(4S):BS -> mu+:BS mu-:BS',
'9.5<M<11.5', path=rec_path_1)
91 collector_bs = register_module(
'BeamSpotCollector', Y4SPListName=
'Upsilon(4S):BS')
92 algorithm_bs = BeamSpotAlgorithm()
93 algorithm_bs.setOuterLoss(kwargs[
'expert_config'][
'outerLoss'])
94 algorithm_bs.setInnerLoss(kwargs[
'expert_config'][
'innerLoss'])
97 collector=collector_bs,
98 algorithms=algorithm_bs,
99 input_files=input_files_physics,
100 pre_collector_path=rec_path_1)
102 calibration_bs.strategies = SingleIOV
106 for algorithm
in calibration_bs.algorithms:
107 algorithm.params = {
"iov_coverage": output_iov}
113 return [calibration_bs]