cleaning and formatting

Author: lionel42

examples/blitting.py

@@ -1,7 +1,7 @@
 import matplotlib
 import matplotlib.pyplot as plt
 
-matplotlib.use('pygame')
+matplotlib.use("pygame")
 import numpy as np
 
 x = np.linspace(0, 2 * np.pi, 100)
@@ -45,4 +45,4 @@
     # flush any pending GUI events, re-painting the screen if needed
     fig.canvas.flush_events()
     # you can put a pause in if you want to slow things down
-    plt.pause(.01)
+    plt.pause(0.01)

examples/blitting_class.py

@@ -1,7 +1,9 @@
 import matplotlib.pyplot as plt
 import numpy as np
 import matplotlib
-matplotlib.use('pygame')
+
+matplotlib.use("pygame")
+
 
 class BlitManager:
     def __init__(self, canvas, animated_artists=()):
@@ -75,6 +77,7 @@ def update(self):
         # let the GUI event loop process anything it has to do
         cv.flush_events()
 
+
 x = np.linspace(0, 2 * np.pi, 100)
 # make a new figure
 fig, ax = plt.subplots()
@@ -102,4 +105,4 @@ def update(self):
     fr_number.set_text("frame: {j}".format(j=j))
     # tell the blitting manager to do its thing
     bm.update()
-    # plt.pause(1)
+    # plt.pause(1)

examples/gui_window.py

@@ -1,6 +1,5 @@
 """Simple script to test from https://pygame-gui.readthedocs.io/en/latest/quick_start.html"""
 
-
 import pygame
 import pygame_gui
 
@@ -11,39 +10,37 @@
 
 pygame.init()
 
-pygame.display.set_caption('Test')
+pygame.display.set_caption("Test")
 window_surface = pygame.display.set_mode((800, 600))
 
 background = pygame.Surface((800, 600))
-background.fill(pygame.Color('#000000'))
+background.fill(pygame.Color("#000000"))
 
 manager = pygame_gui.UIManager((800, 600))
 
-fig, axes = plt.subplots(1, 1,)
-axes.plot([1,2], [1,2], color='green', label='test')
+fig, axes = plt.subplots(1, 1)
+axes.plot([1, 2], [1, 2], color="green", label="test")
 fig.canvas.draw()
 
 
-fig2, axes2 = plt.subplots(1, 1,)
-axes2.plot([1,2], [1,2], color='blue', label='test')
+fig2, axes2 = plt.subplots(1, 1)
+axes2.plot([1, 2], [1, 2], color="blue", label="test")
 fig2.canvas.draw()
 
 
 plot_window = UIPlotWindow(
     rect=pygame.Rect((350, 275), (300, 200)),
     manager=manager,
     figuresurface=fig,
-    resizable=True
+    resizable=True,
 )
 
 
-
-
 plot_window2 = UIPlotWindow(
     rect=pygame.Rect((350, 275), (200, 200)),
     manager=manager,
     figuresurface=fig2,
-    resizable=False
+    resizable=False,
 )
 
 
@@ -52,12 +49,11 @@
 is_running = True
 
 while is_running:
-    time_delta = clock.tick(60)/1000.0
+    time_delta = clock.tick(60) / 1000.0
     for event in pygame.event.get():
         if event.type == pygame.QUIT:
             is_running = False
 
-
         manager.process_events(event)
 
     manager.update(time_delta)
@@ -66,7 +62,6 @@
     window_surface.blit(background, (0, 0))
     manager.draw_ui(window_surface)
 
-
     # print(plot_window2.get_container().get_size())
 
     pygame.display.update()

examples/save_to_file.py

@@ -1,24 +1,18 @@
-import sys
 import os
-from numpy.ma.core import size
-import pygame
-
 import matplotlib
-matplotlib.use('pygame')
-#matplotlib.use('Qt4Agg')
 
-import matplotlib.pyplot as plt
-import matplotlib.figure as fg
+matplotlib.use("pygame")
 
 import matplotlib.pyplot as plt
+
 fig = plt.figure()
 
 print(fig.canvas.get_supported_filetypes())
 
-plt.plot([1,2], [1,2], color='green')
-plt.text(1.5, 1.5, '2', size=50)
-plt.xlabel('swag')
+plt.plot([1, 2], [1, 2], color="green")
+plt.text(1.5, 1.5, "2", size=50)
+plt.xlabel("swag")
 
-plt.savefig('images' + os.sep + 'test.jpg')
-plt.savefig('images' + os.sep + 'test.bmp')
-plt.savefig('images' + os.sep + 'test2.jpg')
+plt.savefig("images" + os.sep + "test.jpg")
+plt.savefig("images" + os.sep + "test.bmp")
+plt.savefig("images" + os.sep + "test2.jpg")

examples/show.py

@@ -1,19 +1,15 @@
-import sys
 import os
-from numpy.ma.core import size
-import pygame
 import numpy as np
 
 import matplotlib
-#matplotlib.use('pygame')
-matplotlib.use('pygame')
 
 import matplotlib.pyplot as plt
-import matplotlib.figure as fg
-
 import matplotlib.image as mpimg
 
 
+matplotlib.use("pygame")
+
+
 def plot_error_bars_ex(ax):
     # example data
     x = np.array([0.5, 1.0, 1.5, 2.0, 2.5, 3.0, 3.5, 4.0, 4.5, 5.0])
@@ -24,24 +20,29 @@ def plot_error_bars_ex(ax):
     # lower & upper limits of the error
     lolims = np.array([0, 0, 1, 0, 1, 0, 0, 0, 1, 0], dtype=bool)
     uplims = np.array([0, 1, 0, 0, 0, 1, 0, 0, 0, 1], dtype=bool)
-    ls = 'dotted'
+    ls = "dotted"
 
     # standard error bars
     ax.errorbar(x, y, xerr=xerr, yerr=yerr, linestyle=ls)
 
     # including upper limits
-    ax.errorbar(x, y + 0.5, xerr=xerr, yerr=yerr, uplims=uplims,
-                linestyle=ls)
+    ax.errorbar(x, y + 0.5, xerr=xerr, yerr=yerr, uplims=uplims, linestyle=ls)
 
     # including lower limits
-    ax.errorbar(x, y + 1.0, xerr=xerr, yerr=yerr, lolims=lolims,
-                linestyle=ls)
+    ax.errorbar(x, y + 1.0, xerr=xerr, yerr=yerr, lolims=lolims, linestyle=ls)
 
     # including upper and lower limits
-    ax.errorbar(x, y + 1.5, xerr=xerr, yerr=yerr,
-                lolims=lolims, uplims=uplims,
-                marker='o', markersize=8,
-                linestyle=ls)
+    ax.errorbar(
+        x,
+        y + 1.5,
+        xerr=xerr,
+        yerr=yerr,
+        lolims=lolims,
+        uplims=uplims,
+        marker="o",
+        markersize=8,
+        linestyle=ls,
+    )
 
     # Plot a series with lower and upper limits in both x & y
     # constant x-error with varying y-error
@@ -58,15 +59,23 @@ def plot_error_bars_ex(ax):
     uplims[[3]] = True  # only limited at this index
 
     # do the plotting
-    ax.errorbar(x, y + 2.1, xerr=xerr, yerr=yerr,
-                xlolims=xlolims, xuplims=xuplims,
-                uplims=uplims, lolims=lolims,
-                marker='o', markersize=8,
-                linestyle='none')
+    ax.errorbar(
+        x,
+        y + 2.1,
+        xerr=xerr,
+        yerr=yerr,
+        xlolims=xlolims,
+        xuplims=xuplims,
+        uplims=uplims,
+        lolims=lolims,
+        marker="o",
+        markersize=8,
+        linestyle="none",
+    )
 
     # tidy up the figure
     ax.set_xlim((0, 5.5))
-    ax.set_title('Errorbar upper and lower limits')
+    ax.set_title("Errorbar upper and lower limits")
 
 
 def plot_violin(ax):
@@ -78,60 +87,58 @@ def adjacent_values(vals, q1, q3):
         lower_adjacent_value = np.clip(lower_adjacent_value, vals[0], q1)
         return lower_adjacent_value, upper_adjacent_value
 
-
     def set_axis_style(ax, labels):
-        ax.xaxis.set_tick_params(direction='out')
-        ax.xaxis.set_ticks_position('bottom')
+        ax.xaxis.set_tick_params(direction="out")
+        ax.xaxis.set_ticks_position("bottom")
         ax.set_xticks(np.arange(1, len(labels) + 1))
         ax.set_xticklabels(labels)
         ax.set_xlim(0.25, len(labels) + 0.75)
-        ax.set_xlabel('Sample name')
-
+        ax.set_xlabel("Sample name")
 
     # create test data
     np.random.seed(19680801)
     data = [sorted(np.random.normal(0, std, 100)) for std in range(1, 5)]
 
+    ax.set_title("Customized violin plot")
+    parts = ax.violinplot(data, showmeans=False, showmedians=False, showextrema=False)
 
-
-    ax.set_title('Customized violin plot')
-    parts = ax.violinplot(
-            data, showmeans=False, showmedians=False,
-            showextrema=False)
-
-    for pc in parts['bodies']:
-        pc.set_facecolor('#D43F3A')
-        pc.set_edgecolor('black')
+    for pc in parts["bodies"]:
+        pc.set_facecolor("#D43F3A")
+        pc.set_edgecolor("black")
         pc.set_alpha(1)
 
     quartile1, medians, quartile3 = np.percentile(data, [25, 50, 75], axis=1)
-    whiskers = np.array([
-        adjacent_values(sorted_array, q1, q3)
-        for sorted_array, q1, q3 in zip(data, quartile1, quartile3)])
+    whiskers = np.array(
+        [
+            adjacent_values(sorted_array, q1, q3)
+            for sorted_array, q1, q3 in zip(data, quartile1, quartile3)
+        ]
+    )
     whiskers_min, whiskers_max = whiskers[:, 0], whiskers[:, 1]
 
     inds = np.arange(1, len(medians) + 1)
-    ax.scatter(inds, medians, marker='o', color='white', s=30, zorder=3)
-    ax.vlines(inds, quartile1, quartile3, color='k', linestyle='-', lw=5)
-    ax.vlines(inds, whiskers_min, whiskers_max, color='k', linestyle='-', lw=1)
+    ax.scatter(inds, medians, marker="o", color="white", s=30, zorder=3)
+    ax.vlines(inds, quartile1, quartile3, color="k", linestyle="-", lw=5)
+    ax.vlines(inds, whiskers_min, whiskers_max, color="k", linestyle="-", lw=1)
 
     # set style for the axes
-    labels = ['A', 'B', 'C', 'D']
+    labels = ["A", "B", "C", "D"]
     set_axis_style(ax, labels)
 
-fig, axes = plt.subplots(3,2,figsize=(16, 12))
+
+fig, axes = plt.subplots(3, 2, figsize=(16, 12))
 print(type(fig))
 
-axes[0, 0].plot([1,2], [1,2], color='green', label='test')
-axes[0, 0].plot([1,2], [1,1], color='orange', lw=5, label='test other larger')
+axes[0, 0].plot([1, 2], [1, 2], color="green", label="test")
+axes[0, 0].plot([1, 2], [1, 1], color="orange", lw=5, label="test other larger")
 # axes[0, 0].legend()
-axes[0, 1].text(0.5, 0.5, '2', size=50)
-axes[1, 0].set_xlabel('swag')
+axes[0, 1].text(0.5, 0.5, "2", size=50)
+axes[1, 0].set_xlabel("swag")
 axes[1, 0].fill_between([0, 1, 2], [1, 2, 3], [3, 4, 5])
 axes[1, 0].scatter([0, 1, 2], [2, 3, 4], s=50)
-axes[1, 1].imshow(mpimg.imread('images' + os.sep + 'long_dog.jpg'))
+axes[1, 1].imshow(mpimg.imread("images" + os.sep + "long_dog.jpg"))
 plot_error_bars_ex(axes[2, 1])
 
 plot_violin(axes[2, 0])
 
-plt.show()
+plt.show()

examples/show_in_gameloop.py

@@ -1,16 +1,15 @@
-
-
 import matplotlib
-matplotlib.use('pygame')
-#matplotlib.use('Qt4Agg')
+
+matplotlib.use("pygame")
+# matplotlib.use('Qt4Agg')
 
 import matplotlib.pyplot as plt
 
 import pygame
 import pygame.display
 
-fig, axes = plt.subplots(1, 1,)
-axes.plot([1,2], [1,2], color='green', label='test')
+fig, axes = plt.subplots(1, 1)
+axes.plot([1, 2], [1, 2], color="green", label="test")
 
 fig.canvas.draw()
 
@@ -23,4 +22,4 @@
         if event.type == pygame.QUIT:
             # Stop showing when quit
             show = False
-    pygame.display.update()
+    pygame.display.update()

pygame_matplotlib/__init__.py

@@ -4,6 +4,4 @@
 def pygame_color_to_plt(color: pygame.Color):
     """Convert a pygame Color to a matplot lib value."""
     # Interval from 0 to 1 in matplotlib
-    return tuple(
-        value / 255.0 for value in [color.r, color.g, color.b, color.a]
-    )
+    return tuple(value / 255.0 for value in [color.r, color.g, color.b, color.a])