feat: makeDereferencable

Author: iwoplaza

packages/typegpu/src/tgsl/makeDereferencable.ts

@@ -0,0 +1,36 @@
+import { snip, type Origin } from '../data/snippet.ts';
+import type { BaseData } from '../data/wgslTypes.ts';
+import { $gpuValueOf, $internal, $ownSnippet, $resolve } from '../shared/symbols.ts';
+import { valueProxyHandler } from '../core/valueProxyUtils.ts';
+import type { SelfResolvable } from '../types.ts';
+
+export function makeDereferencable<T extends SelfResolvable>(
+  value: T,
+  options: makeDereferencable.Options,
+): T {
+  Object.defineProperty(value, $gpuValueOf, {
+    value: function () {
+      const [dataType, origin] = options.getDataTypeAndOrigin();
+
+      return new Proxy(
+        {
+          [$internal]: true,
+          get [$ownSnippet]() {
+            return snip(this, dataType, origin);
+          },
+          [$resolve]: (ctx) => ctx.resolve(this),
+          toString: () => `${this.toString()}.$`,
+        },
+        valueProxyHandler,
+      );
+    },
+  });
+
+  return value;
+}
+
+export namespace makeDereferencable {
+  export interface Options {
+    getDataTypeAndOrigin(): [dataType: BaseData, origin: Origin];
+  }
+}

packages/typegpu/src/tgsl/makeResolvable.ts

@@ -0,0 +1,44 @@
+import { snip, type Origin, type ResolvedSnippet } from '../data/snippet.ts';
+import { type BaseData } from '../data/wgslTypes.ts';
+import { $internal, $resolve, isMarkedInternal } from '../shared/symbols.ts';
+import type { ResolutionCtx } from '../types.ts';
+
+/**
+ *
+ * @param value
+ * @param options
+ * @returns
+ */
+export function makeResolvable<T>(
+  value: T,
+  options: makeResolvable.Options,
+): T & { [$resolve](ctx: ResolutionCtx): ResolvedSnippet } {
+  if (!isMarkedInternal(value)) {
+    Object.defineProperty(value, $internal, {
+      value: true,
+    });
+  }
+
+  Object.defineProperty(value, $resolve, {
+    value: (ctx: ResolutionCtx): ResolvedSnippet => {
+      const protoSnippet = options.resolve(ctx);
+
+      return snip(protoSnippet.value, protoSnippet.dataType, protoSnippet.origin);
+    },
+  });
+
+  return value as T & { [$resolve](ctx: ResolutionCtx): ResolvedSnippet };
+}
+
+interface ProtoSnippet {
+  value: string;
+  dataType: BaseData;
+  origin: Origin;
+}
+
+export namespace makeResolvable {
+  export interface Options {
+    resolve: (ctx: ResolutionCtx) => ProtoSnippet;
+    asString(): string;
+  }
+}

packages/typegpu/src/tgsl/shaderGenerator_members.ts

@@ -4,6 +4,8 @@ import type { FunctionArgument, TgpuShaderStage } from '../types.ts';
 
 export { UnknownData } from '../data/dataTypes.ts';
 export { getName } from '../shared/meta.ts';
+export { makeDereferencable } from './makeDereferencable.ts';
+export { makeResolvable } from './makeResolvable.ts';
 
 // types
 export type { ResolutionCtx, FunctionArgument, TgpuShaderStage } from '../types.ts';