kbasic 0.1.6__tar.gz → 0.1.7__tar.gz

{kbasic-0.1.6 → kbasic-0.1.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: kbasic
-Version: 0.1.6
+Version: 0.1.7
 Summary: Keyan's basic utility functions.
 Requires-Dist: numpy>=2.4.2
 Requires-Dist: pylatexenc>=2.10

{kbasic-0.1.6 → kbasic-0.1.7}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "kbasic"
-version = "0.1.6"
+version = "0.1.7"
 description = "Keyan's basic utility functions."
 readme = "README.md"
 requires-python = ">=3.11"

{kbasic-0.1.6 → kbasic-0.1.7}/src/kbasic/array.py

@@ -56,7 +56,8 @@ def nan_clip(*args):
     return nanless_args
 def interpolate2d(
         data, factor: int, 
-        method: str = 'linear'
+        method: str = 'linear',
+        periodic: bool = True
 ) -> ndarray:
     """interpolate 2d data on a regular grid by an even factor
 
@@ -68,8 +69,16 @@ def interpolate2d(
     Returns:
         ndarray[ndarray]: an interpolated grid of data
     """
+    assert periodic, "non-periodic version not implemented yet! make one but be careful because the shapes will be awkward"
     Nx, Ny = array(data).shape
-    return RegularGridInterpolator((arange(Nx), arange(Ny)), data, method=method)(mgrid[:Nx+1/factor:1/factor, :Ny+1/factor:1/factor].T).T
+    grid = (arange(Nx+1), arange(Ny+1))
+    data_repeat = np.zeros((Nx+1, Ny+1))
+    data_repeat[:Nx, :Ny] = data 
+    data_repeat[-1,:-1] = data[0]
+    data_repeat[:-1,-1] = data[:,0]
+    data_repeat[-1,-1] = data[-1,-1]
+    new_grid = mgrid[:Nx:1/factor, :Ny:1/factor].T
+    return RegularGridInterpolator(grid, data_repeat, method=method)(new_grid).T
 
 # !==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==
 # >-|===|>                            Decorators                           <|===|-<

{kbasic-0.1.6 → kbasic-0.1.7}/src/kbasic/shell.py

@@ -59,6 +59,19 @@ qs = anvil_queue
 # !==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==
 # >-|===|>                             Classes                             <|===|-<
 # !==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!== 
+class Color:
+    HEADER = '\033[95m'
+    OKBLUE = '\033[94m'
+    OKCYAN = '\033[96m'
+    OKGREEN = '\033[92m'
+    WARNING = '\033[93m'
+    FAIL = '\033[91m'
+    ENDC = '\033[0m'
+    BOLD = '\033[1m'
+    UNDERLINE = '\033[4m'
+    def warn(self, text: str):
+        print(self.WARNING + text + self.ENC)
+
 class AnvilJob:
     def __init__(self, queue_row: str, sep="DISTINCTSEPERATOR"):
         self.sep = sep