import { MLKitVisionOptions } from '../';
import { MLKitCameraView, MLKitVisionResult } from '../index';

export interface MLKitObjectDetectionResultItemBounds {
	origin: {
		x: number;
		y: number;
	};
	size: {
		width: number;
		height: number;
	};
}

export interface MLKitObjectDetectionResultItem {
	/**
	 * Only set when detecting objects in a stream (not when using a still image)
	 */
	id?: number;
	confidence: number;
	category: any;
	bounds: MLKitObjectDetectionResultItemBounds;
	image: {
		width: number;
		height: number;
	};
	ios?: any;
	android?: any;
}

export interface MLKitObjectDetectionResult extends MLKitVisionResult {
	objects: Array<MLKitObjectDetectionResultItem>;
}

export interface MLKitObjectDetectionOptions extends MLKitVisionOptions {
	/**
	 * Detect and track multiple objects in the image.
	 * Default false.
	 */
	multiple?: boolean;

	/**
	 * Try to classify the detected object(s).
	 * Default false.
	 */
	classify?: boolean;
}

export declare function detectObjects(options: MLKitObjectDetectionOptions): Promise<MLKitObjectDetectionResult>;

export declare class MLKitObjectDetection extends MLKitCameraView {}