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
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
|
/*
* Distributed under the OSI-approved Apache License, Version 2.0. See
* accompanying file Copyright.txt for details.
*
* py11Variable.cpp :
*
* Created on: Dec 11, 2018
* Author: William F Godoy godoywf@ornl.gov
*/
#include "py11Attribute.h"
#include "py11types.h"
#include <string.h> //std::memcpy
#include "adios2/helper/adiosFunctions.h"
namespace adios2
{
namespace py11
{
Attribute::Attribute(core::AttributeBase *attribute) : m_Attribute(attribute) {}
Attribute::operator bool() const noexcept { return (m_Attribute == nullptr) ? false : true; }
std::string Attribute::Name() const
{
helper::CheckForNullptr(m_Attribute, "in call to Attribute::Name");
return m_Attribute->m_Name;
}
std::string Attribute::Type() const
{
helper::CheckForNullptr(m_Attribute, "in call to Attribute::Type");
return ToString(m_Attribute->m_Type);
}
bool Attribute::SingleValue() const
{
helper::CheckForNullptr(m_Attribute, "in call to Attribute::SingleValue");
return m_Attribute->m_IsSingleValue;
}
std::vector<std::string> Attribute::DataString()
{
helper::CheckForNullptr(m_Attribute, "in call to Attribute::DataStrings");
const adios2::DataType type = m_Attribute->m_Type;
std::vector<std::string> data;
if (type == helper::GetDataType<std::string>())
{
const core::Attribute<std::string> *attribute =
dynamic_cast<core::Attribute<std::string> *>(m_Attribute);
data.reserve(attribute->m_Elements);
if (attribute->m_IsSingleValue)
{
data.push_back(attribute->m_DataSingleValue);
}
else
{
data = attribute->m_DataArray;
}
}
else
{
throw std::invalid_argument("ERROR: data type for attribute " + m_Attribute->m_Name +
" is not string, in call to Attribute::DataStrings\n");
}
return data;
}
pybind11::array Attribute::Data()
{
helper::CheckForNullptr(m_Attribute, "in call to Attribute::Data");
const adios2::DataType type = m_Attribute->m_Type;
if (type == adios2::DataType::Struct)
{
// not supported
}
#define declare_type(T) \
else if (type == helper::GetDataType<T>()) \
{ \
pybind11::array pyArray(pybind11::dtype::of<T>(), m_Attribute->m_Elements); \
if (m_Attribute->m_IsSingleValue) \
{ \
const T value = dynamic_cast<core::Attribute<T> *>(m_Attribute)->m_DataSingleValue; \
std::memcpy(const_cast<void *>(pyArray.data()), &value, sizeof(T)); \
} \
else \
{ \
const std::vector<T> &values = \
dynamic_cast<core::Attribute<T> *>(m_Attribute)->m_DataArray; \
std::memcpy(const_cast<void *>(pyArray.data()), values.data(), \
sizeof(T) * m_Attribute->m_Elements); \
} \
return pyArray; \
}
ADIOS2_FOREACH_NUMPY_ATTRIBUTE_TYPE_1ARG(declare_type)
#undef declare_type
return pybind11::array();
}
} // end namespace py11
} // end namespace adios2
|
