Skip to content

Commit 4056def

Browse files
franzpoeschelfranzpoeschel
committed
Use magic number instead of API call
1 parent 74a83c3 commit 4056def

File tree

6 files changed

+95
-54
lines changed

6 files changed

+95
-54
lines changed

‎include/openPMD/Dataset.hpp‎

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,7 @@
2222

2323
#include "openPMD/Datatype.hpp"
2424

25+
#include <limits>
2526
#include <memory>
2627
#include <optional>
2728
#include <string>
@@ -38,6 +39,11 @@ class Dataset
3839
friend class RecordComponent;
3940

4041
public:
42+
enum : std::uint64_t
43+
{
44+
JOINED_DIMENSION = std::numeric_limits<std::uint64_t>::max()
45+
};
46+
4147
Dataset(Datatype, Extent, std::string options = "{}");
4248

4349
/**
@@ -53,9 +59,10 @@ class Dataset
5359
Extent extent;
5460
Datatype dtype;
5561
uint8_t rank;
56-
std::optional<size_t> joinedDimension;
5762
std::string options = "{}"; //!< backend-dependent JSON configuration
5863

5964
bool empty() const;
65+
66+
std::optional<size_t> joinedDimension() const;
6067
};
6168
} // namespace openPMD

‎src/Dataset.cpp‎

Lines changed: 32 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@
1919
* If not, see <http://www.gnu.org/licenses/>.
2020
*/
2121
#include "openPMD/Dataset.hpp"
22+
#include "openPMD/Error.hpp"
2223

2324
#include <cstddef>
2425
#include <iostream>
@@ -30,7 +31,11 @@ Dataset::Dataset(Datatype d, Extent e, std::string options_in)
3031
, dtype{d}
3132
, rank{static_cast<uint8_t>(e.size())}
3233
, options{std::move(options_in)}
33-
{}
34+
{
35+
// Call this in order to have early error message in case of wrong
36+
// specification of joined dimensions
37+
joinedDimension();
38+
}
3439

3540
Dataset::Dataset(Extent e) : Dataset(Datatype::UNDEFINED, std::move(e))
3641
{}
@@ -52,14 +57,38 @@ Dataset &Dataset::extend(Extent newExtents)
5257

5358
bool Dataset::empty() const
5459
{
60+
auto jd = joinedDimension();
5561
for (size_t i = 0; i < extent.size(); ++i)
5662
{
57-
if (extent[i] == 0 &&
58-
(!joinedDimension.has_value() || joinedDimension.value() != i))
63+
if (extent[i] == 0 && (!jd.has_value() || jd.value() != i))
5964
{
6065
return true;
6166
}
6267
}
6368
return false;
6469
}
70+
71+
std::optional<size_t> Dataset::joinedDimension() const
72+
{
73+
std::optional<size_t> res;
74+
for (size_t i = 0; i < extent.size(); ++i)
75+
{
76+
if (extent[i] == JOINED_DIMENSION)
77+
{
78+
if (res.has_value())
79+
{
80+
throw error::WrongAPIUsage(
81+
"Must specify JOINED_DIMENSION at most once (found at "
82+
"indices " +
83+
std::to_string(res.value()) + " and " + std::to_string(i) +
84+
")");
85+
}
86+
else
87+
{
88+
res = i;
89+
}
90+
}
91+
}
92+
return res;
93+
}
6594
} // namespace openPMD

‎src/backend/BaseRecordComponent.cpp‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -70,7 +70,7 @@ std::optional<size_t> BaseRecordComponent::joinedDimension() const
7070
auto &rc = get();
7171
if (rc.m_dataset.has_value())
7272
{
73-
return rc.m_dataset.value().joinedDimension;
73+
return rc.m_dataset.value().joinedDimension();
7474
}
7575
else
7676
{

‎src/binding/python/Dataset.cpp‎

Lines changed: 44 additions & 37 deletions
Original file line numberDiff line numberDiff line change
@@ -31,43 +31,50 @@ using namespace openPMD;
3131

3232
void init_Dataset(py::module &m)
3333
{
34-
py::class_<Dataset>(m, "Dataset")
34+
auto pyDataset =
35+
py::class_<Dataset>(m, "Dataset")
36+
.def(
37+
py::init<Datatype, Extent>(),
38+
py::arg("dtype"),
39+
py::arg("extent"))
40+
.def(py::init<Extent>(), py::arg("extent"))
41+
.def(
42+
py::init([](py::dtype dt, Extent e) {
43+
auto const d = dtype_from_numpy(dt);
44+
return new Dataset{d, e};
45+
}),
46+
py::arg("dtype"),
47+
py::arg("extent"))
48+
.def(
49+
py::init<Datatype, Extent, std::string>(),
50+
py::arg("dtype"),
51+
py::arg("extent"),
52+
py::arg("options"))
53+
.def(
54+
py::init([](py::dtype dt, Extent e, std::string options) {
55+
auto const d = dtype_from_numpy(dt);
56+
return new Dataset{d, e, std::move(options)};
57+
}),
58+
py::arg("dtype"),
59+
py::arg("extent"),
60+
py::arg("options"))
3561

36-
.def(py::init<Datatype, Extent>(), py::arg("dtype"), py::arg("extent"))
37-
.def(py::init<Extent>(), py::arg("extent"))
38-
.def(
39-
py::init([](py::dtype dt, Extent e) {
40-
auto const d = dtype_from_numpy(dt);
41-
return new Dataset{d, e};
42-
}),
43-
py::arg("dtype"),
44-
py::arg("extent"))
45-
.def(
46-
py::init<Datatype, Extent, std::string>(),
47-
py::arg("dtype"),
48-
py::arg("extent"),
49-
py::arg("options"))
50-
.def(
51-
py::init([](py::dtype dt, Extent e, std::string options) {
52-
auto const d = dtype_from_numpy(dt);
53-
return new Dataset{d, e, std::move(options)};
54-
}),
55-
py::arg("dtype"),
56-
py::arg("extent"),
57-
py::arg("options"))
62+
.def(
63+
"__repr__",
64+
[](const Dataset &d) {
65+
return "<openPMD.Dataset of rank '" +
66+
std::to_string(d.rank) + "'>";
67+
})
5868

59-
.def(
60-
"__repr__",
61-
[](const Dataset &d) {
62-
return "<openPMD.Dataset of rank '" + std::to_string(d.rank) +
63-
"'>";
64-
})
65-
66-
.def_readonly("extent", &Dataset::extent)
67-
.def("extend", &Dataset::extend)
68-
.def_readonly("rank", &Dataset::rank)
69-
.def_property_readonly(
70-
"dtype", [](const Dataset &d) { return dtype_to_numpy(d.dtype); })
71-
.def_readwrite("options", &Dataset::options)
72-
.def_readwrite("joined_dimension", &Dataset::joinedDimension);
69+
.def_property_readonly(
70+
"joined_dimension", &Dataset::joinedDimension)
71+
.def_readonly("extent", &Dataset::extent)
72+
.def("extend", &Dataset::extend)
73+
.def_readonly("rank", &Dataset::rank)
74+
.def_property_readonly(
75+
"dtype",
76+
[](const Dataset &d) { return dtype_to_numpy(d.dtype); })
77+
.def_readwrite("options", &Dataset::options);
78+
pyDataset.attr("JOINED_DIMENSION") =
79+
py::int_(uint64_t(Dataset::JOINED_DIMENSION));
7380
}

‎test/ParallelIOTest.cpp‎

Lines changed: 5 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1887,8 +1887,8 @@ void joined_dim(std::string const &ext)
18871887

18881888
auto it = s.writeIterations()[100];
18891889

1890-
Dataset numParticlesDS(determineDatatype<patchType>(), {0});
1891-
numParticlesDS.joinedDimension = 0;
1890+
Dataset numParticlesDS(
1891+
determineDatatype<patchType>(), {Dataset::JOINED_DIMENSION});
18921892
auto numParticles =
18931893
it.particles["e"]
18941894
.particlePatches["numParticles"][RecordComponent::SCALAR];
@@ -1900,8 +1900,8 @@ void joined_dim(std::string const &ext)
19001900

19011901
auto patchOffset = it.particles["e"].particlePatches["offset"]["x"];
19021902
auto patchExtent = it.particles["e"].particlePatches["extent"]["x"];
1903-
Dataset particlePatchesDS(determineDatatype<float>(), {0});
1904-
particlePatchesDS.joinedDimension = 0;
1903+
Dataset particlePatchesDS(
1904+
determineDatatype<float>(), {Dataset::JOINED_DIMENSION});
19051905
patchOffset.resetDataset(particlePatchesDS);
19061906
patchExtent.resetDataset(particlePatchesDS);
19071907

@@ -1919,8 +1919,7 @@ void joined_dim(std::string const &ext)
19191919
}
19201920

19211921
auto epx = it.particles["e"]["position"]["x"];
1922-
Dataset ds(determineDatatype<type>(), {1});
1923-
ds.joinedDimension = 0;
1922+
Dataset ds(determineDatatype<type>(), {Dataset::JOINED_DIMENSION});
19241923
epx.resetDataset(ds);
19251924

19261925
size_t counter = 0;

‎test/SerialIOTest.cpp‎

Lines changed: 5 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -7263,8 +7263,8 @@ void joined_dim(std::string const &ext)
72637263

72647264
auto it = s.writeIterations()[100];
72657265

7266-
Dataset numParticlesDS(determineDatatype<patchType>(), {0});
7267-
numParticlesDS.joinedDimension = 0;
7266+
Dataset numParticlesDS(
7267+
determineDatatype<patchType>(), {Dataset::JOINED_DIMENSION});
72687268
auto numParticles =
72697269
it.particles["e"]
72707270
.particlePatches["numParticles"][RecordComponent::SCALAR];
@@ -7276,8 +7276,8 @@ void joined_dim(std::string const &ext)
72767276

72777277
auto patchOffset = it.particles["e"].particlePatches["offset"]["x"];
72787278
auto patchExtent = it.particles["e"].particlePatches["extent"]["x"];
7279-
Dataset particlePatchesDS(determineDatatype<float>(), {0});
7280-
particlePatchesDS.joinedDimension = 0;
7279+
Dataset particlePatchesDS(
7280+
determineDatatype<float>(), {Dataset::JOINED_DIMENSION});
72817281
patchOffset.resetDataset(particlePatchesDS);
72827282
patchExtent.resetDataset(particlePatchesDS);
72837283

@@ -7294,8 +7294,7 @@ void joined_dim(std::string const &ext)
72947294
}
72957295

72967296
auto epx = it.particles["e"]["position"]["x"];
7297-
Dataset ds(determineDatatype<type>(), {1});
7298-
ds.joinedDimension = 0;
7297+
Dataset ds(determineDatatype<type>(), {Dataset::JOINED_DIMENSION});
72997298
epx.resetDataset(ds);
73007299

73017300
size_t counter = 0;

0 commit comments

Comments
 (0)