d8/d70/InDetAlignCog_8h_source.html

/*

  Copyright (C) 2002-2020 CERN for the benefit of the ATLAS collaboration

*/


#ifndef INDETALIGNGENALGS_INDETALIGNCOG_H

#define INDETALIGNGENALGS_INDETALIGNCOG_H


#include "AthenaBaseComps/AthAlgorithm.h"

#include "GaudiKernel/AlgTool.h"

#include "GaudiKernel/ToolHandle.h"

#include "GaudiKernel/ServiceHandle.h"

#include "InDetAlignGenTools/IInDetAlignDBTool.h"

#include "TRT_ConditionsServices/ITRT_AlignDbSvc.h"


#include <GeoPrimitives/GeoPrimitives.h>

#include <EventPrimitives/EventPrimitives.h>


#include "InDetReadoutGeometry/SiDetectorElementCollection.h"

#include "TRT_ReadoutGeometry/TRT_DetElementContainer.h"

#include "StoreGate/ReadCondHandleKey.h"


class PixelID;

class SCT_ID;

class TRT_ID;

class ITRT_AlignDbSvc;


namespace InDetDD {

  class TRT_BaseElement;

  class SiDetectorElement;

}


class InDetAlignCog : public AthAlgorithm {

 public:

  InDetAlignCog(const std::string& name, ISvcLocator* pSvcLocator);

  StatusCode initialize();

  StatusCode execute();

  StatusCode finalize();


 private:

  class Params_t

  {

  public:

    Params_t () {

      m_V1.setZero();

      m_V2.setZero();

      m_A1.setZero();

      m_A2.setZero();

      m_M1.setZero();

      m_M2.setZero();

    }


    AmgVector(6) m_V1;

    AmgVector(6) m_V2;

    AmgVector(6) m_A1;

    AmgVector(6) m_A2;

    AmgSymMatrix(6) m_M1;

    AmgSymMatrix(6) m_M2;

  };


  StatusCode getSiElements(const InDetDD::SiDetectorElementCollection*,const bool, InDetAlignCog::Params_t &params);

  StatusCode getTRT_Elements(const InDetDD::TRT_DetElementCollection*,const bool, InDetAlignCog::Params_t &params);

  StatusCode shiftIDbyCog();

  void addL1();


  void enableCoG(Amg::Transform3D&, bool, bool, bool, bool, bool, bool);

  StatusCode normalizeTransform(Amg::Transform3D&, const int);

  void scaleTransform(Amg::Transform3D&, const float);


  Amg::Transform3D sumTransforms(const Amg::Transform3D&,

                   const Amg::Transform3D&) const;


  void prepareDerivative(const Amg::Transform3D&, const bool=false);

  void accumulate(const Amg::Transform3D&, double* );

  void accumulateChi2(const Amg::Transform3D&, AmgSymMatrix(6)&, AmgVector(6)&, const double* );


  std::string printTransform(const Amg::Transform3D&) const;

  bool testIdentity(const Amg::Transform3D&,double,double) const;

  const Amg::Transform3D getL1Transform(int bec);


 private:


  // helpers

  const PixelID *m_pixid;

  const SCT_ID *m_sctid;

  const TRT_ID *m_trtid;


  // IInDetAlignDBTool

  ToolHandle <IInDetAlignDBTool> m_IDAlignDBTool;


  // TRTAlignDbTool

  // ToolHandle<ITRTAlignDbTool> m_TRTAlignDbTool;

  ServiceHandle<ITRT_AlignDbSvc> m_TRTAlignDbTool;


  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_trtDetEleContKey{this, "TRTDetEleContKey", "TRT_DetElementContainer", "Key of TRT_DetElementContainer for TRT"};


  // Select which detectors will be considered for cog calculation

  int m_det;

  int m_Si_bec;

  int m_TRT_bec;

  int m_Si_layer;

  int m_TRT_layer;


  bool m_firstEvent;

  bool m_fullMatrix;

  bool m_printDB;

  bool m_SiTxtOutput;

  bool m_TRT_TxtOutput;

  bool m_useChi2;      // if to use the Chi2 fit


  double  m_errTrans;

  double  m_errRot;

  double  m_sigXpixB;

  double  m_sigYpixB;

  double  m_sigZpixB;

  double  m_sigXpixE;

  double  m_sigYpixE;

  double  m_sigZpixE;

  double  m_sigXsctB;

  double  m_sigYsctB;

  double  m_sigZsctB;

  double  m_sigXsctE;

  double  m_sigYsctE;

  double  m_sigZsctE;

  double  m_sigXtrtB;

  double  m_sigYtrtB;

  double  m_sigZtrtB;

  double  m_sigXtrtE;

  double  m_sigYtrtE;

  double  m_sigZtrtE;


  bool    m_doTX;

  bool    m_doTY;

  bool    m_doTZ;

  bool    m_doRX;

  bool    m_doRY;

  bool    m_doRZ;


  bool    m_doCoG;

  bool    m_doL1;


  double  m_traX;

  double  m_traY;

  double  m_traZ;

  double  m_rotX;

  double  m_rotY;

  double  m_rotZ;


  int     m_counter;


  std::string m_sitxtfile;

  std::string m_trt_txtfile;

  std::string m_SQLiteTag;


  // infinitesimal global transformations used to calculate derivatives

  Amg::Transform3D m_glob_x;

  Amg::Transform3D m_glob_y;

  Amg::Transform3D m_glob_z;

  Amg::Transform3D m_grot_x;

  Amg::Transform3D m_grot_y;

  Amg::Transform3D m_grot_z;


  // Center-of-gravity

  double         m_cog[6]{};

  Amg::Transform3D m_CoG;


  // residual global transformation

  double         m_resglob[6]{};

  Amg::Transform3D m_ResGlob;


  // Chi2 specific stuff:


};


#endif // INDETALIGNGENALGS_INDETALIGNCOG_H