From de05ae03ef3d09160708b227a26ffac7e4be75a8 Mon Sep 17 00:00:00 2001
From: David Rafferty <drafferty@hs.uni-hamburg.de>
Date: Fri, 11 Oct 2019 15:50:44 +0200
Subject: [PATCH] Add limits

---
 lsmtool/operations/group.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/lsmtool/operations/group.py b/lsmtool/operations/group.py
index b03c991..244ea49 100644
--- a/lsmtool/operations/group.py
+++ b/lsmtool/operations/group.py
@@ -258,9 +258,12 @@ def group(LSM, algorithm, targetFlux=None, weightBySize=False, numClusters=100,
             names = LSM.getPatchNames()
             fluxes = LSM.getColValues('I', aggregate='sum', units=units, applyBeam=applyBeam)
             if weightBySize:
-                sizes = LSM.getPatchSizes(weight=True, applyBeam=applyBeam)
+                sizes = LSM.getPatchSizes(units='arcsec', weight=True, applyBeam=applyBeam)
+                sizes[sizes < 1.0] = 1.0
                 meanSize = np.mean(sizes)
                 weights = meanSize / sizes
+                weights[weights > 2.0] = 2.0
+                weights[weights < 0.5] = 0.5
                 fluxes *= weights
             for name, flux in zip(names, fluxes):
                 if flux >= targetFlux:
-- 
GitLab