SurveyConstraint.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef SURVEYCONSTRAINTTOOLS_SURVEYCONSTRAINT_H
#define SURVEYCONSTRAINTTOOLS_SURVEYCONSTRAINT_H
 
#include "AthenaBaseComps/AthAlgTool.h"
#include "InDetSurveyConstraintTool/ISurveyConstraint.h"
#include "InDetReadoutGeometry/SiDetectorElementCollection.h"
#include "StoreGate/ReadCondHandleKey.h"
#include <vector>
#include <map>
 
class AtlasDetectorID;
class IInDetAlignDBTool;
class PixelID;
class SCT_ID;
 
class SurveyConstraint : public extends <AthAlgTool, ISurveyConstraint>{
 public:
  SurveyConstraint(const std::string& type,const std::string& name,const IInterface* parent);
  
  virtual ~SurveyConstraint() = default;
  
  virtual StatusCode initialize() override;
  virtual StatusCode finalize() override;
  
  virtual StatusCode computeConstraint(const Identifier& ModuleID,
                       Amg::VectorX& dparams,
                       double& deltachisq,
                       Amg::VectorX& dchisqdparams,
                       Amg::MatrixX& d2chisqdpdp) override;
  virtual void setup_SurveyConstraintModules() override;
 
  virtual void MMap(std::map<Identifier, SurveyConstraintModule*, std::less<Identifier> >&  ModuleMap) override;
 
  virtual int getWeightPixEC(//const Identifier& ModuleID,
                Amg::MatrixX& weight) override;
  virtual int getWeightPixB(//const Identifier& ModuleID,
                Amg::MatrixX& weight) override;
  virtual int getWeightSCTEC(//const Identifier& ModuleID,
                Amg::MatrixX& weight) override;
  virtual int getWeightSCTB(//const Identifier& ModuleID,
                Amg::MatrixX& weight) override;
  virtual void getSurveyCoordsPixEC(//const Identifier& ModuleID,
                    std::vector< Amg::Vector3D > & coords) override;
  virtual void getSurveyCoordsPixB(//const Identifier& ModuleID,
                   std::vector< Amg::Vector3D > & coords) override;
  virtual void getSurveyCoordsSCTEC(//const Identifier& ModuleID,
                    std::vector< Amg::Vector3D > & coords) override;
  virtual void getSurveyCoordsSCTB(//const Identifier& ModuleID,
                   std::vector< Amg::Vector3D > & coords) override;
  virtual void GlobalToLocal(SurveyConstraintModule* mut,std::vector<SurveyConstraintPoint>& points) override;
  virtual int SectorNumber(int phi_module) override;
  virtual double PhiModuleToSector(int phi_module) override;
  virtual void TransformSector(Identifier Pixel_ModuleID,
                   SurveyConstraintModule* mut,
                   Amg::Vector3D& current,
                   Amg::Transform3D CurrentTransRandSect) override;
  
 private :
  const AtlasDetectorID*                 m_idHelper;
  const PixelID*                         m_pixid;
  const SCT_ID*                          m_sctid;
  
 
  IInDetAlignDBTool*                     m_current_IDAlignDBTool;
  IInDetAlignDBTool*                     m_survey_IDAlignDBTool;
 
  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"};
 
  // dictionary of module constrain objects, indexed by moduleID
  std::map<Identifier, SurveyConstraintModule*, std::less<Identifier> >  m_ModuleMap;        
 
  // algorithm parameters, possible to declare at runtime
  double                                 m_SurveyWeightX;              
                                                                       // (e.g. bow), set to 1 for now
  double                                 m_SurveyWeightY;              
  double                                 m_SurveyWeightZ;              
  double                                 m_SurveyWeightPhiX;           
  double                                 m_SurveyWeightPhiY;           
  double                                 m_SurveyWeightPhiZ;           
  double                                 m_TransX;                     
  double                                 m_TransY;                     
  double                                 m_TransZ;                     
  double                                 m_RotX;                       
  double                                 m_RotX2;                      
  double                                 m_RotY;                       
  double                                 m_RotZ;                       
  double                                 m_TransXRand;                 
  double                                 m_TransYRand;                 
  double                                 m_TransZRand;                 
  double                                 m_RotXRand;                   
  double                                 m_RotYRand;                   
  double                                 m_RotZRand;                   
  double                                 m_TransXRandPixB;             
  double                                 m_TransYRandPixB;             
  double                                 m_TransZRandPixB;             
  double                                 m_RotXRandPixB;               
  double                                 m_RotYRandPixB;               
  double                                 m_RotZRandPixB;               
  double                                 m_TransXRandSCTEC;             
  double                                 m_TransYRandSCTEC;             
  double                                 m_TransZRandSCTEC;             
  double                                 m_RotXRandSCTEC;               
  double                                 m_RotYRandSCTEC;               
  double                                 m_RotZRandSCTEC;               
  double                                 m_TransXRandSCTB;             
  double                                 m_TransYRandSCTB;             
  double                                 m_TransZRandSCTB;             
  double                                 m_RotXRandSCTB;               
  double                                 m_RotYRandSCTB;               
  double                                 m_RotZRandSCTB;               
  bool                                   m_gaus;                       
  double                                 m_TransXRandSect;             
  double                                 m_TransYRandSect;             
  double                                 m_TransZRandSect;             
  double                                 m_RotXRandSect;               
  double                                 m_RotYRandSect;               
  double                                 m_RotZRandSect;               
  double                                 m_TransLayerRand;             
  int                                    m_misaligncase;               
  bool                                   m_gausSect;                   
  bool                                   m_FullDisk;                   
  double                                 m_scaleZ;                     
  double                                 m_proximity;                  
  std::string                            m_aligndbtoolinst; 
  std::string                            m_surveydbtoolinst; 
  std::string                            m_surveywfile;
  std::string                            m_surveyrfile;
  bool                                   m_ntuple;
 
};
 
#endif // SURVEYCONSTRAINTTOOLS_SURVEYCONSTRAINT_H