TruthSvc.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef ISF_SERVICES_TRUTHSVC_H
#define ISF_SERVICES_TRUTHSVC_H 1
 
// STL includes
#include <string>
 
// FrameWork includes
#include "GaudiKernel/ToolHandle.h"
#include "GaudiKernel/ServiceHandle.h"
#include "AthenaBaseComps/AthService.h"
 
// ISF include
#include "ISF_Interfaces/ITruthSvc.h"
#include "ISF_HepMC_Interfaces/ITruthStrategy.h"
#include "ISF_Event/ITruthIncident.h"
 
// DetectorDescription
#include "AtlasDetDescr/AtlasRegion.h"
 
// Barcode
#include "BarcodeInterfaces/IBarcodeSvc.h"
 
// McEventCollection
#include "GeneratorObjects/McEventCollection.h"
 
// forward declarations
class StoreGateSvc;
 
#include "AtlasHepMC/GenEvent_fwd.h"
#include "AtlasHepMC/GenVertex.h"
 
namespace ISFTesting {
  class TruthSvc_test;
}
 
namespace ISF {

  class TruthSvc final : public extends<AthService, ITruthSvc> {
 
    // allow test to access private data
    friend ISFTesting::TruthSvc_test;
 
  public:
 
    //** Constructor with parameters */
    TruthSvc( const std::string& name, ISvcLocator* pSvcLocator );

    virtual ~TruthSvc() = default;

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

    void registerTruthIncident( ITruthIncident& truthincident, bool saveAllChildren=false) const override;

    StatusCode initializeTruthCollection(int largestGeneratedParticleBC=0, int largestGeneratedVertexBC=0) override;

    StatusCode releaseEvent() override;
 
  private:
    void recordIncidentToMCTruth( ITruthIncident& truthincident, bool passWholeVertex) const;
    HepMC::GenVertexPtr  createGenVertexFromTruthIncident( ITruthIncident& truthincident) const;
 
    ServiceHandle<Barcode::IBarcodeSvc> m_barcodeSvc{this, "BarcodeSvc", "BarcodeSvc", ""};           

    ToolHandleArray<ITruthStrategy> m_truthStrategies{this, "TruthStrategies", {}, ""};
    ITruthStrategy**              m_geoStrategies[AtlasDetDescr::fNumAtlasRegions];
    unsigned short                 m_numStrategies[AtlasDetDescr::fNumAtlasRegions];

    Gaudi::Property<bool>   m_skipIfNoChildren{this, "SkipIfNoChildren", true, ""};       
    Gaudi::Property<bool>   m_skipIfNoParentId{this, "SkipIfNoParentId", true, ""};  
    Gaudi::Property<bool>   m_ignoreUndefinedBarcodes{this, "IgnoreUndefinedBarcodes",  false, ""};
 
    Gaudi::Property<bool>   m_passWholeVertex{this, "PassWholeVertices", true, ""};
 
    Gaudi::Property<std::vector<unsigned int> >  m_forceEndVtxRegionsVec{this, "ForceEndVtxInRegions", {}, ""}; 
    //   to write end-vtx
    std::array<bool, AtlasDetDescr::fNumAtlasRegions> m_forceEndVtx{}; 
    //   all parent particles if they die
 
    Gaudi::Property<bool>   m_quasiStableParticlesIncluded{this, "QuasiStableParticlesIncluded", false, ""}; 
    Gaudi::Property<bool>   m_quasiStableParticleOverwrite{this, "QuasiStableParticleOverwrite", true, ""}; 
 
  };
}
 
 
#endif //> !ISF_SERVICES_TRUTHSVC_H