File: CppReader.cpp

package info (click to toggle)
adios2 2.10.2%2Bdfsg1-3
  • links: PTS, VCS
  • area: main
  • in suites: sid
  • size: 33,764 kB
  • sloc: cpp: 175,964; ansic: 160,510; f90: 14,630; yacc: 12,668; python: 7,275; perl: 7,126; sh: 2,825; lisp: 1,106; xml: 1,049; makefile: 579; lex: 557
file content (78 lines) | stat: -rw-r--r-- 2,161 bytes parent folder | download | duplicates (2)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
/*
 * Distributed under the OSI-approved Apache License, Version 2.0.  See
 * accompanying file Copyright.txt for details.
 *
 * CppReader.cpp
 *
 *  Created on: Jan 26, 2018
 *      Author: William F Godoy godoywf@ornl.gov
 */

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

#include <iostream>

int main(int argc, char *argv[])
{
    int provided;

    // MPI_THREAD_MULTIPLE is only required if you enable the SST MPI_DP
    MPI_Init_thread(&argc, &argv, MPI_THREAD_MULTIPLE, &provided);
    int rank, size;
    MPI_Comm_rank(MPI_COMM_WORLD, &rank);
    MPI_Comm_size(MPI_COMM_WORLD, &size);

    if (rank == 0)
    {
        /** ADIOS class factory of IO class objects */
        adios2::ADIOS adios(MPI_COMM_SELF);

        /*** IO class object: settings and factory of Settings: Variables,
         * Parameters, Transports, and Execution: Engines */
        adios2::IO bpIO = adios.DeclareIO("CppReader");

        /** Engine derived class, spawned to start IO operations */
        adios2::Engine bpReader = bpIO.Open("FWriter.bp", adios2::Mode::Read);

        bpReader.BeginStep();

        const std::map<std::string, adios2::Params> variables = bpIO.AvailableVariables();

        for (const auto &variablePair : variables)
        {
            std::cout << "Name: " << variablePair.first;

            for (const auto &parameter : variablePair.second)
            {
                std::cout << "\t" << parameter.first << ": " << parameter.second << "\n";
            }
        }

        auto bpData = bpIO.InquireVariable<float>("data2D");
        bpData.SetSelection({{1, 1}, {3, 2}});
        std::vector<float> data(bpData.SelectionSize());

        bpReader.Get(bpData, data.data());

        bpReader.EndStep();

        std::cout << "Selection size: " << bpData.SelectionSize() << "\n";
        for (size_t i = 0; i < bpData.Count()[0]; ++i)
        {
            for (size_t j = 0; j < bpData.Count()[1]; ++j)
            {
                std::cout << data[i * bpData.Count()[1] + j] << " ";
            }

            std::cout << "\n";
        }
        std::cout << "\n";

        bpReader.Close();
    }

    MPI_Finalize();

    return 0;
}