Belle II Software  release-05-01-25
toCDCCKFTrackingValidationBkg.py
1 #!/usr/bin/env python3
2 # -*- coding: utf-8 -*-
3 
4 """
5 <header>
6  <contact>software-tracking@belle2.org</contact>
7  <input>EvtGenSim.root</input>
8  <output>toCDCCKFTrackingValidationBkg.root</output>
9  <description>This module validates ToCDCCKF module using Y(4S) runs with background.</description>
10 </header>
11 """
12 
13 from tracking.validation.run import TrackingValidationRun
14 import tracking
15 import logging
16 import basf2
17 import svd
18 
19 VALIDATION_OUTPUT_FILE = 'toCDCCKFTrackingValidationBkg.root'
20 N_EVENTS = 1000
21 ACTIVE = True
22 
23 basf2.set_random_seed(1337)
24 
25 
26 class toCDCCKFValidationBkg(TrackingValidationRun):
27  """Validate the ToCDCCKF Kalman finder/filter algorithm with Y(4S) events"""
28 
29  n_events = N_EVENTS
30 
31  generator_module = 'generic'
32 
33  root_input_file = '../EvtGenSim.root'
34 
35  @staticmethod
36  def finder_module(path):
37  """Add the ToCDCCKF module and related modules to the basf2 path"""
38 
39  path.add_module('SetupGenfitExtrapolation',
40  energyLossBrems=False, noiseBrems=False)
41 
42  svd.add_svd_reconstruction(path)
43  tracking.add_vxd_track_finding_vxdtf2(path, reco_tracks="RecoTracksSVD", components=["SVD"])
44  path.add_module("DAFRecoFitter", recoTracksStoreArrayName="RecoTracksSVD")
45 
46  path.add_module("TFCDC_WireHitPreparer",
47  wirePosition="aligned",
48  useSecondHits=False,
49  flightTimeEstimation="outwards")
50  path.add_module("ToCDCCKF",
51  inputWireHits="CDCWireHitVector",
52  inputRecoTrackStoreArrayName="RecoTracksSVD",
53  relatedRecoTrackStoreArrayName="CKFCDCRecoTracks",
54  relationCheckForDirection="backward",
55  outputRecoTrackStoreArrayName="CKFCDCRecoTracks",
56  outputRelationRecoTrackStoreArrayName="RecoTracksSVD",
57  writeOutDirection="backward",
58  stateBasicFilterParameters={"maximalHitDistance": 0.75},
59  stateExtrapolationFilterParameters={"direction": "forward"},
60  pathFilter="arc_length")
61 
62  path.add_module("RelatedTracksCombiner",
63  CDCRecoTracksStoreArrayName="CKFCDCRecoTracks",
64  VXDRecoTracksStoreArrayName="RecoTracksSVD",
65  recoTracksStoreArrayName="RecoTracks")
66 
67  path.add_module("DAFRecoFitter", recoTracksStoreArrayName="RecoTracks")
68 
69  path.add_module('TrackCreator', recoTrackColName='RecoTracks')
70 
71 
72  tracking_coverage = {
73  'UsePXDHits': False,
74  'UseSVDHits': True,
75  'UseCDCHits': True,
76  'WhichParticles': [],
77  }
78 
79 
80  fit_tracks = False
81 
82  use_fit_information = True
83 
84  pulls = True
85 
86  resolution = True
87 
88  output_file_name = VALIDATION_OUTPUT_FILE
89 
90 
91 def main():
92  validation_run = toCDCCKFValidationBkg()
93  validation_run.configure_and_execute_from_commandline()
94 
95 
96 if __name__ == '__main__':
97  logging.basicConfig(level=logging.INFO)
98  if ACTIVE:
99  main()
svd.add_svd_reconstruction
def add_svd_reconstruction(path, isROIsimulation=False, useNN=False, useCoG=True, applyMasking=False)
Definition: __init__.py:9
tracking.validation.run
Definition: run.py:1
main
int main(int argc, char **argv)
Run all tests.
Definition: test_main.cc:77
tracking.run.tracked_event_generation.ReadOrGenerateTrackedEventsRun.finder_module
finder_module
Name of the finder module to be used - can be everything that is accepted by tracking....
Definition: tracked_event_generation.py:25
toCDCCKFTrackingValidationBkg.toCDCCKFValidationBkg
Definition: toCDCCKFTrackingValidationBkg.py:26
tracking.validation.run.TrackingValidationRun
Definition: run.py:18