From ca12cb4acbbde6adee61f26ced4d28790ea4076e Mon Sep 17 00:00:00 2001
From: landmanbester <lbester@sarao.ac.za>
Date: Tue, 12 Nov 2024 10:16:54 +0200
Subject: [PATCH] tweak fluxtractor

---
 pfb/opt/pcg.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/pfb/opt/pcg.py b/pfb/opt/pcg.py
index 416151a6..6afee248 100644
--- a/pfb/opt/pcg.py
+++ b/pfb/opt/pcg.py
@@ -426,6 +426,7 @@ def pcg_dds(ds_name,
     else:
         j = ds.DIRTY.values * mask * ds.BEAM.values
 
+    beam = mask*ds.BEAM.values
     psf = ds.PSF.values
     nx_psf, py_psf = psf.shape
     nx, ny = j.shape
@@ -497,10 +498,13 @@ def pcg_dds(ds_name,
                    weight=ds.WEIGHT.values,
                    vis_mask=ds.MASK.values,
                    freq=ds.FREQ.values,
-                   beam=ds.BEAM.values,
+                   beam=beam,
                    cell=ds.cell_rad,
                    x0=ds.x0,
                    y0=ds.y0,
+                   flip_u=ds.flip_u,
+                   flip_v=ds.flip_v,
+                   flip_w=ds.flip_w,
                    do_wgridding=do_wgridding,
                    epsilon=epsilon,
                    double_accum=double_accum,