Belle II Software  release-05-01-25
dqmNeuro_generateSWinput.py
1 from basf2 import *
2 from ROOT import Belle2
3 import glob
4 from reconstruction import add_reconstruction
5 
6 from simulation import add_simulation
7 import os
8 
9 set_random_seed(1)
10 bkgdir = "/remote/neurobelle/data/bckg/OfficialBKG/15thCampaign/phase3/set0/"
11 thrange = [10, 170]
12 particlegun_params = {
13  'pdgCodes': [-13, 13], # muons
14  'nTracks': 1, # single tracks
15  'momentumGeneration': 'inversePt', # uniform in the track curvature
16  'momentumParams': [0.3, 10.], # 0.3: minimum pt for standard track finder
17  'thetaGeneration': 'uniformCos', # uniform in solid angle
18  'thetaParams': thrange, # hit SL 6 from z in [-50, 50]
19  'phiGeneration': 'uniform', # uniform in solid angle
20  'phiParams': [0, 360], # full phi
21  'vertexGeneration': 'uniform', # uniform vertex distribution
22  'xVertexParams': [0, 0.0], # vertex on z-axis
23  'yVertexParams': [0, 0.0], # vertex on z-axis
24  'zVertexParams': [0.0, 0.0]} # target range for training
25 
26 main = create_path()
27 main.add_module('EventInfoSetter', evtNumList=5000) # , expList=[7], runList=[3525])
28 main.add_module('Gearbox')
29 main.add_module('Geometry')
30 main.add_module('BeamBkgMixer',
31  backgroundFiles=glob.glob(os.path.join(bkgdir, '*usual*.root')),
32  overallScaleFactor=1,
33  components=['CDC'])
34 particlegun = register_module('ParticleGun')
35 particlegun.param(particlegun_params)
36 main.add_module(particlegun)
37 
38 add_simulation(main)
39 # main.add_module('FullSim')
40 main.add_module('CDCDigitizer')
41 main.add_module('CDCTriggerTSF',
42  InnerTSLUTFile=Belle2.FileSystem.findFile("trg/cdc/data/innerLUT_v2.2.coe"),
43  OuterTSLUTFile=Belle2.FileSystem.findFile("trg/cdc/data/outerLUT_v2.2.coe"),
44  TSHitCollectionName='CDCTriggerNNInputSegmentHits')
45 main.add_module('CDCTriggerTSF',
46  InnerTSLUTFile=Belle2.FileSystem.findFile("trg/cdc/data/innerLUT_v2.2.coe"),
47  OuterTSLUTFile=Belle2.FileSystem.findFile("trg/cdc/data/outerLUT_v2.2.coe"),
48  TSHitCollectionName='CDCTriggerSegmentHits')
49 main.add_module('CDCTrigger2DFinder',
50  minHits=4,
51  minHitsShort=4,
52  minPt=0.3,
53  hitCollectionName='CDCTriggerSegmentHits',
54  outputCollectionName='CDCTrigger2DFinderTracks')
55 main.add_module('CDCTrigger2DFinder',
56  minHits=4,
57  minHitsShort=4,
58  minPt=0.3,
59  hitCollectionName='CDCTriggerNNInputSegmentHits',
60  outputCollectionName='CDCTriggerNNInput2DFinderTracks')
61 main.add_module('CDCTriggerETF',
62  hitCollectionName='CDCTriggerNNInputSegmentHits')
63 main.add_module('CDCTriggerNeuro',
64  hitCollectionName='CDCTriggerNNInputSegmentHits',
65  outputCollectionName='CDCTriggerNeuroTracks',
66  inputCollectionName='CDCTriggerNNInput2DFinderTracks',
67  fixedPoint=True,
68  writeMLPinput=True)
69 add_reconstruction(main)
70 
71 
72 main.add_module('RootOutput', outputFileName="phase3bckg-0-reco_sim.root")
73 main.add_module('Progress')
74 process(main)
75 print(statistics)
Belle2::FileSystem::findFile
static std::string findFile(const std::string &path, bool silent=false)
Search for given file or directory in local or central release directory, and return absolute path if...
Definition: FileSystem.cc:147