LoggingBufferHandler.h

Go to the documentation of this file.

#ifndef IPPL_LOGGING_BUFFER_HANDLER_H
#define IPPL_LOGGING_BUFFER_HANDLER_H
 
#include <chrono>
#include <map>
#include <memory>
#include <string>
#include <vector>
 
#include "Communicate/BufferHandler.h"
#include "Communicate/LogEntry.h"
 
namespace ippl {

    template <typename MemorySpace>
    class LoggingBufferHandler : public BufferHandler<MemorySpace> {
    public:
        using buffer_type = typename BufferHandler<MemorySpace>::buffer_type;
        using size_type   = typename BufferHandler<MemorySpace>::size_type;

        LoggingBufferHandler(std::shared_ptr<BufferHandler<MemorySpace>> handler, int rank);

        LoggingBufferHandler();

        buffer_type getBuffer(size_type size, double overallocation) override;

        void freeBuffer(buffer_type buffer) override;

        void freeAllBuffers() override;

        void deleteAllBuffers() override;

        size_type getUsedSize() const override;

        size_type getFreeSize() const override;

        const std::vector<LogEntry>& getLogs() const;
 
    private:
        std::shared_ptr<BufferHandler<MemorySpace>>
            handler_m;                       
        std::vector<LogEntry> logEntries_m;  
        int rank_m;                          

        void logMethod(const std::string& methodName,
                       const std::map<std::string, std::string>& parameters);
    };
}  // namespace ippl
 
#include "Communicate/LoggingBufferHandler.hpp"
 
#endif