Skip to content
Snippets Groups Projects
HDF5WriterP.cpp 1.78 KiB
Newer Older
guj's avatar
guj committed
/*
guj's avatar
guj committed
 * Distributed under the OSI-approved Apache License, Version 2.0.  See
 * accompanying file Copyright.txt for details.
 *
guj's avatar
guj committed
 * HDF5WriterP.cpp
 *
guj's avatar
guj committed
 *  Created on: March 20, 2017
 *      Author: Junmin
guj's avatar
guj committed
 */

#include "HDF5WriterP.h"
guj's avatar
guj committed

#include "adios2/ADIOSMPI.h"
#include "adios2/helper/adiosFunctions.h" //CSVToVector
guj's avatar
guj committed

namespace adios2
guj's avatar
guj committed
{
guj's avatar
guj committed

HDF5WriterP::HDF5WriterP(IO &io, const std::string &name,
                         const Mode openMode, MPI_Comm mpiComm)
: Engine("HDF5Writer", io, name, openMode, mpiComm), m_H5File(io.m_DebugMode)
guj's avatar
guj committed
{
    m_EndMessage = ", in call to IO HDF5Writer Open " + m_Name + "\n";
guj's avatar
guj committed
    Init();
guj's avatar
guj committed
}

HDF5WriterP::~HDF5WriterP() { Close(); }
guj's avatar
guj committed

// PRIVATE
void HDF5WriterP::Init()
guj's avatar
guj committed
{
    if (m_OpenMode != Mode::Write && m_OpenMode != Mode::Append)
guj's avatar
guj committed
    {
        throw std::invalid_argument(
William F Godoy's avatar
William F Godoy committed
            "ERROR: HDF5Writer only support OpenMode::Write or "
            "OpenMode::Append "
guj's avatar
guj committed
            ", in call to ADIOS Open or HDF5Writer constructor\n");
    }
guj's avatar
guj committed

    m_H5File.Init(m_Name, m_MPIComm, true);
guj's avatar
guj committed
}

#define declare_type(T)                                                        \
    void HDF5WriterP::DoWrite(Variable<T> &variable, const T *values)          \
    {                                                                          \
        DoWriteCommon(variable, values);                                       \
    }
guj's avatar
guj committed

ADIOS2_FOREACH_TYPE_1ARG(declare_type)
#undef declare_type
guj's avatar
guj committed

void HDF5WriterP::Advance(const float timeoutSeconds) { m_H5File.Advance(); }
guj's avatar
guj committed

void HDF5WriterP::Close(const int transportIndex) { m_H5File.Close(); }
guj's avatar
guj committed

template <class T>
void HDF5WriterP::DoWriteCommon(Variable<T> &variable, const T *values)
guj's avatar
guj committed
{
    variable.m_AppValues = values;
    m_WrittenVariables.insert(variable.m_Name);
    m_H5File.Write(variable, values);
guj's avatar
guj committed
}

} // end namespace adios