From 3bfdb2b54065bc5d8a1e1998628320e197e68c1e Mon Sep 17 00:00:00 2001
From: Duncan Macleod <duncan.macleod@ligo.org>
Date: Sat, 26 Nov 2022 01:19:36 +0000
Subject: [PATCH] gwsumm.plot: add support for sortbycolor (#333)

in trigger plots

Co-authored-by: Alex Urban <alexander.urban@ligo.org>

Co-authored-by: Alex Urban <alexander.urban@ligo.org>
---
 gwsumm/plot/triggers/__main__.py | 13 +++++++++++--
 gwsumm/plot/triggers/core.py     |  2 +-
 setup.cfg                        |  2 +-
 3 files changed, 13 insertions(+), 4 deletions(-)

diff --git a/gwsumm/plot/triggers/__main__.py b/gwsumm/plot/triggers/__main__.py
index 7de232eb..3ebefc18 100644
--- a/gwsumm/plot/triggers/__main__.py
+++ b/gwsumm/plot/triggers/__main__.py
@@ -196,6 +196,7 @@ def main(args=None):
         'cmap': get_plot('triggers').defaults.get('cmap', 'YlGnBu'),
         'clim': (3, 50),
         'colorlabel': 'Signal-to-noise ratio (SNR)',
+        'sortbycolor': True,
     }
 
     # update with user params
@@ -234,8 +235,16 @@ def main(args=None):
 
     # format keywords for plot creation
     plot_kw = OrderedDict(
-        (key, params.pop(key)) for
-        key in ('xscale', 'xlim', 'epoch', 'yscale', 'ylabel'))
+        (key, params.pop(key))
+        for key in (
+            'xscale',
+            'xlim',
+            'epoch',
+            'yscale',
+            'ylabel',
+            'sortbycolor',
+        )
+    )
 
     # create plot
     if args.tiles:
diff --git a/gwsumm/plot/triggers/core.py b/gwsumm/plot/triggers/core.py
index 22ded144..45442c65 100644
--- a/gwsumm/plot/triggers/core.py
+++ b/gwsumm/plot/triggers/core.py
@@ -166,7 +166,7 @@ def draw(self):
             plotargs.append(dict())
         # get plot arguments
         for key in ['vmin', 'vmax', 'edgecolor', 'facecolor', 'cmap', 's',
-                    'marker', 'rasterized']:
+                    'marker', 'rasterized', 'sortbycolor']:
             try:
                 val = self.pargs.pop(key)
             except KeyError:
diff --git a/setup.cfg b/setup.cfg
index 426eef62..521adb2d 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -48,7 +48,7 @@ install_requires =
 	astropy >=3.0.0
 	gwdatafind >=1.1.1
 	gwdetchar >=2.0.0
-	gwpy >=2.0.0
+	gwpy >=3.0.0
 	gwtrigfind
 	lalsuite
 	ligo-segments