AlignModuleTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef TRKALIGNGENTOOLS_ALIGNMODULETOOL_H
#define TRKALIGNGENTOOLS_ALIGNMODULETOOL_H
 
#include "AthenaBaseComps/AthAlgTool.h"
#include "GaudiKernel/ToolHandle.h"
#include "AthContainers/DataVector.h"
 
#include "TrkAlignInterfaces/IAlignModuleTool.h"
#include "TrkAlignEvent/AlignModule.h"
 
class AtlasDetectorID;
class TRT_ID;
 
namespace Trk {
 
  class TrkDetElementBase;
  class RIO_OnTrack;
 
  class AlignModuleTool :  public AthAlgTool, virtual public IAlignModuleTool {
 
  public:
    typedef DataVector<AlignPar>    AlignParVec;
    typedef DataVector<AlignParVec> AlignPar2DVec;
 
    AlignModuleTool(const std::string& type, const std::string& name,
            const IInterface* parent);
    
    virtual StatusCode initialize() override;
    virtual StatusCode finalize() override;
 
    AlignModuleListType alignModuleListType() const override{
      return static_cast<AlignModuleListType>(m_alignModuleListType.value());
    }
 
    //
    // GeometryManagerTool methods
    //
 
    virtual void setAlignModules(AlignModuleList* alignmods,
                                 std::vector<AlignModuleList*>* idHashMaps) override;
    
    virtual void setAlignParLists(AlignPar2DVec* fullAlignParList, 
                                  AlignPar2DVec* alignParList=0) override;
 
    virtual void setSubDetElement(AlignModule::DetectorType detType, bool status) override { m_subDetElement[detType]=status; }
    virtual bool subDetElement(AlignModule::DetectorType detType) const override { return m_subDetElement[detType]; }
 
    //
    // AlignTrackCreator methods
    //
 
    virtual AlignModule * findAlignModule(const RIO_OnTrack * rio,
                                          AlignModule::DetectorType detType) const override;
 
    virtual AlignModule * findAlignModule(const TrkDetElementBase* det,
                                          AlignModule::DetectorType detType) const override;
 
    //
    // MatrixTool methods
    //
 
    virtual int nAlignParameters() const override { return m_alignParList1D.size(); }
    
    virtual const AlignModuleList* alignModules1D() override { return &m_alignModules1D; }
    
    virtual AlignParVec* alignParList1D() override { return &m_alignParList1D; }
    
    
    //
    // DerivCalcTool methods
    //
 
    virtual AlignParVec* getAlignPars(const AlignModule* alignModule) const override;
    
    virtual void shiftModule(const AlignModule*, const Track*, 
                 AlignModule::TransformParameters, double) const override {}
    virtual void restoreModule(const AlignModule*) const override {}
    
    
    //
    // AlignDBTool methods
    // 
    
    virtual void fillDetAlignmentParameters(const TrkDetElementBase* det,
                        DataVector<AlignPar>* apVec,
                        AlignModule::DetectorType detType) const override;
 
    virtual const AlignModuleList* getAlignModuleCollection() const override
    { return m_alignModules; }
 
    //
    // FillNtuple methods
    //
 
    AlignParVec* getFullAlignPars(const AlignModule* alignModule) const override;
 
 
    //
    // miscellaneous methods
    //
 
    AlignModule::DetectorType getDetectorType(const TrkDetElementBase* det) const override;
 
    virtual int subDetElementIDHash(Identifier ) const { return 0; }
 
    AlignPar2DVec* alignParList() const { return m_alignParList; }
 
    AlignPar2DVec* fullAlignParList() const { return m_fullAlignParList; }
 
  protected:
    const AtlasDetectorID* m_idHelper = nullptr;          
 
  private:
 
    IntegerProperty m_alignModuleListType{this, "AlignModuleListType", Trk::L3}; 
 
    AlignModuleList*                    m_alignModules = nullptr;    
    std::vector<const AlignModuleList*> m_alignModuleMaps; 
 
    AlignPar2DVec*            m_alignParList = nullptr;     
    AlignPar2DVec*            m_fullAlignParList = nullptr; 
 
    std::vector<AlignModule*> m_alignModules1D;    
    AlignParVec               m_alignParList1D;    
 
    std::vector<bool>         m_subDetElement;     
  };
 
 MsgStream& operator << ( MsgStream& sl, AlignModuleTool& alignModTool);
  
}
 
#endif // TRKALIGNGENTOOLS_ALIGNMODULETOOL_H