Matrix interface and functionality is ready.

TODO:
- Examples comboBox

Author: alhazacod

MatrixButtons.py

@@ -40,8 +40,8 @@ def __init__(self):
         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])
+        self.plotButton = QtWidgets.QPushButton('Plot')
+        self.addToLayout(self.plotButton,[1,3],[1,1])
 
 
     def addToLayout(self, widget, position:list, size:list):
@@ -63,4 +63,4 @@ def setMatrix(self,matrix):
         # Third row
         self.matrix02.setText(str(matrix[2,0]))
         self.matrix12.setText(str(matrix[2,1]))
-        self.matrix22.setText(str(matrix[2,2]))
+        self.matrix22.setText(str(matrix[2,2]))

main.py

@@ -4,17 +4,17 @@
 Date: January 27 2021
 """
 from sympy.matrices import Matrix       #It's easier work with matrices in sympy than numpy
-from sympy import sin,cos,pi            #Trigonometric functions
+from sympy import sin,cos,pi,N          #Trigonometric functions
 import matplotlib.pyplot as plt         #Plotting
 from numpy import linspace,column_stack,array #For create the linspace for the plotting
 
 import PyQt5.QtWidgets as QtWidgets     #Python GUI
 
 from Plotting import PlotWidget         #The plot object
 
-from MatrixButtons import MatrixButtons
+from MatrixButtons import MatrixButtons # Interface class
 
-def transformPoint(x,y,z):
+def transformPoint(x,y,z,transformationMatrix):
     """Return the vector transformed
     
     """
@@ -60,7 +60,8 @@ def get_transformedGrid(originalGrid, transformationMatrix):
     transformationMatrix
         
     """
-    transformedGrid = transformPoint(originalGrid[0],originalGrid[1],originalGrid[2])
+    transformedGrid = transformPoint(originalGrid[0],originalGrid[1],originalGrid[2],transformationMatrix)
+
     return array([transformedGrid[0,:][0,:],     #
                   transformedGrid[1,:][0,:],     # Save the transformed grid as a numpy array 
                   transformedGrid[2,:][0,:]])    #
@@ -73,14 +74,37 @@ def adjust_plot(axes):
     axes.set_xlabel('x')   #
     axes.set_ylabel('y')   # Axes names
     axes.set_zlabel('z')   #
-    axes.grid(False)       # Don't show the grid
+    axes.grid(True)       # Don't show the grid
     axes.view_init(10,5)   # View init at 15 and 5 degrees
 
+def exceptionDialog(msg):
+    dlg = QtWidgets.QErrorMessage()
+    dlg.showMessage(msg)
+    dlg.exec_()
+
+
+def transformAndPlot(interface, plot):
+    try:
+        transformationMatrix = Matrix([[N(interface.matrix00.text()), N(interface.matrix10.text()), N(interface.matrix20.text())],
+                                       [N(interface.matrix01.text()), N(interface.matrix11.text()), N(interface.matrix21.text())],
+                                       [N(interface.matrix02.text()), N(interface.matrix12.text()), N(interface.matrix22.text())]])
+
+        
+        xyzgrid = get_transformedGrid(originalGrid, transformationMatrix) # get a transformed grid
+
+        plot.scatter(xyzgrid,colors)
+        adjust_plot(plot.axes)
+        plot.fig.canvas.draw()
+        plot.fig.canvas.flush_events()
+    except ValueError as err:
+        print(err)
+        exceptionDialog('An error has ocurred. Remember that you only can write operations between numbers or trigonometric functions in the matrix boxes. ex. sin(3*pi/2) or 30+0.5')
+
 if __name__ == "__main__":
     """Transformation matrix"""
     transformationMatrix = Matrix([[1, 0, 0],
                                    [0, 1, 0],
-                                   [0, 1, 1]])
+                                   [0, 0, 1]])
 
     originalGrid = get_grid(-2,2) # Create a straigh grid
     xyzgrid = get_transformedGrid(originalGrid, transformationMatrix) # get a transformed grid
@@ -94,10 +118,14 @@ def adjust_plot(axes):
 
     adjust_plot(plot.axes) # Set the limits of the axes
 
+    plot.addToLayout(QtWidgets.QLabel('Titulo'),[11,0],[1,1])
+
     interface = MatrixButtons() # Get the interface
     plot.addToLayout(interface,[11,4],[10,2]) # add the interface to the plot layout
 
     interface.setMatrix(transformationMatrix)
 
+    interface.plotButton.clicked.connect(lambda: transformAndPlot(interface,plot))
+
     plot.show() 
     app.exec()