ATLAS Offline Software
TrkDetDescrTPCnvTest.cxx
Go to the documentation of this file.
1 /*
2  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
3 */
4 
6 // TrkDetDescrTPCnvTest.cxx, (c) ATLAS Detector software
8 
9 // Trk includes
10 #include "TrkDetDescrUnitTests/TrkDetDescrTPCnvTest.h"
11 // general Trk & TrkGeometry includes
12 #include "TrkDetDescrUtils/BinUtility.h"
13 #include "TrkGeometry/MaterialStepCollection.h"
14 #include "TrkGeometry/LayerMaterialMap.h"
15 #include "TrkGeometry/MaterialProperties.h"
16 #include "TrkGeometry/BinnedLayerMaterial.h"
17 #include "TrkGeometry/CompressedLayerMaterial.h"
18 #include "TrkGeometry/CompoundLayerMaterial.h"
19 #include "TrkGeometry/HomogeneousLayerMaterial.h"
20 #include "TrkGeometry/ElementTable.h"
21 #include "TrkGeometry/Material.h"
22 #include <climits>
23 #include <memory>
24 
25 Trk::TrkDetDescrTPCnvTest::TrkDetDescrTPCnvTest(const std::string& name, ISvcLocator* pSvcLocator) :
26  Trk::TrkDetDescrUnitTestBase(name, pSvcLocator),
27  m_writeMode(true),
28  m_materialStepCollectionName("RandomMaterialSteps"),
29  m_layerMaterialCollectionName("RandomLayerMaterialMap"),
30  m_elementTableName("RandomElementTable")
31 {
32 
33  declareProperty("WriteMode", m_writeMode);
34  // collection names
35  declareProperty("MaterialStepCollection", m_materialStepCollectionName);
36  declareProperty("LayerMaterialMap", m_layerMaterialCollectionName);
37  declareProperty("ElementTable", m_elementTableName);
38 }
39 
40 StatusCode Trk::TrkDetDescrTPCnvTest::runTest()
41 {
42  ATH_MSG_VERBOSE("Running the TrkDetDescrTPCnvTest Test");
43 
44  // ----------------- WRITE MODE -----------------
45  if (m_writeMode){
46 
47  ATH_MSG_VERBOSE("Test is running in write mode - writing out a file.");
48 
49  // The Material STEP collection
50  Trk::MaterialStepCollection* msCollection = new Trk::MaterialStepCollection;
51  for (size_t it = 0; it < TrkDetDescrUnitTestBase::m_numTests; ++it ){
52  msCollection->push_back(new Trk::MaterialStep(TrkDetDescrUnitTestBase::m_flatDist->shoot(),
53  TrkDetDescrUnitTestBase::m_flatDist->shoot(),
54  TrkDetDescrUnitTestBase::m_flatDist->shoot(),
55  TrkDetDescrUnitTestBase::m_flatDist->shoot(),
56  TrkDetDescrUnitTestBase::m_flatDist->shoot(),
57  TrkDetDescrUnitTestBase::m_flatDist->shoot(),
58  TrkDetDescrUnitTestBase::m_flatDist->shoot(),
59  TrkDetDescrUnitTestBase::m_flatDist->shoot(),
60  TrkDetDescrUnitTestBase::m_flatDist->shoot()));
61  }
62 
63 
64  ATH_MSG_VERBOSE("Recording MaterialStepCollection '" << m_materialStepCollectionName << "'.");
65 
66  if (!m_materialStepCollectionName.empty() && (evtStore()->record(msCollection, m_materialStepCollectionName)).isFailure() ) {
67  ATH_MSG_WARNING("Could not record the material collection " << m_materialStepCollectionName);
68  return StatusCode::FAILURE;
69  }
70 
71  ATH_MSG_VERBOSE("Recording LayerMaterialMap '" << m_layerMaterialCollectionName << "'.");
72 
73  // The LayerMaterialMap
74  std::unique_ptr<Trk::LayerMaterialMap> lmMap(new Trk::LayerMaterialMap);
75 
76  // (e) element Table
77  Trk::ElementTable* eTable = new Trk::ElementTable();
78  eTable->addElement(Trk::Material(TrkDetDescrUnitTestBase::m_flatDist->shoot(),
79  TrkDetDescrUnitTestBase::m_flatDist->shoot(),
80  TrkDetDescrUnitTestBase::m_flatDist->shoot(),
81  1,
82  TrkDetDescrUnitTestBase::m_flatDist->shoot()),"TestMaterial1");
83  eTable->addElement(Trk::Material(TrkDetDescrUnitTestBase::m_flatDist->shoot(),
84  TrkDetDescrUnitTestBase::m_flatDist->shoot(),
85  TrkDetDescrUnitTestBase::m_flatDist->shoot(),
86  2,
87  TrkDetDescrUnitTestBase::m_flatDist->shoot()),"TestMaterial2");
88  eTable->addElement(Trk::Material(TrkDetDescrUnitTestBase::m_flatDist->shoot(),
89  TrkDetDescrUnitTestBase::m_flatDist->shoot(),
90  TrkDetDescrUnitTestBase::m_flatDist->shoot(),
91  3,
92  TrkDetDescrUnitTestBase::m_flatDist->shoot()),"TestMaterial3");
93 
94  if ( !m_elementTableName.empty() && (evtStore()->record(eTable, m_elementTableName)).isFailure() ) {
95  ATH_MSG_WARNING("Could not record the element table " << m_elementTableName);
96  return StatusCode::FAILURE;
97  }
98 
99  // (a) homogeneous material
100  Trk::HomogeneousLayerMaterial* hml = new Trk::HomogeneousLayerMaterial(Trk::MaterialProperties(TrkDetDescrUnitTestBase::m_flatDist->shoot(),
101  TrkDetDescrUnitTestBase::m_flatDist->shoot(),
102  TrkDetDescrUnitTestBase::m_flatDist->shoot(),
103  TrkDetDescrUnitTestBase::m_flatDist->shoot(),
104  TrkDetDescrUnitTestBase::m_flatDist->shoot(),
105  TrkDetDescrUnitTestBase::m_flatDist->shoot()),
106  TrkDetDescrUnitTestBase::m_flatDist->shoot());
107  (*lmMap)[Trk::LayerIndex(1)] = hml;
108 
109  // (b) binned material
110  Trk::BinUtility rPhiZUtility(200,-M_PI, M_PI, Trk::closed, Trk::binRPhi);
111  rPhiZUtility += Trk::BinUtility(300,-250.,250., Trk::open, Trk::binZ);
112  std::vector< std::vector< const Trk::MaterialProperties* > > materialMatrix(300, std::vector<const Trk::MaterialProperties*>(200, (Trk::MaterialProperties*)nullptr) );
113  for (size_t iob = 0; iob < 300; ++iob)
114  for (size_t iib = 0; iib < 200; ++iib)
115  materialMatrix[iob][iib] = new Trk::MaterialProperties(TrkDetDescrUnitTestBase::m_flatDist->shoot(),
116  TrkDetDescrUnitTestBase::m_flatDist->shoot(),
117  TrkDetDescrUnitTestBase::m_flatDist->shoot(),
118  TrkDetDescrUnitTestBase::m_flatDist->shoot(),
119  TrkDetDescrUnitTestBase::m_flatDist->shoot(),
120  TrkDetDescrUnitTestBase::m_flatDist->shoot());
121  (*lmMap)[Trk::LayerIndex(2)] = new Trk::BinnedLayerMaterial(rPhiZUtility, materialMatrix, TrkDetDescrUnitTestBase::m_flatDist->shoot()) ;
122 
123  // (c) compressed material
124  Trk::BinUtility zUtility(200,-100,100, Trk::open, Trk::binZ);
125 
126  std::vector< const Trk::MaterialProperties*> materialVector;
127  std::vector< unsigned short int > materialIndices;
128  materialVector.reserve(200);
129  materialIndices.reserve(200);
130  for (size_t iib = 0; iib < 200; ++iib){
131  materialIndices.push_back(int(TrkDetDescrUnitTestBase::m_flatDist->shoot()*200));
132  materialVector.push_back(new Trk::MaterialProperties(TrkDetDescrUnitTestBase::m_flatDist->shoot(),
133  TrkDetDescrUnitTestBase::m_flatDist->shoot(),
134  TrkDetDescrUnitTestBase::m_flatDist->shoot(),
135  TrkDetDescrUnitTestBase::m_flatDist->shoot(),
136  TrkDetDescrUnitTestBase::m_flatDist->shoot(),
137  TrkDetDescrUnitTestBase::m_flatDist->shoot()));
138 
139  }
140  (*lmMap)[Trk::LayerIndex(3)] = new Trk::CompressedLayerMaterial(zUtility, materialVector, materialIndices);
141 
142  // (d) compound material
143  Trk::BinUtility zUtilityC(200,-100,100, Trk::open, Trk::binZ);
144 
145 
146  Trk::ValueMatrix thicknessMatrix;
147  Trk::ValueMatrix x0Matrix;
148  Trk::ValueMatrix l0Matrix;
149  Trk::ValueMatrix aMatrix;
150  Trk::ValueMatrix zMatrix;
151  Trk::ValueMatrix rhoMatrix;
152  thicknessMatrix.reserve(25);
153  x0Matrix.reserve(25);
154  l0Matrix.reserve(25);
155  aMatrix.reserve(25);
156  zMatrix.reserve(25);
157  rhoMatrix.reserve(25);
158 
159  std::vector< std::vector< Trk::MaterialComposition > > materialCompositionMatrix;
160 
161  for (size_t iob = 0; iob < 200; ++iob){
162  // the vectors
163  std::vector<unsigned char> thicknessVector; thicknessVector.reserve(25);
164  std::vector<unsigned char> x0Vector; x0Vector.reserve(25);
165  std::vector<unsigned char> l0Vector; l0Vector.reserve(25);
166  std::vector<unsigned char> aVector; aVector.reserve(25);
167  std::vector<unsigned char> zVector; zVector.reserve(25);
168  std::vector<unsigned char> rhoVector; rhoVector.reserve(25);
169  std::vector< Trk::MaterialComposition > materialCompositionVector;
170  materialCompositionVector.reserve(25);
171  for (size_t iib = 0; iib < 25; ++iib){
172  x0Vector.push_back((unsigned char)(TrkDetDescrUnitTestBase::m_flatDist->shoot()*UCHAR_MAX));
173  l0Vector.push_back((unsigned char)(TrkDetDescrUnitTestBase::m_flatDist->shoot()*UCHAR_MAX));
174  aVector.push_back((unsigned char)(TrkDetDescrUnitTestBase::m_flatDist->shoot()*UCHAR_MAX));
175  zVector.push_back((unsigned char)(TrkDetDescrUnitTestBase::m_flatDist->shoot()*UCHAR_MAX));
176  rhoVector.push_back((unsigned char)(TrkDetDescrUnitTestBase::m_flatDist->shoot()*UCHAR_MAX));
177  std::vector< Trk::ElementFraction > elements;
178  elements.reserve(5);
179  for (size_t iic = 0; iic < 5; ++iic){
180  elements.emplace_back((unsigned char)(TrkDetDescrUnitTestBase::m_flatDist->shoot()*UCHAR_MAX),
181  (unsigned char)(TrkDetDescrUnitTestBase::m_flatDist->shoot()*UCHAR_MAX));
182  }
183  materialCompositionVector.emplace_back(elements);
184  }
185  thicknessMatrix.push_back(thicknessVector);
186  x0Matrix.push_back(x0Vector);
187  l0Matrix.push_back(l0Vector);
188  aMatrix.push_back(aVector);
189  zMatrix.push_back(zVector);
190  rhoMatrix.push_back(rhoVector);
191  materialCompositionMatrix.push_back(materialCompositionVector);
192  }
193 
194  Trk::ValueStore thicknessBins;
195  thicknessBins.valueMin = 0.;
196  thicknessBins.valueStep = TrkDetDescrUnitTestBase::m_flatDist->shoot();
197  thicknessBins.valueBinMatrix = thicknessMatrix;
198 
199  Trk::ValueStore x0Bins;
200  x0Bins.valueMin = 0.;
201  x0Bins.valueStep = TrkDetDescrUnitTestBase::m_flatDist->shoot();
202  x0Bins.valueBinMatrix = x0Matrix;
203 
204 
205  Trk::ValueStore l0Bins;
206  l0Bins.valueMin = 0.;
207  l0Bins.valueStep = TrkDetDescrUnitTestBase::m_flatDist->shoot();
208  l0Bins.valueBinMatrix = l0Matrix;
209 
210  Trk::ValueStore aBins;
211  aBins.valueMin = 0.;
212  aBins.valueStep = TrkDetDescrUnitTestBase::m_flatDist->shoot();
213  aBins.valueBinMatrix = aMatrix;
214 
215  Trk::ValueStore zBins;
216  zBins.valueMin = 0.;
217  zBins.valueStep = TrkDetDescrUnitTestBase::m_flatDist->shoot();
218  zBins.valueBinMatrix = zMatrix;
219 
220  Trk::ValueStore rhoBins;
221  rhoBins.valueMin = 0.;
222  rhoBins.valueStep = TrkDetDescrUnitTestBase::m_flatDist->shoot();
223  rhoBins.valueBinMatrix = rhoMatrix;
224 
225  (*lmMap)[Trk::LayerIndex(3)] = new Trk::CompoundLayerMaterial(zUtilityC,
226  thicknessBins,
227  x0Bins,
228  l0Bins,
229  aBins,
230  zBins,
231  rhoBins,
232  materialCompositionMatrix);
233 
234  auto sharedElementTable = std::make_shared<Trk::ElementTable>(*eTable);
235  lmMap->updateElementTable(sharedElementTable);
236 
237  if ( (evtStore()->record(lmMap.release(), m_layerMaterialCollectionName)).isFailure() ) {
238  ATH_MSG_WARNING("Could not record the material collection " << m_layerMaterialCollectionName);
239  return StatusCode::FAILURE;
240  }
241  } else {
242  //
243  ATH_MSG_VERBOSE("Test is running in READ mode, reading in a file.");
244 
245  // ----------------- READ MODE -----------------
246  const Trk::LayerMaterialMap* lmMap = nullptr;
247  if ( (evtStore()->retrieve(lmMap, m_layerMaterialCollectionName)).isFailure() ) {
248  ATH_MSG_WARNING("Could not read the material collection " << m_layerMaterialCollectionName);
249  return StatusCode::FAILURE;
250  } else
251  ATH_MSG_INFO("Successfully read a LayerMaterialMap with size " << lmMap->size());
252 
253  }
254 
255  return StatusCode::SUCCESS;
256 }
python.PyKernel.retrieve
def retrieve(aClass, aKey=None)
Definition: PyKernel.py:110
ElementTable.h
Muon::MuonStationIndex::LayerIndex
LayerIndex
enum to classify the different layers in the muon spectrometer
Definition: MuonStationIndex.h:38
Trk::TrkDetDescrUnitTestBase::m_flatDist
Rndm::Numbers * m_flatDist
Definition: TrkDetDescrUnitTestBase.h:58
Trk::TrkDetDescrUnitTestBase::m_numTests
size_t m_numTests
number of tests
Definition: TrkDetDescrUnitTestBase.h:61
Trk::ValueStore::valueStep
float valueStep
Definition: CompoundLayerMaterial.h:34
ATH_MSG_INFO
#define ATH_MSG_INFO(x)
Definition: AthMsgStreamMacros.h:31
MaterialProperties.h
TrkDetDescrTPCnvTest.h
Trk::binZ
@ binZ
Definition: BinningType.h:49
Trk::TrkDetDescrUnitTestBase
Definition: TrkDetDescrUnitTestBase.h:27
BinUtility.h
skel.it
it
Definition: skel.GENtoEVGEN.py:407
M_PI
#define M_PI
Definition: ActiveFraction.h:11
MaterialStepCollection.h
Trk::TrkDetDescrTPCnvTest::m_layerMaterialCollectionName
std::string m_layerMaterialCollectionName
Definition: TrkDetDescrTPCnvTest.h:40
Trk::closed
@ closed
Definition: BinningType.h:41
HomogeneousLayerMaterial.h
Trk::MaterialStepCollection
DataVector< Trk::MaterialStep > MaterialStepCollection
Definition: MaterialStepCollection.h:17
Trk::CompoundLayerMaterial
Definition: CompoundLayerMaterial.h:60
Trk::TrkDetDescrTPCnvTest::m_elementTableName
std::string m_elementTableName
Definition: TrkDetDescrTPCnvTest.h:41
ATH_MSG_VERBOSE
#define ATH_MSG_VERBOSE(x)
Definition: AthMsgStreamMacros.h:28
Trk::LayerMaterialMap
Definition: LayerMaterialMap.h:32
CompoundLayerMaterial.h
Trk::CompressedLayerMaterial
Definition: CompressedLayerMaterial.h:31
Trk::MaterialStep
Definition: MaterialStep.h:34
Trk::ValueStore
Definition: CompoundLayerMaterial.h:32
Trk::ValueStore::valueBinMatrix
ValueMatrix valueBinMatrix
Definition: CompoundLayerMaterial.h:35
EL::StatusCode
::StatusCode StatusCode
StatusCode definition for legacy code.
Definition: PhysicsAnalysis/D3PDTools/EventLoop/EventLoop/StatusCode.h:22
Trk::LayerMaterialMap::updateElementTable
void updateElementTable(const std::shared_ptr< const ElementTable > &eTable)
update method
Definition: LayerMaterialMap.h:66
AthCommonDataStore< AthCommonMsg< Algorithm > >::declareProperty
Gaudi::Details::PropertyBase & declareProperty(Gaudi::Property< T, V, H > &t)
Definition: AthCommonDataStore.h:145
BinnedLayerMaterial.h
Trk::BinnedLayerMaterial
Definition: BinnedLayerMaterial.h:33
DataVector
Derived DataVector<T>.
Definition: DataVector.h:794
Trk::TrkDetDescrTPCnvTest::TrkDetDescrTPCnvTest
TrkDetDescrTPCnvTest(const std::string &name, ISvcLocator *pSvcLocator)
Standard Athena-Algorithm Constructor.
Definition: TrkDetDescrTPCnvTest.cxx:25
MakeTH3DFromTH2Ds.zBins
list zBins
Definition: MakeTH3DFromTH2Ds.py:86
Trk::BinUtility
Definition: BinUtility.h:39
Trk::TrkDetDescrTPCnvTest::m_writeMode
bool m_writeMode
Definition: TrkDetDescrTPCnvTest.h:37
Trk
Ensure that the ATLAS eigen extensions are properly loaded.
Definition: FakeTrackBuilder.h:9
name
std::string name
Definition: Control/AthContainers/Root/debug.cxx:240
DataVector::push_back
value_type push_back(value_type pElem)
Add an element to the end of the collection.
LayerMaterialMap.h
Trk::TrkDetDescrTPCnvTest::m_materialStepCollectionName
std::string m_materialStepCollectionName
Definition: TrkDetDescrTPCnvTest.h:39
Trk::open
@ open
Definition: BinningType.h:40
Trk::MaterialProperties
Definition: MaterialProperties.h:40
Trk::ElementTable
Definition: ElementTable.h:24
Trk::ValueMatrix
std::vector< ValueVector > ValueMatrix
Definition: CompoundLayerMaterial.h:30
ATH_MSG_WARNING
#define ATH_MSG_WARNING(x)
Definition: AthMsgStreamMacros.h:32
Trk::binRPhi
@ binRPhi
Definition: BinningType.h:52
Trk::ValueStore::valueMin
float valueMin
Definition: CompoundLayerMaterial.h:33
Trk::Material
Definition: Material.h:117
Trk::HomogeneousLayerMaterial
Definition: HomogeneousLayerMaterial.h:53
Trk::TrkDetDescrTPCnvTest::runTest
StatusCode runTest()
Definition: TrkDetDescrTPCnvTest.cxx:40
Material.h
Trk::ElementTable::addElement
void addElement(const Material &mat, const std::string &mname="")
Add material to the Table - if the elment is already filled ignore.
Definition: ElementTable.h:75
CompressedLayerMaterial.h
Generated on Tue Sep 2 2025 21:24:56 for ATLAS Offline Software by   doxygen 1.8.18