db/d8d/SurveyConstraint_8h_source.html

/*

  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