Use enum properties to determine plot types

SasView · Dec 3, 2024 · 66a1a77 · 66a1a77
1 parent 31055cd
commit 66a1a77
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/src/sas/qtgui/MainWindow/DataExplorer.py b/src/sas/qtgui/MainWindow/DataExplorer.py
@@ -1123,10 +1123,8 @@ def displayData(self, data_list, id=None):
 
             plot_name = plot_to_show.name
             role = plot_to_show.plot_role
-            stand_alone_types = [DataRole.ROLE_RESIDUAL, DataRole.ROLE_RESIDUAL_SESANS, DataRole.ROLE_STAND_ALONE,
-                                 DataRole.ROLE_LIN_LIN, DataRole.ROLE_POLYDISPERSITY]
 
-            if (role in stand_alone_types and shown) or role == DataRole.ROLE_DELETABLE:
+            if (role.stand_alone and shown) or role == DataRole.ROLE_DELETABLE:
                 # Nothing to do if stand-alone plot already shown or plot to be deleted
                 continue
             elif role in [DataRole.ROLE_RESIDUAL, DataRole.ROLE_RESIDUAL_SESANS] and config.DISABLE_RESIDUAL_PLOT:
@@ -1135,7 +1133,7 @@ def displayData(self, data_list, id=None):
             elif role in [DataRole.ROLE_POLYDISPERSITY] and config.DISABLE_POLYDISPERSITY_PLOT:
                 # Nothing to do if polydispersity plot is not plotted
                 continue
-            elif role in stand_alone_types:
+            elif role.stand_alone:
                 # Stand-alone plots should always be separate
                 self.plotData([(plot_item, plot_to_show)])
             elif append: