updated rois in master file

python/src/file.hpp

@@ -190,8 +190,14 @@ void define_file_io_bindings(py::module &m) {
                      "<ROI: xmin: {} xmax: {} ymin: {} ymax: {}>", self.xmin,
                      self.xmax, self.ymin, self.ymax);
              })
-        .def("__iter__", [](const ROI &self) {
-            return py::make_iterator(&self.xmin, &self.ymax + 1); // NOLINT
+        .def("__iter__",
+             [](const ROI &self) {
+                 return py::make_iterator(&self.xmin, &self.ymax + 1); // NOLINT
+             })
+
+        .def("__eq__", [](const ROI &self, const ROI &other) {
+            return self.xmin == other.xmin && self.xmax == other.xmax &&
+                   self.ymin == other.ymin && self.ymax == other.ymax;
         });
 
 #pragma GCC diagnostic pop

python/tests/test_RawFile.py

@@ -1,6 +1,6 @@
 # SPDX-License-Identifier: MPL-2.0
 import pytest
-from aare import RawFile
+from aare import RawFile, ROI
 import numpy as np
 
 @pytest.mark.withdata
@@ -120,12 +120,19 @@ def test_read_eiger_udp_port_disabled(test_data_path):
         assert(len(frame) == 2)
         assert frame[0].shape == (256, 512)
         assert frame[1].shape == (256, 1024)
+        rois = f.master.rois
+        assert len(rois) == 2
+        assert rois[0] == ROI(512, 1024, 0, 256)
+        assert rois[1] == ROI(0, 1024, 256, 512)
 
     with RawFile(test_data_path / "raw/eiger/quad_eiger_disabled_bottom_port_master_0.json") as f:
         _, frame = f.read_frame() 
 
         assert frame.shape == (256, 512)
         assert(f.master.disabled_udp_ports == [1])
+        rois = f.master.rois
+        assert len(rois) == 1
+        assert rois[0] == ROI(0, 512, 256, 512)
 
     with RawFile(test_data_path / "raw/eiger/2_modules_eiger_disabled_udp_port_master_0.json") as f:
         _, frame = f.read_rois() 
@@ -134,3 +141,8 @@ def test_read_eiger_udp_port_disabled(test_data_path):
         assert frame[0].shape == (512, 512)
         assert frame[1].shape == (512, 512)
         assert (f.master.disabled_udp_ports == [1, 3, 5, 7])
+
+        rois = f.master.rois
+        assert len(rois) == 2
+        assert rois[0] == ROI(0, 512, 0, 512)
+        assert rois[1] == ROI(1024, 1536, 0, 512)