oxfmt

Author: cieplypolar

apps/typegpu-docs/src/examples/algorithms/probability/executor.ts

@@ -1,8 +1,4 @@
-import {
-  randf,
-  randomGeneratorSlot,
-  type StatefulGenerator,
-} from '@typegpu/noise';
+import { randf, randomGeneratorSlot, type StatefulGenerator } from '@typegpu/noise';
 import type {
   StorageFlag,
   TgpuBindGroup,
@@ -28,16 +24,10 @@ export class Executor {
   readonly #bindGroupLayout: TgpuBindGroupLayout;
   readonly #bufferCache: Map<
     number,
-    [
-      TgpuBuffer<d.WgslArray<d.Vec3f>> & StorageFlag,
-      TgpuBuffer<d.WgslArray<d.F32>> & StorageFlag,
-    ]
+    [TgpuBuffer<d.WgslArray<d.Vec3f>> & StorageFlag, TgpuBuffer<d.WgslArray<d.F32>> & StorageFlag]
   >;
   // they can be WeakMaps, because we always have reference to distribution and PRNG
-  readonly #pipelineCache: WeakMap<
-    TgpuFn,
-    WeakMap<StatefulGenerator, TgpuComputePipeline>
-  >;
+  readonly #pipelineCache: WeakMap<TgpuFn, WeakMap<StatefulGenerator, TgpuComputePipeline>>;
 
   constructor(root: TgpuRoot) {
     this.#root = root;
@@ -60,15 +50,12 @@ export class Executor {
       const id = input.gid.x;
       if (id >= bindGroupLayoutTempAlias.$.samplesBuffer.length) return;
       randf.seed(bindGroupLayoutTempAlias.$.seedBuffer[id]);
-      bindGroupLayoutTempAlias.$.samplesBuffer[id] = distributionSlotTempAlias
-        .$();
+      bindGroupLayoutTempAlias.$.samplesBuffer[id] = distributionSlotTempAlias.$();
     });
   }
 
   reseed() {
-    this.#seedBuffer.write(
-      Array.from({ length: this.#count }, () => Math.random()),
-    );
+    this.#seedBuffer.write(Array.from({ length: this.#count }, () => Math.random()));
   }
 
   set count(value: number) {
@@ -77,8 +64,7 @@ export class Executor {
     if (cacheEntry) {
       [this.#samplesBuffer, this.#seedBuffer] = cacheEntry;
     } else {
-      this.#samplesBuffer = this.#root.createBuffer(d.arrayOf(d.vec3f, value))
-        .$usage('storage');
+      this.#samplesBuffer = this.#root.createBuffer(d.arrayOf(d.vec3f, value)).$usage('storage');
       this.#seedBuffer = this.#root
         .createBuffer(
           d.arrayOf(d.f32, value),
@@ -130,9 +116,7 @@ export class Executor {
   ): Promise<d.v3f[]> {
     const pipeline = this.pipelineCacheGet(distribution, generator);
 
-    pipeline.with(this.#bindGroup).dispatchWorkgroups(
-      Math.ceil(this.#count / 64),
-    );
+    pipeline.with(this.#bindGroup).dispatchWorkgroups(Math.ceil(this.#count / 64));
 
     return await this.#samplesBuffer.read();
   }

apps/typegpu-docs/src/examples/algorithms/probability/helpers.ts

@@ -100,5 +100,4 @@ const GENERATOR_MAP = {
   [Generator.LCG]: LCG,
 };
 
-export const getGenerator = (gen: Generator): StatefulGenerator =>
-  GENERATOR_MAP[gen];
+export const getGenerator = (gen: Generator): StatefulGenerator => GENERATOR_MAP[gen];

apps/typegpu-docs/src/examples/algorithms/probability/index.ts

@@ -70,27 +70,19 @@ export const controls = defineControls({
   Reseed: {
     async onButtonClick() {
       executor.reseed();
-      await replot(
-        currentDistribution,
-        currentGenerator,
-        true,
-      );
+      await replot(currentDistribution, currentGenerator, true);
       plotter.resetView(getCameraPosition(currentDistribution));
     },
   },
-  'Generator': {
+  Generator: {
     initial: c.initialGenerator,
     options: c.generators,
     onSelectChange: async (value: Generator) => {
       if (currentGenerator === value) {
         return;
       }
       currentGenerator = value;
-      await replot(
-        currentDistribution,
-        currentGenerator,
-        true,
-      );
+      await replot(currentDistribution, currentGenerator, true);
       plotter.resetView(getCameraPosition(currentDistribution));
     },
   },
@@ -103,11 +95,7 @@ export const controls = defineControls({
       }
 
       currentDistribution = value;
-      await replot(
-        currentDistribution,
-        currentGenerator,
-        true,
-      );
+      await replot(currentDistribution, currentGenerator, true);
       plotter.resetView(getCameraPosition(currentDistribution));
     },
   },
@@ -116,15 +104,16 @@ export const controls = defineControls({
     options: c.numSamplesOptions,
     async onSelectChange(value) {
       executor.count = value;
-      await replot(
-        currentDistribution,
-        currentGenerator,
-      );
+      await replot(currentDistribution, currentGenerator);
     },
   },
   'Test Resolution': import.meta.env.DEV && {
     onButtonClick() {
-      for (const [i, j] of [[0, 0], [8, 1], [10, 0]]) {
+      for (const [i, j] of [
+        [0, 0],
+        [8, 1],
+        [10, 0],
+      ]) {
         const code = tgpu.resolve({
           externals: {
             p: executor.pipelineCacheGet(

apps/typegpu-docs/src/examples/algorithms/probability/types.ts

@@ -75,4 +75,4 @@ export const Generator = {
   LCG: 'lcg',
 } as const;
 
-export type Generator = typeof Generator[keyof typeof Generator];
+export type Generator = (typeof Generator)[keyof typeof Generator];

packages/typegpu-noise/src/generator.ts

@@ -54,9 +54,12 @@ export const BPETER: StatefulGenerator = (() => {
 export const LCG: StatefulGenerator = (() => {
   const seed = tgpu.privateVar(d.u32);
 
-  const u32To01Float = tgpu.fn([d.u32], d.f32)((value) => {
+  const u32To01Float = tgpu.fn(
+    [d.u32],
+    d.f32,
+  )((value) => {
     const mantissa = value >> 9;
-    const bits = 0x3F800000 | mantissa;
+    const bits = 0x3f800000 | mantissa;
     const f = bitcastU32toF32(bits);
     return f - 1;
   });
@@ -71,16 +74,12 @@ export const LCG: StatefulGenerator = (() => {
     }),
 
     seed3: tgpu.fn([d.vec3f])((value) => {
-      seed.$ = d.u32(
-        value.x * pow(32, 3) + value.y * pow(32, 2) +
-          value.z * pow(32, 1),
-      );
+      seed.$ = d.u32(value.x * pow(32, 3) + value.y * pow(32, 2) + value.z * pow(32, 1));
     }),
 
     seed4: tgpu.fn([d.vec4f])((value) => {
       seed.$ = d.u32(
-        value.x * pow(32, 3) + value.y * pow(32, 2) +
-          value.z * pow(32, 1) + value.w * pow(32, 0),
+        value.x * pow(32, 3) + value.y * pow(32, 2) + value.z * pow(32, 1) + value.w * pow(32, 0),
       );
     }),
 
@@ -95,6 +94,4 @@ export const LCG: StatefulGenerator = (() => {
 // The default (Can change between releases to improve uniformity).
 export const DefaultGenerator: StatefulGenerator = BPETER;
 
-export const randomGeneratorSlot: TgpuSlot<StatefulGenerator> = tgpu.slot(
-  DefaultGenerator,
-);
+export const randomGeneratorSlot: TgpuSlot<StatefulGenerator> = tgpu.slot(DefaultGenerator);

packages/typegpu/tests/examples/individual/probability.test.ts

@@ -10,12 +10,15 @@ describe('probability distribution plot example', () => {
   setupCommonMocks();
 
   it('should produce valid code', async ({ device }) => {
-    const shaderCodes = await runExampleTest({
-      category: 'algorithms',
-      name: 'probability',
-      controlTriggers: ['Test Resolution'],
-      expectedCalls: 3,
-    }, device);
+    const shaderCodes = await runExampleTest(
+      {
+        category: 'algorithms',
+        name: 'probability',
+        controlTriggers: ['Test Resolution'],
+        expectedCalls: 3,
+      },
+      device,
+    );
 
     expect(shaderCodes).toMatchInlineSnapshot(`
       "@group(0) @binding(1) var<storage, read_write> samplesBuffer: array<vec3f>;