RawFile.hpp

Go to the documentation of this file.

#pragma once
#include "aare/core/Frame.hpp"
#include "aare/file_io/FileInterface.hpp"
#include "aare/file_io/SubFile.hpp"
 
namespace aare {
 
class RawFile : public FileInterface {
  public:
    RawFile(const std::filesystem::path &fname, const std::string &mode = "r", const FileConfig &cfg = {});
 
    void write(Frame &frame) override { throw std::runtime_error("Not implemented"); };
    Frame read() override { return get_frame(this->current_frame++); };
    std::vector<Frame> read(size_t n_frames) override;
    void read_into(std::byte *image_buf) override { return get_frame_into(this->current_frame++, image_buf); };
    void read_into(std::byte *image_buf, size_t n_frames) override;
    size_t frame_number(size_t frame_index) override;
 
    size_t bytes_per_frame() override { return m_rows * m_cols * m_bitdepth / 8; }
 
    size_t pixels() override { return m_rows * m_cols; }
 
    // goto frame number
    void seek(size_t frame_number) override { this->current_frame = frame_number; };
 
    // return the position of the file pointer (in number of frames)
    size_t tell() override { return this->current_frame; };
 
    static bool is_master_file(std::filesystem::path fpath);
 
    inline void set_config(int row, int col) {
        cfg.module_gap_row = row;
        cfg.module_gap_col = col;
    }
    // TODO! Deal with fast quad and missing files
 
    void find_number_of_subfiles();
 
    inline std::filesystem::path master_fname();
    inline std::filesystem::path data_fname(int mod_id, int file_id);
 
    ~RawFile();
 
    size_t total_frames() const override { return m_total_frames; }
    ssize_t rows() const override { return m_rows; }
    ssize_t cols() const override { return m_cols; }
    ssize_t bitdepth() const override { return m_bitdepth; }
 
  private:
    void get_frame_into(size_t frame_number, std::byte *image_buf);
 
    Frame get_frame(size_t frame_number);
 
    void parse_fname();
 
    void parse_metadata();
 
    void parse_raw_metadata();
 
    void parse_json_metadata();
 
    void find_geometry();
 
    sls_detector_header read_header(const std::filesystem::path &fname);
 
    void open_subfiles();
 
    size_t n_subfiles;
    size_t n_subfile_parts;
    std::vector<std::vector<SubFile *>> subfiles;
    int subfile_rows, subfile_cols;
    xy geometry;
    std::vector<xy> positions;
    RawFileConfig cfg{0, 0};
    TimingMode timing_mode;
    bool quad{false};
};
 
} // namespace aare