AlgorithmMemoryWrapper.h

Go to the documentation of this file.

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

 
 
 
#ifndef EVENT_LOOP__ALGORITHM_MEMORY_WRAPPER_H
#define EVENT_LOOP__ALGORITHM_MEMORY_WRAPPER_H
 
#include <EventLoop/Global.h>
 
#include <AnaAlgorithm/IAlgorithmWrapper.h>
#include <Rtypes.h>
#include <chrono>
 
namespace EL
{
 
  class AlgorithmMemoryWrapper final : public IAlgorithmWrapper
  {
 
  public:

    using clock_type = std::chrono::high_resolution_clock;

    void testInvariant () const;

    AlgorithmMemoryWrapper () {};

    AlgorithmMemoryWrapper (std::unique_ptr<IAlgorithmWrapper>&& val_algorithm);
 
 

 
  public:
 
    virtual std::string_view getName () const override;
 
    virtual bool hasName (const std::string& name) const override;
 
    virtual std::unique_ptr<IAlgorithmWrapper> makeClone() const override;
 
    virtual Algorithm *getLegacyAlg () override;
 
    virtual StatusCode initialize (const AlgorithmWorkerData& workerData) override;
 
    virtual StatusCode execute () override;
 
    virtual StatusCode postExecute () override;
 
    virtual StatusCode finalize () override;
 
    virtual StatusCode fileExecute () override;
 
    virtual StatusCode beginInputFile () override;
 
    virtual StatusCode endInputFile () override;
 
 

 
  private:

    std::unique_ptr<IAlgorithmWrapper> m_algorithm;

    std::vector<Long_t> m_mem_resident;
    std::vector<Long_t> m_mem_virtual;
 
    Long_t m_preMem_resident = 0;
    Long_t m_preMem_virtual = 0;
 
    StatusCode recordPreMemory();
    StatusCode recordPostMemory();
  };
}
 
#endif