Files and structure for python interface

src/File.cpp

@@ -0,0 +1,70 @@
+#include "aare/File.hpp"
+#include "aare/NumpyFile.hpp"
+#include "aare/RawFile.hpp"
+
+#include <fmt/format.h>
+
+namespace aare {
+
+File::File(const std::filesystem::path &fname, const std::string &mode, const FileConfig &cfg)
+    : file_impl(nullptr), is_npy(true) {
+    if (mode != "r" && mode != "w" && mode != "a") {
+        throw std::invalid_argument("Unsupported file mode");
+    }
+
+    if ((mode == "r" || mode == "a") && !std::filesystem::exists(fname)) {
+        throw std::runtime_error(fmt::format("File does not exist: {}", fname.string()));
+    }
+
+    if (fname.extension() == ".raw" || fname.extension() == ".json") {
+        // aare::logger::debug("Loading raw file");
+        file_impl = new RawFile(fname, mode, cfg);
+        is_npy = false;
+    }
+    // check if extension is numpy
+    else if (fname.extension() == ".npy") {
+        // aare::logger::debug("Loading numpy file");
+        file_impl = new NumpyFile(fname, mode, cfg);
+    } else {
+        throw std::runtime_error("Unsupported file type");
+    }
+}
+
+void File::write(Frame &frame, sls_detector_header header) {
+    if (is_npy) {
+        // aare::logger::info("ignoring header for npy file");
+        dynamic_cast<NumpyFile *>(file_impl)->write(frame);
+    } else {
+        dynamic_cast<RawFile *>(file_impl)->write(frame, header);
+    }
+}
+Frame File::read() { return file_impl->read(); }
+size_t File::total_frames() const { return file_impl->total_frames(); }
+std::vector<Frame> File::read(size_t n_frames) { return file_impl->read(n_frames); }
+void File::read_into(std::byte *image_buf) { file_impl->read_into(image_buf); }
+void File::read_into(std::byte *image_buf, size_t n_frames) { file_impl->read_into(image_buf, n_frames); }
+size_t File::frame_number(size_t frame_index) { return file_impl->frame_number(frame_index); }
+size_t File::bytes_per_frame() { return file_impl->bytes_per_frame(); }
+size_t File::pixels_per_frame() { return file_impl->pixels_per_frame(); }
+void File::seek(size_t frame_number) { file_impl->seek(frame_number); }
+size_t File::tell() const { return file_impl->tell(); }
+size_t File::rows() const { return file_impl->rows(); }
+size_t File::cols() const { return file_impl->cols(); }
+size_t File::bitdepth() const { return file_impl->bitdepth(); }
+void File::set_total_frames(size_t total_frames) { return file_impl->set_total_frames(total_frames); }
+File::~File() { delete file_impl; }
+DetectorType File::detector_type() const { return file_impl->detector_type(); }
+xy File::geometry() const {
+    if (is_npy) {
+        return {1, 1};
+    }
+    return reinterpret_cast<RawFile *>(file_impl)->geometry();
+}
+
+Frame File::iread(size_t frame_number) { return file_impl->iread(frame_number); }
+
+File::File(File &&other) noexcept : file_impl(other.file_impl), is_npy(other.is_npy) { other.file_impl = nullptr; }
+
+// write move assignment operator
+
+} // namespace aare