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
79
80
81
82
|
/*
* Distributed under the OSI-approved Apache License, Version 2.0. See
* accompanying file Copyright.txt for details.
*
* dataManReader.cpp
*
* Created on: Jan 9, 2017
* Author: Jason Wang
*/
#include <adios2.h>
#include <chrono>
#include <iostream>
#include <mpi.h>
#include <numeric>
#include <thread>
#include <vector>
int mpiRank, mpiSize;
template <class T>
void PrintData(std::vector<T> &data, size_t step)
{
std::cout << "Rank: " << mpiRank << " Step: " << step << " [";
for (size_t i = 0; i < data.size(); ++i)
{
std::cout << data[i] << " ";
}
std::cout << "]" << std::endl;
}
int main(int argc, char *argv[])
{
// initialize MPI
int provided;
// MPI_THREAD_MULTIPLE is only required if you enable the SST MPI_DP
MPI_Init_thread(&argc, &argv, MPI_THREAD_MULTIPLE, &provided);
MPI_Comm_rank(MPI_COMM_WORLD, &mpiRank);
MPI_Comm_size(MPI_COMM_WORLD, &mpiSize);
// initialize adios2
adios2::ADIOS adios(MPI_COMM_WORLD);
adios2::IO io = adios.DeclareIO("whatever");
io.SetEngine("DataMan");
io.SetParameters({{"IPAddress", "127.0.0.1"}, {"Port", "12306"}, {"Timeout", "5"}});
// open stream
adios2::Engine engine = io.Open("HelloDataMan", adios2::Mode::Read);
// define variable
adios2::Variable<float> floatArrayVar;
// read data
std::vector<float> floatVector;
while (true)
{
auto status = engine.BeginStep();
if (status == adios2::StepStatus::OK)
{
floatArrayVar = io.InquireVariable<float>("FloatArray");
auto shape = floatArrayVar.Shape();
size_t datasize =
std::accumulate(shape.begin(), shape.end(), 1, std::multiplies<size_t>());
floatVector.resize(datasize);
engine.Get<float>(floatArrayVar, floatVector.data(), adios2::Mode::Sync);
engine.EndStep();
PrintData(floatVector, engine.CurrentStep());
}
else if (status == adios2::StepStatus::EndOfStream)
{
std::cout << "End of stream" << std::endl;
break;
}
}
// clean up
engine.Close();
MPI_Finalize();
return 0;
}
|
