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
trg
cdc
examples
dqmNeuro_generateSWinput.py
Generated on Fri Nov 5 2021 04:05:08 for Belle II Software by
1.8.17