pymodaq 3.6.12__py3-none-any.whl → 4.0.1__py3-none-any.whl

pymodaq/{daq_utils → utils}/plotting/data_viewers/viewer2D_basic.py

@@ -1,24 +1,35 @@
-from qtpy import QtCore, QtGui, QtWidgets
-from qtpy.QtCore import QObject, Slot, QThread, Signal, QRectF, QRect, QPointF, QLocale
+from typing import Union
+
+from qtpy import QtCore, QtWidgets
+from qtpy.QtCore import QObject, Slot, Signal
 import sys
-from pymodaq.daq_utils.plotting.items.axis_scaled import AXIS_POSITIONS, AxisItem_Scaled
 import pyqtgraph as pg
 import numpy as np
 from easydict import EasyDict as edict
 
+from pymodaq.utils.plotting.widgets import ImageWidget
+from pymodaq.utils.plotting.data_viewers.viewer import ViewerBase
+from pymodaq.utils.data import DataRaw
+from pymodaq.utils.plotting.items.image import UniformImageItem, SpreadImageItem
+
+
+class Viewer2DBasic(ViewerBase):
+    """Very Basic 2D Viewer used as a view to add Image Items into it and use the image object directly
 
-class Viewer2DBasic(QObject):
-    sig_double_clicked = Signal(float, float)
+    Not meant to plot directly data, see Viewer2D for that
+
+    See Also
+    --------
+    Navigator, UniformImageItem, SpreadImageItem
+
+    """
 
     def __init__(self, parent=None, **kwargs):
-        super().__init__()
-        # setting the gui
-        if parent is None:
-            parent = QtWidgets.QWidget()
-        self.parent = parent
+        super().__init__(parent, **kwargs)
+
         self.scaling_options = edict(scaled_xaxis=edict(label="", units=None, offset=0, scaling=1),
                                      scaled_yaxis=edict(label="", units=None, offset=0, scaling=1))
-        self.setupUI()
+        self.setup_ui()
 
     def scale_axis(self, xaxis, yaxis):
         return xaxis * self.scaling_options.scaled_xaxis.scaling + self.scaling_options.scaled_xaxis.offset, yaxis * self.scaling_options.scaled_yaxis.scaling + self.scaling_options.scaled_yaxis.offset
@@ -27,7 +38,7 @@ class Viewer2DBasic(QObject):
     def double_clicked(self, posx, posy):
         self.sig_double_clicked.emit(posx, posy)
 
-    def setupUI(self):
+    def setup_ui(self):
         vlayout = QtWidgets.QVBoxLayout()
         hsplitter = QtWidgets.QSplitter(QtCore.Qt.Horizontal)
 
@@ -77,87 +88,59 @@ class Viewer2DBasic(QObject):
         histo_layout.addWidget(self.histogram_adaptive)
         hsplitter.addWidget(self.histo_widget)
 
+    def _show_data(self, data: DataRaw):
+        """This basic viewer is not meant to plot data directly
 
-class ImageWidget(pg.GraphicsLayoutWidget):
-    """this gives a layout to add imageitems.
-    """
-
-    def __init__(self, parent=None, *args_plotitem, **kwargs_plotitem):
-        
-        super().__init__(parent)
-        self.setupUI(*args_plotitem, **kwargs_plotitem)
-
-    def setAspectLocked(self, lock=True, ratio=1):
-        """
-        Defines the aspect ratio of the view
-        Parameters
-        ----------
-        lock: (bool) if True aspect ratio is set to ratio, else the aspect ratio is varying when scaling the view
-        ratio: (int) aspect ratio between horizontal and vertical axis
-        """
-        self.plotitem.vb.setAspectLocked(lock=True, ratio=1)
-
-    def getAxis(self, position):
-        return self.plotitem.getAxis(position)
-
-    def setupUI(self, *args_plotitem, **kwargs_plotitem):
-        layout = QtWidgets.QGridLayout()
-        # set viewer area
-        self.scene_obj = self.scene()
-        self.view = View_cust()
-        self.plotitem = pg.PlotItem(viewBox=self.view, *args_plotitem, **kwargs_plotitem)
-        self.plotItem = self.plotitem  # for backcompatibility
-        self.setAspectLocked(lock=True, ratio=1)
-        self.setCentralItem(self.plotitem)
-
-    def add_scaled_axis(self, position):
         """
-        Add a AxisItem_Scaled to the given position with respect with the plotitem
-        Parameters
-        ----------
-        position: (str) either 'top', 'bottom', 'right' or 'left'
+        ...
 
-        Returns
-        -------
+    def set_aspect_ratio(self, status=True):
+        self.image_widget.plotitem.vb.setAspectLocked(lock=status, ratio=1)
 
-        """
-        if position not in AXIS_POSITIONS:
-            raise ValueError(f'The Axis position {position} should be in {AXIS_POSITIONS}')
-        axis = AxisItem_Scaled(position)
-        self.plotitem.setAxisItems({position: axis})
-        return axis
+    @property
+    def plotitem(self):
+        return self.image_widget.plotitem
 
-class View_cust(pg.ViewBox):
-    """Custom ViewBox used to enable other properties compared to parent class: pg.ViewBox
+    @property
+    def histograms(self):
+        return [self.histogram_red, self.histogram_green, self.histogram_blue]
 
-    """
-    sig_double_clicked = Signal(float, float)
+    def add_image_item(self, image: Union[SpreadImageItem, UniformImageItem], histogram: pg.HistogramLUTWidget = None):
+        """Adds an image item to this viewer plotitem
 
-    def __init__(self, parent=None, border=None, lockAspect=False, enableMouse=True, invertY=False,
-                 enableMenu=True, name=None, invertX=False):
-        super().__init__(parent, border, lockAspect, enableMouse, invertY, enableMenu, name,
-                                        invertX)
-
-    def mouseClickEvent(self, ev):
-        if ev.button() == QtCore.Qt.RightButton and self.menuEnabled():
-            ev.accept()
-            self.raiseContextMenu(ev)
-        if ev.double():
-            pos = self.mapToView(ev.pos())
-            self.sig_double_clicked.emit(pos.x(), pos.y())
+        Links it to an existing histogram
+        """
+        self.plotitem.addItem(image)
+        if histogram in self.histograms:
+            histogram.item.setImageItem(image)
 
 
 if __name__ == '__main__':  # pragma: no cover
-    from pymodaq.daq_utils.plotting.items.image import SpreadImageItem
+    from pymodaq.utils.plotting.items.image import SpreadImageItem
+    from qtpy import QtSvg, QtGui
 
     app = QtWidgets.QApplication(sys.argv)
     form = QtWidgets.QWidget()
     prog = Viewer2DBasic(form)
     img = SpreadImageItem()
     prog.image_widget.plotItem.addItem(img)
+
+    svg_item = QtSvg.QGraphicsSvgItem()
+    svg_renderer = QtSvg.QSvgRenderer(
+        r'C:\Users\weber\Labo\Projet-Dossier candidature\Technical project\GDSII\wafer.svg')
+    svg_item.setSharedRenderer(svg_renderer)
+
     form.show()
 
-    data = np.load('triangulation_data.npy')
+    data = np.load('../../../resources/triangulation_data.npy')
     img.setImage(data)
 
+    prog.image_widget.plotitem.addItem(svg_item)
+    curr_size = svg_renderer.defaultSize()
+    real_size = svg_item.boundingRect()
+    # tr = QtGui.QTransform()
+    # tr.translate(rect.left(), rect.top())
+    # tr.scale(300/rect.width(), 300/rect.height())
+    # svg_item.setTransform(tr)
+    pass
     sys.exit(app.exec_())