Fixing sampling.

Quasars · Nov 14, 2024 · 2b5fc7b · 2b5fc7b
1 parent c03d052
commit 2b5fc7b
Showing 1 changed file with 13 additions and 3 deletions.
diff --git a/orangecontrib/spectroscopy/widgets/owcos.py b/orangecontrib/spectroscopy/widgets/owcos.py
@@ -417,7 +417,11 @@ def plotCOS(self):
             self.cbarCOS.set_range(-1 * np.nanmax(np.absolute(cosmat)), np.nanmax(np.absolute(cosmat)))
             self.cbarCOS.set_colors(np.array(colorcet.diverging_bwr_40_95_c42) * 255)
 
-            left_indices = np.linspace(0, len(leftSP)-1, 100, dtype=int)
+            if len(leftSP) > 100:
+                left_indices = np.linspace(0, len(leftSP)-1, 100, dtype=int)
+            else:
+                left_indices = np.linspace(0, len(leftSP)-1, len(leftSP), dtype=int)
+
             for s in leftSP[left_indices]:
                 pt = pg.PlotCurveItem(s, leftSPwn, pen=pg.mkPen(color=(50, 50, 50), width=0.5))
                 self.left_plot.addItem(pt, ignoreBounds=True)
@@ -427,7 +431,11 @@ def plotCOS(self):
 
             self.left_plot.setXRange(np.min(leftSP), np.max(leftSP))
 
-            top_indices = np.linspace(0, len(topSP)-1, 100, dtype=int)
+            if len(topSP) > 100:
+                top_indices = np.linspace(0, len(topSP)-1, 100, dtype=int)
+            else:
+                top_indices = np.linspace(0, len(topSP)-1, len(topSP), dtype=int)
+
             for s in topSP[top_indices]:
                 pt = pg.PlotCurveItem(topSPwn, s, pen=pg.mkPen(color=(50, 50, 50), width=0.5))
                 self.top_plot.addItem(pt, ignoreBounds=True)
@@ -464,4 +472,6 @@ def commit(self):
     # pylint: disable=ungrouped-imports
     from Orange.widgets.utils.widgetpreview import WidgetPreview
 
-    WidgetPreview(OWCos).run(set_data1=Orange.data.Table("collagen"), set_data2=None)
+    data1 = Orange.data.Table("collagen")[100:230]
+    data2 = Orange.data.Table("collagen")[600:730]
+    WidgetPreview(OWCos).run(set_data1=data1, set_data2=data2)