kaxe 1.6.1.dev0__tar.gz → 1.6.2__tar.gz

{kaxe-1.6.1.dev0/src/kaxe.egg-info → kaxe-1.6.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kaxe
-Version: 1.6.1.dev0
+Version: 1.6.2
 Summary: A small graphing tool for functions, points, equations and more
 Author-email: Valter Yde Daugberg <valteryde@hotmail.com>
 Project-URL: Homepage, https://github.com/valteryde/kaxe

{kaxe-1.6.1.dev0 → kaxe-1.6.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "kaxe"
-version = "1.6.1.dev0"
+version = "1.6.2"
 authors = [
   { name="Valter Yde Daugberg", email="valteryde@hotmail.com" },
 ]

{kaxe-1.6.1.dev0 → kaxe-1.6.2}/src/kaxe/core/color.py

@@ -134,6 +134,32 @@ class Colormap:
         x -= x0
         return tuple(round(i) for i in (1 - x) * self.colorGradientSteps[x0] + (x) * self.colorGradientSteps[x1])
 
+    def map_array(self, values: np.ndarray, start: Union[int, float], end: Union[int, float]) -> np.ndarray:
+        """
+        Map a numeric array to RGBA colors using this colormap.
+
+        Returns an array with shape ``(*values.shape, 4)`` and dtype ``uint8``.
+        """
+        steps = np.asarray(self.colorGradientSteps, dtype=np.float64)
+        if end == start:
+            return np.broadcast_to(steps[0].astype(np.uint8), (*values.shape, 4)).copy()
+
+        n = len(steps) - 1
+        t = (values - start) / (end - start) * n
+        t = np.clip(t, 0, n)
+        i0 = np.floor(t).astype(np.intp)
+        i1 = np.minimum(i0 + 1, n)
+        frac = (t - i0)[..., np.newaxis]
+        colors = np.round((1 - frac) * steps[i0] + frac * steps[i1]).astype(np.uint8)
+
+        below = values < start
+        above = values > end
+        if np.any(below):
+            colors[below] = steps[0].astype(np.uint8)
+        if np.any(above):
+            colors[above] = steps[-1].astype(np.uint8)
+        return colors
+
     def setAlpha(self, alpha255):
         """
         Update the alpha (transparency) value for each color in the color gradient steps.

{kaxe-1.6.1.dev0 → kaxe-1.6.2}/src/kaxe/objects/d2/map.py

@@ -2,6 +2,7 @@
 import numpy as np
 from ..mapdata import heatcolormap
 import math
+from PIL import Image
 from ...core.shapes import shapes
 from ...core.text import Text
 from ...core.round import koundTeX
@@ -297,37 +298,55 @@ class HeatMap:
         )
 
     def finalize(self, parent):
-
-        # get size of one box
-        width, height = parent.scaled(*self.unitPerPixel)
-        width, height = math.ceil(width), math.ceil(height)
-
-        for rowNum, row in enumerate(self.data):
-
-            for cellNum, cell in enumerate(row):
-                
-                p = parent.pixel(
-                    cellNum*self.unitPerPixel[0] + self.position[0], 
-                    rowNum*self.unitPerPixel[1] + self.position[1]
-                )
-                if not parent.inside(*p): continue
-
-                w, h = width, height
-                p1 = parent.inversepixel(p[0]+w, p[1]+h)
-                p1 = parent.pixel(p1[0], p1[1])
-                if not parent.inside(*p1):
-                    p1 = parent.clamp(*p1)
-                    w = p1[0] - p[0]
-                    h = p1[1] - p[1]
-
-                    if w == 0 or h == 0:
-                        continue
-            
-                shapes.Rectangle(
-                    *p, math.ceil(w), math.ceil(h),
-                    color=self.cmap.getColor(cell, self.minValue, self.maxValue), 
-                    batch=self.batch
-                )
+        nrows = len(self.data)
+        if nrows == 0:
+            return
+        ncols = len(self.data[0])
+        if ncols == 0:
+            return
+
+        x0, y0 = self.position
+        x1 = x0 + ncols * self.unitPerPixel[0]
+        y1 = y0 + nrows * self.unitPerPixel[1]
+
+        px0, py0 = parent.pixel(x0, y0)
+        px1, py1 = parent.pixel(x1, y1)
+
+        left = int(math.floor(px0))
+        bottom = int(math.floor(py0))
+        right = int(math.ceil(px1))
+        top = int(math.ceil(py1))
+        pw = max(right - left, 1)
+        ph = max(top - bottom, 1)
+
+        rgba = self.cmap.map_array(
+            np.asarray(self.data, dtype=np.float64),
+            self.minValue,
+            self.maxValue,
+        )
+        rgba = rgba[::-1]
+
+        img = Image.fromarray(rgba)
+        if (pw, ph) != (ncols, nrows):
+            img = img.resize((pw, ph), Image.NEAREST)
+
+        wb = parent.windowBox
+        clip_l = max(left, wb[0])
+        clip_b = max(bottom, wb[1])
+        clip_r = min(left + pw, wb[2])
+        clip_t = min(bottom + ph, wb[3])
+        if clip_l >= clip_r or clip_b >= clip_t:
+            return
+
+        if (clip_l, clip_b, clip_r, clip_t) != (left, bottom, left + pw, bottom + ph):
+            src_left = clip_l - left
+            src_right = clip_r - left
+            src_top = ph - (clip_t - bottom)
+            src_bottom = ph - (clip_b - bottom)
+            img = img.crop((src_left, src_top, src_right, src_bottom))
+            left, bottom = clip_l, clip_b
+
+        self.img = shapes.ImageArray(np.asarray(img), left, bottom, batch=self.batch)
         
     
     def addColorScale(self, parent, digits=64):

{kaxe-1.6.1.dev0 → kaxe-1.6.2/src/kaxe.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kaxe
-Version: 1.6.1.dev0
+Version: 1.6.2
 Summary: A small graphing tool for functions, points, equations and more
 Author-email: Valter Yde Daugberg <valteryde@hotmail.com>
 Project-URL: Homepage, https://github.com/valteryde/kaxe