feat(dsp): add refG()/refP() wrappers

thi-ng · Sep 27, 2023 · 99f4535 · 99f4535
1 parent 314e9b2
commit 99f4535
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 0 deletions.
diff --git a/packages/dsp/src/index.ts b/packages/dsp/src/index.ts
@@ -50,6 +50,7 @@ export * from "./pipe.js";
 export * from "./power.js";
 export * from "./product.js";
 export * from "./reciprocal.js";
+export * from "./ref.js";
 export * from "./serial.js";
 export * from "./sincos.js";
 export * from "./sum.js";

diff --git a/packages/dsp/src/ref.ts b/packages/dsp/src/ref.ts
@@ -0,0 +1,27 @@
+import { AGen } from "./agen.js";
+import type { IGen, IProc } from "./api.js";
+import { AProc } from "./aproc.js";
+
+export const refG = <T>(src: IGen<T>) => new RefG(src);
+
+export class RefG<T> extends AGen<T> {
+	constructor(protected _src: IGen<T>) {
+		super(_src.deref());
+	}
+
+	next() {
+		return this._src.deref();
+	}
+}
+
+export const refP = <A, B>(src: IProc<A, B>) => new RefP(src);
+
+export class RefP<A, B> extends AProc<A, B> {
+	constructor(protected _src: IProc<A, B>) {
+		super(_src.deref());
+	}
+
+	next() {
+		return this._src.deref();
+	}
+}