polytope-python 2.1.0__tar.gz → 2.1.2__tar.gz

{polytope_python-2.1.0 → polytope_python-2.1.2}/.gitignore

@@ -57,6 +57,7 @@ target
 rust_deployment_venv
 *.so
 *.nc
+*.grb
 
 test_w_qubed
 rust_venv
@@ -75,4 +76,6 @@ performance_plots_qubed
 *.nc
 **/*venv*
 
+bugfix*
+
 icon_grids

{polytope_python-2.1.0/polytope_python.egg-info → polytope_python-2.1.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: polytope-python
-Version: 2.1.0
+Version: 2.1.2
 Summary: Polytope datacube feature extraction library
 Author-email: "European Centre for Medium-Range Weather Forecasts (ECMWF)" <software.support@ecmwf.int>
 Maintainer-email: James Hawkes <James.Hawkes@ecmwf.int>, Mathilde Leuridan <Mathilde.Leuridan@ecmwf.int>

{polytope_python-2.1.0 → polytope_python-2.1.2}/polytope_feature/_version.py

@@ -1,2 +1,2 @@
 # Do not change! Do not track in version control!
-__version__ = "2.1.0"
+__version__ = "2.1.2"

{polytope_python-2.1.0 → polytope_python-2.1.2}/polytope_feature/datacube/tensor_index_tree.py

@@ -205,11 +205,23 @@ class TensorIndexTree(object):
                 old_parent.remove_branch()
 
     def remove_compressed_branch(self, value):
-        if value in self.values:
-            if len(self.values) == 1:
-                self.remove_branch()
-            else:
-                self.values = tuple(val for val in self.values if val != value)
+        if value not in self.values:
+            return
+
+        if len(self.values) == 1:
+            self.remove_branch()
+            return
+
+        self.values = tuple(val for val in self.values if val != value)
+        parent = self.parent
+        if parent is None:
+            return
+
+        for sibling in parent.children:
+            if sibling is not self:
+                if self == sibling:
+                    self.remove_branch()
+                    return
 
     def flatten(self):
         path = DatacubePath()

{polytope_python-2.1.0 → polytope_python-2.1.2}/polytope_feature/datacube/transformations/datacube_mappers/mapper_types/local_regular.py

@@ -1,5 +1,7 @@
 import bisect
 
+import numpy as np
+
 from ..datacube_mappers import DatacubeMapper
 
 
@@ -45,15 +47,28 @@ class LocalRegularGridMapper(DatacubeMapper):
         if self._axis_reversed[mapped_axes[1]]:
             raise NotImplementedError("Local regular grid with second axis in decreasing order is not supported")
 
+        self._second_axis_vals = np.arange(
+            self._second_axis_min,
+            self._second_axis_min + self._second_deg_increment * (self.second_resolution + 1),
+            self._second_deg_increment,
+            dtype=np.float64,
+        )
+
     def first_axis_vals(self):
         if self._axis_reversed[self._mapped_axes[0]]:
-            first_ax_vals = [
-                self._first_axis_max - i * self._first_deg_increment for i in range(self.first_resolution + 1)
-            ]
+            first_ax_vals = np.arange(
+                self._first_axis_max,
+                self._first_axis_max - self._first_deg_increment * (self.first_resolution + 1),
+                -self._first_deg_increment,
+                dtype=np.float64,
+            )
         else:
-            first_ax_vals = [
-                self._first_axis_min + i * self._first_deg_increment for i in range(self.first_resolution + 1)
-            ]
+            first_ax_vals = np.arange(
+                self._first_axis_min,
+                self._first_axis_min + self._first_deg_increment * (self.first_resolution + 1),
+                self._first_deg_increment,
+                dtype=np.float64,
+            )
         return first_ax_vals
 
     def map_first_axis(self, lower, upper):
@@ -62,10 +77,7 @@ class LocalRegularGridMapper(DatacubeMapper):
         return return_vals
 
     def second_axis_vals(self, first_val):
-        second_ax_vals = [
-            self._second_axis_min + i * self._second_deg_increment for i in range(self.second_resolution + 1)
-        ]
-        return second_ax_vals
+        return self._second_axis_vals
 
     def map_second_axis(self, first_val, lower, upper):
         axis_lines = self.second_axis_vals(first_val)
@@ -89,16 +101,26 @@ class LocalRegularGridMapper(DatacubeMapper):
         return first_idx * self.second_resolution
 
     def unmap(self, first_val, second_vals, unmapped_idx=None):
-        tol = 1e-8
-        first_val = [i for i in self._first_axis_vals if first_val[0] - tol <= i <= first_val[0] + tol][0]
-        first_idx = self._first_axis_vals.index(first_val)
-        return_idxs = []
-        for second_val in second_vals:
-            second_val = [i for i in self.second_axis_vals(first_val) if second_val - tol <= i <= second_val + tol][0]
-            second_idx = self.second_axis_vals(first_val).index(second_val)
-            final_index = self.axes_idx_to_regular_idx(first_idx, second_idx)
-            return_idxs.append(final_index)
-        return return_idxs
+        first_array = self._first_axis_vals
+        second_array = self._second_axis_vals
+        second_vals = np.asarray(second_vals)
+        descending = self._axis_reversed[self._mapped_axes[0]]
+        if descending:
+            # right descending order for searchsorted
+            first_idx = np.searchsorted(first_array[::-1], first_val[0])
+            first_idx = len(first_array) - 1 - first_idx
+        else:
+            first_idx = np.searchsorted(first_array, first_val[0])
+        if first_idx > 0 and first_idx < len(first_array):
+            left_val = first_array[first_idx - 1]
+            right_val = first_array[first_idx]
+            if abs(first_val[0] - left_val) < abs(first_val[0] - right_val):
+                first_idx -= 1
+        second_idxs = np.searchsorted(second_array, second_vals)
+
+        # map to grid idx
+        final_idxs = first_idx * (self.second_resolution + 1) + second_idxs
+        return final_idxs
 
 
 # md5 grid hash in form {resolution : hash}

{polytope_python-2.1.0 → polytope_python-2.1.2/polytope_python.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: polytope-python
-Version: 2.1.0
+Version: 2.1.2
 Summary: Polytope datacube feature extraction library
 Author-email: "European Centre for Medium-Range Weather Forecasts (ECMWF)" <software.support@ecmwf.int>
 Maintainer-email: James Hawkes <James.Hawkes@ecmwf.int>, Mathilde Leuridan <Mathilde.Leuridan@ecmwf.int>