face-api/src/dom/getMediaDimensions.ts

15 lines
529 B
TypeScript
Raw Normal View History

2020-08-18 13:54:53 +02:00
import { Dimensions, IDimensions } from '../classes/Dimensions';
2020-12-19 17:46:41 +01:00
import { env } from '../env/index';
2020-08-18 13:54:53 +02:00
export function getMediaDimensions(input: HTMLImageElement | HTMLCanvasElement | HTMLVideoElement | IDimensions): Dimensions {
2020-12-23 17:26:55 +01:00
const { Image, Video } = env.getEnv();
2020-08-18 13:54:53 +02:00
if (input instanceof Image) {
2020-12-23 17:26:55 +01:00
return new Dimensions(input.naturalWidth, input.naturalHeight);
2020-08-18 13:54:53 +02:00
}
if (input instanceof Video) {
2020-12-23 17:26:55 +01:00
return new Dimensions(input.videoWidth, input.videoHeight);
2020-08-18 13:54:53 +02:00
}
2020-12-23 17:26:55 +01:00
return new Dimensions(input.width, input.height);
2020-08-18 13:54:53 +02:00
}