//
//  main.cpp
//  AutoGrader
//
//  Created by Robert LiKamWa on 9/17/13.
//  Copyright (c) 2013 Robert LiKamWa. All rights reserved.
//

#include <iostream>

#include <opencv2/core/core.hpp>
#include <opencv2/highgui/highgui.hpp>
#include <opencv2/features2d/features2d.hpp>
#include "opencv2/imgproc/imgproc.hpp"
#include "dirent.h"
#include <stdio.h>
#include <string>
#include <fstream>
#include <AppKit/NSOpenPanel.h>
//#include <cstdlib>
using namespace cv;
using namespace std;

string corners_window="corners_window";
char	buf[120];

void showim(Mat image){
    namedWindow( "Display window", CV_WINDOW_AUTOSIZE );// Create a window for display.
    imshow( "Display window", image );
    waitKey(0);
}
string dirChooser(NSString *title, NSString * startdir){
    NSOpenPanel *panel = [NSOpenPanel openPanel];
    [panel setCanChooseFiles:NO];
    [panel setCanChooseDirectories:YES];
    [panel setDirectoryURL:[NSURL URLWithString:startdir]];
    //[panel setAllowsMultipleSelection:YES]; // yes if more than one dir is allowed
    [panel setTitle:title];
    NSInteger clicked = [panel runModal];
    
    if (clicked != NSFileHandlingPanelOKButton) {
        //clicked = [panel runModal];
        exit(0);
    }
    NSString * str = [[panel URL] path];
    NSLog(str);
    string s = [str UTF8String];
    cout<<s<<endl;

    return s;
}
string fileChooser(NSString *title){
    NSOpenPanel *panel = [NSOpenPanel openPanel];
    [panel setCanChooseFiles:YES];
    [panel setCanChooseDirectories:NO];
    //[panel setAllowsMultipleSelection:YES]; // yes if more than one dir is allowed
    [panel setTitle:title];
    NSInteger clicked = [panel runModal];
    
    while (clicked != NSFileHandlingPanelOKButton) {
        clicked = [panel runModal];
    }
    NSString * str = [[panel URL] absoluteString];
    string s = [str cStringUsingEncoding:NSUnicodeStringEncoding];
    return s;
}
int main(int argc, const char * argv[])
{
    
    std::string path= argv[0];
    //getcwd(argv[0], sizeof(argv[0]));
    path = path.substr(0,path.find_last_of("/"));
    NSString * execpath = [NSString stringWithUTF8String:path.c_str()];
    //cout<<"asdf"<<execpath<<endl;
    cout <<"Template Generator"<<endl;
    
    string inputpath=dirChooser(@"Template Directory (Input)",execpath);
    cout<<inputpath<<endl;
    string paramspath=inputpath;
    //string paramspath=dirChooser(@"Params Directory (Output)",execpath);
    cout<<paramspath<<endl;
    
    DIR *dir;
    struct dirent *ent;
    
    if ((dir = opendir (inputpath.c_str())) != NULL) {
        /* print all the files and directories within directory */
        while ((ent = readdir (dir)) != NULL) {
            string filename = ent->d_name;
            size_t i = filename.rfind(".png");
            if ((i != string::npos) && (i == (filename.length() - 4))){
                
                cout<<"File: "<<filename<<endl;
                string impath=inputpath+"/"+filename;
                Mat imagein=imread(impath);
                Mat image;
                resize(imagein,image,cv::Size(1275,1650));
                cout<<"Hey"<<endl;
                Mat channel[3];
                split(image, channel);
                cout<<"Heyasdf"<<endl;
                
                Mat blue=channel[0]>channel[1]+100;
                Mat red = channel[2]>channel[0]+50;
                Mat green = channel[1]>channel[0]+100;
                //showim(red);
                //cvNamedWindow("hello");
                //imshow("hello",green);
                //waitKey();
                cv::SimpleBlobDetector::Params params;
                params.minDistBetweenBlobs = 10.0f;
                params.filterByInertia = false;
                params.filterByConvexity = false;
                params.filterByColor = false;
                params.filterByCircularity = false;
                params.filterByArea = true;
                params.minArea = 5.0f;
                params.maxArea = 1000.0f;
                cv::Ptr<cv::FeatureDetector> blob_detector = new cv::SimpleBlobDetector(params);
                blob_detector->create("SimpleBlob");
                
                vector<cv::KeyPoint> keypoints;
                blob_detector->detect(red, keypoints);
                cout<<keypoints.size()<<endl;
                vector<cv::KeyPoint> greenkeypoints;
                blob_detector->detect(green, greenkeypoints);
                //cout<<greenkeypoints.size()<<" green"<<endl;
                reverse( greenkeypoints.begin(), greenkeypoints.end() );
                
                int N = int(keypoints.size());
                Point2f botright=keypoints[0].pt;
                Point2f botleft=keypoints[N-2].pt;
                Point2f top=keypoints[N-1].pt;
                
                
                string paramsstrpath = paramspath + "/Params" + filename +".txt";
                std::ofstream paramstr;
                paramstr.open (paramsstrpath.c_str());
                
                paramstr<<top.x<<" "<<top.y<<" "<<botleft.x<<" "<<botleft.y<<" "<<botright.x<<" "<<botright.y<<" "<<endl;
                
                vector<cv::KeyPoint> keypoints2;
                blob_detector->detect(blue, keypoints2);
                cout<<keypoints2.size()<<endl;

                int numcols=10;
                int numrows=int(keypoints2.size()-numcols+1);
                //blob size
                paramstr<<keypoints2[0].size<<endl;
                cout<<keypoints2[0].size<<endl;
                
                //cols
                paramstr<<numcols;
                cout<<numcols;
                for (int i=0;i<numcols;i++){
                    long ii=keypoints2.size()-i-1;
                    float X=keypoints2[ii].pt.x;
                    paramstr<<" "<<X;
                    cout<<" "<<X;
                }
                paramstr<<endl;
                cout<<endl;
                
                paramstr<<numrows;
                paramstr<<" "<<keypoints2[keypoints2.size()-1].pt.y;
                cout<<numrows;
                float * rowY = new float[numrows];
                rowY[0]=keypoints2[keypoints2.size()-1].pt.y;
                cout<<" "<<keypoints2[keypoints2.size()-1].pt.y;
                for (int ii=0;ii<numrows-1;ii++){
                    long i = keypoints2.size()-ii-numcols-1;
                    float Y=keypoints2[i].pt.y;
                    cout<<" "<<Y;
                    paramstr<<" "<<Y;
                    rowY[ii+1]=Y;
                }
                
                if (greenkeypoints.size()>0){
                    
                    string commentsparamspath = paramspath + "/Comments" + filename+".txt";
                    std::ofstream commentsparams;
                    commentsparams.open (commentsparamspath.c_str());
                    string greenstring="";
                    
                    int minX=5000000;
                    int maxX=-1;
                    for(long j=0; j<greenkeypoints.size(); j++){
                        if (greenkeypoints[j].pt.x<minX){
                            minX=greenkeypoints[j].pt.x;
                        }
                        if (greenkeypoints[j].pt.x>maxX){
                            maxX=greenkeypoints[j].pt.x;
                        }
                    }
                    commentsparams<<minX<<" "<<maxX<<" ";
                    
                    
                    for (int ii=0;ii<numrows;ii++){
                        float Y=rowY[ii];
                        float closeTopY=500000;
                        float closeBottomY=-500000;
                        int g1=0;
                        int g2=1;
                        for(long j=greenkeypoints.size()-1; j>=0; j--){
                            float distance = Y-greenkeypoints[j].pt.y;
                            if (distance>0 && distance<closeTopY){
                                g1 = greenkeypoints[j].pt.y;
                                closeTopY=distance;
                            }
                            if (distance<0 && distance>closeBottomY){
                                g2 = greenkeypoints[j].pt.y;
                                closeBottomY=distance;
                            }
                        }
                        commentsparams<<g1<<" "<<g2<<" ";
                    }
                    
                }else{
                    string commentsparamspath = paramspath + "/Comments" + filename+".txt";
                    std::ofstream commentsparams;
                    commentsparams.open (commentsparamspath.c_str());
                    string greenstring="";
                    commentsparams<<"-1 -1 ";
                }
                paramstr<<endl;
                
                paramstr<<keypoints2.size()<<endl;
                //string bubbles = bubblespath + "Bubbles" + filename +".txt" ;
                //imwrite( bubbles, blue );
                cout <<"Rows: "<<numrows <<endl;
                cout <<"Cols: "<<numcols <<endl;
                cout <<"====================="<<endl;
            }
            
            //printf ("%s\n", );
        }
        closedir (dir);
    } else {
        /* could not open directory */
        perror ("");
        return EXIT_FAILURE;
    }
    
    return 1;
    
    
    }