CreateMisalignAlg.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef INDETALIGNMENT_CREATEMISALIGNALG_H
#define INDETALIGNMENT_CREATEMISALIGNALG_H
 
// CreateMisalignAlg.h
//
 
// Gaudi includes
#include "AthenaBaseComps/AthAlgorithm.h"
 
#include "GaudiKernel/NTuple.h"
#include "GaudiKernel/ToolHandle.h"
#include "GaudiKernel/ServiceHandle.h"
 
// CLHEP includes
#include "CLHEP/Matrix/Vector.h"
#include "CLHEP/Geometry/Transform3D.h"
 
#include "Identifier/Identifier.h"
#include "InDetAlignGenTools/IInDetAlignDBTool.h"
#include "TRT_ConditionsServices/ITRT_AlignDbSvc.h"
#include "InDetReadoutGeometry/SiDetectorElementCollection.h"
#include "TRT_ReadoutGeometry/TRT_DetElementContainer.h"
#include "StoreGate/ReadCondHandleKey.h"
#include <vector>
 
//Forward declaration
class IdentifierHash;
class TTree;
 
class AtlasDetectorID;
class PixelID;
class SCT_ID;
class TRT_ID;
 
namespace InDetAlignment {
class CreateMisalignAlg : public AthAlgorithm {
 
public:
 
    //Gaudi style constructor and execution methods
    CreateMisalignAlg(const std::string& name, ISvcLocator* pSvcLocator);
    ~CreateMisalignAlg();

    StatusCode          initialize();
    StatusCode          execute(const EventContext& ctx);
    StatusCode          finalize();
 
private:
    // Data Members m_*
    const AtlasDetectorID                 *m_idHelper;
    const PixelID                         *m_pixelIdHelper;
    const SCT_ID                          *m_sctIdHelper;
    const TRT_ID                          *m_trtIdHelper;
 
    ToolHandle< IInDetAlignDBTool >        m_IDAlignDBTool;
    ServiceHandle<ITRT_AlignDbSvc>          m_trtaligndbservice;
 
    // algorithm parameters, possible to declare at runtime
    std::string                            m_asciiFileNameBase;  
    std::string                            m_SQLiteTag;          
 
    bool                                   m_firstEvent;         
    bool                                   m_createFreshDB;      
    int                                    m_MisalignmentMode;   
    long int                               m_nEvents;
    std::vector<double>                    m_translation;        
    std::vector<double>                    m_rotation;           
    std::vector<double>                    m_local_translation;        
    std::vector<double>                    m_local_rotation;           
    std::string                            m_index;              
    Gaudi::Property<double> m_Misalign_x {this,"MisalignmentX",0.0,"Fixed X shift (mode 1 and 2)"};
    Gaudi::Property<double> m_Misalign_y {this,"MisalignmentY",0.0,"Fixed Y shif (model 1 and 2)"};       
    Gaudi::Property<double> m_Misalign_z {this,"MisalignmentZ",0.0,"Fixed Z shift (mode 1 and 2)"};
    Gaudi::Property<double> m_Misalign_alpha {this,"MisalignmentAlpha",0.0,"Fixed Alpha shift (model 1 and 2)"};
    Gaudi::Property<double> m_Misalign_beta {this,"MisalignmentBeta",0.0,"Fixed Beta shift (mode 1 and 2)"};
    Gaudi::Property<double> m_Misalign_gamma {this,"MisalignmentGamma",0.0,"Fixed Gamma shift (model 1 and 2)"};     
    Gaudi::Property<double> m_RndmMisalignWidth_x {this,"RandomMisalignmentWidthX",0.1,"Gaussian width of X random misalignments (mode 2)"};
    Gaudi::Property<double> m_RndmMisalignWidth_y {this,"RandomMisalignmentWidthY",0.1,"Gaussian width of Y random misalignments (mode 2)"};
    Gaudi::Property<double> m_RndmMisalignWidth_z {this,"RandomMisalignmentWidthZ",0.1,"Gaussian width of Z random misalignments (mode 2)"};
    Gaudi::Property<double> m_RndmMisalignWidth_alpha {this,"RandomMisalignmentWidthAplha",0.1,"Gaussian width of Alpha random misalignments (mode 2)"};
    Gaudi::Property<double> m_RndmMisalignWidth_beta {this,"RandomMisalignmentWidthBeta",0.1,"Gaussian width of Beta random misalignments (mode 2)"};
    Gaudi::Property<double> m_RndmMisalignWidth_gamma {this,"RandomMisalignmentWidthGamma",0.1,"Gaussian width of Gamma random misalignments (mode 2)"};
    double                                 m_Misalign_maxShift;  
    double                                 m_Misalign_maxShift_Inner;  
    double                                 m_ScalePixelIBL;
    double                                 m_ScalePixelDBM;
    double                                 m_IBLBowingTshift;    
    int                                    m_targetLayer;         
    int                                    m_targetLayerMax;      
    std::string                            m_endcapShiftConvention; 
    double                                 m_radialShift;        
    std::string                            m_radialShiftConvention; 
    std::string                            m_radialSubdetector;  
    double                                 m_ScalePixelBarrel;
    double                                 m_ScalePixelEndcap;
    double                                 m_ScaleSCTBarrel;
    double                                 m_ScaleSCTEndcap;
    double                                 m_ScaleTRTBarrel;
    double                                 m_ScaleTRTEndcap;
 
 
    std::map<Identifier, HepGeom::Point3D<double> >  m_ModuleList;        
 
    // Generated Initial Alignment NTuple
    TTree * m_VisualizationLookupTree;
 
    int         m_AthenaHashedID;
    int         m_HumanReadableID;
 
    NTuple::Item<double> m_AlignResults_x;                    
    NTuple::Item<double> m_AlignResults_y;                    
    NTuple::Item<double> m_AlignResults_z;                    
    NTuple::Item<double> m_AlignResults_alpha;                
    NTuple::Item<double> m_AlignResults_beta;                 
    NTuple::Item<double> m_AlignResults_gamma;                
    NTuple::Item<long>   m_AlignResults_Identifier_ID;        
    NTuple::Item<long>   m_AlignResults_Identifier_PixelSCT;  
    NTuple::Item<long>   m_AlignResults_Identifier_BarrelEC;  
    NTuple::Item<long>   m_AlignResults_Identifier_LayerDisc; 
    NTuple::Item<long>   m_AlignResults_Identifier_Phi;       
    NTuple::Item<long>   m_AlignResults_Identifier_Eta;       
 
    NTuple::Item<double> m_Initial_center_x;             
    NTuple::Item<double> m_Initial_center_y;             
    NTuple::Item<double> m_Initial_center_z;             
 
 
    NTuple::Item<double> m_Global_center_x;             
    NTuple::Item<double> m_Global_center_y;             
    NTuple::Item<double> m_Global_center_z;             
 
    // Member Functions
 
    void setupSCT_AlignModule(int&);
    void setupPixel_AlignModule(int&);
    void setupTRT_AlignModule(int&);

    const HepGeom::Transform3D BuildAlignTransform(const CLHEP::HepVector &);
 
    const Identifier reduceTRTID(Identifier id);
    const Identifier reduceTRTID(IdentifierHash &hash);
 
    // IBL bowing functions
    double getBowingMagParam(double temp_shift);
    double getBowingTx(double p1, double z);
  
    StatusCode GenerateMisaligment();
    
    bool m_doPix;
    bool m_doStrip;
    bool m_doTRT;
 
    SG::ReadCondHandleKey<InDetDD::SiDetectorElementCollection> m_pixelDetEleCollKey{this, "PixelDetEleCollKey", "PixelDetectorElementCollection", "Key of SiDetectorElementCollection for Pixel"};
    SG::ReadCondHandleKey<InDetDD::SiDetectorElementCollection> m_SCTDetEleCollKey{this, "SCTDetEleCollKey", "SCT_DetectorElementCollection", "Key of SiDetectorElementCollection for SCT"};
    SG::ReadCondHandleKey<InDetDD::TRT_DetElementContainer> m_trtDetEleCollKey{this, "TRTDetEleCollKey", "TRT_DetElementContainer", "Key of TRT_DetElementContainer for TRT"};
 
};
 
} // end of namespace bracket
#endif // INDETALIGNMENT_CREATEMISALIGNALG_H