Skip to content

Commit 5db2308

Browse files
authored
Fix y-axis range of main plot in dataviewer from 0 to 1 when P1 data type (#226)
This closes #225. Now, once the P1 data type button is clicked, the y-axis range is set from 0 to 1. ![image](https://github.com/snu-quiqcl/iquip/assets/76851886/aa1b8136-195b-4e0d-bfb6-f3522e7d8551)
2 parents 2b3156d + 410dbf3 commit 5db2308

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

iquip/apps/dataviewer.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -778,11 +778,11 @@ def setData(self, data: np.ndarray, axes: Sequence[AxisInfo]):
778778
logger.error("MainPlotWidget does not support %d-dim data", data.ndim)
779779
return
780780
self.viewers[plotType].setData(data, axes)
781-
self.stack.setCurrentIndex(plotType)
782781
if self.autoRangeBox.isChecked():
783782
plotItem = self.viewers[plotType].plotItem
784783
bounds = plotItem.getViewBox().childrenBoundingRect(items=None)
785784
plotItem.setXRange(bounds.left(), bounds.right())
785+
self.stack.setCurrentIndex(plotType)
786786

787787
def _mouseClicked(self, viewer: NDArrayViewer, event: mouseEvents.MouseClickEvent):
788788
"""Mouse is clicked on the plot.
@@ -1191,6 +1191,8 @@ def updateMainPlot(self, axis: Sequence[int], dataType: DataPointWidget.DataType
11911191
reduce = self._reduceFunction(dataType)
11921192
data, axes = self.policy.extract(axis, reduce)
11931193
self.frame.mainPlotWidget.setData(data, axes)
1194+
if data.ndim == 1 and dataType == DataPointWidget.DataType.P1:
1195+
self.frame.mainPlotWidget.viewer().plotItem.setYRange(0, 1)
11941196
index = self.dataPointIndex
11951197
if data.ndim == len(index) and np.all(np.less(index, data.shape)):
11961198
self.selectDataPoint(index)

0 commit comments

Comments
 (0)