TruthMetaDataWriter.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef DERIVATIONFRAMEWORK_TRUTHMETADATAWRITER_H
#define DERIVATIONFRAMEWORK_TRUTHMETADATAWRITER_H
 
#include "AthenaBaseComps/AthAlgTool.h"
#include "CxxUtils/checker_macros.h"
#include "DerivationFrameworkInterfaces/IAugmentationTool.h"
 
// Handles to services
#include "GaudiKernel/ServiceHandle.h"
 
// Service for the metadata (tag info)
#include "EventInfoMgt/ITagInfoMgr.h"
 
// EDM classes - typedefs, so have to #include them
#include "xAODTruth/TruthMetaDataContainer.h"
 
// Standard library includes
#include <string>
#include <unordered_set>
 
// Forward declarations
class IHepMCWeightSvc;
 
namespace DerivationFramework {
 
  class ATLAS_NOT_THREAD_SAFE TruthMetaDataWriter : public AthAlgTool, public IAugmentationTool {
    //  ^ meta-data handling in addBranches
    public: 
      TruthMetaDataWriter(const std::string& t, const std::string& n, const IInterface* p);
      ~TruthMetaDataWriter();
      virtual StatusCode initialize() override;
      virtual StatusCode addBranches() const override;
 
    private:
      ServiceHandle< StoreGateSvc > m_metaStore;
      ServiceHandle< IHepMCWeightSvc > m_weightSvc;
      xAOD::TruthMetaDataContainer* m_tmd = nullptr;
      std::string m_metaName;
      mutable std::unordered_set<uint32_t> m_existingMetaDataChan; 
      ServiceHandle< ITagInfoMgr > m_tagInfoMgr{
        "TagInfoMgr", name()};
 
  }; 
}
 
#endif