TRTCalibrator.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef TRT_CALIBTOOLS__TRTCALIBRATOR_H
#define TRT_CALIBTOOLS__TRTCALIBRATOR_H
 
/********************************************************************
 
   NAME: TRTCalibrator.h    
PACKAGE: TRTCalibTools 
 
AUTHORS: Johan Lundquist  
CREATED: 27-03-2009 
 
PURPOSE: Tool for calibrating the TRT
          
********************************************************************/
 
 
#include "TRT_CalibTools/ITRTCalibrator.h"
#include "TRT_ConditionsServices/ITRT_StrawNeighbourSvc.h"
#include "TRT_ConditionsServices/ITRT_StrawStatusSummaryTool.h"
#include "TRT_ConditionsServices/ITRT_CalDbTool.h"
#include "AthenaBaseComps/AthAlgTool.h"
#include "GaudiKernel/ToolHandle.h"
#include "GaudiKernel/ServiceHandle.h"
#include "Gaudi/Property.h"
 
#include <string>
#include <vector>
#include <map>
#include <set>
 
class AtlasDetectorID;
class TRT_ID;
class TFile;
namespace InDetDD{
  class TRT_DetectorManager;
}

class TRTCalibrator :  virtual public ITRTCalibrator, public AthAlgTool {
public:

  TRTCalibrator(const std::string& type, const std::string& name, const IInterface* parent);

  virtual ~TRTCalibrator() {}

  virtual StatusCode initialize() override;
  virtual StatusCode finalize() override;
 
  virtual bool fill(const Trk::Track* aTrack, TRT::TrackInfo* output) override;
 
  bool IncludedLevels(std::string, int*);

  std::string SubLev(std::string, int);  

  bool IsSubLev(const std::string&, int, const std::string&);  

  void DumpStrawData(int);  

  int GetSubLevels(const std::string&, int, std::set<int>*);  

  void MakeBDKeys(int, int, int, int, int, int);

  virtual bool calibrate ATLAS_NOT_THREAD_SAFE () override;
 
private:
 
    ToolHandle<ITRT_CalDbTool> m_trtcaldbTool {this, "TRTCalDbTool", "ITRT_CalDbTool", ""};
    ToolHandle<ITRT_StrawStatusSummaryTool> m_TRTStrawSummaryTool {this, "TRTStrawSummaryTool", "ITRT_StrawStatusSummaryTool", ""};
    ServiceHandle<ITRT_StrawNeighbourSvc> m_neighbourSvc {this,"NeighbourSvc","ITRT_StrawNeighbourSvc","retrieve barrel and end-cap straw number" };
    
 
    // Gaudi Properties
    Gaudi::Property<float> m_maxDistance             {this, "maxDistance"     , 2.8 , ""};
    Gaudi::Property<float> m_maxTimeResidual         {this, "maxTimeResidual" , 150.0 , ""};
    Gaudi::Property<float> m_maxTrackChisquarePerDof {this, "maxTrackChisquarePerDof" , 10.0 , ""};
    Gaudi::Property<float> m_t0offset                {this, "T0Offset" , -0.3 , ""};
 
    Gaudi::Property<int> m_minTimebinsOverThreshold {this, "minTimebinsOverThreshold" , 2 , ""};
    Gaudi::Property<int> m_minrt   {this, "MinRt"   , 10000 , "minimum number of hits required to do R-t calibration"    };
    Gaudi::Property<int> m_mint0   {this, "MinT0"   , 200   , "minimum number of hits required to do T0 calibration"     };
    Gaudi::Property<int> m_nevents {this, "Nevents" , -1    , "The number of hits or histograms read from the input file"};
 
    Gaudi::Property<bool> m_SplitBarrel  {this, "SplitBarrel"  , true  , ""};
    Gaudi::Property<bool> m_useP0        {this, "UseP0"        , true  , ""};
    Gaudi::Property<bool> m_floatP3      {this, "FloatP3"      , true  , ""};
    Gaudi::Property<bool> m_DoArXenonSep {this, "DoArXenonSep" , false , ""};
    Gaudi::Property<bool> m_DoShortStrawCorrection {this, "DoShortStrawCorrection" , true , ""};
 
    Gaudi::Property<std::string> m_selstring   {this, "Selstring" , "_*_-_-_-_-_-_-" , ""};
    Gaudi::Property<std::string> m_calsub      {this, "SubPart"   , "user" , ""};
    Gaudi::Property<std::string> m_options     {this, "Options"   , "" , ""};
    Gaudi::Property<std::string> m_hittuple    {this, "Hittuple"  , "merged.root" , ""};
    Gaudi::Property<std::string> m_rtrel       {this, "RtRel"     , "basic" , "the rt relation to use"};
    Gaudi::Property<std::string> m_rtbinning   {this, "RtBinning" , "t" , ""};
    Gaudi::Property<std::string> m_ntrtmanager {this, "TrtManagerLocation" , "TRT" , ""};
 
    Gaudi::Property< std::vector<std::string> > m_doRt    {this, "CalibrateRt"  , {} , ""};
    Gaudi::Property< std::vector<std::string> > m_doT0    {this, "CalibrateT0"  , {} , ""};
    Gaudi::Property< std::vector<std::string> > m_doRes   {this, "FitResidual"  , {} , ""};
    Gaudi::Property< std::vector<std::string> > m_beQuiet {this, "NoHistograms" , {} , ""};
    Gaudi::Property< std::vector<std::string> > m_useBoardRef {this, "UseBoardRef" , {} , ""};
    Gaudi::Property< std::vector<std::string> > m_doLogPrint  {this, "PrintLog"    , {} , ""};
    Gaudi::Property< std::vector<std::string> > m_doOutPrint  {this, "PrintT0Out"  , {} , ""};
    Gaudi::Property< std::vector<std::string> > m_doRtPrint   {this, "PrintRtOut"  , {} , ""};
 
  const AtlasDetectorID* m_DetID{};
  const TRT_ID* m_TRTID{}; 
  const InDetDD::TRT_DetectorManager* m_trtmanager{};
  unsigned int m_numOfHitsTotal{};
  unsigned int m_numOfHitsAccepted{} ;
  unsigned int m_numOfProcessedTracks{} ;
 
  std::string m_comTimeName;
  std::string m_Tkey,m_Dkey,m_Lkey,m_Mkey,m_Bkey,m_Ckey,m_Skey,m_Lkey_acc,m_Dkey_acc;
   class BDzero{public: int z;};  class BDstraw{public: std::map<std::string,BDzero> s;};   class BDchip{public: std::map<std::string,BDstraw> c;};  class BDboard{public: std::map<std::string,BDchip> b;}; class BDmodule{public: std::map<std::string,BDboard> m;}; class BDlayer{public: std::map<std::string,BDmodule> l;}; class BDdetector{public: std::map<std::string,BDlayer> d;}; class BDTRT{public: std::map<std::string,BDdetector> t;}; BDTRT m_trt;
 BDTRT m_trt_acc;
 
  
  class epdata{
  public:
    float run,event,track,nhits,ephase1,ephase2,t0,t,ttrack;
  };
 
  struct conf {
    std::string Selstring, Options;
    std::vector<std::string> CalibrateRt, CalibrateT0, FitResidual, NoHistograms, PrintLog, PrintT0Out, PrintRtOut;
    std::map< std::string,std::set<int> > selections;//TRTsel, detsel, laysel, modsel, brdsel, chpsel, stwsel;
  };
  std::map<std::string,conf> m_config;
 
  enum {CALIB_ALL=-3, CALIB_NONE=-4};
 
  
};
 
#endif //  TRT_CALIBTOOLS__TRTCALIBRATOR_H