GlobalChi2AlignTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef TRKGLOBALCHI2ALIGNTOOLS_GLOBALCHI2ALIGNTOOL_H
#define TRKGLOBALCHI2ALIGNTOOLS_GLOBALCHI2ALIGNTOOL_H
 
#include "GaudiKernel/ToolHandle.h"
#include "AthenaBaseComps/AthAlgTool.h"
 
#include "TrkAlignInterfaces/IAlignTool.h"
#include "TrkAlignInterfaces/IMatrixTool.h"
#include "TrkAlignInterfaces/IAlignModuleTool.h"
 
//#include "TMatrixD.h"
 
#include <string>

 
class TFile;
class TTree;
 
namespace Trk {
  class Track;
 
  class GlobalChi2AlignTool : virtual public IAlignTool, public AthAlgTool {    
  public:
    
    GlobalChi2AlignTool(const std::string& type, const std::string& name,
           const IInterface* parent);
    
    virtual ~GlobalChi2AlignTool();
    
    virtual StatusCode initialize();
    
    virtual StatusCode finalize();

    virtual StatusCode firstEventInitialize();

    virtual StatusCode solve();

    bool accumulate(AlignTrack* alignTrack);

    bool accumulateFromFiles();

    unsigned int nAccumulatedTracks() { return m_ntracks; };

    void setNtuple(TFile* ntuple) { m_ntuple=ntuple; }

    void setLogStream(std::ostream * os);

    StatusCode fillNtuple();
 
  private:
 
    double getMaterialOnTrack(const Trk::Track* track);
  
    ToolHandle<Trk::IMatrixTool> m_matrixTool{this, "MatrixTool", "Trk::MatrixTool",
      "tool for storing and inverting matrix"};

    PublicToolHandle<Trk::IAlignModuleTool> m_alignModuleTool{this,
      "AlignModuleTool", "Trk::AlignModuleTool/AlignModuleTool"};
    
    std::string m_pathbin;            
    std::string m_pathtxt;            
    std::string m_prefixName;         
 
    unsigned int m_ntracks = 0; 
    unsigned int m_nmeas = 0;   
    unsigned int m_nhits = 0;   
    double m_chi2 = 0.;          
    unsigned int m_nDoF = 0;    
 
    Gaudi::Property<double> m_secondDerivativeCut{this, "SecondDerivativeCut", -1e5};
    Gaudi::Property<bool>   m_doTree{this, "DoTree", false};
    Gaudi::Property<bool>   m_writeActualSecDeriv{this, "WriteActualSecDeriv", false};
    Gaudi::Property<bool>   m_storeLocalDerivOnly{this, "StoreLocalDerivOnly", false};
 
    static constexpr int MAXNCHAMBERS=50;
    static constexpr int MAXNINDICES =50*6;

    TFile*   m_ntuple = nullptr;
    TTree*   m_tree = nullptr;
    int      m_run{};
    int      m_event{};
    double   m_materialOnTrack{};
    double   m_momentum{};
    int      m_nChambers = 0;
    int*     m_chamberIds = new int[MAXNCHAMBERS];
    int      m_nMatrixIndices = 0;
    int*     m_matrixIndices = new int[MAXNINDICES];
    int      m_nSecndDeriv = 0;
    double*  m_secndDeriv = new double[MAXNINDICES*MAXNINDICES];
    double*  m_firstDeriv = new double[MAXNINDICES];
    double*  m_actualSecndDeriv = new double[MAXNINDICES];
    double   m_eta{};
    double   m_phi{};
    double   m_perigee_x{};
    double   m_perigee_y{};
    double   m_perigee_z{};
    int      m_trackInfo{};
    int      m_bremFit{};
    int      m_bremFitSuccessful{};
    int      m_straightTrack{};
    int      m_slimmedTrack{};
    int      m_hardScatterOrKink{};
    int      m_lowPtTrack{};
              
    bool     m_fromFiles = false;
 
  }; // end of class
 
} // End of namespace 
 
#endif // TRKALIGNGENTOOLS_MATRIXTOOL_H