helloDataManWriter.cpp

/*
 * Distributed under the OSI-approved Apache License, Version 2.0.  See
 * accompanying file Copyright.txt for details.
 *
 * helloWriter.cpp
 *
 *  Created on: Feb 16, 2017
 *      Author: wfg
 */

#include <iostream>
#include <vector>

#include <mpi.h>

#include "ADIOS_CPP.h"

int main(int argc, char *argv[])
{
    MPI_Init(&argc, &argv);
    int rank;
    MPI_Comm_rank(MPI_COMM_WORLD, &rank);
    const bool adiosDebug = true;
    adios::ADIOS adios(MPI_COMM_WORLD, adios::Verbose::WARN, adiosDebug);

    // Application variable
    std::vector<double> myDoubles = {0, 1, 2, 3, 4, 5, 6, 7, 8, 9};
    const std::size_t Nx = myDoubles.size();

    std::vector<std::complex<float>> myCFloats;
    myCFloats.reserve(3);
    myCFloats.emplace_back(1, 3);
    myCFloats.emplace_back(2, 2);
    myCFloats.emplace_back(3, 1);

    try
    {
        // Define variable and local size
        auto ioMyDoubles = adios.DefineVariable<double>("myDoubles", {Nx});
        auto ioMyCFloats =
            adios.DefineVariable<std::complex<float>>("myCFloats", {3});

        // Define method for engine creation, it is basically straight-forward
        // parameters
        adios::Method &datamanSettings = adios.DeclareMethod("WAN");
        if (!datamanSettings.isUserDefined())
        {
            // if not defined by user, we can change the default settings
            datamanSettings.SetEngine("DataManWriter");
            datamanSettings.SetParameters("peer-to-peer=yes", "real_time=yes",
                                          "compress=no");
            datamanSettings.AddTransport("Mdtm", "localIP=128.0.0.0.1",
                                         "remoteIP=128.0.0.0.2",
                                         "tolerances=1,2,3");
            // datamanSettings.AddTransport( "file", "name=myfile.bp",
            // "remoteIP=128.0.0.0.2", "tolerances=1,2,3" );
            // datamanSettings.AddTransport( "file", "name=myfile.bp",
            // "remoteIP=128.0.0.0.2", "tolerances=1,2,3" );
            // datamanSettings.AddTransport( "Mdtm", "localIP=128.0.0.0.1",
            // "remoteIP=128.0.0.0.2", "tolerances=1,2,3" );
            // datamanSettings.AddTransport( "ZeroMQ", "localIP=128.0.0.0.1",
            // "remoteIP=128.0.0.0.2", "tolerances=1,2,3" );
            // datamanSettings.AddTransport( "ZeroMQ", "localIP=128.0.0.0.1.1",
            // "remoteIP=128.0.0.0.2.1", "tolerances=1,2,3" ); not yet
            // supported, will
            // throw an exception
        }

        // Create engine smart pointer to DataMan Engine due to polymorphism,
        // Open returns a smart pointer to Engine containing the Derived class
        // DataMan

        // adios::DataManWriter datamanWriter;

        auto datamanWriter = adios.Open("myDoubles.bp", "w", datamanSettings,
                                        adios::IOMode::INDEPENDENT);

        if (datamanWriter == nullptr)
            throw std::ios_base::failure(
                "ERROR: failed to create DataMan I/O engine at Open\n");

        datamanWriter->Write<double>(
            ioMyDoubles,
            myDoubles.data()); // Base class Engine own the Write<T>
                               // that will call overloaded Write from
                               // Derived
        datamanWriter->Write<std::complex<float>>(ioMyCFloats,
                                                  myCFloats.data());
        datamanWriter->Close();
    }
    catch (std::invalid_argument &e)
    {
        if (rank == 0)
        {
            std::cout << "Invalid argument exception, STOPPING PROGRAM\n";
            std::cout << e.what() << "\n";
        }
    }
    catch (std::ios_base::failure &e)
    {
        if (rank == 0)
        {
            std::cout << "System exception, STOPPING PROGRAM\n";
            std::cout << e.what() << "\n";
        }
    }
    catch (std::exception &e)
    {
        if (rank == 0)
        {
            std::cout << "Exception, STOPPING PROGRAM\n";
            std::cout << e.what() << "\n";
        }
    }

    MPI_Finalize();

    return 0;
}