ZeroLifetimePositioner.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2021 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef BEAMEFFECTS_ZEROLIFETIMEPOSITIONER_H
#define BEAMEFFECTS_ZEROLIFETIMEPOSITIONER_H 1
 
// FrameWork includes
#include "GaudiKernel/ToolHandle.h"
#include "AthenaBaseComps/AthService.h"
#include "HepMC_Interfaces/IZeroLifetimePatcher.h"
#include <vector>
 
namespace Simulation {

  class ZeroLifetimePositioner : public extends<AthService, Simulation::IZeroLifetimePatcher> {
 
    public:
      ZeroLifetimePositioner( const std::string& name, ISvcLocator* pSvcLocator );

      StatusCode  initialize() override final;
      StatusCode  finalize() override final;

      virtual StatusCode applyWorkaround(HepMC::GenEvent& ge) const override final;

      virtual StatusCode removeWorkaround(HepMC::GenEvent& ge) const override final;
 
    private:
      StatusCode manipulate(HepMC::GenEvent& ge, bool applyPatch, bool removePatch) const;
      Gaudi::Property<bool> m_applyPatch{this, "ApplyPatch", false};
      Gaudi::Property<bool> m_removePatch{this, "RemovePatch", false};
      Gaudi::Property<std::vector<int>> m_pdgCodesToCheck{ this, "PDGCodesToCheck", {421,511,531} }; // Apply fix to neutral mesons with oscillations: B0, B0s, D0
  };
 
}
 
#endif //> !BEAMEFFECTS_ZEROLIFETIMEPOSITIONER_H