Alexandria 2.31.0
SDC-CH common library for the Euclid project
Loading...
Searching...
No Matches
NpyWriter.icpp
Go to the documentation of this file.
1/*
2 * Copyright (C) 2012-2021 Euclid Science Ground Segment
3 *
4 * This library is free software; you can redistribute it and/or modify it under
5 * the terms of the GNU Lesser General Public License as published by the Free
6 * Software Foundation; either version 3.0 of the License, or (at your option)
7 * any later version.
8 *
9 * This library is distributed in the hope that it will be useful, but WITHOUT
10 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
11 * FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more
12 * details.
13 *
14 * You should have received a copy of the GNU Lesser General Public License
15 * along with this library; if not, write to the Free Software Foundation, Inc.,
16 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
17 */
18
19#ifdef NPY_IMPL
20
21#ifdef __APPLE__
22#include <machine/endian.h>
23#else
24#include <endian.h>
25#endif
26#include "NdArray/NdArray.h"
27#include "NpyCommon.h"
28#include <ElementsKernel/Exception.h>
29#include <boost/endian/arithmetic.hpp>
30#include <sstream>
31
32namespace Euclid {
33namespace NdArray {
34
35/*
36 * Implementation of writeNpy
37 */
38template <typename T>
39void writeNpy(std::ostream& out, const NdArray<T>& array) {
40 writeNpyHeader<T>(out, array.shape(), array.attributes());
41 // The header already has the endian type, so just dump the content of the array
42 for (auto v : array) {
43 out.write(reinterpret_cast<const char*>(&v), sizeof(v));
44 }
45}
46
47} // end of namespace NdArray
48} // end of namespace Euclid
49
50#endif // NPY_IMPL