OpenHPS/openhps-opencv

import { ImageObjectClassifierNode } from './ImageObjectClassifierNode';
import { HAAR_EYE, CascadeClassifier } from 'opencv4nodejs';
import { ImageFeatureObject, ImageFrame } from '../../../../common';

export class EyeDetectionNode extends ImageObjectClassifierNode<ImageFeatureObject> {
    constructor() {
        super(new CascadeClassifier(HAAR_EYE), ImageFeatureObject, 2);
    }

    /**
     * Process the data that was pushed or pulled from this layer
     *
     * @param {ImageFrame} data Data frame
     * @returns {Promise<ImageFrame>} Image frame processing promise
     */
    process(data: ImageFrame): Promise<ImageFrame> {
        return new Promise<ImageFrame>((resolve, reject) => {
            let faceDetected = false;
            data.getObjects(ImageFeatureObject).forEach(() => {
                faceDetected = true;
            });
            if (faceDetected) {
                super
                    .process(data)
                    .then((frame) => {
                        resolve(frame);
                    })
                    .catch((ex) => {
                        reject(ex);
                    });
            } else {
                resolve(data);
            }
        });
    }
}