com.raulh82vlc.face_detection_sample.opencv.render.FaceDrawerOpenCV.java Source code

Introduction

Here is the source code for com.raulh82vlc.face_detection_sample.opencv.render.FaceDrawerOpenCV.java

Source

/*
 * Copyright (C) 2017 Raul Hernandez Lopez @raulh82vlc
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.raulh82vlc.face_detection_sample.opencv.render;

import org.opencv.core.Core;
import org.opencv.core.Mat;
import org.opencv.core.Point;
import org.opencv.core.Rect;
import org.opencv.core.Scalar;
import org.opencv.imgproc.Imgproc;

/**
 * <p>Draw face shapes utils for OpenCV</p>
 *
 * @author Raul Hernandez Lopez.
 */
public final class FaceDrawerOpenCV {

    private FaceDrawerOpenCV() {

    }

    private static final Scalar FACE_RECT_COLOR = new Scalar(0, 255, 0, 255);

    public static void drawFaceShapes(Rect face, Mat matrixRGBA) {
        Point start = face.tl();
        int h = (int) start.y + (face.height / 2);
        int w = (int) start.x + (face.width / 2);
        Imgproc.rectangle(matrixRGBA, start, face.br(), FACE_RECT_COLOR, 3);
        Point center = new Point(w, h);
        Imgproc.circle(matrixRGBA, center, 10, new Scalar(255, 0, 0, 255), 3);
    }

    public static void drawEyesRectangles(Rect rightEyeArea, Rect leftEyeArea, Mat matrixRgba) {
        drawEyeRectangle(leftEyeArea, matrixRgba);
        drawEyeRectangle(rightEyeArea, matrixRgba);
    }

    public static void drawEyeRectangle(Rect eyeArea, Mat matrixRgba) {
        Imgproc.rectangle(matrixRgba, eyeArea.tl(), eyeArea.br(), new Scalar(255, 0, 0, 255), 2);
    }

    public static void drawIrisCircle(Mat matrixRgba, Core.MinMaxLocResult minMaxLocResult) {
        Imgproc.circle(matrixRgba, minMaxLocResult.minLoc, 2, new Scalar(255, 255, 255, 255), 2);
    }

    public static void drawMatchedEye(Point matchLocTx, Point matchLocTy, Mat matrixRgba) {
        Imgproc.rectangle(matrixRgba, matchLocTx, matchLocTy, new Scalar(255, 255, 0, 255));
    }
}