adios2.hpp

Go to the documentation of this file.

#ifndef ADIOS2_HPP
#define ADIOS2_HPP

#include <adios2.h>
#include <mpi.h>
#include <memory>

void ADIOS2Init(const std::string configFile, MPI_Comm &comm);

using ADIOS2Ptr = std::shared_ptr<adios2::ADIOS>;
using EnginePtr = std::shared_ptr<adios2::Engine>;
using IOPtr = std::shared_ptr<adios2::IO>;

ADIOS2Ptr GetADIOSPtr2();

// communicator for processes on the same compute node
// used only for IO 
extern MPI_Comm SML_COMM_COMPNODE;
extern int SML_COMM_RANK_COMPNODE;
extern int SML_COMM_SIZE_COMPNODE;

extern MPI_Comm SML_COMM_ONEPERNODE;
extern int SML_COMM_RANK_ONEPERNODE;
extern int SML_COMM_SIZE_ONEPERNODE;

class ADIOS2Stream
{
public:
    IOPtr io;
    EnginePtr engine;
    adios2::Mode mode;

    ADIOS2Stream();
    ADIOS2Stream(const std::string &IOName);
    ~ADIOS2Stream();

    void Init(const std::string &IOName);
    IOPtr DeclareIO(const std::string &IOName);
    void Open(const std::string &StreamName, const adios2::Mode mode);
    void Open(const std::string &StreamName, const adios2::Mode mode, const MPI_Comm &comm);
    void Close();

private:
    const std::string m_StreamName;

};

ADIOS2Stream* ADIOS2Lookup(const std::string &IOName);

#endif