feat(pixel): add/update buffer factory fns

- add fn overrides to simplify userland API
thi-ng · Feb 28, 2021 · ba38e13 · ba38e13
1 parent b98d05d
commit ba38e13
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 4 deletions.
diff --git a/packages/pixel/src/float.ts b/packages/pixel/src/float.ts
@@ -22,12 +22,23 @@ import { clampRegion, ensureChannel, ensureSize, prepRegions } from "./utils";
  * @param fmt -
  * @param pixels -
  */
-export const floatBuffer = (
+export function floatBuffer(
     w: number,
     h: number,
     fmt: FloatFormat | FloatFormatSpec,
     pixels?: Float32Array
-) => new FloatBuffer(w, h, fmt, pixels);
+): FloatBuffer;
+export function floatBuffer(
+    src: PackedBuffer,
+    fmt: FloatFormat | FloatFormatSpec
+): FloatBuffer;
+export function floatBuffer(...args: any[]) {
+    return args[0] instanceof PackedBuffer
+        ? // @ts-ignore
+          FloatBuffer.fromPacked(...args)
+        : // @ts-ignore
+          new FloatBuffer(...args);
+}
 
 export class FloatBuffer
     implements

diff --git a/packages/pixel/src/packed.ts b/packages/pixel/src/packed.ts
@@ -47,12 +47,22 @@ import {
  * @param fmt -
  * @param pixels -
  */
-export const packedBuffer = (
+export function packedBuffer(
     w: number,
     h: number,
     fmt: PackedFormat | PackedFormatSpec,
     pixels?: UIntArray
-) => new PackedBuffer(w, h, fmt, pixels);
+): PackedBuffer;
+export function packedBuffer(
+    src: PackedBuffer,
+    fmt: PackedFormat | PackedFormatSpec
+): PackedBuffer;
+export function packedBuffer(...args: any[]) {
+    return args[0] instanceof PackedBuffer
+        ? args[0].as(args[1])
+        : // @ts-ignore
+          new PackedBuffer(...args);
+}
 
 /**
  * @deprecated use {@link packedBuffer} instead.