/JavaCV/src/com/johnflan/javacv/detectcontours/DetectContours.java

package com.johnflan.javacv.detectcontours;


import java.util.Iterator;
import java.util.Map;

import com.googlecode.javacpp.Loader;
import com.googlecode.javacv.CanvasFrame;
import com.googlecode.javacv.OpenCVFrameGrabber;
import com.googlecode.javacv.cpp.opencv_core.CvBox2D;
import com.googlecode.javacv.cpp.opencv_core.CvContour;
import com.googlecode.javacv.cpp.opencv_core.CvMemStorage;
import com.googlecode.javacv.cpp.opencv_core.CvSeq;
import com.googlecode.javacv.cpp.opencv_core.IplImage;

import static com.googlecode.javacv.cpp.opencv_imgproc.*;
import static com.googlecode.javacv.cpp.opencv_core.*;
import static com.googlecode.javacv.cpp.opencv_video.*;
import static com.googlecode.javacv.cpp.opencv_highgui.*;


public class DetectContours {


	public static void main(String[] args) throws Exception {
		
        OpenCVFrameGrabber grabber = new OpenCVFrameGrabber(0);
        grabber.start();
        
        IplImage frameRaw = grabber.grab();
        //!!IplImage frameRaw = cvLoadImage(
        //!!        "/home/johnflan/workspace/Maple/docs/misc_images/testimg/leaf.png",
        //!!        CV_LOAD_IMAGE_UNCHANGED);
        
        
        IplImage frameProcessed = cvCreateImage( cvGetSize(frameRaw), IPL_DEPTH_8U, 1 );
        
        CanvasFrame canvasFramePre = new CanvasFrame("Preprocessing");
        canvasFramePre.setCanvasSize(frameRaw.width(), frameRaw.height());
        
        CanvasFrame canvasFramePost = new CanvasFrame("Postprocessing");
        canvasFramePost.setCanvasSize(frameRaw.width(), frameRaw.height());
        
        CanvasFrame canvasFramePre1 = new CanvasFrame("Preprocessing");
        canvasFramePre1.setCanvasSize(frameRaw.width(), frameRaw.height());
        
        CanvasFrame canvasFramePost1 = new CanvasFrame("Postprocessing");
        canvasFramePost1.setCanvasSize(frameRaw.width(), frameRaw.height());
        
    	
    	FramePreProcessor framePreProcessor = new FramePreProcessor();
    	ContourProcessor contourProcessor = new ContourProcessor();
    	
    	CvMemStorage storage = CvMemStorage.create();
    	
    	CvSeq contourPtr = new CvSeq(null);
    	CvSeq contourLow;
    	
        
    	//Load Sample contours
    	//CvSeq sampleLeaf = contourProcessor.loadTestLeafContour();
    	Map<String, CvSeq> leafSet = contourProcessor.loadTestLeaves();
    	
    	
    	
    	
    	CvSeq tempContour = leafSet.get("Maple");
    	contourProcessor.printContourPoints(tempContour);
    	IplImage mapleContour = cvLoadImage(
    	                "/home/johnflan/workspace/Maple/docs/misc_images/testimg/leaf.png",
    	                CV_LOAD_IMAGE_UNCHANGED);
    	cvDrawContours(mapleContour, tempContour, CV_RGB( 0,  255,  0 ), CV_RGB( 0,  255,  0 ), -1, 2,8, cvPoint(0,0));
    	
    	
    	canvasFramePre1.setTitle("Maple contour raw");
    	canvasFramePre1.showImage(mapleContour);

		

    	
    	System.out.println("Low");
    	CvSeq tempContourLow = cvApproxPoly(tempContour, Loader.sizeof(CvContour.class), storage,CV_POLY_APPROX_DP,1,1);
    	contourProcessor.printContourPoints(tempContourLow);
    	IplImage mapleContourPolyApprox = cvLoadImage(
                "/home/johnflan/workspace/Maple/docs/misc_images/testimg/leaf.png",
                CV_LOAD_IMAGE_UNCHANGED);
    	cvDrawContours(mapleContourPolyApprox, tempContourLow, CV_RGB( 0,  255,  0 ), CV_RGB( 0,  255,  0 ), -1, 2,8, cvPoint(0,0));
    	
		canvasFramePost1.setTitle("Maple contour poly approx");
    	canvasFramePost1.showImage(mapleContourPolyApprox);
    	
    	Iterator leafItr = leafSet.entrySet().iterator();
		
		while (leafItr.hasNext()){
			Map.Entry<String, CvSeq> leaf = (Map.Entry)leafItr.next();
    	
			leaf.setValue( cvApproxPoly( (CvSeq) leaf.getValue(), Loader.sizeof(CvContour.class), storage,CV_POLY_APPROX_DP,1,1) );
			//contourProcessor.printContourPoints(sampleLeafLow);
		}
        
        while (canvasFramePre.isVisible() && canvasFramePost.isVisible() &&  (frameRaw = grabber.grab()) != null) {

        	//convert to grayscale and equalise levels
        	framePreProcessor.prepareImage(frameRaw, frameProcessed);	
        	
        	canvasFramePre1.setTitle("After Equalisation");
        	canvasFramePre1.showImage(frameProcessed);
        	
    		framePreProcessor.applyThreshold(frameProcessed, frameProcessed);
    		cvNot(frameProcessed, frameProcessed); //invert the image
    		
    		canvasFramePost1.setTitle("After threshold");
        	canvasFramePost1.showImage(frameProcessed);

    		framePreProcessor.dilate(frameProcessed);
    		
    		canvasFramePost.setTitle("Final image");
    		canvasFramePost.showImage(frameProcessed);
        	//Detect and Draw contours
        	cvFindContours(cvCloneImage(frameProcessed), storage, contourPtr, Loader.sizeof(CvContour.class), CV_RETR_EXTERNAL, CV_CHAIN_APPROX_SIMPLE, cvPoint(0, 0));
        	
        	//Approx contours
        	contourLow = cvApproxPoly(contourPtr, Loader.sizeof(CvContour.class), storage,CV_POLY_APPROX_DP,1,1);
        	
        	
        	for (; contourLow != null; contourLow = contourLow.h_next()){
        		        		
        		//CvScalar colour = CV_RGB( Math.random() * 255,  Math.random() * 255,  Math.random() * 255 );
        		CvScalar colour = CV_RGB( 0,  255,  0 );
        		
        		//Find bounding box
        		CvRect rect;
        		CvPoint p1 = new CvPoint();
        		CvPoint p2 = new CvPoint();
        		
        		rect = cvBoundingRect(contourLow, 1);
        		
        		//If the width is less than 1/10 of the image width skip it
        		if ((frameRaw.width() * .2) > rect.width() )
        			continue;
        		
        		
        		//if (contourLow == null || sampleLeaf == null){
        		//	System.out.println("Error contourLow or sampleLeaf contain null value");
        		//	continue;
        		//}
        		
        		
        		leafItr = leafSet.entrySet().iterator();
        		double matchValue = 100;
        		String matchName = "";
        		
        		while (leafItr.hasNext()){
        			Map.Entry<String, CvSeq> leaf = (Map.Entry)leafItr.next();
        			
        			double tmpMatchValue = cvMatchShapes(contourLow, leaf.getValue(), CV_CONTOURS_MATCH_I1, 0);


        			if (tmpMatchValue <= matchValue){
        				
        				matchValue = tmpMatchValue;
        				matchName = leaf.getKey();
        			}
        			
        			//contourProcessor.printContourPoints(sampleLeafLow);
        		}
        		
        		//double matchValue = cvMatchShapes(contourLow, sampleLeafLow, CV_CONTOURS_MATCH_I1, 0);
        		//double matchValue = .1;
        		
        		if (matchValue > .04)
        			continue;
        		
        		//Draw bounding box
        		p1.x(rect.x());
        		p2.x(rect.x() + rect.width());
        		p1.y(rect.y());
        		p2.y(rect.y() + rect.height());
        		cvRectangle(frameRaw, p1, p2, colour, 2, 8, 0);
        		      		
        		//Draw contour 
        		//cvDrawContours(frameRaw, contourLow, colour, colour, -1, 0,8, cvPoint(0,0));
        		
        		
        		//----Calculate the objects features using hu moments
        		//cvMoments(contourLow, moments, 0);
        		//cvGetHuMoments(moments, humoments);
        		
    			System.out.println("Found a " + matchName + " leaf, with a value of: " + matchValue);        		
        	}
        	     	
        	
        	canvasFramePre.showImage(frameRaw);
        	canvasFramePost.showImage(frameProcessed);
//        	cvReleaseMemStorage(storage); 
//        	storage = new CvMemStorage();
        	
        }
        
        grabber.stop();
        canvasFramePre.dispose();
        canvasFramePost.dispose();

	}
}