ATLAS Offline Software
Loading...
Searching...
No Matches
python.CaloThinCellsByClusterAlg_test Namespace Reference

Classes

class  CheckThinningAlg
class  CreateDataAlg

Functions

 make_calo_cells (mgr)
 make_clusters (mgr, ccc, hashes, ctx)
 testCfg (flags)

Variables

 cell_hashes = set()
 flags = initConfigFlags()
 Files
 TimeStamps
 GeometryLAr
 GeometryTile
 acc = MainServicesCfg(flags)

Function Documentation

◆ make_calo_cells()

python.CaloThinCellsByClusterAlg_test.make_calo_cells ( mgr)

Definition at line 20 of file CaloThinCellsByClusterAlg_test.py.

20def make_calo_cells (mgr):
21 ccc = ROOT.CaloCellContainer()
22 for i in range (mgr.element_size()):
23 elt = mgr.get_element (ROOT.IdentifierHash (i))
24 if not elt: break
25 cc = ROOT.CaloCell (elt, 0, 0, 0, 0)
26 ccc.push_back (cc)
27 ROOT.SetOwnership (cc, False)
28 ccc.order()
29 ccc.updateCaloIterators()
30 return ccc
31
32

◆ make_clusters()

python.CaloThinCellsByClusterAlg_test.make_clusters ( mgr,
ccc,
hashes,
ctx )

Definition at line 33 of file CaloThinCellsByClusterAlg_test.py.

33def make_clusters (mgr, ccc, hashes, ctx):
34 clc = ROOT.xAOD.CaloClusterContainer()
35 clc_store = ROOT.xAOD.CaloClusterAuxContainer()
36 clc.setStore (clc_store)
37 cellLinks = ROOT.CaloClusterCellLinkContainer()
38 ids = ROOT.vector(ROOT.IdentifierHash)()
39
40 for i in range(2):
41 cl = ROOT.xAOD.CaloCluster()
42 clc.push_back (cl)
43 ROOT.SetOwnership (cl, False)
44 eta = 0.5 - i # 0.5 or -0.5
45 phi = 1
46 cl.setEta (eta)
47 cl.setPhi (phi)
48 cl.setClusterSize (ROOT.xAOD.CaloCluster.SW_37ele)
49 links = ROOT.CaloClusterCellLink (ccc)
50 cl.addCellLink (links)
51 ROOT.SetOwnership (links, False)
52
53 mgr.cellsInZone (eta - 0.05, eta + 0.05, phi - 0.05, phi + 0.05, ids)
54 for hash in ids:
55 elt = mgr.get_element (hash)
56 s = elt.getSampling()
57 if s == 0 or s == 2:
58 idx = ccc.findIndex (hash)
59 if idx < 0:
60 print ("??? Can't find cell with hash ", hash)
61 else:
62 hashes.add (hash.value())
63 cl.addCell (idx, 1)
64 cl.setLink(cellLinks, ctx)
65
66 return (clc, clc_store, cellLinks)
67
68

◆ testCfg()

python.CaloThinCellsByClusterAlg_test.testCfg ( flags)

Definition at line 113 of file CaloThinCellsByClusterAlg_test.py.

113def testCfg (flags):
114 result = ComponentAccumulator()
115
116 from LArGeoAlgsNV.LArGMConfig import LArGMCfg
117 from TileGeoModel.TileGMConfig import TileGMCfg
118 result.merge(LArGMCfg(flags))
119 result.merge(TileGMCfg(flags))
120
121 from LArCabling.LArCablingConfig import LArOnOffIdMappingCfg
122 result.merge(LArOnOffIdMappingCfg(flags))
123
124 result.addEventAlgo (CreateDataAlg ('CreateDataAlg'))
125
126 CaloThinCellsByClusterAlg=CompFactory.CaloThinCellsByClusterAlg
127 result.addEventAlgo (CaloThinCellsByClusterAlg ('thinAlg',
128 StreamName = 'StreamAOD',
129 Clusters = 'Clusters',
130 SamplingCellsName = ['EMB3']))
131
132 result.addEventAlgo (CheckThinningAlg ('CheckThinningAlg'))
133 return result
134
135
136# Work around issue with cling in root 6.20.06 getting confused
137# by forward declarations.

Variable Documentation

◆ acc

python.CaloThinCellsByClusterAlg_test.acc = MainServicesCfg(flags)

Definition at line 153 of file CaloThinCellsByClusterAlg_test.py.

◆ cell_hashes

python.CaloThinCellsByClusterAlg_test.cell_hashes = set()

Definition at line 17 of file CaloThinCellsByClusterAlg_test.py.

◆ Files

python.CaloThinCellsByClusterAlg_test.Files

Definition at line 144 of file CaloThinCellsByClusterAlg_test.py.

◆ flags

python.CaloThinCellsByClusterAlg_test.flags = initConfigFlags()

Definition at line 143 of file CaloThinCellsByClusterAlg_test.py.

◆ GeometryLAr

python.CaloThinCellsByClusterAlg_test.GeometryLAr

Definition at line 146 of file CaloThinCellsByClusterAlg_test.py.

◆ GeometryTile

python.CaloThinCellsByClusterAlg_test.GeometryTile

Definition at line 147 of file CaloThinCellsByClusterAlg_test.py.

◆ TimeStamps

python.CaloThinCellsByClusterAlg_test.TimeStamps

Definition at line 145 of file CaloThinCellsByClusterAlg_test.py.