14 #include "Acts/Definitions/Algebra.hpp"
15 #include "Acts/Definitions/Units.hpp"
16 #include "Acts/Geometry/ApproachDescriptor.hpp"
17 #include "Acts/Geometry/GenericApproachDescriptor.hpp"
18 #include "Acts/Geometry/GeometryContext.hpp"
19 #include "Acts/Geometry/CylinderLayer.hpp"
20 #include "Acts/Geometry/LayerCreator.hpp"
21 #include "Acts/Geometry/ProtoLayer.hpp"
22 #include "Acts/Material/ProtoSurfaceMaterial.hpp"
23 #include "Acts/Surfaces/CylinderSurface.hpp"
24 #include "Acts/Surfaces/DiscSurface.hpp"
25 #include "Acts/Utilities/BinningType.hpp"
26 #include "Acts/Surfaces/AnnulusBounds.hpp"
27 #include "Acts/Surfaces/SurfaceArray.hpp"
28 #include "Acts/Visualization/GeometryView3D.hpp"
29 #include "Acts/Visualization/ObjVisualization3D.hpp"
30 #include "Acts/Utilities/AxisDefinitions.hpp"
33 #include <unordered_map>
37 using Acts::Transform3;
38 using Acts::Translation3;
40 using namespace Acts::UnitLiterals;
43 std::unique_ptr<const Acts::Logger>
logger)
45 m_logger(std::move(
logger))
48 throw std::invalid_argument(
"ALB mode is undefined");
52 const Acts::LayerVector
54 ACTS_VERBOSE(
"Building negative layers");
55 Acts::LayerVector nVector;
60 const Acts::LayerVector
62 ACTS_VERBOSE(
"Building central layers");
63 Acts::LayerVector cVector;
68 const Acts::LayerVector
70 ACTS_VERBOSE(
"Building positive layers");
71 Acts::LayerVector pVector;
76 std::vector<std::shared_ptr<const ActsDetectorElement>>
78 ACTS_VERBOSE(
"Retrieving detector elements from detector manager");
80 ACTS_ERROR(
"Manager is null");
81 throw std::runtime_error{
"Detector manager is null"};
84 ACTS_VERBOSE(
"Detector manager has "
86 siDetMng->getDetectorElementEnd())
89 std::vector<std::shared_ptr<const ActsDetectorElement>> elements;
92 for (
iter = siDetMng->getDetectorElementBegin();
93 iter != siDetMng->getDetectorElementEnd(); ++
iter) {
96 if (siDetElement ==
nullptr) {
97 ACTS_ERROR(
"Detector element was nullptr");
98 throw std::runtime_error{
"Corrupt detector element collection"};
101 std::make_shared<const ActsDetectorElement>(*siDetElement));
103 ACTS_VERBOSE(
"Retrieved " << elements.size() <<
" elements");
109 Acts::LayerVector &layersOutput)
const {
110 using enum Acts::AxisDirection;
112 ACTS_VERBOSE(
"Build layers: BARREL");
113 std::vector<std::shared_ptr<const ActsDetectorElement>> elements =
116 std::map<int, std::vector<std::shared_ptr<const Acts::Surface>>>
layers{};
118 for (
const auto &element : elements) {
125 if(
id.layer_disk() >= 2)
continue;
128 if(
id.layer_disk() <= 1)
continue;
134 elementLayer =
id.layer_disk();
136 layers[elementLayer].push_back(element->surface().getSharedPtr());
142 for (
auto &[
key, layerSurfaces] :
layers) {
144 std::stringstream
name;
148 name <<
"obj/" <<
m_cfg.
mode <<
"_brl_" << std::setfill(
'0') << std::setw(2)
152 std::ofstream ofs{
name.str()};
153 Acts::ObjVisualization3D vis{};
154 Acts::ViewConfig vc = Acts::s_viewSensitive;
155 vc.quarterSegments = 200;
156 for (
const auto &surface : layerSurfaces) {
157 Acts::GeometryView3D::drawSurface(vis, *surface, gctx,
158 Acts::Transform3::Identity(), vc);
165 ACTS_VERBOSE(
"Found " <<
layers.size() <<
" barrel layers");
169 for (
const auto &[
key, surfaces] :
layers) {
170 Acts::ProtoLayer pl(gctx, surfaces);
171 ACTS_VERBOSE(
"Layer #" <<
n <<
" with layerKey: (" <<
key <<
")");
172 ACTS_VERBOSE(
" -> at rMin / rMax: " << pl.min(AxisR) <<
" / "
174 ACTS_VERBOSE(
" -> at zMin / zMax: " << pl.min(AxisZ) <<
" / "
181 for (
const auto &[
key, surfaces] :
layers) {
183 std::unique_ptr<Acts::ApproachDescriptor> approachDescriptor =
nullptr;
184 std::shared_ptr<const Acts::ProtoSurfaceMaterial> materialProxy =
nullptr;
187 Acts::ProtoLayer pl(gctx, surfaces);
191 double layerZ = pl.medium(AxisZ,
true);
192 double layerHalfZ = 0.5 * pl.range(AxisZ);
194 Acts::Transform3
transform(Translation3(0., 0., -layerZ));
197 std::shared_ptr<Acts::CylinderSurface> innerBoundary =
198 Acts::Surface::makeShared<Acts::CylinderSurface>(
201 std::shared_ptr<Acts::CylinderSurface> outerBoundary =
202 Acts::Surface::makeShared<Acts::CylinderSurface>(
205 std::shared_ptr<Acts::CylinderSurface> centralSurface =
206 Acts::Surface::makeShared<Acts::CylinderSurface>(
207 transform, (pl.min(AxisR) + pl.max(AxisR)) / 2.,
215 materialBinUtil += Acts::BinUtility(
binsZ, -layerHalfZ, layerHalfZ,
219 std::make_shared<const Acts::ProtoSurfaceMaterial>(materialBinUtil);
221 ACTS_VERBOSE(
"[L] Layer is marked to carry support material on Surface ( "
222 "inner=0 / center=1 / outer=2 ) : "
224 ACTS_VERBOSE(
"with binning: [" << binsPhi <<
", " <<
binsZ <<
"]");
226 ACTS_VERBOSE(
"Created ApproachSurfaces for cylinder layer at:");
227 ACTS_VERBOSE(
" - inner: R=" << pl.min(AxisR));
228 ACTS_VERBOSE(
" - central: R=" << (pl.min(AxisR) + pl.max(AxisR)) /
230 ACTS_VERBOSE(
" - outer: R=" << pl.max(AxisR));
233 innerBoundary->assignSurfaceMaterial(materialProxy);
235 std::vector<std::shared_ptr<const Acts::Surface>> aSurfaces;
236 aSurfaces.push_back(std::move(innerBoundary));
237 aSurfaces.push_back(std::move(centralSurface));
238 aSurfaces.push_back(std::move(outerBoundary));
241 std::make_unique<Acts::GenericApproachDescriptor>(std::move(aSurfaces));
244 auto phiEqual = [
this](
const Acts::Surface &
a,
245 const Acts::Surface &
b) {
246 Acts::GeometryContext gctx;
250 auto zEqual = [
this](
const Acts::Surface &
a,
251 const Acts::Surface &
b) {
252 Acts::GeometryContext gctx;
258 auto& xsurfaces = surfaces;
259 auto countKey = [&xsurfaces](
auto equal) ->
size_t {
260 std::vector<const Acts::Surface *> keySurfaces;
261 for (
const auto &surface : xsurfaces) {
263 for (
const auto* existing : keySurfaces) {
264 if (
equal(*surface, *existing)) {
270 keySurfaces.push_back(surface.get());
273 return keySurfaces.size();
276 size_t nModPhi = countKey(phiEqual);
277 size_t nModZ = countKey(zEqual);
279 ACTS_VERBOSE(
"Found " << nModPhi <<
" modules in phi " << nModZ
282 std::shared_ptr<Acts::Layer>
layer;
288 std::move(approachDescriptor));
303 std::move(approachDescriptor));
307 std::move(approachDescriptor));
310 layersOutput.push_back(
layer);
315 ACTS_DEBUG(
"Configured to build " << numPassiveLayers <<
" passive central layers.");
316 if (numPassiveLayers != 0
u) {
317 for (std::size_t icl = 0; icl < numPassiveLayers; ++icl) {
318 ACTS_VERBOSE(
"- build layer " << icl
324 auto cBounds = std::make_shared<const Acts::CylinderBounds>(
328 std::shared_ptr<Acts::Layer>
layer =
331 layersOutput.push_back(
layer);
337 Acts::LayerVector &layersOutput,
int type)
const
339 using enum Acts::AxisDirection;
341 ACTS_VERBOSE(
"Build layers: " << (
type < 0 ?
"NEGATIVE" :
"POSITIVE")
343 std::vector<std::shared_ptr<const ActsDetectorElement>> elements =
345 std::map<std::tuple<int, int, int>, std::vector<const Acts::Surface *>>
348 for (
const auto &element : elements) {
356 if(
id.layer_disk() >= 3)
continue;
359 if(
id.layer_disk() <= 2)
continue;
364 std::tuple<int, int, int>
key{
id.bec(),
id.layer_disk(),
id.eta_module()};
366 initialLayers[
key].push_back(&element->surface());
369 ACTS_VERBOSE(
"Found " << initialLayers.size() <<
" "
370 << (
type < 0 ?
"NEGATIVE" :
"POSITIVE")
371 <<
" ENDCAP inital layers");
373 std::vector<Acts::ProtoLayer> protoLayers;
374 protoLayers.reserve(initialLayers.size());
376 for (
const auto &[
key, surfaces] : initialLayers) {
377 auto &pl = protoLayers.emplace_back(gctx, surfaces);
383 std::sort(protoLayers.begin(), protoLayers.end(),
384 [
type](
const Acts::ProtoLayer &
a,
const Acts::ProtoLayer &
b) {
385 double midA = (a.min(AxisZ) + a.max(AxisZ)) / 2.0;
386 double midB = (b.min(AxisZ) + b.max(AxisZ)) / 2.0;
394 auto plPrintZ = [](
const auto &pl) -> std::string {
395 std::stringstream
ss;
396 double zMid = (pl.min(AxisZ) + pl.max(AxisZ)) / 2.0;
397 ss <<
" < " << pl.min(AxisZ) <<
" | " << zMid <<
" | "
398 << pl.max(AxisZ) <<
" > ";
402 for (
const auto &pl : protoLayers) {
404 ACTS_VERBOSE(
" -> at < zMin | zMid | zMax >: " << plPrintZ(pl));
406 ACTS_VERBOSE(
" -> at rMin / rMax: " << pl.min(AxisR) <<
" / "
411 std::vector<Acts::ProtoLayer> mergedProtoLayers;
413 if (m_cfg.doEndcapLayerMerging) {
414 mergedProtoLayers.push_back(protoLayers.front());
415 std::vector<const Acts::Surface *> surfaces;
416 for (
size_t i = 1;
i < protoLayers.size();
i++) {
417 auto &pl = protoLayers[
i];
418 auto &pl_prev = mergedProtoLayers.back();
420 ACTS_VERBOSE(
"Compare: " << plPrintZ(pl_prev) <<
" and " << plPrintZ(pl));
421 bool overlap = (pl.min(AxisZ) <= pl_prev.max(AxisZ) &&
422 pl.max(AxisZ) >= pl_prev.min(AxisZ));
423 ACTS_VERBOSE(
" -> overlap? " << (overlap ?
"yes" :
"no"));
425 ACTS_VERBOSE(
" ===> merging");
427 surfaces.reserve(pl.surfaces().size() + pl_prev.surfaces().size());
428 surfaces.insert(surfaces.end(), pl.surfaces().begin(),
429 pl.surfaces().end());
430 surfaces.insert(surfaces.end(), pl_prev.surfaces().begin(),
431 pl_prev.surfaces().end());
432 mergedProtoLayers.pop_back();
434 mergedProtoLayers.emplace_back(gctx, std::move(surfaces));
435 new_pl.envelope[AxisR] = pl.envelope[AxisR];
436 new_pl.envelope[AxisZ] = pl.envelope[AxisZ];
438 mergedProtoLayers.push_back(std::move(pl));
442 ACTS_VERBOSE(
"" << mergedProtoLayers.size() <<
" "
443 << (
type < 0 ?
"NEGATIVE" :
"POSITIVE")
444 <<
" ENDCAP layers remain after merging");
446 mergedProtoLayers = protoLayers;
449 if (m_cfg.objDebugOutput) {
450 for (
size_t i = 0;
i < mergedProtoLayers.size();
i++) {
452 std::stringstream
ss;
453 ss <<
"obj/" << m_cfg.mode <<
"_" << (
type < 0 ?
"neg" :
"pos")
454 <<
"_disk_" << std::setfill(
'0') << std::setw(2) <<
i <<
".obj";
456 std::ofstream ofs{
ss.str()};
457 Acts::ObjVisualization3D vis{};
458 Acts::ViewConfig vc = Acts::s_viewSensitive;
459 vc.quarterSegments = 200;
460 for (
const auto &surface : mergedProtoLayers[
i].surfaces()) {
461 Acts::GeometryView3D::drawSurface(vis, *surface, gctx,
462 Acts::Transform3::Identity(), vc);
469 std::vector<std::shared_ptr<const Surface>> ownedSurfaces;
470 for (
const auto &pl : mergedProtoLayers) {
472 std::unique_ptr<Acts::ApproachDescriptor> approachDescriptor =
nullptr;
473 std::shared_ptr<const Acts::ProtoSurfaceMaterial> materialProxy =
nullptr;
475 double layerZ = pl.medium(AxisZ);
476 double layerHalfZ = 0.5 * pl.range(AxisZ);
477 double layerThickness = pl.range(AxisZ);
479 double layerZInner = layerZ - layerHalfZ;
480 double layerZOuter = layerZ + layerHalfZ;
482 if (std::abs(layerZInner) > std::abs(layerZOuter))
485 std::vector<std::shared_ptr<const Acts::Surface>> aSurfaces;
487 Acts::Transform3 transformNominal(Translation3(0., 0., layerZ));
488 Acts::Transform3 transformInner(Translation3(0., 0., layerZInner));
489 Acts::Transform3 transformOuter(Translation3(0., 0., layerZOuter));
491 std::shared_ptr<Acts::DiscSurface> innerBoundary =
492 Acts::Surface::makeShared<Acts::DiscSurface>(
493 transformInner, pl.min(AxisR), pl.max(AxisR));
494 aSurfaces.push_back(innerBoundary);
496 std::shared_ptr<Acts::DiscSurface> nominalSurface =
497 Acts::Surface::makeShared<Acts::DiscSurface>(
498 transformNominal, pl.min(AxisR), pl.max(AxisR));
499 aSurfaces.push_back(nominalSurface);
501 std::shared_ptr<Acts::DiscSurface> outerBoundary =
502 Acts::Surface::makeShared<Acts::DiscSurface>(
503 transformOuter, pl.min(AxisR), pl.max(AxisR));
504 aSurfaces.push_back(outerBoundary);
506 if(layerThickness > 2_mm) {
507 ACTS_VERBOSE(
"Wide disc layer ("<< layerThickness <<
") => adding cylinder like approach surfaces");
508 Acts::Transform3
trf{Translation3{0, 0, layerZ}};
510 Acts::Surface::makeShared<Acts::CylinderSurface>(
511 trf, pl.min(AxisR), layerHalfZ);
512 aSurfaces.push_back(cylinderInner);
515 Acts::Surface::makeShared<Acts::CylinderSurface>(
516 trf, pl.max(AxisR), layerHalfZ);
517 aSurfaces.push_back(cylinderOuter);
521 size_t matBinsPhi = m_cfg.endcapMaterialBins.first;
522 size_t matBinsR = m_cfg.endcapMaterialBins.second;
527 Acts::BinUtility(matBinsR, pl.min(AxisR), pl.max(AxisR),
531 std::make_shared<const Acts::ProtoSurfaceMaterial>(materialBinUtil);
533 ACTS_VERBOSE(
"[L] Layer is marked to carry support material on Surface ( "
534 "inner=0 / center=1 / outer=2 ) : "
536 ACTS_VERBOSE(
"with binning: [" << matBinsPhi <<
", " << matBinsR <<
"]");
538 ACTS_VERBOSE(
"Created ApproachSurfaces for disc layer at:");
539 ACTS_VERBOSE(
" - inner: Z=" << layerZInner);
540 ACTS_VERBOSE(
" - central: Z=" << layerZ);
541 ACTS_VERBOSE(
" - outer: Z=" << layerZOuter);
544 innerBoundary->assignSurfaceMaterial(materialProxy);
548 bool isITk = m_cfg.mode == Mode::ITkPixelInner ||
549 m_cfg.mode == Mode::ITkPixelOuter ||
550 m_cfg.mode == Mode::ITkStrip;
552 std::map<int, std::set<int>> phiModuleByRing;
555 for (
const auto &srf : pl.surfaces()) {
557 srf->associatedDetectorElement());
561 if(m_cfg.mode == Mode::ITkStrip || !isITk) {
565 ring_number =
id.layer_disk();
567 phiModuleByRing[ring_number].insert(
id.phi_module());
571 for(
const auto& [ring, phiModules] : phiModuleByRing) {
572 nModPhi =
std::min(nModPhi, phiModules.size());
575 size_t nModR = phiModuleByRing.size();
577 ACTS_VERBOSE(
"Identifier reports: " << nModPhi <<
" is lowest for " << nModR
580 size_t nBinsPhi = nModPhi * m_cfg.numberOfBinsFactor;
581 size_t nBinsR = nModR * m_cfg.numberOfBinsFactor;
583 ACTS_VERBOSE(
"Creating r x phi binned layer with " << nBinsR <<
" x "
584 << nBinsPhi <<
" bins");
588 std::make_unique<Acts::GenericApproachDescriptor>(aSurfaces);
591 ownedSurfaces.clear();
592 ownedSurfaces.reserve(pl.surfaces().size());
594 std::back_inserter(ownedSurfaces),
595 [](
const auto &
s) { return s->getSharedPtr(); });
597 auto layer = m_cfg.layerCreator->discLayer(gctx, ownedSurfaces, nBinsR,
598 nBinsPhi, pl, Transform3::Identity(),
599 std::move(approachDescriptor));
601 layersOutput.push_back(
layer);
621 case Mode::ITkPixelInner:
622 os <<
"ITkPixelInner";
624 case Mode::ITkPixelOuter:
625 os <<
"ITkPixelOuter";