AlignAlg.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef TRKALIGNGENALGS_ALIGNALG_H
#define TRKALIGNGENALGS_ALIGNALG_H
 
#include "GaudiKernel/ToolHandle.h"
#include "AthenaBaseComps/AthAlgorithm.h"
 
#include "TrkAlignInterfaces/ITrackCollectionProvider.h"
#include "TrkAlignInterfaces/IAlignTrackPreProcessor.h"
#include "TrkAlignInterfaces/IAlignTrackCreator.h"
#include "TrkAlignInterfaces/IAlignTrackDresser.h"
#include "TrkAlignInterfaces/IAlignTool.h"
#include "TrkAlignInterfaces/IGeometryManagerTool.h"
#include "TrkAlignInterfaces/ITrkAlignDBTool.h"
#include "TrkAlignInterfaces/IFillNtupleTool.h"
 
 
#include <string>
#include <fstream>
  
 
namespace Trk {
  
  class IFillNtupleTool;
 
  class AlignAlg : public AthAlgorithm {
    
  public: 

    using AthAlgorithm::AthAlgorithm;

    virtual ~AlignAlg();
    
    virtual StatusCode initialize();

    virtual StatusCode start();

    virtual StatusCode execute();

    virtual StatusCode stop();

    virtual StatusCode finalize();

    void showStatistics();
    
  private:
    
    ToolHandle <ITrackCollectionProvider> m_trackCollectionProvider{
      this, "TrackCollectionProvider", "Trk::TrackCollectionProvider",
    "tool for getting track collection from StoreGate"};

    ToolHandle<IAlignTrackPreProcessor> m_alignTrackPreProcessor{
      this, "AlignTrackPreProcessor", "Trk::AlignTrackPreProcessor",
      "tool for converting Trk::Track to AlignTrack after processing if necessary"};

    ToolHandle <IAlignTrackCreator>  m_alignTrackCreator{
      this, "AlignTrackCreator", "Trk::AlignTrackCreator",
      "tool for creating AlignTSOSCollection to store on AlignTrack"};
    
    ToolHandle <IAlignTrackDresser>  m_alignTrackDresser{
      this, "AlignTrackDresser", "Trk::AlignTrackDresser",
      "tool for dressing AlignTrack with residuals, derivatives, etc."};
    
    ToolHandle <IAlignTool>  m_alignTool{
      this, "AlignTool", "Trk::GlobalChi2AlignTool",
      "alignment algorithm-specific tool"};
        
    PublicToolHandle <IGeometryManagerTool> m_geometryManagerTool{
      this, "GeometryManagerTool", "InDet::InDetGeometryManagerTool",
      "tool for configuring geometry"};
    
    ToolHandle <ITrkAlignDBTool> m_trkAlignDBTool{
      this, "AlignDBTool", "Trk::TrkAlignDBTool", "tool for handling DB stuff"};
    
    ToolHandle <IFillNtupleTool> m_fillNtupleTool{
      this, "FillNtupleTool", "",
      "tool for storing Trk::Track information into the ntuple"};
 
    // various job options
    StringProperty m_filename{this, "FileName", "Align.root", "name of ntuple file"};
    StringProperty m_filepath{this, "FilePath", "./", "path to ntuple file"};
 
    BooleanProperty m_solveOnly{this, "SolveOnly", false,
      "only do the solving (accumulate from binaries)"};
    BooleanProperty m_writeNtuple{this, "WriteNtuple", true,
      "write track and event information to ntuple"};
 
    IntegerProperty m_alignSolveLevel{this, "AlignSolveLevel", 3,
      "Set the Alignment Solve Level"};
 
    TFile*         m_ntuple = nullptr;        
    BooleanProperty m_writeLogfile{this, "WriteLogFile", true,
      "write a logfile for solving"};
    StringProperty m_logfileName{this, "LogFileName", "alignlogfile.txt",
      "name of the logfile"};
    std::ostream * m_logStream = nullptr;     
 
    int m_nevents = 0;    
    int m_ntracks = 0;    
    int m_ntracksSel = 0; 
    int m_ntracksProc = 0;  
    int m_ntracksDress = 0; 
    int m_ntracksAccum = 0; 
 
    StringProperty m_alignTracksName{this, "AlignTracksName", "AlignTracks",
      "name of the AlignTrack collection in the StoreGate"};
 
   };
 
} // end namespace
 
#endif // TRKALIGNGENALGS_ALIGNALG_H