BookkeeperTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef EVENT_BOOKKEEPER_TOOLS__BOOKKEEPER_TOOL_H
#define EVENT_BOOKKEEPER_TOOLS__BOOKKEEPER_TOOL_H

 
 
 
#include <AsgTools/AsgMetadataTool.h>
#include <AsgTools/PropertyWrapper.h>
#include <AsgTools/SourceID.h>
 
#ifndef XAOD_STANDALONE
#include <AthenaKernel/ICutFlowSvc.h>
#include <AthenaKernel/IMetaDataTool.h>
#include <GaudiKernel/ServiceHandle.h>
#endif // XAOD_STANDALONE
 
#include <xAODCutFlow/CutBookkeeperContainer.h>
#include <xAODCutFlow/CutBookkeeperAuxContainer.h>
 
#include <EventBookkeeperTools/CutBookkeepersLocalCache.h>
#include <memory>
#include <string>
#include <unordered_map>
 
#ifndef XAOD_STANDALONE
 
class CutFlowSvc;
 
#endif // XAOD_STANDALONE
 
 
class BookkeeperTool : public asg::AsgMetadataTool
#ifndef XAOD_STANDALONE
                     , public virtual ::IMetaDataTool
#endif // XAOD_STANDALONE
{
#ifndef XAOD_STANDALONE
  ASG_TOOL_CLASS( BookkeeperTool, IMetaDataTool )
#endif
 
public: // Constructor and Destructor
  BookkeeperTool(const std::string& name = "BookkeeperTool");
  virtual ~BookkeeperTool() = default;

public:
  virtual StatusCode initialize() override;
  virtual StatusCode metaDataStop() override;
#ifdef XAOD_STANDALONE
  virtual StatusCode beginInputFile() override;
  virtual StatusCode endInputFile() override;
#else
  virtual StatusCode beginInputFile() override {return StatusCode::SUCCESS;}
  virtual StatusCode endInputFile() override {return StatusCode::SUCCESS;}
  virtual StatusCode beginInputFile(const SG::SourceID &source) override;
  virtual StatusCode endInputFile(const SG::SourceID &source) override;
#endif
 
private:
  StatusCode copyInputContainersToOutput(CutBookkeepersLocalCache &target,
                                         const SG::SourceID &source = "");

  StatusCode copyCutflowFromService();

  StatusCode loadXAODMetaData();
  StatusCode loadPOOLMetaData();
 
#ifndef XAOD_STANDALONE
  ServiceHandle<ICutFlowSvc> m_cutFlowSvc{ this, "CutFlowSvc", "CutFlowSvc/CutFlowSvc", "Pointer to the CutFlowSvc"};
  CutFlowSvc *m_cutFlowSvcPrivate{};
#endif // XAOD_STANDALONE

  Gaudi::Property<std::string> m_inputCollName{this, "InputCollName", "CutBookkeepers",
    "The default name of the xAOD::CutBookkeeperContainer for input files"};

  Gaudi::Property<std::string> m_outputCollName{this, "OutputCollName", "CutBookkeepers",
    "The default name of the xAOD::CutBookkeeperContainer for output files"};

  size_t m_numberOfWeightVariations{};

  std::unordered_map<SG::SourceID, CutBookkeepersLocalCache> m_inputContainers;

  CutBookkeepersLocalCache m_completeContainers;
  CutBookkeepersLocalCache m_incompleteContainers;
};
 
#endif // EVENT_BOOKKEEPER_TOOLS__BOOKKEEPER_TOOL_H