Basic interface created

Everything is ready for start with the functionality

Author: alhazacod

MatrixButtons.py

@@ -0,0 +1,66 @@
+import PyQt5.QtWidgets as QtWidgets
+from matplotlib.figure import Figure
+from matplotlib.backends.backend_qt5agg import FigureCanvasQTAgg as FigureCanvas
+
+class MatrixButtons(QtWidgets.QWidget):
+    """Widget for PyQt5 with the Matrix input and the buttons
+
+    ...
+
+    Attributes
+    ----------
+    layout: QtWidgets.QGridLayout
+    """
+    def __init__(self):
+        super().__init__()
+        self.layout = QtWidgets.QGridLayout(self) # Set the layout
+
+        # First row
+        self.matrix00 = QtWidgets.QLineEdit()
+        self.matrix10 = QtWidgets.QLineEdit()
+        self.matrix20 = QtWidgets.QLineEdit()
+        self.addToLayout(self.matrix00,[0,0],[1,1])
+        self.addToLayout(self.matrix10,[1,0],[1,1])
+        self.addToLayout(self.matrix20,[2,0],[1,1])
+
+        # Second row
+        self.matrix01 = QtWidgets.QLineEdit()
+        self.matrix11 = QtWidgets.QLineEdit()
+        self.matrix21 = QtWidgets.QLineEdit()
+        self.addToLayout(self.matrix01,[0,1],[1,1])
+        self.addToLayout(self.matrix11,[1,1],[1,1])
+        self.addToLayout(self.matrix21,[2,1],[1,1])
+
+        # Third row
+        self.matrix02 = QtWidgets.QLineEdit()
+        self.matrix12 = QtWidgets.QLineEdit()
+        self.matrix22 = QtWidgets.QLineEdit()
+        self.addToLayout(self.matrix02,[0,2],[1,1])
+        self.addToLayout(self.matrix12,[1,2],[1,1])
+        self.addToLayout(self.matrix22,[2,2],[1,1])
+
+        # Button for plot the grid transformed by the matrix
+        plotButton = QtWidgets.QPushButton('Plot')
+        self.addToLayout(plotButton,[1,3],[1,1])
+
+        
+    def addToLayout(self, widget, position:list, size:list):
+        self.layout.addWidget(widget,position[1],position[0],size[1],size[0])
+
+    def setMatrix(self,matrix):
+        """Add widget to layout
+        """
+        # First row
+        self.matrix00.setText(str(matrix[0,0]))
+        self.matrix10.setText(str(matrix[0,1]))
+        self.matrix20.setText(str(matrix[0,2]))
+
+        # Second row
+        self.matrix01.setText(str(matrix[1,0]))
+        self.matrix11.setText(str(matrix[1,1]))
+        self.matrix21.setText(str(matrix[1,2]))
+
+        # Third row
+        self.matrix02.setText(str(matrix[2,0]))
+        self.matrix12.setText(str(matrix[2,1]))
+        self.matrix22.setText(str(matrix[2,2]))

Plotting.py

@@ -40,8 +40,11 @@ def __init__(self):
         self.canvas = FigureCanvas(self.fig) # Get the plot image
         self.axes = self.fig.add_subplot(111,projection='3d') # Create matpotlib axes
 
-        layout = QtWidgets.QVBoxLayout(self) 
-        layout.addWidget(self.canvas)
+        self.setWindowTitle('Linear transformations')
+
+        self.layout = QtWidgets.QGridLayout(self) # Set the layout
+
+        self.addToLayout(self.canvas, [0,0],[10,10]) # add the plot to the layout
 
     def scatter(self,xyzgrid:list,colors:list):
         """Scatter the data
@@ -59,4 +62,9 @@ def scatter(self,xyzgrid:list,colors:list):
             colors
         """
         self.axes.clear()
-        self.axes.scatter(xyzgrid[0],xyzgrid[1],xyzgrid[2],c = colors,alpha=0.7)
+        self.axes.scatter(xyzgrid[0],xyzgrid[1],xyzgrid[2],c = colors,alpha=0.7)
+
+    def addToLayout(self, widget, position:list = [0,0], size:list = [1,1]):
+        """Add widget to layout
+        """
+        self.layout.addWidget(widget,position[1],position[0],size[1],size[0])

main.py

@@ -12,6 +12,8 @@
 
 from Plotting import PlotWidget         #The plot object
 
+from MatrixButtons import MatrixButtons
+
 def transformPoint(x,y,z):
     """Return the vector transformed
     
@@ -85,12 +87,17 @@ def adjust_plot(axes):
 
     colors = list(map(colorizer, originalGrid[0], originalGrid[1], originalGrid[2])) # Asign the colors to the points
 
-    app = QtWidgets.QApplication([])
-    plot = PlotWidget()
+    app = QtWidgets.QApplication([]) # Create the app
+    plot = PlotWidget() # Get the widget where the plot is
+
+    plot.scatter(xyzgrid,colors) # Scatter
+
+    adjust_plot(plot.axes) # Set the limits of the axes
 
-    plot.scatter(xyzgrid,colors)
+    interface = MatrixButtons() # Get the interface
+    plot.addToLayout(interface,[11,4],[10,2]) # add the interface to the plot layout
 
-    adjust_plot(plot.axes)
+    interface.setMatrix(transformationMatrix)
 
-    plot.show()
+    plot.show() 
     app.exec()