Reconhecimento.Circulo.java Source code

Introduction

Here is the source code for Reconhecimento.Circulo.java

Source

/*
 * To change this license header, choose License Headers in Project Properties.
 * To change this template file, choose Tools | Templates
 * and open the template in the editor.
 */
package Reconhecimento;

import Layout.Video;
import java.awt.event.WindowAdapter;
import java.awt.event.WindowEvent;
import java.io.File;
import javax.swing.ImageIcon;
import javax.swing.JFrame;
import javax.swing.JLabel;
import javax.swing.JOptionPane;
import org.opencv.core.Core;
import org.opencv.core.Mat;
import org.opencv.core.Point;
import org.opencv.core.Scalar;
import org.opencv.core.Size;
import org.opencv.highgui.Highgui;
import org.opencv.highgui.VideoCapture;
import org.opencv.imgproc.Imgproc;
import static org.opencv.imgproc.Imgproc.THRESH_BINARY_INV;

/**
 *
 */
public class Circulo {

    public static char ind = (char) 96;

    /**
    segmenta o circulo para acompanhamento do movimento 
    **/
    public static void segmentarCirculo(int minRaio, int maxRaio, int minThreshold, int maxThreshold,
            int medianBlurKernel) {

        class threadSegmentar extends Thread {

            public boolean closed = false;
            public double CentroX;
            public double CentroY;

            @Override
            public void run() {
                int contador = 0;

                File folder = new File("imagens/frames");
                if (folder.exists() == false) {
                    folder.mkdir();
                }
                for (String file : folder.list()) {
                    new File(folder, file).delete();
                }

                ind = (char) ((int) ind + 1);

                JFrame frame = new JFrame();
                JLabel label = new JLabel();
                frame.add(label);
                frame.setBounds(10, 10, 640, 480);
                label.setSize(640, 480);
                frame.setLocation(250, 250);
                frame.setVisible(true);
                closed = false;

                frame.addWindowListener(new WindowAdapter() {
                    @Override
                    public void windowClosing(WindowEvent e) {
                        closed = true;
                    }
                });

                Mat img = new Mat();
                Mat circles = new Mat();
                Mat grayImg = new Mat();
                Mat gravar = new Mat();
                Mat element = new Mat();

                VideoCapture cap = new VideoCapture(Video.videoAtual);

                // capturar primeiro frame do video 
                cap.read(img);

                Imgproc.cvtColor(img, grayImg, Imgproc.COLOR_BGR2GRAY);

                Imgproc.medianBlur(grayImg, grayImg, 5);

                Imgproc.HoughCircles(grayImg, circles, Imgproc.CV_HOUGH_GRADIENT, 1, 100, 220, 10, minRaio,
                        maxRaio);

                double Circle[] = circles.get(0, 0);

                Point center = new Point(Math.round(Circle[0]), Math.round(Circle[1]));

                int radius = (int) Math.round(Circle[2]);

                CentroX = center.x;
                CentroY = center.y;

                cap.read(img);

                boolean continuar = true;

                while (continuar) {

                    // passar a imagem para tons de cinza
                    Imgproc.cvtColor(img, grayImg, Imgproc.COLOR_BGR2GRAY);

                    // limiarizacao
                    Imgproc.threshold(grayImg, grayImg, minThreshold, maxThreshold, THRESH_BINARY_INV);

                    Core.bitwise_not(grayImg, grayImg);

                    // filtro da mediana
                    Imgproc.medianBlur(grayImg, grayImg, medianBlurKernel);

                    // deteccao de vertices
                    Imgproc.Canny(grayImg, grayImg, 100, 255);

                    // aplicar transformada circular de hough
                    Imgproc.HoughCircles(grayImg, circles, Imgproc.CV_HOUGH_GRADIENT, 1, 100, 220, 9, minRaio,
                            maxRaio);

                    try {
                        for (int x = 0; x < circles.cols(); x++) {
                            double vCircle[] = circles.get(0, x);

                            center = new Point(Math.round(vCircle[0]), Math.round(vCircle[1]));
                            radius = (int) Math.round(vCircle[2]);

                            // analisa a distancia entre o circulo do frame anterior e atual
                            if (((center.x <= CentroX) || (center.x - CentroX <= 5))
                                    && (Math.sqrt(CentroX * CentroX + CentroY * CentroY)
                                            - Math.sqrt(center.x * center.x + center.y * center.y) <= 70.0)
                                    && (Math.sqrt(CentroX * CentroX + CentroY * CentroY)
                                            - Math.sqrt(center.x * center.x + center.y * center.y) >= -70.0)) {

                                Core.circle(img, center, radius, new Scalar(0, 0, 255), 3, 8, 0);

                                CentroX = center.x;
                                CentroY = center.y;
                            }
                        }
                    } catch (Exception e) {
                    }

                    Imgproc.resize(img, gravar, new Size(640, 480));
                    Highgui.imwrite("imagens/frames/houghcircles" + contador + ind + ".jpg", gravar);

                    label.setIcon(new ImageIcon("imagens/frames/houghcircles" + contador + ind + ".jpg"));

                    contador++;

                    continuar = cap.read(img) && !closed;
                }
            }
        }

        if (Video.videoAtual == null) {
            JOptionPane.showMessageDialog(null, "Selecione um arquivo de video!", "Nenhum vdeo selecionado",
                    JOptionPane.WARNING_MESSAGE);
            Video.abrirVideo();
        }

        threadSegmentar t = new threadSegmentar();
        t.start();
    }
}