Everything setted up for start with the interactive functions

Author: alhazacod

Plotting.py

@@ -4,12 +4,10 @@
 
 """
 TODO:
-    - I should add the option of default colors and alpha but i couldn't, it's something to fix
+    -
 """
 class PlotWidget(QtWidgets.QWidget):
-    """
-
-    Widget for PyQt5 with the plot
+    """Widget for PyQt5 with the plot
 
     ...
 
@@ -36,13 +34,29 @@ class PlotWidget(QtWidgets.QWidget):
 
     canvas: matplotlib.backends.backend_qt5agg.FigureCanvasQTAgg
     """
-    def __init__(self, x: list, y: list, z: list, colors: tuple, alpha: float):
+    def __init__(self):
         super().__init__()
-        self.fig = Figure()
-        self.canvas = FigureCanvas(self.fig)
-        self.axes = self.fig.add_subplot(111,projection='3d')
+        self.fig = Figure() # Create a matplotlib figure
+        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 = QtWidgets.QVBoxLayout(self) 
         layout.addWidget(self.canvas)
 
-        self.axes.scatter(x, y, z, c = colors, alpha = alpha)
+    def scatter(self,xyzgrid:list,colors:list):
+        """Scatter the data
+    
+        ...
+
+        Parameters
+        ----------
+        xyzgrid:list
+            data for plotting
+            xyzgrid[0] the x data
+            xyzgrid[1] the y data
+            xyzgrid[2] the z
+        colors:list
+            colors
+        """
+        self.axes.clear()
+        self.axes.scatter(xyzgrid[0],xyzgrid[1],xyzgrid[2],c = colors,alpha=0.7)

main.py

@@ -12,11 +12,6 @@
 
 from Plotting import PlotWidget         #The plot object
 
-"""Transformation matrix"""
-transformationMatrix = Matrix([[1, 0, 0],
-                               [0, 0.5, 0],
-                               [0, 0, 1]])
-
 def transformPoint(x,y,z):
     """Return the vector transformed
     
@@ -51,28 +46,51 @@ def get_grid(min,max):
     xyzgrid = column_stack([[x, y,z] for x in xvals for y in yvals for z in zvals])
     return xyzgrid
 
-if __name__ == "__main__":
+def get_transformedGrid(originalGrid, transformationMatrix):
+    """Return a grid in x, y and z transformed
+    
+    ...
 
-    originalGrid = get_grid(-2,2)
+    Parameters
+    ----------
+    originalGrid
+        
+    transformationMatrix
+        
+    """
     transformedGrid = transformPoint(originalGrid[0],originalGrid[1],originalGrid[2])
-    xyzgrid = array([transformedGrid[0,:][0,:],     #
-                     transformedGrid[1,:][0,:],     # Save the transformed grid as a numpy array 
-                     transformedGrid[2,:][0,:]])    #
+    return array([transformedGrid[0,:][0,:],     #
+                  transformedGrid[1,:][0,:],     # Save the transformed grid as a numpy array 
+                  transformedGrid[2,:][0,:]])    #
+
+def adjust_plot(axes):
+    axes.set_xlim([-2, 2]) #
+    axes.set_ylim([-2, 2]) # Set the limits of the plot
+    axes.set_zlim([-2, 2]) #
+    axes.axis('on')        # Show the axis
+    axes.set_xlabel('x')   #
+    axes.set_ylabel('y')   # Axes names
+    axes.set_zlabel('z')   #
+    axes.grid(False)       # Don't show the grid
+    axes.view_init(10,5)   # View init at 15 and 5 degrees
+
+if __name__ == "__main__":
+    """Transformation matrix"""
+    transformationMatrix = Matrix([[1, 0, 0],
+                                   [0, 1, 0],
+                                   [0, 1, 1]])
+
+    originalGrid = get_grid(-2,2) # Create a straigh grid
+    xyzgrid = get_transformedGrid(originalGrid, transformationMatrix) # get a transformed grid
 
     colors = list(map(colorizer, originalGrid[0], originalGrid[1], originalGrid[2])) # Asign the colors to the points
 
     app = QtWidgets.QApplication([])
-    plot = PlotWidget(xyzgrid[0], xyzgrid[1], xyzgrid[2], colors = colors, alpha = 0.7)
-
-    plot.axes.set_xlim([-2, 2]) #
-    plot.axes.set_ylim([-2, 2]) # Set the limits of the plot
-    plot.axes.set_zlim([-2, 2]) #
-    plot.axes.axis('on')        # Show the axis
-    plot.axes.set_xlabel('x')   #
-    plot.axes.set_ylabel('y')   # Axes names
-    plot.axes.set_zlabel('z')   #
-    plot.axes.grid(False)       # Don't show the grid
-    plot.axes.view_init(10,5)   # View init at 15 and 5 degrees
+    plot = PlotWidget()
+
+    plot.scatter(xyzgrid,colors)
+
+    adjust_plot(plot.axes)
 
     plot.show()
     app.exec()