mergechannels 0.3.0__cp313-cp313-musllinux_1_2_armv7l.whl → 0.4.0__cp313-cp313-musllinux_1_2_armv7l.whl

mergechannels/__init__.py

@@ -3,11 +3,12 @@ from .mergechannels import (  # type: ignore
 	dispatch_single_channel,
 	dispatch_multi_channel,
 )
-from ._internal import merge
+from ._internal import merge, apply_color_map
 
 
 __all__ = [
 	'dispatch_single_channel',
 	'dispatch_multi_channel',
 	'merge',
+	'apply_color_map',
 ]

mergechannels/__init__.pyi

@@ -6,11 +6,20 @@ from python.mergechannels._blending import BLENDING_OPTIONS
 from python.mergechannels._luts import COLORMAPS
 
 
+def apply_color_map(
+	arr: np.ndarray,
+	color: COLORMAPS,
+	percentiles: tuple[float, float] | None = None,
+	saturation_limits: tuple[float, float] | None = None,
+) -> np.ndarray:
+    ...
+
 def merge(
-    arrs: np.ndarray | Sequence[np.ndarray],
-    colors: Sequence[COLORMAPS] = (),
-    blending: Literal[BLENDING_OPTIONS] = 'max',
-    saturation_limits: tuple[float, float] = (0.2, 99.8),
+	arrs: Sequence[np.ndarray],
+	colors: Sequence[COLORMAPS],
+	blending: BLENDING_OPTIONS = 'max',
+	percentiles: tuple[float, float] | Sequence[tuple[float, float]] | None = None,
+	saturation_limits: tuple[float, float] | Sequence[tuple[float, float]] | None = None,
 ) -> np.ndarray:
     ...
 

mergechannels/_internal.py

@@ -1,4 +1,7 @@
-from typing import Sequence
+from typing import (
+	Sequence,
+	Union,
+)
 
 import numpy as np
 
@@ -9,57 +12,48 @@ from mergechannels import (
 from ._luts import COLORMAPS
 from ._blending import BLENDING_OPTIONS
 
+def apply_color_map(
+	arr: np.ndarray,
+	color: COLORMAPS,
+	percentiles: Union[tuple[float, float], None] = None,
+	saturation_limits: Union[tuple[float, float], None] = None,
+) -> np.ndarray:
+	'''
+	apply a color map to an array
+	'''
+	if saturation_limits is None:
+		if percentiles is None:
+			percentiles = (1.1, 99.9)
+		low, high = np.percentile(arr, percentiles)
+		saturation_limits = (low, high)
+
+	return dispatch_single_channel(
+		array_reference=arr,
+		cmap_name=color,
+		limits=saturation_limits,
+	)
+
+
 def merge(
 	arrs: Sequence[np.ndarray],
 	colors: Sequence[COLORMAPS],
 	blending: BLENDING_OPTIONS = 'max',
-    saturation_limits: tuple[float, float] = (0.011, 0.999),
+	percentiles: Union[Sequence[tuple[float, float]], None] = None,
+	saturation_limits: Union[Sequence[tuple[float, float]], None] = None,
 ) -> np.ndarray:
 	'''
 	apply cmaps to arrays and blend the colors
 	'''
-	# region validation
-	n_arrs = len(arrs)
-	n_colors = len(colors)
-	if not n_arrs == n_colors and n_arrs > 0:
-		raise ValueError(
-			'Expected an equal number of arrays to colorize and colormap names to apply. '
-			f'Got {n_arrs} arrays and {n_colors} colors'
-		)
-	arr_shapes = [arr.shape for arr in arrs]
-	if not len(set(arr_shapes)) == 1:
-		raise ValueError(
-			f'Expected every array to have the same shape, got {arr_shapes}'
-		)
-	if len(arr_shapes[0]) not in (2, 3):
-		raise ValueError(
-			f'Expected every array to be 2D or 3D, got {arr_shapes[0]}'
-		)
-	arr_dtypes = [arr.dtype for arr in arrs]
-	if not len(set(arr_dtypes)) == 1:
-		raise ValueError(
-			f'Expected every array to have the same dtype, got {arr_dtypes}'
-		)
-	# endregion
-	if n_arrs == 1:
-		if arrs[0].dtype == 'uint8':
-			limits = (0, 255)
-		else:
-			low, high = np.percentile(arrs[0], np.array(saturation_limits) * 100)
-			limits = (low, high)
-		return dispatch_single_channel(
-			array_reference=arrs[0],
-			cmap_name=colors[0],
-			limits=limits,
-		)
-	else:
-		if all(arr.dtype == 'uint8' for arr in arrs):
-			limits = (0, 255)
-		else:
-			limits = tuple(np.percentile(arr, np.array(saturation_limits) * 100) for arr in arrs)
-		return dispatch_multi_channel(
-			array_references=arrs,
-			cmap_names=colors,
-			blending=blending,
-			limits=limits,  # type: ignore
+	if saturation_limits is None:
+		if percentiles is None:
+			percentiles = [(1.1, 99.9)] * len(arrs)
+		saturation_limits = tuple(
+			np.percentile(arr, ch_percentiles)
+			for arr, ch_percentiles in zip(arrs, percentiles)  # type: ignore
 		)
+	return dispatch_multi_channel(
+		array_references=arrs,
+		cmap_names=colors,
+		blending=blending,
+		limits=saturation_limits,  # type: ignore
+	)

{mergechannels-0.3.0.dist-info → mergechannels-0.4.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mergechannels
-Version: 0.3.0
+Version: 0.4.0
 Classifier: Programming Language :: Rust
 Classifier: Programming Language :: Python :: Implementation :: CPython
 Classifier: Programming Language :: Python :: Implementation :: PyPy
@@ -50,6 +50,7 @@ pip install git+https://github.com/zacswider/mergechannels.git
 
 ```python
 from skimage import data
+import matplotlib.pyplot as plt
 import mergechannels as mc
 
 img = data.camera()
@@ -116,10 +117,10 @@ b.imshow(
         channels,
         colormaps,
         blending='min',
-        saturation_limits=(
-            0.01,  # bottom 1% of pixels set to black point
-            0.97,  # top 3% of pixels set to white point
-        ),
+        percentiles=[(
+            1,  # bottom 1% of pixels set to black point
+            97,  # top 3% of pixels set to white point
+        )]*len(channels),
     ),
 )
 fig.tight_layout()

mergechannels-0.4.0.dist-info/RECORD

@@ -0,0 +1,12 @@
+mergechannels-0.4.0.dist-info/METADATA,sha256=UFSUXd5ow06vtucS9sA9wum226fAXxzK8oQignieJtY,5762
+mergechannels-0.4.0.dist-info/WHEEL,sha256=8qbut5OV21MVsJh-6koZCKZnnuRHvhBkNK3ct3shD1c,107
+mergechannels-0.4.0.dist-info/licenses/LICENSE,sha256=csvD60rgtSorbYEM3f8867qNyPCzmIXyFNj8h01Bd6c,1071
+mergechannels.libs/libgcc_s-5b5488a6.so.1,sha256=HGKUsVmTeNAxEdSy7Ua5Vh_I9FN3RCbPWzvZ7H_TrwE,2749061
+mergechannels/__init__.py,sha256=87haN6eeG02qnJJerNCb4Tfn-3GpChD89DKsFAYIH3Y,269
+mergechannels/__init__.pyi,sha256=FZYQH4ue5M9r3zeUMHVZoU2D_RYwd3UiVD-GkcReY7M,1011
+mergechannels/_blending.py,sha256=mE5Cr9wvVJRcwfymg_UUZUjdIKx2mldfLU8QJT95bVM,84
+mergechannels/_internal.py,sha256=tMQtEG3s-N8eamqLUPUHy_fTO-7FqR8Q0Xng-4RSAC0,1421
+mergechannels/_luts.py,sha256=RswhkrCGAFtDthmVHt-Kk7lxROwS-kdsQy1EboyUUc8,3009
+mergechannels/mergechannels.cpython-313-arm-linux-musleabihf.so,sha256=2OCUpdal_0DX1GFm_i4k0aY7OPWanXCkJwfAKtPrMcc,1157721
+mergechannels/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+mergechannels-0.4.0.dist-info/RECORD,,

mergechannels-0.3.0.dist-info/RECORD

@@ -1,12 +0,0 @@
-mergechannels-0.3.0.dist-info/METADATA,sha256=5KI3P3u9O0prtayGxavf_PooNdXYHHnyLp_fjrgW1Pw,5725
-mergechannels-0.3.0.dist-info/WHEEL,sha256=8qbut5OV21MVsJh-6koZCKZnnuRHvhBkNK3ct3shD1c,107
-mergechannels-0.3.0.dist-info/licenses/LICENSE,sha256=csvD60rgtSorbYEM3f8867qNyPCzmIXyFNj8h01Bd6c,1071
-mergechannels.libs/libgcc_s-5b5488a6.so.1,sha256=HGKUsVmTeNAxEdSy7Ua5Vh_I9FN3RCbPWzvZ7H_TrwE,2749061
-mergechannels/__init__.py,sha256=TTNiDGOOJ6_-yR-qvX_nXph646Sz6-MEORc8ItksCRE,232
-mergechannels/__init__.pyi,sha256=tiDx6v14J0cedFdk6fzVRRyA21b4Z8eB_rtSTUjI804,749
-mergechannels/_blending.py,sha256=mE5Cr9wvVJRcwfymg_UUZUjdIKx2mldfLU8QJT95bVM,84
-mergechannels/_internal.py,sha256=ZcdBUFEq8k2Ve_ASOZLTuPBaFEEPqadc9LqAGmBqOas,1773
-mergechannels/_luts.py,sha256=RswhkrCGAFtDthmVHt-Kk7lxROwS-kdsQy1EboyUUc8,3009
-mergechannels/mergechannels.cpython-313-arm-linux-musleabihf.so,sha256=Xe8MZJb541OTiHF8xZFZ6T5Sty6kAzmp8UMQDILKHXE,1157721
-mergechannels/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-mergechannels-0.3.0.dist-info/RECORD,,