mergechannels 0.1.0__tar.gz → 0.1.2__tar.gz

mergechannels-0.1.2/.gitattributes

@@ -0,0 +1,7 @@
+# LFS for image files
+*.png filter=lfs diff=lfs merge=lfs -text
+*.jpg filter=lfs diff=lfs merge=lfs -text
+*.jpeg filter=lfs diff=lfs merge=lfs -text
+*.gif filter=lfs diff=lfs merge=lfs -text
+*.tif filter=lfs diff=lfs merge=lfs -text
+*.tiff filter=lfs diff=lfs merge=lfs -text

{mergechannels-0.1.0 → mergechannels-0.1.2}/.github/workflows/CI.yml

@@ -8,8 +8,7 @@ name: CI
 on:
   push:
     branches:
-      - main
-      - mvp
+      - '*'
     tags:
       - '*'
   pull_request:
@@ -24,50 +23,73 @@ jobs:
     runs-on: ubuntu-latest
     steps:
       - uses: actions/checkout@v4
-      
+
       # Setup Rust toolchain
       - name: Install Rust stable
         uses: dtolnay/rust-toolchain@stable
         with:
           components: rustfmt, clippy
-      
-      # Run Rust format check
+
       - name: Check Rust formatting
         run: cargo fmt --all -- --check
-      
-      # Run Clippy
+
       - name: Run Clippy
         run: cargo clippy -- -D warnings
-      
-      # Run Cargo tests
+
       - name: Run Cargo tests
         run: cargo test
-      
-      # # Setup Python for testing
-      # - uses: actions/setup-python@v5
-      #   with:
-      #     python-version: 3.x
-      
-      # Install UV
+
       - name: Install UV
         uses: astral-sh/setup-uv@v5
-      
-      # Create virtual environment and install the project
-      - name: Create virtual environment and install
-        run: |
-          uv venv .venv --python 3.11
-          source .venv/bin/activate
-          pip install -e .
-      
-      # Run pytest
-      - name: Run pytest
+        with:
+          enable-cache: true
+
+      - name: Install the project
+        run: uv sync --all-extras --dev
+
+      - name: Run pre-commit
+        shell: bash -l {0}
+        run: uv run pre-commit run --all-files
+
+  tox_tests:
+    name: Tox Tests (${{ matrix.python-version }}) # Add python version to name for clarity
+    runs-on: ubuntu-latest
+    strategy:
+      matrix:
+        python-version: ["3.9", "3.10", "3.11", "3.12", "3.13"]
+      fail-fast: true # Optional: prevent other matrix jobs from cancelling if one fails
+    steps:
+      - uses: actions/checkout@v4
+
+      # Set up the specific Python version for this matrix job
+      - name: Set up Python ${{ matrix.python-version }}
+        uses: actions/setup-python@v5
+        with:
+          python-version: ${{ matrix.python-version }}
+
+      # Install UV using the official action, enabling caching
+      - name: Install UV
+        uses: astral-sh/setup-uv@v5
+        with:
+          # Enable uv's cache for faster subsequent runs
+          enable-cache: true
+
+      # Install tox and tox-uv using uv tool install
+      # This installs them into an isolated environment managed by uv
+      - name: Install Tox with tox-uv plugin using UV
+        run: uv tool install tox --with tox-uv
+
+      # Run Tox using uvx
+      # uvx ensures that the 'tox' command installed via 'uv tool install' is used
+      - name: Run Tox tests via uvx
         run: |
-          source .venv/bin/activate
-          pip install pytest
-          pytest
+          PY_VERSION=$(echo ${{ matrix.python-version }} | tr -d '.')
+          uvx tox -e py$PY_VERSION
 
-  linux:
-    needs: rust_checks_and_python_tests
+
+  build:
+    needs: [rust_checks_and_python_tests, tox_tests]
+    if: ${{ startsWith(github.ref, 'refs/tags/') || github.event_name == 'workflow_dispatch' }}
     runs-on: ${{ matrix.platform.runner }}
     strategy:
       matrix:
@@ -104,6 +126,7 @@ jobs:
 
   musllinux:
     needs: rust_checks_and_python_tests
+    if: ${{ startsWith(github.ref, 'refs/tags/') || github.event_name == 'workflow_dispatch' }}
     runs-on: ${{ matrix.platform.runner }}
     strategy:
       matrix:
@@ -136,6 +159,7 @@ jobs:
 
   windows:
     needs: rust_checks_and_python_tests
+    if: ${{ startsWith(github.ref, 'refs/tags/') || github.event_name == 'workflow_dispatch' }}
     runs-on: ${{ matrix.platform.runner }}
     strategy:
       matrix:
@@ -164,6 +188,7 @@ jobs:
 
   macos:
     needs: rust_checks_and_python_tests
+    if: ${{ startsWith(github.ref, 'refs/tags/') || github.event_name == 'workflow_dispatch' }}
     runs-on: ${{ matrix.platform.runner }}
     strategy:
       matrix:
@@ -191,6 +216,7 @@ jobs:
 
   sdist:
     needs: rust_checks_and_python_tests
+    if: ${{ startsWith(github.ref, 'refs/tags/') || github.event_name == 'workflow_dispatch' }}
     runs-on: ubuntu-latest
     steps:
       - uses: actions/checkout@v4
@@ -209,7 +235,7 @@ jobs:
     name: Release
     runs-on: ubuntu-latest
     if: ${{ startsWith(github.ref, 'refs/tags/') || github.event_name == 'workflow_dispatch' }}
-    needs: [linux, musllinux, windows, macos, sdist]
+    needs: [build, musllinux, windows, macos, sdist]
     permissions:
       # Use to sign the release artifacts
       id-token: write

{mergechannels-0.1.0 → mergechannels-0.1.2}/.gitignore

@@ -72,4 +72,8 @@ docs/_build/
 .python-version
 
 # ignore images
-*.tif*
+*.tif*
+
+# ignore notebooks
+*.ipynb
+*.ju.py

mergechannels-0.1.2/.pre-commit-config.yaml

@@ -0,0 +1,16 @@
+repos:
+  - repo: https://github.com/pre-commit/pre-commit-hooks
+    rev: v4.6.0
+    hooks:
+      - id: trailing-whitespace
+      - id: end-of-file-fixer
+      - id: check-yaml
+      - id: check-toml
+      - id: check-added-large-files
+
+  - repo: https://github.com/astral-sh/ruff-pre-commit
+    rev: v0.3.3
+    hooks:
+      - id: ruff
+        types_or: [ python, pyi, jupyter ]
+        args: [ --fix ]

{mergechannels-0.1.0 → mergechannels-0.1.2}/Cargo.lock

@@ -59,7 +59,7 @@ dependencies = [
 
 [[package]]
 name = "mergechannels"
-version = "0.1.0"
+version = "0.1.2"
 dependencies = [
  "lazy_static",
  "ndarray",

{mergechannels-0.1.0 → mergechannels-0.1.2}/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "mergechannels"
-version = "0.1.0"
+version = "0.1.2"
 edition = "2021"
 
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html

{mergechannels-0.1.0 → mergechannels-0.1.2}/LICENSE

@@ -18,4 +18,4 @@ FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
-SOFTWARE.
+SOFTWARE.

mergechannels-0.1.2/PKG-INFO

@@ -0,0 +1,115 @@
+Metadata-Version: 2.4
+Name: mergechannels
+Version: 0.1.2
+Classifier: Programming Language :: Rust
+Classifier: Programming Language :: Python :: Implementation :: CPython
+Classifier: Programming Language :: Python :: Implementation :: PyPy
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Requires-Dist: numpy>1.25.0
+License-File: LICENSE
+License: MIT
+Requires-Python: >=3.9, <=3.13
+Description-Content-Type: text/markdown; charset=UTF-8; variant=GFM
+
+[![CI](https://github.com/zacswider/mergechannels/actions/workflows/CI.yml/badge.svg)](https://github.com/zacswider/mergechannels/actions/workflows/CI.yml)
+![PyPI - License](https://img.shields.io/pypi/l/mergechannels)
+![PyPI - Python Version](https://img.shields.io/pypi/pyversions/mergechannels)
+![PyPI](https://img.shields.io/pypi/v/mergechannels)
+
+# mergechannels
+
+This project was originally conceived because I often find myself wanting to apply and blend colormaps to images while working from Python, and for historical reasons many of my favorite colormaps are distributed as [FIJI](https://imagej.net/software/fiji/) lookup tables. I also care about things likes speed and memory usage (e.g., not casting large arrays to floating point dtypes, not creating multiple whole arrays just to add them together), so I was interested in seeing if I could at least match matplotlib's colormapping performance with my own hand-rolled colorizer in Rust.
+
+
+
+## Installation
+
+Install pre-compiled binaries from [PyPI](https://pypi.org/project/mergechannels/):
+```bash
+pip install mergechannels
+```
+
+Build from source on your own machine:
+```bash
+pip install git+https://github.com/zacswider/mergechannels.git
+```
+
+## Usage
+*NOTE*: scikit-image is not a dependency of this project, but is used in the examples below to fetch images.
+
+### colorize a single image
+
+```python
+from skimage import data
+import mergechannels as mc
+
+img = data.camera()
+colorized = mc.apply_color_map(img, 'Red/Green')
+fig, axes = plt.subplots(1, 2, figsize=(6, 3), dpi=150)
+for ax in axes: ax.axis('off')
+(a, b) = axes
+a.imshow(img, cmap='gray')
+b.imshow(colorized)
+plt.show()
+print(colorized.shape, colorized.dtype)
+>> (512, 512, 3) uint8
+```
+![colorize a single image](https://raw.githubusercontent.com/zacswider/README_Images/main/camera_red-green.png
+)
+
+
+### apply a different colormap to each channel
+```python
+from skimage import data
+import matplotlib.pyplot as plt
+import mergechannels as mc
+
+cells, nuclei = data.cells3d().max(axis=0)
+cells = to_uint8(cells)     # normalize your own arrays, mergechannels doesn't currently handle this
+nuclei = to_uint8(nuclei)   # normalize your own arrays, mergechannels doesn't currently handle this
+
+fig, axes = plt.subplots(2, 2, figsize=(6, 6), dpi=150)
+for ax in axes.ravel(): ax.axis('off')
+(a, b, c, d) = axes.ravel()
+a.imshow(cells, cmap='gray')
+b.imshow(nuclei, cmap='gray')
+c.imshow(
+    mc.merge(
+        [cells, nuclei],
+        ['Orange Hot', 'Cyan Hot'],  # maximum blending is the default
+    ),
+)
+d.imshow(
+    mc.merge(
+        [cells, nuclei],
+        ['I Blue', 'I Forest'],
+        blending='min',     # use minimum blending with inverted colormaps
+    ),
+)
+fig.tight_layout()
+plt.show()
+```
+![max and min multicolor blending](https://raw.githubusercontent.com/zacswider/README_Images/main/cells_multicolor.png)
+
+
+## Roadmap
+mergechannels is currently incredibly simple. It can apply one or more colormaps to one or more 2D 8-bit images and that's it.
+- Add support for any numerical dtype
+- Add support for 3D images
+- Add option to return any colormap as a matplotlib colormap
+- Add support for directly passing matplotlib colormaps instead of colormap names
+- Parallelize colormap application on large images (if it's helpful)
+- Add option to overlay binary or instance masks onto colorized images
+
+## Acknowledgements
+
+There are other great colormapping libraries available (e.g., [microfilm](https://github.com/guiwitz/microfilm), [cmap](https://github.com/pyapp-kit/cmap)) that are more feature-rich than this one, but which don't address my goals. My hope is that this project can fill an un-met niche and otherwise maintain full compatibility with these and similar libraries.
+
+This project incorporates a number of colormaps that were hand-crafted by Christophe Leterrier and were originally distributed here under the MIT license: https://github.com/cleterrier/ChrisLUTs
+

mergechannels-0.1.2/README.md

@@ -0,0 +1,95 @@
+[![CI](https://github.com/zacswider/mergechannels/actions/workflows/CI.yml/badge.svg)](https://github.com/zacswider/mergechannels/actions/workflows/CI.yml)
+![PyPI - License](https://img.shields.io/pypi/l/mergechannels)
+![PyPI - Python Version](https://img.shields.io/pypi/pyversions/mergechannels)
+![PyPI](https://img.shields.io/pypi/v/mergechannels)
+
+# mergechannels
+
+This project was originally conceived because I often find myself wanting to apply and blend colormaps to images while working from Python, and for historical reasons many of my favorite colormaps are distributed as [FIJI](https://imagej.net/software/fiji/) lookup tables. I also care about things likes speed and memory usage (e.g., not casting large arrays to floating point dtypes, not creating multiple whole arrays just to add them together), so I was interested in seeing if I could at least match matplotlib's colormapping performance with my own hand-rolled colorizer in Rust.
+
+
+
+## Installation
+
+Install pre-compiled binaries from [PyPI](https://pypi.org/project/mergechannels/):
+```bash
+pip install mergechannels
+```
+
+Build from source on your own machine:
+```bash
+pip install git+https://github.com/zacswider/mergechannels.git
+```
+
+## Usage
+*NOTE*: scikit-image is not a dependency of this project, but is used in the examples below to fetch images.
+
+### colorize a single image
+
+```python
+from skimage import data
+import mergechannels as mc
+
+img = data.camera()
+colorized = mc.apply_color_map(img, 'Red/Green')
+fig, axes = plt.subplots(1, 2, figsize=(6, 3), dpi=150)
+for ax in axes: ax.axis('off')
+(a, b) = axes
+a.imshow(img, cmap='gray')
+b.imshow(colorized)
+plt.show()
+print(colorized.shape, colorized.dtype)
+>> (512, 512, 3) uint8
+```
+![colorize a single image](https://raw.githubusercontent.com/zacswider/README_Images/main/camera_red-green.png
+)
+
+
+### apply a different colormap to each channel
+```python
+from skimage import data
+import matplotlib.pyplot as plt
+import mergechannels as mc
+
+cells, nuclei = data.cells3d().max(axis=0)
+cells = to_uint8(cells)     # normalize your own arrays, mergechannels doesn't currently handle this
+nuclei = to_uint8(nuclei)   # normalize your own arrays, mergechannels doesn't currently handle this
+
+fig, axes = plt.subplots(2, 2, figsize=(6, 6), dpi=150)
+for ax in axes.ravel(): ax.axis('off')
+(a, b, c, d) = axes.ravel()
+a.imshow(cells, cmap='gray')
+b.imshow(nuclei, cmap='gray')
+c.imshow(
+    mc.merge(
+        [cells, nuclei],
+        ['Orange Hot', 'Cyan Hot'],  # maximum blending is the default
+    ),
+)
+d.imshow(
+    mc.merge(
+        [cells, nuclei],
+        ['I Blue', 'I Forest'],
+        blending='min',     # use minimum blending with inverted colormaps
+    ),
+)
+fig.tight_layout()
+plt.show()
+```
+![max and min multicolor blending](https://raw.githubusercontent.com/zacswider/README_Images/main/cells_multicolor.png)
+
+
+## Roadmap
+mergechannels is currently incredibly simple. It can apply one or more colormaps to one or more 2D 8-bit images and that's it.
+- Add support for any numerical dtype
+- Add support for 3D images
+- Add option to return any colormap as a matplotlib colormap
+- Add support for directly passing matplotlib colormaps instead of colormap names
+- Parallelize colormap application on large images (if it's helpful)
+- Add option to overlay binary or instance masks onto colorized images
+
+## Acknowledgements
+
+There are other great colormapping libraries available (e.g., [microfilm](https://github.com/guiwitz/microfilm), [cmap](https://github.com/pyapp-kit/cmap)) that are more feature-rich than this one, but which don't address my goals. My hope is that this project can fill an un-met niche and otherwise maintain full compatibility with these and similar libraries.
+
+This project incorporates a number of colormaps that were hand-crafted by Christophe Leterrier and were originally distributed here under the MIT license: https://github.com/cleterrier/ChrisLUTs

{mergechannels-0.1.0 → mergechannels-0.1.2}/assets/builtin_luts/phase.lut

@@ -254,4 +254,3 @@
 248   4   4
 252   0   0
 252   0   0
-

{mergechannels-0.1.0 → mergechannels-0.1.2}/assets/christ_luts/Turbo.lut

@@ -254,4 +254,4 @@ Index	Red	Green	Blue
 252	133	7	2
 253	129	6	2
 254	126	5	2
-255	122	4	3
+255	122	4	3

mergechannels-0.1.2/pyproject.toml

@@ -0,0 +1,33 @@
+[build-system]
+requires = ["maturin>=1.8,<2.0"]
+build-backend = "maturin"
+
+[project]
+name = "mergechannels"
+license = { text = "MIT" }
+requires-python = ">=3.9, <=3.13"
+classifiers = [
+    "Programming Language :: Rust",
+    "Programming Language :: Python :: Implementation :: CPython",
+    "Programming Language :: Python :: Implementation :: PyPy",
+    "License :: OSI Approved :: MIT License",
+    "Programming Language :: Python :: 3",
+    "Programming Language :: Python :: 3.9",
+    "Programming Language :: Python :: 3.10",
+    "Programming Language :: Python :: 3.11",
+    "Programming Language :: Python :: 3.12",
+    "Programming Language :: Python :: 3.13",
+]
+dynamic = ["version"]
+dependencies = ["numpy>1.25.0"]
+[tool.maturin]
+features = ["pyo3/extension-module"]
+python-source = "python"
+
+[dependency-groups]
+dev = [
+    "matplotlib>=3.7.5",
+    "pre-commit>=3.5.0",
+    "pytest-benchmark>=4.0.0",
+    "pytest>=8.3.5",
+]

{mergechannels-0.1.0 → mergechannels-0.1.2}/pytests/test_colorize.py

@@ -3,6 +3,9 @@ import mergechannels as mc
 
 
 def test_apply_color_map():
+    '''
+    Test that the color map is applied correctly
+    '''
     x = np.ones((1, 1), dtype=np.uint8)
     rgb = mc.apply_color_map(x, 'betterBlue')
     assert rgb.shape == (1,1,3)
@@ -23,6 +26,9 @@ def test_apply_color_map():
     )
 
 def test_apply_colors_and_merge_low_sum():
+    '''
+    Test that the colors are merged correctly with sum blending and low values
+    '''
     x = np.ones((1, 1), dtype=np.uint8)
     y = np.ones((1, 1), dtype=np.uint8)
     rgb_sum = mc.apply_colors_and_merge_nc([x, y], ['betterBlue', 'betterOrange'], 'sum')
@@ -36,6 +42,9 @@ def test_apply_colors_and_merge_low_sum():
     )
 
 def test_apply_colors_and_merge_high_sum():
+    '''
+    Test that the colors are merged correctly with sum blending and high values
+    '''
     x = np.ones((1, 1), dtype=np.uint8) * 255
     y = np.ones((1, 1), dtype=np.uint8) * 255
     rgb_sum = mc.apply_colors_and_merge_nc([x, y], ['betterBlue', 'betterOrange'], 'sum')
@@ -49,6 +58,9 @@ def test_apply_colors_and_merge_high_sum():
     )
 
 def test_apply_colors_and_merge_low_max():
+    '''
+    Test that the colors are merged correctly with max blending and low values
+    '''
     x = np.ones((1, 1), dtype=np.uint8)
     y = np.ones((1, 1), dtype=np.uint8)
     rgb_max = mc.apply_colors_and_merge_nc([x, y], ['betterBlue', 'betterOrange'], 'max')
@@ -62,6 +74,9 @@ def test_apply_colors_and_merge_low_max():
     )
 
 def test_apply_colors_and_merge_high_max():
+    '''
+    Test that the colors are merged correctly with max blending and high values
+    '''
     x = np.ones((1, 1), dtype=np.uint8) * 255
     y = np.ones((1, 1), dtype=np.uint8) * 255
     rgb_max = mc.apply_colors_and_merge_nc([x, y], ['betterBlue', 'betterOrange'], 'max')
@@ -75,6 +90,9 @@ def test_apply_colors_and_merge_high_max():
     )
 
 def test_apply_colors_and_merge_low_min():
+    '''
+    Test that the colors are merged correctly with min blending and low values
+    '''
     x = np.ones((1, 1), dtype=np.uint8)
     y = np.ones((1, 1), dtype=np.uint8)
     rgb_min = mc.apply_colors_and_merge_nc([x, y], ['betterBlue', 'betterOrange'], 'min')
@@ -88,6 +106,9 @@ def test_apply_colors_and_merge_low_min():
     )
 
 def test_apply_colors_and_merge_high_min():
+    '''
+    Test that the colors are merged correctly with min blending and high values
+    '''
     x = np.ones((1, 1), dtype=np.uint8) * 255
     y = np.ones((1, 1), dtype=np.uint8) * 255
     rgb_min = mc.apply_colors_and_merge_nc([x, y], ['betterBlue', 'betterOrange'], 'min')
@@ -101,6 +122,9 @@ def test_apply_colors_and_merge_high_min():
     )
 
 def test_apply_colors_and_merge_low_mean():
+    '''
+    Test that the colors are merged correctly with mean blending and low values
+    '''
     x = np.ones((1, 1), dtype=np.uint8)
     y = np.ones((1, 1), dtype=np.uint8)
     rgb_mean = mc.apply_colors_and_merge_nc([x, y], ['betterBlue', 'betterOrange'], 'mean')
@@ -116,6 +140,9 @@ def test_apply_colors_and_merge_low_mean():
     )
 
 def test_apply_colors_and_merge_high_mean():
+    '''
+    Test that the colors are merged correctly with mean blending and high values
+    '''
     x = np.ones((1, 1), dtype=np.uint8) * 255
     y = np.ones((1, 1), dtype=np.uint8) * 255
     rgb_mean = mc.apply_colors_and_merge_nc([x, y], ['betterBlue', 'betterOrange'], 'mean')
@@ -128,4 +155,4 @@ def test_apply_colors_and_merge_high_mean():
         np.array(
             [[[127, 168, 127]]]
         )
-    )
+    )

{mergechannels-0.1.0 → mergechannels-0.1.2}/python/mergechannels/__init__.py

@@ -2,12 +2,12 @@
 from .mergechannels import (  # type: ignore
 	apply_color_map,
 	apply_colors_and_merge_nc,
-)  
+)
 
 from ._internal import merge
 
 __all__ = [
-	"apply_color_map",
-	"apply_colors_and_merge_nc",
-	"merge",
+	'apply_color_map',
+	'apply_colors_and_merge_nc',
+	'merge',
 ]

{mergechannels-0.1.0 → mergechannels-0.1.2}/python/mergechannels/__init__.pyi

@@ -9,13 +9,19 @@ def merge(
 	arrs: Sequence[np.ndarray],
 	colors: Sequence[COLORMAPS] = (),
 	blending: Literal[BLENDING_OPTIONS] = 'max',
+    saturation_limits: tuple[float, float] = (0.2, 99.8),
 ) -> np.ndarray:
     ...
 
-def apply_color_map(arr: np.ndarray, cmap_name: COLORMAPS) -> np.ndarray: ...
+def apply_color_map(
+    arr: np.ndarray,
+    cmap_name: COLORMAPS,
+    saturation_limits: tuple[float, float] = (0.2, 99.8),
+) -> np.ndarray: ...
 
 def apply_colors_and_merge_nc(
     arrs: Sequence[np.ndarray],
     colors: Sequence[COLORMAPS] = (),
     blending: Literal[BLENDING_OPTIONS] = 'max',
-) -> np.ndarray: ...
+    saturation_limits: Sequence[tuple[float, float]] = (),
+) -> np.ndarray: ...

{mergechannels-0.1.0 → mergechannels-0.1.2}/python/mergechannels/_internal.py

@@ -11,24 +11,16 @@ from ._blending import BLENDING_OPTIONS
 
 def merge(
 	arrs: Sequence[np.ndarray],
-	colors: Sequence[COLORMAPS] = (),
+	colors: Sequence[COLORMAPS],
 	blending: BLENDING_OPTIONS = 'max',
 ) -> np.ndarray:
 	'''
 	apply cmaps to arrays and blend the colors
 	'''
+	# region validation
 	n_arrs = len(arrs)
 	n_colors = len(colors)
-	if n_arrs > 0 and n_colors == 0:
-		arr_shapes = [arr.shape for arr in arrs]
-		if not all([len(arr_shape) == 3 for arr_shape in arr_shapes]):
-			raise ValueError(
-				'Expected a sequence of pre-colorized arrays, '
-				f'got {n_arrs} arrays of shapes {arr_shapes}'
-			)
-		# call merge rgb arrs here
-		return np.zeros((3,3))
-	if not n_arrs == n_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'
@@ -38,19 +30,18 @@ def merge(
 		raise ValueError(
 			f'Expected every array to have the same shape, got {arr_shapes}'
 		)
-	# call apply and merge rgb arrs here
-	match n_arrs:
-		case 1:
-			return apply_color_map(arr=arrs[0], cmap_name=colors[0])
-		case _:
-			return apply_colors_and_merge_nc(arrs, colors, blending)
-
-
-
-
-
-
-
-
-
+	if not len(arr_shapes[0]) == 2:
+		raise ValueError(
+			f'Expected every array to be 2D, 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:
+		return apply_color_map(arr=arrs[0], cmap_name=colors[0])
+	else:
+		return apply_colors_and_merge_nc(arrs, colors, blending)