IGeometryManagerTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef TRKALIGNINTERFACES_IGEOMETRYMANAGERTOOL_H
#define TRKALIGNINTERFACES_IGEOMETRYMANAGERTOOL_H
 
#include "GaudiKernel/IAlgTool.h"
 
#include "TrkAlignEvent/AlignModuleList.h"
#include "AthContainers/DataVector.h"

 
#include <fstream>
 
class TFile;
 
static const InterfaceID 
IID_TRKALIGNINTERFACES_IGeometryManagerTool("IGeometryManagerTool",1,0);
 
namespace Trk {
 
  class AlignModule;
  class AlignPar;
 
class IGeometryManagerTool : virtual public IAlgTool {
 
 public:
  virtual ~IGeometryManagerTool() {}
 
  static const InterfaceID& interfaceID();
 
  virtual StatusCode initialize() = 0;
  virtual StatusCode finalize() = 0;

  virtual int ReadGeometry(int solvelevel) = 0;
 
  virtual void PrintModPositions() = 0;
  virtual void StoreModPositions(std::string filename) = 0;

  virtual void setNtuple(TFile*) {} 

  virtual void setLogStream(std::ostream * os) { m_logStream = os; }

  virtual void buildGeometry() {};

  virtual void addModuleParameters(Trk::AlignModule * , DataVector< DataVector<Trk::AlignPar> > * , DataVector< DataVector<Trk::AlignPar> > *) {};

  virtual int getNextIDHash() { return m_hashCounter++; };

  virtual void setFirstIDHash(unsigned int idhash) { m_hashCounter=idhash; };

  virtual void setModuleList(Trk::AlignModuleList * list) { m_alignModuleListPtr=list; }

  virtual void setHashMaps(std::vector<Trk::AlignModuleList *> * maps) { m_idHashToAlignModuleMapsPtr=maps; }

  virtual bool checkAlignLevel() { return true; }

  virtual int alignLevel() { return m_alignLevel; }

  virtual void setAlignLevel(int level) { m_alignLevel = level; }

  virtual int alignLevelBarrel() { return m_alignLevelBarrel; }

  virtual void setAlignLevelBarrel(int level) { m_alignLevelBarrel = level; }

  virtual bool checkAlignLevelBarrel() { return true; }

  virtual int alignLevelEndcaps() { return m_alignLevelEndcaps; }

  virtual void setAlignLevelEndcaps(int level) { m_alignLevelEndcaps = level; }

  virtual bool checkAlignLevelEndcaps() { return true; }
 
 protected:

  Trk::AlignModuleList * m_alignModuleListPtr = nullptr;

  std::vector<Trk::AlignModuleList *> * m_idHashToAlignModuleMapsPtr = nullptr;
 
  int m_hashCounter = 0;  
 
  std::ostream * m_logStream = nullptr; 
 
  int m_alignLevel = 0;         
  int m_alignLevelBarrel = 0;   
  int m_alignLevelEndcaps = 0;  
 
}; // end class definition
  
inline const InterfaceID& IGeometryManagerTool::interfaceID() {
  return   IID_TRKALIGNINTERFACES_IGeometryManagerTool;
}
 
} // end namespace
  
#endif // TRKALIGNINTERFACES_IGEOMETRYMANAGERTOOL_H