Reuse single PlaybackWorker between all Playbacks

Signed-off-by: Michael Telatynski <[email protected]>

src/PlaybackEncoder.ts

@@ -0,0 +1,34 @@
+/*
+Copyright 2024 The Matrix.org Foundation C.I.C.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+
+// @ts-ignore - `.ts` is needed here to make TS happy
+import { Request, Response } from "./workers/playback.worker";
+import { WorkerManager } from "./WorkerManager";
+import playbackWorkerFactory from "./workers/blurhashWorkerFactory";
+
+export class PlaybackEncoder {
+    private static internalInstance = new PlaybackEncoder();
+
+    public static get instance(): PlaybackEncoder {
+        return PlaybackEncoder.internalInstance;
+    }
+
+    private readonly worker = new WorkerManager<Request, Response>(playbackWorkerFactory());
+
+    public getPlaybackWaveform(input: Float32Array): Promise<number[]> {
+        return this.worker.call({ data: Array.from(input) }).then((resp) => resp.waveform);
+    }
+}

src/WorkerManager.ts

@@ -43,12 +43,4 @@ export class WorkerManager<Request extends {}, Response> {
         this.worker.postMessage({ seq, ...request });
         return deferred.promise;
     }
-
-    /**
-     * Terminate the worker.
-     * This will prevent any further messages from being sent to the worker and clean up its resources.
-     */
-    public terminate(): void {
-        this.worker.terminate();
-    }
 }

src/audio/Playback.ts

@@ -19,17 +19,14 @@ import { SimpleObservable } from "matrix-widget-api";
 import { logger } from "matrix-js-sdk/src/logger";
 import { defer } from "matrix-js-sdk/src/utils";
 
-// @ts-ignore - `.ts` is needed here to make TS happy
-import { Request, Response } from "../workers/playback.worker.ts";
 import { UPDATE_EVENT } from "../stores/AsyncStore";
 import { arrayFastResample } from "../utils/arrays";
 import { IDestroyable } from "../utils/IDestroyable";
 import { PlaybackClock } from "./PlaybackClock";
 import { createAudioContext, decodeOgg } from "./compat";
 import { clamp } from "../utils/numbers";
-import { WorkerManager } from "../WorkerManager";
 import { DEFAULT_WAVEFORM, PLAYBACK_WAVEFORM_SAMPLES } from "./consts";
-import playbackWorkerFactory from "../workers/playbackWorkerFactory";
+import { PlaybackEncoder } from "../PlaybackEncoder";
 
 export enum PlaybackState {
     Decoding = "decoding",
@@ -64,7 +61,6 @@ export class Playback extends EventEmitter implements IDestroyable, PlaybackInte
     private waveformObservable = new SimpleObservable<number[]>();
     private readonly clock: PlaybackClock;
     private readonly fileSize: number;
-    private readonly worker = new WorkerManager<Request, Response>(playbackWorkerFactory());
 
     /**
      * Creates a new playback instance from a buffer.
@@ -145,7 +141,6 @@ export class Playback extends EventEmitter implements IDestroyable, PlaybackInte
         this.removeAllListeners();
         this.clock.destroy();
         this.waveformObservable.close();
-        this.worker.terminate();
         if (this.element) {
             URL.revokeObjectURL(this.element.src);
             this.element.remove();
@@ -210,7 +205,9 @@ export class Playback extends EventEmitter implements IDestroyable, PlaybackInte
 
             // Update the waveform to the real waveform once we have channel data to use. We don't
             // exactly trust the user-provided waveform to be accurate...
-            this.resampledWaveform = await this.makePlaybackWaveform(this.audioBuf.getChannelData(0));
+            this.resampledWaveform = await PlaybackEncoder.instance.getPlaybackWaveform(
+                this.audioBuf.getChannelData(0),
+            );
         }
 
         this.waveformObservable.update(this.resampledWaveform);
@@ -223,10 +220,6 @@ export class Playback extends EventEmitter implements IDestroyable, PlaybackInte
         this.emit(PlaybackState.Stopped); // signal that we're not decoding anymore
     }
 
-    private makePlaybackWaveform(input: Float32Array): Promise<number[]> {
-        return this.worker.call({ data: Array.from(input) }).then((resp) => resp.waveform);
-    }
-
     private onPlaybackEnd = async (): Promise<void> => {
         await this.context.suspend();
         this.emit(PlaybackState.Stopped);