Spaces:
Runtime error
Runtime error
File size: 2,388 Bytes
92189dd |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 |
/**
* Copyright (c) Meta Platforms, Inc. and affiliates.
*
* 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.
*/
import {RefObject, useEffect, useMemo, useRef} from 'react';
import VideoWorkerBridge from './VideoWorkerBridge';
type Options = {
createVideoWorker?: () => Worker;
createWorkerBridge?: CreateWorkerBridgeFunction;
};
const DEFAULT_OPTIONS: Options = {
createVideoWorker: () =>
new Worker(new URL('./VideoWorker', import.meta.url), {
type: 'module',
}),
};
type WorkerFactory = () => Worker;
type CreateWorkerBridgeFunction = (
workerFactory: WorkerFactory,
) => VideoWorkerBridge;
export default function useVideoWorker(
src: string,
canvasRef: RefObject<HTMLCanvasElement>,
options: Options = {},
) {
const isControlTransferredToOffscreenRef = useRef(false);
const mergedOptions = useMemo(() => {
const definedProps = (o: Options) =>
Object.fromEntries(
Object.entries(o).filter(([_k, v]) => v !== undefined),
);
return Object.assign(
DEFAULT_OPTIONS,
definedProps(options),
) as Required<Options>;
}, [options]);
const worker = useMemo(() => {
if (mergedOptions.createWorkerBridge) {
return mergedOptions.createWorkerBridge(mergedOptions.createVideoWorker);
}
return VideoWorkerBridge.create(mergedOptions.createVideoWorker);
}, [mergedOptions]);
useEffect(() => {
const canvas = canvasRef.current;
if (canvas == null) {
return;
}
if (isControlTransferredToOffscreenRef.current) {
return;
}
isControlTransferredToOffscreenRef.current = true;
worker.setCanvas(canvas);
return () => {
// Cannot terminate worker in DEV mode
// workerRef.current?.terminate();
};
}, [canvasRef, mergedOptions, worker]);
useEffect(() => {
worker.setSource(src);
}, [src, worker]);
return worker;
}
|