AlignModuleTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 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 ~AlignModuleTool();
    
    virtual StatusCode initialize();
    virtual StatusCode finalize();
 
 
    //
    // GeometryManagerTool methods
    //
 
    void setAlignModules(AlignModuleList* alignmods,
             std::vector<AlignModuleList*>* idHashMaps);
    
    void setAlignParLists(AlignPar2DVec* fullAlignParList, 
              AlignPar2DVec* alignParList=0);
 
    void setSubDetElement(AlignModule::DetectorType detType, bool status) { m_subDetElement[detType]=status; }
    bool subDetElement(AlignModule::DetectorType detType) const { return m_subDetElement[detType]; }
 
    //
    // AlignTrackCreator methods
    //
 
    virtual AlignModule * findAlignModule(const RIO_OnTrack * rio,
                                          AlignModule::DetectorType detType) const;
 
    virtual AlignModule * findAlignModule(const TrkDetElementBase* det,
                                          AlignModule::DetectorType detType) const;
 
    //
    // MatrixTool methods
    //
 
    int nAlignParameters() const { return m_alignParList1D.size(); }
    
    const AlignModuleList* alignModules1D() { return &m_alignModules1D; }
    
    AlignParVec* alignParList1D() { return &m_alignParList1D; }
    
    
    //
    // DerivCalcTool methods
    //
 
    AlignParVec* getAlignPars(const AlignModule* alignModule) const;
    
    virtual void shiftModule(const AlignModule*, const Track*, 
                 AlignModule::TransformParameters, double) const {}
    virtual void restoreModule(const AlignModule*) const {}
    
    
    //
    // AlignDBTool methods
    // 
    
    virtual void fillDetAlignmentParameters(const TrkDetElementBase* det,
                        DataVector<AlignPar>* apVec,
                        AlignModule::DetectorType detType) const;
 
    virtual const AlignModuleList* getAlignModuleCollection() const 
    { return m_alignModules; }
 
    //
    // FillNtuple methods
    //
 
    AlignParVec* getFullAlignPars(const AlignModule* alignModule) const;
 
 
    //
    // miscellaneous methods
    //
 
    AlignModule::DetectorType getDetectorType(const TrkDetElementBase* det) const;
 
    virtual int subDetElementIDHash(Identifier ) const { return 0; }
 
    AlignPar2DVec* alignParList() const { return m_alignParList; }
 
    AlignPar2DVec* fullAlignParList() const { return m_fullAlignParList; }
 
  protected:
    const AtlasDetectorID* m_idHelper;          
 
  private:
 
    AlignModuleList*                    m_alignModules;    
    std::vector<const AlignModuleList*> m_alignModuleMaps; 
 
    AlignPar2DVec*            m_alignParList;     
    AlignPar2DVec*            m_fullAlignParList; 
 
    std::vector<AlignModule*> m_alignModules1D;    
    AlignParVec               m_alignParList1D;    
 
    std::vector<bool>         m_subDetElement;     
  };
 
 MsgStream& operator << ( MsgStream& sl, AlignModuleTool& alignModTool);
  
}
 
#endif // TRKALIGNGENTOOLS_ALIGNMODULETOOL_H