adding upie

Author: jposada202020

circuitpython_uplot/ubar.py

@@ -17,6 +17,11 @@
 
 # pylint: disable=too-many-arguments, invalid-name, protected-access
 # pylint: disable=too-few-public-methods, no-self-use
+
+__version__ = "0.0.0+auto.0"
+__repo__ = "https://github.com/adafruit/CircuitPython_uplot.git"
+
+
 class ubar:
     """
     Main class to display different graphics

circuitpython_uplot/upie.py

@@ -0,0 +1,76 @@
+# SPDX-FileCopyrightText: Copyright (c) 2023 Jose D. Montoya
+#
+# SPDX-License-Identifier: MIT
+
+"""
+
+`upie`
+================================================================================
+
+CircuitPython pie graph
+
+* Author(s): Jose D. Montoya
+
+
+"""
+
+import math
+import bitmaptools
+
+__version__ = "0.0.0+auto.0"
+__repo__ = "https://github.com/adafruit/CircuitPython_uplot.git"
+
+# pylint: disable=too-many-arguments, invalid-name, no-self-use
+class upie:
+    """
+    Class to draw pie
+    """
+
+    def __init__(self, plot, data: list, x: int = 130, y: int = 130, radius: int = 50):
+        """
+
+        :param plot: Plot object for the upie to be drawn
+        :param list data: data to make the pie
+        :param x: pie center x coordinate
+        :param y: pie center y coordinate
+        :param int radius: pie radius
+
+        """
+
+        plot._drawbox()
+
+        step = 2
+        total = sum(data)
+        per = [int(i / total * 360) for i in data]
+
+        start = 0
+        end = 0
+        index_color = 4
+
+        for pie in per:
+            end = end + pie
+            for i in range(start, end, step):
+                self.draw_lines(plot._plotbitmap, x, y, radius, i, index_color)
+            start = start + pie
+            index_color = index_color + 1
+
+    def draw_lines(self, bitmap, x, y, radius, angle, color):
+        """
+
+        :param bitmap: bitmap to be drawn in
+        :param x: center x coordinate
+        :param y: center y coordinate
+        :param radius: pie radius in pixels
+        :param angle: line angle
+        :param color: color index
+        :return: None
+
+        """
+        return bitmaptools.draw_line(
+            bitmap,
+            x,
+            y,
+            x + int(radius * math.cos(angle * math.pi / 180)),
+            y + int(radius * math.sin(angle * math.pi / 180)),
+            color,
+        )

circuitpython_uplot/uplot.py

@@ -68,26 +68,37 @@ def __init__(
 
         self.padding = padding
         self._newxmin = padding
-        self._newxmax = width - 1 * padding
-        self._newymin = height - 1 * padding
+        self._newxmax = width - padding
+        self._newymin = height - padding
         self._newymax = padding
 
         self._tickheight = 8
         self._tickcolor = 0xFFFFFF
         self._showticks = False
         self._tickgrid = False
 
+        self._barcolor = 0xFFFFFF
+
+        self._piecolor = 0xFFFFFF
+
         self._width = width
         self._height = height
 
         self._axeslinethikness = 1
 
-        self._plotbitmap = displayio.Bitmap(width, height, 4)
+        self._plotbitmap = displayio.Bitmap(width, height, 10)
 
-        self._plot_palette = displayio.Palette(4)
+        self._plot_palette = displayio.Palette(10)
         self._plot_palette[0] = 0x000000
         self._plot_palette[1] = 0xFFFFFF
         self._plot_palette[2] = self._tickcolor
+        self._plot_palette[3] = self._barcolor
+        self._plot_palette[4] = 0xFFFF00
+        self._plot_palette[5] = 0xFF0000
+        self._plot_palette[6] = 0x7428EF
+        self._plot_palette[7] = 0x005E99
+        self._plot_palette[8] = 0x00A76D
+        self._plot_palette[9] = 0x2C4971
         self.append(
             displayio.TileGrid(
                 self._plotbitmap, pixel_shader=self._plot_palette, x=x, y=y

circuitpython_uplot/uscatter.py

@@ -18,6 +18,9 @@
 import displayio
 from vectorio import Circle
 
+__version__ = "0.0.0+auto.0"
+__repo__ = "https://github.com/adafruit/CircuitPython_uplot.git"
+
 # pylint: disable=too-few-public-methods, invalid-name
 class uscatter:
     """

docs/api.rst

@@ -10,6 +10,8 @@
 .. automodule:: circuitpython_uplot.uscatter
     :members:
 
-
 .. automodule:: circuitpython_uplot.ubar
     :members:
+
+.. automodule:: circuitpython_uplot.upie
+    :members:

docs/examples.rst

@@ -70,3 +70,12 @@ ubar example
     :caption: examples/uplot_ubar_example.py
     :linenos:
 .. image:: ../docs/uplot_ex8.jpg
+
+Upie Example
+----------------
+
+upie example
+
+.. literalinclude:: ../examples/uplot_upie_example.py
+    :caption: examples/uplot_upie_example.py
+    :linenos:

examples/uplot_upie_example.py

@@ -0,0 +1,27 @@
+# SPDX-FileCopyrightText: Copyright (c) 2023 Jose D. Montoya
+#
+# SPDX-License-Identifier: Unlicense
+
+import time
+import board
+from circuitpython_uplot.uplot import Uplot
+from circuitpython_uplot.upie import upie
+
+
+# Setting up the display
+display = board.DISPLAY
+
+plot = Uplot(0, 0, display.width, display.height)
+
+# Setting up tick parameters
+plot.axs_params(axstype="box")
+a = [5, 2, 7]
+
+upie(plot, a)
+
+# Plotting and showing the plot
+display.show(plot)
+
+# Adding some wait time
+while True:
+    time.sleep(1)