squarenet 0.2.2__tar.gz → 0.2.4__tar.gz

{squarenet-0.2.2/src/squarenet.egg-info → squarenet-0.2.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: squarenet
-Version: 0.2.2
+Version: 0.2.4
 Summary: Sparse local operations for point clouds in any dimension.
 Author-email: ArmanddeCacqueray <armanddecacqueray@sfr.fr>
 Project-URL: Homepage, https://github.com/ArmanddeCacqueray/SquareNet

{squarenet-0.2.2 → squarenet-0.2.4}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "squarenet"
-version = "0.2.2"
+version = "0.2.4"
 description = "Sparse local operations for point clouds in any dimension."
 authors = [{ name = "ArmanddeCacqueray", email = "armanddecacqueray@sfr.fr" }]
 readme = "README.md"

{squarenet-0.2.2 → squarenet-0.2.4}/src/squarenet/squarenet.py

@@ -2,7 +2,7 @@ import numpy as np
 from .core import sort_increasing, carthesian_heuristics
 from .boards import checkerboard, checkerboard2D, checkerboard3D
 from .views import localview, lazylocalview
-from .utils import initpoint, dualgrid
+from .utils import initpoint, dualgrid, dualgridflat
 from warnings import warn
 
 class SquareNet:
@@ -37,6 +37,7 @@ class SquareNet:
             self.N, dtype = np.int32
         ).reshape(IJ_) #grid to sort heuritics
         self.invert_grid = dualgrid(self.grid) #for invertibility
+        self.invgridflat = dualgridflat(self.grid) #flat version
         self.packed = False #Flag for faster computations if possible
 
     def fit(self, points):
@@ -84,6 +85,7 @@ class SquareNet:
         # Save results
         self.grid = grid
         self.invert_grid = dualgrid(grid)
+        self.invgridflat = dualgridflat(grid)
         self.learning_curve = learning_curve
     
     def map(self, features):
@@ -102,7 +104,7 @@ class SquareNet:
         if not self.packed:
             return features.reshape(
                 -1, *features.shape[self.D:]
-                )[self.invert_grid.flatten()]
+                )[self.invgridflat]
         
         C = features.shape[self.D:]
         return features.reshape(-1, *C)

{squarenet-0.2.2 → squarenet-0.2.4}/src/squarenet/utils.py

@@ -16,6 +16,16 @@ def dualgrid(grid):
         dual_grid = np.ascontiguousarray(dual_grid)
         return dual_grid
 
+def dualgridflat(grid):
+    gr= grid.flatten()
+    N = len(grid)
+    dgf = np.zeros(N, dtype = np.int32)
+    identity = np.arange(N, dtype = np.int32)
+    dgf[gr] = identity
+    dgf = np.ascontiguousarray(dgf)
+    return dgf
+
+
 def ball(n, d):
     points = np.random.randn(n, d)
     radius = np.random.uniform(0, 1, size=n)**(1/d)

{squarenet-0.2.2 → squarenet-0.2.4/src/squarenet.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: squarenet
-Version: 0.2.2
+Version: 0.2.4
 Summary: Sparse local operations for point clouds in any dimension.
 Author-email: ArmanddeCacqueray <armanddecacqueray@sfr.fr>
 Project-URL: Homepage, https://github.com/ArmanddeCacqueray/SquareNet