Improve plotting

tests/JFJochReceiverProcessingTest.cpp

@@ -241,3 +241,93 @@ TEST_CASE("JFJochIntegrationTest_ZMQ_lysozyme_resolution", "[JFJochReceiver]") {
     CHECK(receiver_out.status.images_sent == experiment.GetImageNum());
     CHECK(!receiver_out.status.cancelled);
 }
+
+
+TEST_CASE("JFJochIntegrationTest_ZMQ_ROI", "[JFJochReceiver]") {
+    Logger logger("JFJochIntegrationTest_ZMQ_ROI");
+
+    RegisterHDF5Filter();
+
+    const uint16_t nthreads = 4;
+
+    DiffractionExperiment experiment(DetectorGeometry(8,2,8,36));
+    experiment.ImagesPerTrigger(5).NumTriggers(1).DataFileCount(1).UseInternalPacketGenerator(true)
+            .FilePrefix("lyso_test").ConversionOnFPGA(false)
+            .DetectorDistance_mm(75).BeamY_pxl(1136).BeamX_pxl(1090).PhotonEnergy_keV(12.4)
+            .SetUnitCell(UnitCell{.a = 36.9, .b = 78.95, .c = 78.95, .alpha =90, .beta = 90, .gamma = 90});
+
+    experiment.ROI().SetROIBox({ROIBox("roi0", 100, 120, 20,30)});
+    experiment.ROI().SetROICircle({ROICircle("roi1", 500, 800, 10)});
+
+    // Load example image
+    HDF5ReadOnlyFile data("../../tests/test_data/compression_benchmark.h5");
+    HDF5DataSet dataset(data, "/entry/data/data");
+    HDF5DataSpace file_space(dataset);
+
+    REQUIRE(file_space.GetDimensions()[2] == experiment.GetXPixelsNum());
+    REQUIRE(file_space.GetDimensions()[1] == experiment.GetYPixelsNum());
+    std::vector<int16_t> image_conv (file_space.GetDimensions()[1] * file_space.GetDimensions()[2]);
+
+    std::vector<hsize_t> start = {5,0,0};
+    std::vector<hsize_t> file_size = {1, file_space.GetDimensions()[1], file_space.GetDimensions()[2]};
+    dataset.ReadVector(image_conv, start, file_size);
+
+    uint64_t roi_value = 0;
+    uint64_t pixels = 0;
+    for (int y = 20; y <= 30; y++) {
+        for (int x = 100; x <= 120; x++) {
+            int16_t val = image_conv[experiment.GetXPixelsNum() * y + x];
+            if ((val != INT16_MIN) && (val != INT16_MAX)) {
+                pixels += 1;
+                roi_value += val;
+            }
+        }
+    }
+
+    std::vector<int16_t> image_raw_geom(experiment.GetModulesNum() * RAW_MODULE_SIZE);
+    ConvertedToRawGeometry(experiment, image_raw_geom.data(), image_conv.data());
+    logger.Info("Loaded image");
+
+    // Setup acquisition device
+    AcquisitionDeviceGroup aq_devices;
+    std::unique_ptr<HLSSimulatedDevice> test = std::make_unique<HLSSimulatedDevice>(0, 64);
+    for (int m = 0; m < experiment.GetModulesNum(); m++)
+        test->SetInternalGeneratorFrame((uint16_t *) image_raw_geom.data() + m * RAW_MODULE_SIZE, m);
+
+    aq_devices.Add(std::move(test));
+
+    JFCalibration calib(experiment);
+    HDF5ImagePusher pusher;
+
+    ZMQContext context;
+    context.NumThreads(4);
+
+    JFJochReceiverService service(aq_devices, logger, pusher);
+    service.NumThreads(nthreads);
+
+    service.Start(experiment, nullptr);
+    auto receiver_out = service.Stop();
+
+    std::string jpeg;
+    PreviewJPEGSettings jpeg_settings{
+            .saturation_value = 10,
+            .jpeg_quality = 100,
+            .show_spots = true,
+            .show_roi = true
+    };
+    REQUIRE_NOTHROW(jpeg = service.GetJPEG(jpeg_settings));
+    std::ofstream f("lyso_processing_test_roi.jpeg", std::ios::binary);
+    f.write(jpeg.data(), jpeg.size());
+
+    auto plot = service.GetDataProcessingPlot(PlotRequest{.type = PlotType::ROISum, .binning = 1});
+
+    REQUIRE(plot.size() == 2);
+    CHECK(plot[0].title == "roi0");
+    CHECK(plot[0].x[0] == 0);
+    CHECK(plot[0].y[0] == roi_value);
+    CHECK(plot[1].title == "roi1");
+
+    CHECK(receiver_out.efficiency == 1.0);
+    CHECK(receiver_out.status.images_sent == experiment.GetImageNum());
+    CHECK(!receiver_out.status.cancelled);
+}