face-api/build/dom/createCanvas.js

27 lines
1.0 KiB
JavaScript
Raw Normal View History

2020-08-20 02:10:42 +02:00
import { env } from '../env';
import { getContext2dOrThrow } from './getContext2dOrThrow';
import { getMediaDimensions } from './getMediaDimensions';
import { isMediaLoaded } from './isMediaLoaded';
export function createCanvas({ width, height }) {
const { createCanvasElement } = env.getEnv();
2020-08-18 14:04:33 +02:00
const canvas = createCanvasElement();
canvas.width = width;
canvas.height = height;
return canvas;
}
2020-08-20 02:10:42 +02:00
export function createCanvasFromMedia(media, dims) {
const { ImageData } = env.getEnv();
if (!(media instanceof ImageData) && !isMediaLoaded(media)) {
2020-08-18 14:04:33 +02:00
throw new Error('createCanvasFromMedia - media has not finished loading yet');
}
2020-08-20 02:10:42 +02:00
const { width, height } = dims || getMediaDimensions(media);
2020-08-18 14:04:33 +02:00
const canvas = createCanvas({ width, height });
if (media instanceof ImageData) {
2020-08-20 02:10:42 +02:00
getContext2dOrThrow(canvas).putImageData(media, 0, 0);
2020-08-18 14:04:33 +02:00
}
else {
2020-08-20 02:10:42 +02:00
getContext2dOrThrow(canvas).drawImage(media, 0, 0, width, height);
2020-08-18 14:04:33 +02:00
}
return canvas;
}
//# sourceMappingURL=createCanvas.js.map