StatCollector.h

#ifndef STATCOLLECTOR_H
#define STATCOLLECTOR_H
#include <string>
#include <json/json.h>
#include <iostream>
#include <opencv2/core/core.hpp>

enum ComparisonMode{COMP_BUBBLE_VALS, COMP_BUBBLE_OFFSETS};

class StatCollector
{
int tp, fp, tn, fn;
int correctFields, incorrectFields;
int missedSegments, numSegments;
std::vector<double> times;
std::vector<cv::Point> offsets;
std::map<int, int> histogram;
private:
	void compareItems(const Json::Value& foundSeg, const Json::Value& actualSeg);
	void compareFields(const Json::Value& foundField, const Json::Value& actualField, ComparisonMode mode);
public:
	int errors, numImages;

	StatCollector():
		tp(0), fp(0), tn(0), fn(0),
		correctFields(0), incorrectFields(0),
		errors(0), numImages(0),
		missedSegments(0), numSegments(0)
	{}

	void incrErrors(){ errors++; }
	void incrImages(){ numImages++; }
	void addTime(double t){ times.push_back(t); }
	void compareFiles(const std::string& foundPath, const std::string& actualPath, ComparisonMode mode);
	void recomputeFieldValues(const std::string& inPath, const std::string& outPath) const;

	float formAlignmentRatio() const { return 1.f * (numImages - errors) / numImages; }
	float segmentAlignmentRatio() const { return 1.f * (numSegments - missedSegments) / numSegments; }
	float correctClassificationRatio() const { return 1.f * (tp + tn) / (tp+fp+tn+fn); }
	
	void printAsRow(std::ostream& myOut) const;
	void printHistRows(const std::string& condition, std::ostream& myOut) const;
	void print(std::ostream& myOut) const;
	
	StatCollector& operator+=(const StatCollector& sc){
		this->tp += sc.tp;
		this->fp += sc.fp;
		this->tn += sc.tn;
		this->fn += sc.fn;
		this->correctFields += sc.correctFields;
		this->incorrectFields += sc.incorrectFields;
		this->errors += sc.errors;
		this->numImages += sc.numImages;
		this->missedSegments += sc.missedSegments;
		this->numSegments += sc.numSegments;
		this->times.insert(this->times.end(), sc.times.begin(), sc.times.end());
		this->offsets.insert(this->offsets.end(), sc.offsets.begin(), sc.offsets.end());

		std::map<int, int>::const_iterator mapit;
		for (mapit=sc.histogram.begin() ; mapit != sc.histogram.end(); mapit++ ){
			this->histogram[(*mapit).first]+= (*mapit).second;
		}

		return *this;
	}

	StatCollector operator+(const StatCollector& sc) const {
		std::vector <double> newTimes;
		newTimes.insert(newTimes.end(), times.begin(), times.end());
		newTimes.insert(newTimes.end(), sc.times.begin(), sc.times.end());
		std::vector <cv::Point> newOffsets;
		newOffsets.insert(newOffsets.end(), offsets.begin(), offsets.end());
		newOffsets.insert(newOffsets.end(), sc.offsets.begin(), sc.offsets.end());
		return StatCollector() + sc;
	}
};

std::ostream& operator<<(std::ostream& os, const StatCollector& sc);

const std::string linebreak("________________________________________________________");

#endif