This commit is contained in:
Erik Fröjdh 2024-10-31 10:29:07 +01:00
parent cee0d71b9c
commit ec61132296
4 changed files with 110 additions and 33 deletions

View File

@ -24,6 +24,20 @@ struct ModuleConfig {
* @note documentation can also be found in the FileInterface class * @note documentation can also be found in the FileInterface class
*/ */
class RawFile : public FileInterface { class RawFile : public FileInterface {
size_t n_subfiles{};
size_t n_subfile_parts{};
std::vector<std::vector<SubFile *>> subfiles;
size_t subfile_rows{}, subfile_cols{};
xy m_geometry{};
std::vector<xy> positions;
ModuleConfig cfg{0, 0};
TimingMode timing_mode{};
bool quad{false};
//Stuff that we might need with Ctb files
uint32_t m_analog_samples{};
uint32_t m_digital_samples{};
public: public:
/** /**
* @brief RawFile constructor * @brief RawFile constructor
@ -176,15 +190,7 @@ class RawFile : public FileInterface {
void open_subfiles(); void open_subfiles();
void parse_config(const FileConfig &config); void parse_config(const FileConfig &config);
size_t n_subfiles{};
size_t n_subfile_parts{};
std::vector<std::vector<SubFile *>> subfiles;
size_t subfile_rows{}, subfile_cols{};
xy m_geometry{};
std::vector<xy> positions;
ModuleConfig cfg{0, 0};
TimingMode timing_mode{};
bool quad{false};
}; };
} // namespace aare } // namespace aare

View File

@ -17,8 +17,9 @@
/** /**
* @brief LOCATION macro to get the current location in the code * @brief LOCATION macro to get the current location in the code
*/ */
#define LOCATION std::string(__FILE__) + std::string(":") + std::to_string(__LINE__) + ":" + std::string(__func__) + ":" #define LOCATION \
std::string(__FILE__) + std::string(":") + std::to_string(__LINE__) + \
":" + std::string(__func__) + ":"
namespace aare { namespace aare {
@ -34,12 +35,15 @@ class Cluster {
std::byte *m_data; std::byte *m_data;
public: public:
Cluster(int cluster_sizeX_, int cluster_sizeY_, Dtype dt_ = Dtype(typeid(int32_t))) Cluster(int cluster_sizeX_, int cluster_sizeY_,
: cluster_sizeX(cluster_sizeX_), cluster_sizeY(cluster_sizeY_), dt(dt_) { Dtype dt_ = Dtype(typeid(int32_t)))
: cluster_sizeX(cluster_sizeX_), cluster_sizeY(cluster_sizeY_),
dt(dt_) {
m_data = new std::byte[cluster_sizeX * cluster_sizeY * dt.bytes()]{}; m_data = new std::byte[cluster_sizeX * cluster_sizeY * dt.bytes()]{};
} }
Cluster() : Cluster(3, 3) {} Cluster() : Cluster(3, 3) {}
Cluster(const Cluster &other) : Cluster(other.cluster_sizeX, other.cluster_sizeY, other.dt) { Cluster(const Cluster &other)
: Cluster(other.cluster_sizeX, other.cluster_sizeY, other.dt) {
if (this == &other) if (this == &other)
return; return;
x = other.x; x = other.x;
@ -54,18 +58,23 @@ class Cluster {
return *this; return *this;
} }
Cluster(Cluster &&other) noexcept Cluster(Cluster &&other) noexcept
: cluster_sizeX(other.cluster_sizeX), cluster_sizeY(other.cluster_sizeY), x(other.x), y(other.y), dt(other.dt), : cluster_sizeX(other.cluster_sizeX),
m_data(other.m_data) { cluster_sizeY(other.cluster_sizeY), x(other.x), y(other.y),
dt(other.dt), m_data(other.m_data) {
other.m_data = nullptr; other.m_data = nullptr;
other.dt = Dtype(Dtype::TypeIndex::ERROR); other.dt = Dtype(Dtype::TypeIndex::ERROR);
} }
~Cluster() { delete[] m_data; } ~Cluster() { delete[] m_data; }
template <typename T> T get(int idx) { template <typename T> T get(int idx) {
(sizeof(T) == dt.bytes()) ? 0 : throw std::invalid_argument("[ERROR] Type size mismatch"); (sizeof(T) == dt.bytes())
? 0
: throw std::invalid_argument("[ERROR] Type size mismatch");
return *reinterpret_cast<T *>(m_data + idx * dt.bytes()); return *reinterpret_cast<T *>(m_data + idx * dt.bytes());
} }
template <typename T> auto set(int idx, T val) { template <typename T> auto set(int idx, T val) {
(sizeof(T) == dt.bytes()) ? 0 : throw std::invalid_argument("[ERROR] Type size mismatch"); (sizeof(T) == dt.bytes())
? 0
: throw std::invalid_argument("[ERROR] Type size mismatch");
return memcpy(m_data + idx * dt.bytes(), &val, (size_t)dt.bytes()); return memcpy(m_data + idx * dt.bytes(), &val, (size_t)dt.bytes());
} }
// auto x() const { return x; } // auto x() const { return x; }
@ -74,10 +83,15 @@ class Cluster {
// auto y(int16_t y_) { return y = y_; } // auto y(int16_t y_) { return y = y_; }
template <typename T> std::string to_string() const { template <typename T> std::string to_string() const {
(sizeof(T) == dt.bytes()) ? 0 : throw std::invalid_argument("[ERROR] Type size mismatch"); (sizeof(T) == dt.bytes())
std::string s = "x: " + std::to_string(x) + " y: " + std::to_string(y) + "\nm_data: ["; ? 0
: throw std::invalid_argument("[ERROR] Type size mismatch");
std::string s = "x: " + std::to_string(x) + " y: " + std::to_string(y) +
"\nm_data: [";
for (int i = 0; i < cluster_sizeX * cluster_sizeY; i++) { for (int i = 0; i < cluster_sizeX * cluster_sizeY; i++) {
s += std::to_string(*reinterpret_cast<T *>(m_data + i * dt.bytes())) + " "; s += std::to_string(
*reinterpret_cast<T *>(m_data + i * dt.bytes())) +
" ";
} }
s += "]"; s += "]";
return s; return s;
@ -85,10 +99,12 @@ class Cluster {
/** /**
* @brief size of the cluster in bytes when saved to a file * @brief size of the cluster in bytes when saved to a file
*/ */
size_t size() const { return cluster_sizeX * cluster_sizeY ; } size_t size() const { return cluster_sizeX * cluster_sizeY; }
size_t bytes() const { return cluster_sizeX * cluster_sizeY * dt.bytes(); } size_t bytes() const { return cluster_sizeX * cluster_sizeY * dt.bytes(); }
auto begin() const { return m_data; } auto begin() const { return m_data; }
auto end() const { return m_data + cluster_sizeX * cluster_sizeY * dt.bytes(); } auto end() const {
return m_data + cluster_sizeX * cluster_sizeY * dt.bytes();
}
std::byte *data() { return m_data; } std::byte *data() { return m_data; }
}; };
@ -117,28 +133,49 @@ struct sls_detector_header {
} }
packetMaskStr += "]"; packetMaskStr += "]";
return "frameNumber: " + std::to_string(frameNumber) + "\n" + "expLength: " + std::to_string(expLength) + "\n" + return "frameNumber: " + std::to_string(frameNumber) + "\n" +
"packetNumber: " + std::to_string(packetNumber) + "\n" + "bunchId: " + std::to_string(bunchId) + "\n" + "expLength: " + std::to_string(expLength) + "\n" +
"timestamp: " + std::to_string(timestamp) + "\n" + "modId: " + std::to_string(modId) + "\n" + "packetNumber: " + std::to_string(packetNumber) + "\n" +
"row: " + std::to_string(row) + "\n" + "column: " + std::to_string(column) + "\n" + "bunchId: " + std::to_string(bunchId) + "\n" +
"reserved: " + std::to_string(reserved) + "\n" + "debug: " + std::to_string(debug) + "\n" + "timestamp: " + std::to_string(timestamp) + "\n" +
"roundRNumber: " + std::to_string(roundRNumber) + "\n" + "detType: " + std::to_string(detType) + "\n" + "modId: " + std::to_string(modId) + "\n" +
"version: " + std::to_string(version) + "\n" + "packetMask: " + packetMaskStr + "\n"; "row: " + std::to_string(row) + "\n" +
"column: " + std::to_string(column) + "\n" +
"reserved: " + std::to_string(reserved) + "\n" +
"debug: " + std::to_string(debug) + "\n" +
"roundRNumber: " + std::to_string(roundRNumber) + "\n" +
"detType: " + std::to_string(detType) + "\n" +
"version: " + std::to_string(version) + "\n" +
"packetMask: " + packetMaskStr + "\n";
} }
}; };
template <typename T> struct t_xy { template <typename T> struct t_xy {
T row; T row;
T col; T col;
bool operator==(const t_xy &other) const { return row == other.row && col == other.col; } bool operator==(const t_xy &other) const {
return row == other.row && col == other.col;
}
bool operator!=(const t_xy &other) const { return !(*this == other); } bool operator!=(const t_xy &other) const { return !(*this == other); }
std::string to_string() const { return "{ x: " + std::to_string(row) + " y: " + std::to_string(col) + " }"; } std::string to_string() const {
return "{ x: " + std::to_string(row) + " y: " + std::to_string(col) +
" }";
}
}; };
using xy = t_xy<uint32_t>; using xy = t_xy<uint32_t>;
using dynamic_shape = std::vector<int64_t>; using dynamic_shape = std::vector<int64_t>;
enum class DetectorType { Jungfrau, Eiger, Mythen3, Moench, ChipTestBoard, Unknown }; enum class DetectorType {
Jungfrau,
Eiger,
Mythen3,
Moench,
Moench03,
Moench03_old,
ChipTestBoard,
Unknown
};
enum class TimingMode { Auto, Trigger }; enum class TimingMode { Auto, Trigger };

View File

@ -223,6 +223,28 @@ void RawFile::parse_json_metadata() {
} catch (const json::out_of_range &e) { } catch (const json::out_of_range &e) {
m_bitdepth = 16; m_bitdepth = 16;
} }
try {
m_analog_samples = j.at("Analog Samples");
}catch (const json::out_of_range &e) {
m_analog_samples = 0;
}
try {
m_digital_samples = j.at("Digital Samples");
}catch (const json::out_of_range &e) {
m_digital_samples = 0;
}
//Update detector type for Moench
if (m_type == DetectorType::Moench && m_analog_samples == 0 && m_rows == 400) {
m_type = DetectorType::Moench03;
}else if (m_type == DetectorType::Moench && m_rows == 400 && m_analog_samples == 5000) {
m_type = DetectorType::Moench03_old;
}else{
throw std::runtime_error(LOCATION + "Could not determine Moench detector type");
}
// only Eiger had quad // only Eiger had quad
if (m_type == DetectorType::Eiger) { if (m_type == DetectorType::Eiger) {
quad = (j["Quad"] == 1); quad = (j["Quad"] == 1);
@ -301,6 +323,10 @@ Frame RawFile::get_frame(size_t frame_index) {
auto f = Frame(this->m_rows, this->m_cols, Dtype::from_bitdepth(this->m_bitdepth)); auto f = Frame(this->m_rows, this->m_cols, Dtype::from_bitdepth(this->m_bitdepth));
std::byte *frame_buffer = f.data(); std::byte *frame_buffer = f.data();
get_frame_into(frame_index, frame_buffer); get_frame_into(frame_index, frame_buffer);
//here would be the place to run a transform before returning the frame
if (m_type == DetectorType::Moench03_old)
fmt::print("Moench03_old\n");
return f; return f;
} }

View File

@ -19,6 +19,10 @@ template <> std::string toString(DetectorType arg) {
return "Mythen3"; return "Mythen3";
case DetectorType::Moench: case DetectorType::Moench:
return "Moench"; return "Moench";
case DetectorType::Moench03:
return "Moench03";
case DetectorType::Moench03_old:
return "Moench03_old";
case DetectorType::ChipTestBoard: case DetectorType::ChipTestBoard:
return "ChipTestBoard"; return "ChipTestBoard";
default: default:
@ -41,6 +45,10 @@ template <> DetectorType StringTo(const std::string &arg) {
return DetectorType::Mythen3; return DetectorType::Mythen3;
if (arg == "Moench") if (arg == "Moench")
return DetectorType::Moench; return DetectorType::Moench;
if (arg == "Moench03")
return DetectorType::Moench03;
if (arg == "Moench03_old")
return DetectorType::Moench03_old;
if (arg == "ChipTestBoard") if (arg == "ChipTestBoard")
return DetectorType::ChipTestBoard; return DetectorType::ChipTestBoard;
throw std::runtime_error("Could not decode dector from: \"" + arg + "\""); throw std::runtime_error("Could not decode dector from: \"" + arg + "\"");