cassdlcm/mediapipe
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Invoke vector listeners for zero-element vectors

Browse Source 
PiperOrigin-RevId: 513662913
This commit is contained in:
Sebastian Schmidt 2023-03-02 15:55:44 -08:00 committed by Copybara-Service
parent 7664e0ef64
commit c3adf19706
1 changed files with 8 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
mediapipe/web/graph_runner/graph_runner.ts
View File

@ -29,10 +29,12 @@ export type EmptyPacketListener = (timestamp: number) => void;
/**
* A listener that receives a single element of vector-returning output packet.
* Receives one element at a time (in order). Once all elements are processed,
* the listener is invoked with `data` set to `unknown` and `done` set to true.
* Intended for internal usage.
*/
export type VectorListener<T> =
(data: T, index: number, length: number, timestamp: number) => void;
export type VectorListener<T> = (data: T, done: boolean, timestamp: number) =>
void;
/**
* Declarations for Emscripten's WebAssembly Module behavior, so TS compiler
@ -454,17 +456,12 @@ export class GraphRunner {
let buffer: T[] = [];
this.wasmModule.simpleListeners = this.wasmModule.simpleListeners || {};
this.wasmModule.simpleListeners[outputStreamName] =
(data: unknown, index: number, length: number, timestamp: number) => {
// The Wasm listener gets invoked once for each element. Once we
// receive all elements, we invoke the registered callback with
// the full array.
buffer[index] = data as T;
if (index === length - 1) {
// Invoke the user callback directly, as the Wasm layer may
// clean up the underlying data elements once we leave the scope
// of the listener.
(data: unknown, done: boolean, timestamp: number) => {
if (done) {
callbackFcn(buffer, timestamp);
buffer = [];
} else {
buffer.push(data as T);
}
};
}