From ca3b42a7972e81d1bdc480fcea1c274395d4c319 Mon Sep 17 00:00:00 2001
From: Sven Augustin <sven.augustin@psi.ch>
Date: Fri, 2 Aug 2024 16:11:57 +0200
Subject: [PATCH] use the boolean pixel_mask_pf

---
 dap/algos/mask.py | 2 +-
 dap/worker.py     | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/dap/algos/mask.py b/dap/algos/mask.py
index febd40f..7ba422f 100644
--- a/dap/algos/mask.py
+++ b/dap/algos/mask.py
@@ -5,7 +5,7 @@ def calc_mask_pixels(data, pixel_mask_pf):
     if pixel_mask_pf is None:
         return
 
-    data[pixel_mask_pf == 0] = np.nan #TODO: boolean mask
+    data[~pixel_mask_pf] = np.nan
 
 
 
diff --git a/dap/worker.py b/dap/worker.py
index 5c1c4b5..b3bf7f7 100644
--- a/dap/worker.py
+++ b/dap/worker.py
@@ -201,7 +201,7 @@ def work(backend_address, accumulator_host, accumulator_port, visualisation_host
                         forceSendVisualisation = True
                         data_summed = None
                         n_aggregated_images = 1
-                data[pixel_mask_pf == 0] = np.NaN #TODO: boolean mask
+                data[~pixel_mask_pf] = np.nan
 
             else:
                 data = image