AFPTDBasicTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/

#ifndef AFP_LOCRECO_AFPTDBASICKALMANTOOL_H
#define AFP_LOCRECO_AFPTDBASICKALMANTOOL_H 1
 
 
// FrameWork includes
#include "AthenaBaseComps/AthAlgTool.h"
#include "GaudiKernel/ServiceHandle.h"
 
// xAOD includes
#include "AFP_DBTools/ToFLocParamData.h"
#include "AFP_DBTools/IToFLocParamDBTool.h"
#include "xAODForward/AFPToFHit.h"
#include "xAODForward/AFPToFHitContainer.h"
#include "xAODForward/AFPToFTrack.h"
#include "xAODForward/AFPToFTrackContainer.h"
#include "xAODForward/AFPToFTrackAuxContainer.h"
 
// AFP_LocReco includes
#include "AFP_LocReco/IAFPTDLocRecoTrackAlgTool.h"
#include "AFP_LocReco/AFPLocRecoStationBasicObj.h"
#include "AFP_LocReco/AFPTDBasicToolTrack.h"
 
// STL includes
#include "nlohmann/json.hpp"
#include <string>
#include <list>
#include <vector>

class AFPTDBasicTool
  : public extends<AthAlgTool, IAFPTDLocRecoTrackAlgTool>
{
public:
  AFPTDBasicTool(const std::string& type,
                        const std::string& name,
                        const IInterface* parent);
 
  virtual ~AFPTDBasicTool() override {}

  virtual StatusCode initialize() override;

  virtual StatusCode finalize() override;

  StatusCode reconstructTracks(std::unique_ptr<xAOD::AFPToFTrackContainer>& outputContainer, const EventContext& ctx) const override;

  const std::string& outputContainerName () const override {return m_tracksContainerName;}
 
 
private:
  Gaudi::Property<std::string> m_tracksContainerName{this, "tracksContainerName", "AFPToFTrackContainer", "Name of the container in which ToF tracks are saved"};

  Gaudi::Property<int> m_stationID{this, "stationID", 0, "ID number of station for which tracks should be reconstructed"};

  SG::ReadHandleKey<xAOD::AFPToFHitContainer> m_hitContainerKey{this, "AFPToFHitContainerKey", "AFPToFHitContainer", "Name of the container with ToF hits from which ToF tracks are to be reconstructed"};

  Gaudi::Property<unsigned int> m_minHitsNumber{this, "minHitsNumber", 3, "Minimal number of bars in track. If there are less bars, track is rejected"};
 
  Gaudi::Property<double> m_maxAllowedLength{this, "maxAllowedLength", 100, "Maximal length of the bar signal at which bar can be joined to the track"};

  ToolHandle<AFP::IToFLocParamDBTool> m_tofLocParamDBTool {this, "tofLocParamDBTool", "AFP__ToFLocParamDBTool", "Tool to access DB to get the local ToF parameters"};

  void fillTrainWithBars(std::vector<const xAOD::AFPToFHit*> my_trainBars[4], SG::ReadHandle<xAOD::AFPToFHitContainer>& hitContainer) const;

  void clearTrains(std::vector<const xAOD::AFPToFHit*> my_trainBars[4]) const
  {my_trainBars[0].clear(); my_trainBars[1].clear(); my_trainBars[2].clear(); my_trainBars[3].clear();}  
  
  void saveToXAOD (const AFPTDBasicToolTrack& recoTrack, std::unique_ptr<xAOD::AFPToFTrackContainer>& containerToFill, SG::ReadHandle<xAOD::AFPToFHitContainer>& hitContainer) const;
};
 
 
#endif // > !AFP_LOCRECO_AFPTDBASICKALMANTOOL_H