MaterialMapping.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/

// MaterialMapping.h, (c) ATLAS Detector software
 
#ifndef TRKDETDESCRALGS_MATERIALMAPPING_H
#define TRKDETDESCRALGS_MATERIALMAPPING_H
 
// Athena & Gaudi includes
#include "AthenaBaseComps/AthAlgorithm.h"
#include "GaudiKernel/ServiceHandle.h"
#include "GaudiKernel/ToolHandle.h"
#include "GeoPrimitives/GeoPrimitives.h"
#include "StoreGate/ReadHandleKey.h"
#include "TString.h"
#include "TrkGeometry/MaterialStepCollection.h"
#include <fstream>
#include <iostream>
#include <map>
#include <string>
// TrkDetDescr Algs, Interfaces, Utils
#include "TrkDetDescrInterfaces/ILayerMaterialAnalyser.h"
#include "TrkDetDescrInterfaces/ILayerMaterialCreator.h"
#include "TrkDetDescrInterfaces/IMaterialMapper.h"
// TrkExtrapolation
#include "TrkExInterfaces/IExtrapolationEngine.h"
// TrkGeometry
#include "TrkGeometry/TrackingGeometry.h"
 
#ifdef TRKDETDESCR_MEMUSAGE
#include "TrkDetDescrUtils/MemoryLogger.h"
#endif
 
#ifndef UCHARCONV
#define UCHARCONV
#define ucharbin 0.00392157 // 1./double(1.*UCHAR_MAX)
// int to unsigned char and vv
#define uchar2uint(uchar) static_cast<unsigned int>(uchar)
#define uint2uchar(unint) static_cast<unsigned char>(unint)
// double to unsigned char and vv
#define uchar2dfrac(uchar) double(uchar * ucharbin)
#define dfrac2uchar(dfrac) lrint(dfrac* UCHAR_MAX)
#endif
 
class TTree;
 
namespace Trk {
 
class Layer;
class TrackingVolume;
class SurfaceMaterialRecord;
class LayerMaterialRecord;
class LayerMaterialMap;
class Material;
class MaterialProperties;
class BinnedLayerMaterial;
class CompressedLayerMaterial;
class ElementTable;

 
class MaterialMapping : public AthAlgorithm
{
 
public:
  MaterialMapping(const std::string& name, ISvcLocator* pSvcLocator);

  ~MaterialMapping();

  StatusCode initialize();

  StatusCode execute();

  StatusCode finalize();
 
private:
  bool associateHit(const Trk::Layer& tvol,
                    const Amg::Vector3D& pos,
                    const Amg::Vector3D& layerHitPosition,
                    double stepl,
                    const Trk::Material& mat);

  void registerVolume(const Trk::TrackingVolume& tvol, int lvl);

  void assignLayerMaterialProperties(Trk::TrackingVolume& tvol,
                                     Trk::LayerMaterialMap* propSet);

  void insertLayerMaterialRecord(const Trk::Layer& lay);

  StatusCode handleTrackingGeometry();
 
  const TrackingGeometry& trackingGeometry() const;
 
  bool
    m_checkForEmptyHits; 
  ToolHandle<IExtrapolationEngine> m_extrapolationEngine{
    this,
    "ExtrapolationEngine",
    "",
    "Extrapolation Engine"
  };
 
  std::string m_mappingVolumeName;
  const Trk::TrackingVolume* m_mappingVolume;

  SG::ReadHandleKey<MaterialStepCollection> m_inputMaterialStepCollection;
  std::string m_outputLayerMaterialSetName;

  double m_etaCutOff;
  int m_etaSide; 
  bool m_useLayerThickness; 
  int m_associationType;
 
  ToolHandle<ILayerMaterialAnalyser> m_layerMaterialRecordAnalyser{
    this,
    "LayerMaterialRecordAnalyser",
    "",
    "Layer material analyser for the layer material record"
  };
  ToolHandleArray<ILayerMaterialAnalyser> m_layerMaterialAnalysers{
    this,
    "LayerMaterialAnalysers",
    {},
    "Layer material analysers per creator (if wanted)"
  };
  ToolHandleArray<ILayerMaterialCreator> m_layerMaterialCreators{
    this,
    "LayerMaterialCreators",
    {},
    "Layer material creators"
  };

  bool m_mapMaterial;
  ToolHandle<IMaterialMapper> m_materialMapper{ this,
                                                "MaterialMapper",
                                                "",
                                                "IMaterialMapper algTool" };
  bool m_mapComposition;           
  double m_minCompositionFraction; 
 
  Trk::ElementTable* m_elementTable; 
  SG::ReadHandleKey<Trk::ElementTable>
    m_inputEventElementTable; 
 
  // the material maps ordered with layer keys
  std::map<const Layer*, LayerMaterialRecord>
    m_layerRecords; 
  std::map<const Layer*, bool>
    m_layersRecordedPerEvent;      
  double m_accumulatedMaterialXX0; 
  double m_accumulatedRhoS;
 
  // statistics for steps
  size_t m_mapped;
  size_t m_unmapped;
  size_t m_skippedOutside;
 
  int m_layerMaterialScreenOutput;
 
  void throwFailedToGetTrackingGeometry() const;
  const TrackingGeometry* retrieveTrackingGeometry(
    const EventContext& ctx) const
  {
    SG::ReadCondHandle<TrackingGeometry> handle(m_trackingGeometryReadKey, ctx);
    if (!handle.isValid()) {
      ATH_MSG_FATAL("Could not load TrackingGeometry with name '"
                    << m_trackingGeometryReadKey.key() << "'. Aborting.");
      throwFailedToGetTrackingGeometry();
    }
    return handle.cptr();
  }
 
  SG::ReadCondHandleKey<TrackingGeometry> m_trackingGeometryReadKey{
    this,
    "TrackingGeometryReadKey",
    "",
    "Key of the TrackingGeometry conditions data."
  };
 
#ifdef TRKDETDESCR_MEMUSAGE
  MemoryLogger m_memoryLogger; 
#endif
};
 
inline const Trk::TrackingGeometry&
Trk::MaterialMapping::trackingGeometry() const
{
  const Trk::TrackingGeometry* tracking_geometry =
    retrieveTrackingGeometry(Gaudi::Hive::currentContext());
  if (!tracking_geometry) {
    ATH_MSG_FATAL("Did not get valid TrackingGeometry. Aborting.");
    throw GaudiException("MaterialMapping",
                         "Problem with TrackingGeometry loading.",
                         StatusCode::FAILURE);
  }
  return *tracking_geometry;
}
}
 
#endif