aare/examples/numpy_write_example.cpp
2024-04-09 17:30:42 +02:00

28 lines
645 B
C++

// Your First C++ Program
#include "aare/core/Frame.hpp"
#include "aare/file_io/File.hpp"
#include <iostream>
#define AARE_ROOT_DIR_VAR "PROJECT_ROOT_DIR"
using aare::File;
using aare::FileConfig;
using aare::Frame;
int main() {
auto path = std::filesystem::path("/tmp/test.npy");
auto dtype = aare::DType(typeid(uint32_t));
FileConfig cfg = {path, dtype, 100, 100};
File npy(path, "w", cfg);
Frame f(100, 100, dtype.bitdepth());
for (int i = 0; i < 10000; i++) {
f.set<uint32_t>(i / 100, i % 100, i);
}
npy.write(f);
f.set<uint32_t>(0, 0, 77);
npy.write(f);
npy.write(f);
return 0;
}