68 lines
2.0 KiB
C++
68 lines
2.0 KiB
C++
// Copyright (2019-2024) Paul Scherrer Institute
|
|
|
|
#include "ROIMask.h"
|
|
#include "JFJochException.h"
|
|
|
|
ROIMask::ROIMask(int64_t in_xpixel, int64_t in_ypixel)
|
|
: mask(in_xpixel * in_ypixel, 0),
|
|
xpixel(in_xpixel), ypixel(in_ypixel) {
|
|
}
|
|
|
|
ROIMask::ROIMask(const DetectorSetup &setup)
|
|
: ROIMask(setup.GetGeometry().GetWidth(), setup.GetGeometry().GetHeight()) {}
|
|
|
|
void ROIMask::UpdateMask() {
|
|
roi_name_map.clear();
|
|
|
|
for (auto &i: mask)
|
|
i = UINT16_MAX;
|
|
|
|
for (int i = 0; i < boxes.size(); i++) {
|
|
boxes[i].MarkROI(mask, i, xpixel, ypixel);
|
|
roi_name_map[boxes[i].GetName()] = i;
|
|
}
|
|
|
|
for (int i = 0; i < circles.size(); i++) {
|
|
size_t id = boxes.size() + i;
|
|
circles[i].MarkROI(mask, id, xpixel, ypixel);
|
|
roi_name_map[circles[i].GetName()] = id;
|
|
}
|
|
}
|
|
|
|
const std::vector<uint16_t>& ROIMask::GetMask() const {
|
|
return mask;
|
|
}
|
|
|
|
const std::map<std::string, uint16_t> &ROIMask::GetROINameMap() const {
|
|
return roi_name_map;
|
|
}
|
|
|
|
void ROIMask::SetROIBox(const std::vector<ROIBox> &input) {
|
|
if (boxes.size() > box_array_size)
|
|
throw JFJochException(JFJochExceptionCategory::ArrayOutOfBounds, "Limit of box ROIs exceeded");
|
|
for (const auto &i: input) {
|
|
if (i.GetXMax() >= xpixel)
|
|
throw JFJochException(JFJochExceptionCategory::ArrayOutOfBounds, "ROI box X coordinate out of detector bounds");
|
|
if (i.GetYMax() >= ypixel)
|
|
throw JFJochException(JFJochExceptionCategory::ArrayOutOfBounds, "ROI box Y coordinate out of detector bounds");
|
|
}
|
|
boxes = input;
|
|
UpdateMask();
|
|
|
|
}
|
|
|
|
void ROIMask::SetROICircle(const std::vector<ROICircle> &input) {
|
|
if (circles.size() > circle_array_size)
|
|
throw JFJochException(JFJochExceptionCategory::ArrayOutOfBounds, "Limit of circular ROIs exceeded");
|
|
circles = input;
|
|
UpdateMask();
|
|
}
|
|
|
|
const std::vector<ROIBox> &ROIMask::GetROIBox() const {
|
|
return boxes;
|
|
}
|
|
|
|
const std::vector<ROICircle> &ROIMask::GetROICircle() const {
|
|
return circles;
|
|
}
|