EMBremCollectionBuilder.h

Go to the documentation of this file.

/*
 Copyright (C) 2002-2021 CERN for the benefit of the ATLAS collaboration
 */
 
#ifndef EGAMMAALGS_EMBREMCOLLECTIONBUILDER_H
#define EGAMMAALGS_EMBREMCOLLECTIONBUILDER_H
 
#include "TrkToolInterfaces/ITrackParticleCreatorTool.h"
#include "TrkToolInterfaces/ITrackSlimmingTool.h"
#include "TrkToolInterfaces/ITrackSummaryTool.h"
#include "egammaInterfaces/IEMExtrapolationTools.h"
#include "egammaInterfaces/IegammaTrkRefitterTool.h"
 
#include "AthenaBaseComps/AthReentrantAlgorithm.h"
#include "GaudiKernel/EventContext.h"
#include "GaudiKernel/ToolHandle.h"
#include "StoreGate/ReadDecorHandleKey.h"
#include "StoreGate/ReadHandleKey.h"
#include "StoreGate/WriteHandleKey.h"
#include "TrkTrack/TrackCollection.h"
 
#include "InDetReadoutGeometry/SiDetectorElementCollection.h"
#include "TrkTrack/Track.h"
#include "xAODTracking/TrackParticleContainerFwd.h"
#include "xAODTracking/TrackParticleFwd.h"
 
#include <memory>

class EMBremCollectionBuilder : public AthReentrantAlgorithm
{
 
public:
  EMBremCollectionBuilder(const std::string& name, ISvcLocator* pSvcLocator);
 
  virtual StatusCode initialize() override final;
  virtual StatusCode finalize() override final;
  virtual StatusCode execute(const EventContext& ctx) const override final;

  struct TrackWithIndex
  {
 
    TrackWithIndex(std::unique_ptr<Trk::Track> intrack, size_t index)
      : track(std::move(intrack))
      , origIndex(index)
    {}
 
    TrackWithIndex(const TrackWithIndex&) = delete;
    TrackWithIndex(TrackWithIndex&&) = default;
    TrackWithIndex& operator=(const TrackWithIndex&) = delete;
    TrackWithIndex& operator=(TrackWithIndex&&) = default;
    ~TrackWithIndex() = default;
 
    std::unique_ptr<Trk::Track> track;
    size_t origIndex;
  };
 
private:
  StatusCode refitTracks(const EventContext& ctx,
                         const std::vector<const xAOD::TrackParticle*>& input,
                         std::vector<TrackWithIndex>& refitted,
                         std::vector<TrackWithIndex>& failedfit) const;
 
  StatusCode createCollections(
    const EventContext& ctx,
    std::vector<TrackWithIndex>& refitted,
    std::vector<TrackWithIndex>& failedfit,
    std::vector<TrackWithIndex>& trtAlone,
    TrackCollection* finalTracks,
    xAOD::TrackParticleContainer* finalTrkPartContainer,
    const xAOD::TrackParticleContainer* inputTrkPartContainer) const;
 
  StatusCode createNew(
    const EventContext& ctx,
    TrackWithIndex& Info,
    TrackCollection* finalTracks,
    xAOD::TrackParticleContainer* finalTrkPartContainer,
    const xAOD::TrackParticleContainer* inputTrkPartContainer,
    bool isRefitted) const;
 
  void copyOverInfo(xAOD::TrackParticle& created,
                    const xAOD::TrackParticle& original,
                    bool isRefitted) const;

  Gaudi::Property<bool> m_doTruth{ this, "DoTruth", false, "do truth" };

  Gaudi::Property<bool> m_doSCT{ this, "useSCT", true, "do SCT" };

  Gaudi::Property<bool> m_doPix{ this, "usePixel", true, "do Pix" };

  Gaudi::Property<bool> m_doTRT{ this, "useTRT", true, "do TRT" };

  Gaudi::Property<bool> m_doSlimTrkTracks{ this,
                                           "slimTrkTracks",
                                           true,
                                           "Slim the GSF Trk::Tracks" };

  Gaudi::Property<bool> m_doHGTD{ this, "useHGTD", false, "do HGTD" };

  Gaudi::Property<int> m_MinNoSiHits{ this,
                                      "minNoSiHits",
                                      4,
                                      "Minimum number of silicon hits on track "
                                      "before it is allowed to be refitted" };

  ToolHandle<IegammaTrkRefitterTool> m_trkRefitTool{ this,
                                                     "TrackRefitTool",
                                                     "ElectronRefitterTool",
                                                     "Track refitter tool" };

  ToolHandle<Trk::ITrackParticleCreatorTool> m_particleCreatorTool{
    this,
    "TrackParticleCreatorTool",
    "TrackParticleCreatorTool",
    "TrackParticle creator tool"
  };

  ToolHandle<Trk::ITrackSlimmingTool> m_slimTool{ this,
                                                  "TrackSlimmingTool",
                                                  "TrkTrackSlimmingTool",
                                                  "Track slimming tool" };
 
  SG::ReadHandleKey<xAOD::TrackParticleContainer> m_trackParticleContainerKey{
    this,
    "TrackParticleContainerName",
    "InDetTrackParticles",
    "Input InDet TrackParticles"
  };

  SG::ReadHandleKey<xAOD::TrackParticleContainer>
    m_selectedTrackParticleContainerKey{ this,
                                         "SelectedTrackParticleContainerName",
                                         "egammaSelectedTrackParticles",
                                         "Input of Selected TrackParticles" };
 
  SG::WriteHandleKey<xAOD::TrackParticleContainer> m_OutputTrkPartContainerKey{
    this,
    "OutputTrkPartContainerName",
    "GSFTrackParticles",
    "Output GSF TrackParticles"
  };
 
  SG::WriteHandleKey<TrackCollection> m_OutputTrackContainerKey{
    this,
    "OutputTrackContainerName",
    "GSFTracks",
    "Output GSF Trk::Tracks"
  };
 
  // counters for final statistics
  mutable std::atomic_uint m_FailedFitTracks{ 0 };
  mutable std::atomic_uint m_RefittedTracks{ 0 };
  mutable std::atomic_uint m_FailedSiliconRequirFit{ 0 };
};
#endif //