face-api/src/dom/getMediaDimensions.ts

16 lines
526 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 {
const { Image, Video } = env.getEnv()
if (input instanceof Image) {
return new Dimensions(input.naturalWidth, input.naturalHeight)
}
if (input instanceof Video) {
return new Dimensions(input.videoWidth, input.videoHeight)
}
return new Dimensions(input.width, input.height)
}