Add TypeScript definitions for applyColorMap function and colormap constants (#88)

* Initial plan

* Add applyColorMap TypeScript definitions and tests

Co-authored-by: ttt43ttt <132509+ttt43ttt@users.noreply.github.com>

* Complete applyColorMap implementation with comprehensive tests

Co-authored-by: ttt43ttt <132509+ttt43ttt@users.noreply.github.com>

---------

Co-authored-by: copilot-swe-agent[bot] <198982749+Copilot@users.noreply.github.com>
Co-authored-by: ttt43ttt <132509+ttt43ttt@users.noreply.github.com>

Author: Copilot

src/types/opencv/_types.ts

@@ -19,6 +19,7 @@ export * from "./fisheye";
 export * from "./FlannBasedMatcher";
 export * from "./HOGDescriptor";
 export * from "./imgproc_color_conversions";
+export * from "./imgproc_colormap";
 export * from "./imgproc_draw";
 export * from "./imgproc_feature";
 export * from "./imgproc_filter";

src/types/opencv/imgproc_colormap.ts

@@ -0,0 +1,60 @@
+import type { InputArray, int, OutputArray } from "./_types";
+
+/*
+ * # Colormap Transformations
+ *
+ */
+
+/**
+ * Applies a colormap on a given image.
+ *
+ * @param src The source image, which should be grayscale. Should be 8-bit, 16-bit, or floating-point.
+ * @param dst The result is the colored image.
+ * @param colormap The colormap to apply.
+ */
+export declare function applyColorMap(
+  src: InputArray,
+  dst: OutputArray,
+  colormap: int,
+): void;
+
+/**
+ * Applies a user colormap on a given image.
+ *
+ * @param src The source image, which should be grayscale. Should be 8-bit, 16-bit, or floating-point.
+ * @param dst The result is the colored image.
+ * @param userColor The colormap to apply of type CV_8UC1 or CV_8UC3 and size 256.
+ */
+export declare function applyColorMap(
+  src: InputArray,
+  dst: OutputArray,
+  userColor: InputArray,
+): void;
+
+/**
+ * Colormap types used by the applyColorMap function.
+ */
+export type ColormapTypes = any;
+
+export declare const COLORMAP_AUTUMN: ColormapTypes; // initializer: = 0
+export declare const COLORMAP_BONE: ColormapTypes; // initializer: = 1
+export declare const COLORMAP_JET: ColormapTypes; // initializer: = 2
+export declare const COLORMAP_WINTER: ColormapTypes; // initializer: = 3
+export declare const COLORMAP_RAINBOW: ColormapTypes; // initializer: = 4
+export declare const COLORMAP_OCEAN: ColormapTypes; // initializer: = 5
+export declare const COLORMAP_SUMMER: ColormapTypes; // initializer: = 6
+export declare const COLORMAP_SPRING: ColormapTypes; // initializer: = 7
+export declare const COLORMAP_COOL: ColormapTypes; // initializer: = 8
+export declare const COLORMAP_HSV: ColormapTypes; // initializer: = 9
+export declare const COLORMAP_PINK: ColormapTypes; // initializer: = 10
+export declare const COLORMAP_HOT: ColormapTypes; // initializer: = 11
+export declare const COLORMAP_PARULA: ColormapTypes; // initializer: = 12
+export declare const COLORMAP_MAGMA: ColormapTypes; // initializer: = 13
+export declare const COLORMAP_INFERNO: ColormapTypes; // initializer: = 14
+export declare const COLORMAP_PLASMA: ColormapTypes; // initializer: = 15
+export declare const COLORMAP_VIRIDIS: ColormapTypes; // initializer: = 16
+export declare const COLORMAP_CIVIDIS: ColormapTypes; // initializer: = 17
+export declare const COLORMAP_TWILIGHT: ColormapTypes; // initializer: = 18
+export declare const COLORMAP_TWILIGHT_SHIFTED: ColormapTypes; // initializer: = 19
+export declare const COLORMAP_TURBO: ColormapTypes; // initializer: = 20
+export declare const COLORMAP_DEEPGREEN: ColormapTypes; // initializer: = 21

test/applyColorMap.test.ts

@@ -0,0 +1,90 @@
+import { setupOpenCv } from "./cv";
+
+beforeAll(setupOpenCv);
+
+describe("applyColorMap", () => {
+  it("should apply COLORMAP_JET to a grayscale image", async () => {
+    // Create a simple grayscale image
+    const src = new cv.Mat(100, 100, cv.CV_8UC1);
+    
+    // Fill with gradient values
+    for (let i = 0; i < 100; i++) {
+      for (let j = 0; j < 100; j++) {
+        src.ucharPtr(i, j)[0] = Math.floor((i + j) * 255 / 200);
+      }
+    }
+    
+    const dst = new cv.Mat();
+    
+    // Apply JET colormap
+    cv.applyColorMap(src, dst, cv.COLORMAP_JET);
+    
+    // Verify the output is a 3-channel color image
+    expect(dst.channels()).toBe(3);
+    expect(dst.rows).toBe(100);
+    expect(dst.cols).toBe(100);
+    expect(dst.type()).toBe(cv.CV_8UC3);
+    
+    // Clean up
+    src.delete();
+    dst.delete();
+  });
+
+  it("should have all COLORMAP constants available", () => {
+    // Test that all colormap constants are defined
+    expect(typeof cv.COLORMAP_JET).toBe('number');
+    expect(typeof cv.COLORMAP_AUTUMN).toBe('number');
+    expect(typeof cv.COLORMAP_BONE).toBe('number');
+    expect(typeof cv.COLORMAP_WINTER).toBe('number');
+    expect(typeof cv.COLORMAP_RAINBOW).toBe('number');
+    expect(typeof cv.COLORMAP_OCEAN).toBe('number');
+    expect(typeof cv.COLORMAP_SUMMER).toBe('number');
+    expect(typeof cv.COLORMAP_SPRING).toBe('number');
+    expect(typeof cv.COLORMAP_COOL).toBe('number');
+    expect(typeof cv.COLORMAP_HSV).toBe('number');
+    expect(typeof cv.COLORMAP_PINK).toBe('number');
+    expect(typeof cv.COLORMAP_HOT).toBe('number');
+    expect(typeof cv.COLORMAP_PARULA).toBe('number');
+    expect(typeof cv.COLORMAP_MAGMA).toBe('number');
+    expect(typeof cv.COLORMAP_INFERNO).toBe('number');
+    expect(typeof cv.COLORMAP_PLASMA).toBe('number');
+    expect(typeof cv.COLORMAP_VIRIDIS).toBe('number');
+    expect(typeof cv.COLORMAP_CIVIDIS).toBe('number');
+    expect(typeof cv.COLORMAP_TWILIGHT).toBe('number');
+    expect(typeof cv.COLORMAP_TWILIGHT_SHIFTED).toBe('number');
+    expect(typeof cv.COLORMAP_TURBO).toBe('number');
+    expect(typeof cv.COLORMAP_DEEPGREEN).toBe('number');
+  });
+
+  it("should apply different colormaps correctly", async () => {
+    // Create a simple grayscale image
+    const src = new cv.Mat(50, 50, cv.CV_8UC1, new cv.Scalar(128));
+    const dst1 = new cv.Mat();
+    const dst2 = new cv.Mat();
+    
+    // Apply different colormaps
+    cv.applyColorMap(src, dst1, cv.COLORMAP_JET);
+    cv.applyColorMap(src, dst2, cv.COLORMAP_VIRIDIS);
+    
+    // Both should be 3-channel color images
+    expect(dst1.channels()).toBe(3);
+    expect(dst2.channels()).toBe(3);
+    
+    // Different colormaps should produce different results
+    const data1 = dst1.data;
+    const data2 = dst2.data;
+    let different = false;
+    for (let i = 0; i < Math.min(data1.length, data2.length); i++) {
+      if (data1[i] !== data2[i]) {
+        different = true;
+        break;
+      }
+    }
+    expect(different).toBe(true);
+    
+    // Clean up
+    src.delete();
+    dst1.delete();
+    dst2.delete();
+  });
+});