Belle II Software development
caf_ecl_cluster_energy.py
1
8
9"""ECL cluster crystal energy calibration using single photon MCrd."""
10
11from prompt import CalibrationSettings, INPUT_DATA_FILTERS
12
13# --------------------------------------------------------------
14# ..Tell the automated script some required details
15settings = CalibrationSettings(
16 name="ecl_cluster_energy",
17 expert_username="hearty",
18 subsystem="ecl",
19 description=__doc__,
20 input_data_formats=["mdst"],
21 input_data_names=["single_gamma_mc"],
22 input_data_filters={"single_gamma_mc": [INPUT_DATA_FILTERS["Data Tag"]["single_gamma_mc"]]},
23 depends_on=[],
24
25 # ..The expert_config settings should agree with the parameters of the jobs used
26 # to create the single_gamma_mc samples.
27 # Number of energy points, and the values of these for each of the three ECL regions.
28 # The mc samples include a reduced set of showers and CalDigits, with these names.
29 expert_config={"number_energies": 8,
30 "forward_energies": [0.030, 0.050, 0.100, 0.200, 0.483, 1.166, 2.816, 6.800],
31 "barrel_energies": [0.030, 0.050, 0.100, 0.200, 0.458, 1.049, 2.402, 5.500],
32 "backward_energies": [0.030, 0.050, 0.100, 0.200, 0.428, 0.917, 1.962, 4.200],
33 "digitArrayName": "ECLTrimmedDigits",
34 "showerArrayName": "ECLTrimmedShowers",
35 "nGroupPerThetaID": 8,
36 "lowEnergyThreshold": 0.},
37 produced_payloads=["ECLnOptimal", "ECLLeakageCorrections"])
38
39#
40# --------------------------------------------------------------
41# ..The calibration functions
42
43
44def get_calibrations(input_data, **kwargs):
45 import basf2
46 from ROOT import Belle2
47 from caf.utils import IoV
48 from caf.framework import Calibration
49
50 # --------------------------------------------------------------
51 # ..Input data, same for both calibrations
52 file_to_iov = input_data["single_gamma_mc"]
53 input_files_single_gamma = list(file_to_iov.keys())
54
55 # --------------------------------------------------------------
56 # ..nOptimal calibration first
57
58 # ..eclNOptimal collector
59 ecl_nOptimal_collector = basf2.register_module("eclNOptimalCollector")
60 ecl_nOptimal_collector.param("granularity", "all")
61
62 # ..Number of energy points and their values can be set via expert_config
63 expert_config = kwargs.get("expert_config")
64 number_energies = expert_config["number_energies"]
65 ecl_nOptimal_collector.param("numberEnergies", number_energies)
66
67 forward_energies = expert_config["forward_energies"]
68 ecl_nOptimal_collector.param("energiesForward", forward_energies)
69
70 barrel_energies = expert_config["barrel_energies"]
71 ecl_nOptimal_collector.param("energiesBarrel", barrel_energies)
72
73 backward_energies = expert_config["backward_energies"]
74 ecl_nOptimal_collector.param("energiesBackward", backward_energies)
75
76 digitArrayName = expert_config["digitArrayName"]
77 ecl_nOptimal_collector.param("digitArrayName", digitArrayName)
78
79 showerArrayName = expert_config["showerArrayName"]
80 ecl_nOptimal_collector.param("showerArrayName", showerArrayName)
81
82 nGroupPerThetaID = expert_config["nGroupPerThetaID"]
83 ecl_nOptimal_collector.param("nGroupPerThetaID", nGroupPerThetaID)
84
85 # ..eclNOptimal algorithm
86 algo_nOptimal = Belle2.ECL.eclNOptimalAlgorithm()
87
88 # ..The calibration
89 cal_ecl_nOptimal = Calibration(
90 name="ecl_nOptimal",
91 collector=ecl_nOptimal_collector,
92 algorithms=algo_nOptimal,
93 input_files=input_files_single_gamma)
94
95 # ..pre_path is empty
96 ecl_nOptimal_pre_path = basf2.create_path()
97 cal_ecl_nOptimal.pre_collector_path = ecl_nOptimal_pre_path
98
99 # --------------------------------------------------------------
100 # ..Leakage correction, which depends on nOptimal
101
102 # ..eclLeakage collector
103 ecl_leakage_collector = basf2.register_module("eclLeakageCollector")
104 ecl_leakage_collector.param("granularity", "all")
105 ecl_leakage_collector.param("position_bins", 29)
106
107 # ..Number of energy points and their values can be set via expert_config
108 ecl_leakage_collector.param("number_energies", number_energies)
109 ecl_leakage_collector.param("energies_forward", forward_energies)
110 ecl_leakage_collector.param("energies_barrel", barrel_energies)
111 ecl_leakage_collector.param("energies_backward", backward_energies)
112 ecl_leakage_collector.param("showerArrayName", showerArrayName)
113
114 # ..eclLeakage algorithm
115 algo_leakage = Belle2.ECL.eclLeakageAlgorithm()
116 lowEnergyThreshold = expert_config["lowEnergyThreshold"]
117 algo_leakage.setLowEnergyThreshold(lowEnergyThreshold)
118
119 # ..The calibration
120 cal_ecl_leakage = Calibration(
121 name="ecl_leakage",
122 collector=ecl_leakage_collector,
123 algorithms=algo_leakage,
124 input_files=input_files_single_gamma)
125
126 # ..Depends on nOptimal
127 cal_ecl_leakage.depends_on(cal_ecl_nOptimal)
128
129 # ..pre_path is empty
130 ecl_leakage_pre_path = basf2.create_path()
131 cal_ecl_leakage.pre_collector_path = ecl_leakage_pre_path
132
133 # --------------------------------------------------------------
134 # ..Force the output iovs to be open
135 requested_iov = kwargs.get("requested_iov", None)
136 output_iov = IoV(requested_iov.exp_low, requested_iov.run_low, -1, -1)
137 for algorithm in cal_ecl_nOptimal.algorithms:
138 algorithm.params = {"apply_iov": output_iov}
139 for algorithm in cal_ecl_leakage.algorithms:
140 algorithm.params = {"apply_iov": output_iov}
141
142 # --------------------------------------------------------------
143 # ..Return the calibrations
144 return [cal_ecl_nOptimal, cal_ecl_leakage]
Calculate ECL energy leakage corrections.
Algorithm that works with eclNOptimalCollector to find the number of crystals to be summed to get the...