main.cpp

#include <iostream>
#include <fstream>
#include <sstream>

//#include <string>
#include <cmath>
#include <random>
#include <dirent.h>

#include "data.h"
#include "test.h"

int numberOfLayers = 3;
string imagePath = "data/images/";
string locationPath = "data/locations/";
string rotationPath = "data/rotations/";

void copyChars(char *source, char *target){
	int i= 0;
	while(source[i] != '\0'){
		target[i] = source[i];
		++i;
	}
	target[i] = '\0';
}

void setRotationAngle(data * data,  string currentLocation){
	//Look, if rotation has already been calculated and stored in data/locations/
	ifstream location(currentLocation);
	if (location.is_open()){
	//	cout << "Previous calulation of rotation-angle found."<< endl;

		std::string line;
		std::getline(location, line);
		std::stringstream iss(line);
	//	cout << line << endl;
		double x = 0;
		if (!(iss >> data->angle)){
			cout << " oh no, no parsing "<< endl;
		}
	}
	else{
		cout << "No previous calulation of rotation-angle found. Calculate Rotation."<< endl;

		data->calc_best_rot_angle();
		std::ofstream outfile (currentLocation);
		outfile << (double)data->angle << std::endl;
		outfile.close();
	}
}

void showStomata(data data){
	string windowName = "Image ";
	cv::namedWindow(windowName);
	cv::Mat temp = data.getImage();
	for(int j = 0; j < data.numberOfStomata(); ++j)
		cv::circle(temp, data.getCoordinate(j), 20, CV_RGB(255,255,255),10);
	cv::imshow(windowName , temp);
	cv::waitKey(-1);
	cv::destroyWindow(windowName);
}


std::vector<data> getData(){
	std::vector<data> datasets;
	struct dirent *entry;
	DIR *dir = opendir(imagePath.c_str());
	if (dir != NULL){
		//Skip first 2 entries . & ..
		entry = readdir(dir);
		entry = readdir(dir);
		while ((entry = readdir(dir))){
			ifstream location(locationPath + entry->d_name + ".txt");
			//Add the picture only if there is ground-truth data
			if (location.is_open()){
				char *name = new char[128];
				copyChars(entry->d_name, name);
				cv::Mat image= cv::imread(imagePath + name, CV_LOAD_IMAGE_GRAYSCALE);
				std::vector<cv::Point> coordinates;
				std::string line;
				while (std::getline(location, line)){
					std::istringstream iss(line);
					int x, y;
					char c;
					if (!(iss >> x >> c >> y)){
						cout << " oh no, no parsing "<< endl;
						break;
					}
					coordinates.push_back(cv::Point(x,y));
				}
				data currentData = data(image, coordinates, name);

				setRotationAngle(&currentData,  rotationPath + entry->d_name + ".rot");
				//showStomata(currentData);
				/*
				string windowName = "Image ";
				cv::namedWindow(windowName);
				image = currentData.getImage();
				cv::Point center(image.cols/2.0, image.rows/2.0);
				//cv::Mat rot = cv::getRotationMatrix2D(center, currentData.angle, 1.0);
				//cv::warpAffine(image, image, rot , image.size());
				for(int j = 0; j < currentData.numberOfStomata(); ++j)
					cv::circle(image, currentData.getCoordinate(j), 2, CV_RGB(255,255,255),2);
				cv::imshow(windowName + currentData.name, image);
				cv::waitKey(-1);
				cv::destroyWindow(windowName + currentData.name);
				*/
				datasets.push_back(currentData);
				location.close();
			}
		}
	}
	cout << datasets.size() << " images with corresponding stomata coordinates found." << endl;
	return datasets;
}


int main(void){
	cout << "OpenCV version : " << CV_VERSION << endl;

	std::vector<data> datasets = getData();
	test testInstance = test(&datasets, 240, 6360);
	testInstance.startTesting();
	return EXIT_SUCCESS;
}