/*! \file GaussProbModel.h
	\brief A brief file description.
  	
	A more elaborated file description.
	
	Created: 2015/06/22, author: Jin Bingwen.
*/

#ifndef __GaussProbModel_h_
#define __GaussProbModel_h_

#include <math.h>
#include <QFile>
#include <QTextStream>

class GaussProbModel
{
public:
	GaussProbModel()
		: mMean(0), mStddev(1) {}

	GaussProbModel(double mean, double stddev)
		: mMean(mean), mStddev(stddev) {}

	double prob(double x) const
	{
        if (mStddev < 0.00001) {
            printf("stddev < 0.00001\n");
            return 0;
        }
		double normX = (x - mMean) / mStddev;
		return exp(-0.5 * normX * normX) / mStddev * 0.39894228;
	}

	double normProb(double x) const{
		return prob(x) * mStddev;
	}

	static void save2file(const vector<GaussProbModel>& vec, std::string fileName)
	{
        QFile file(fileName.data());
		if (!file.open(QIODevice::WriteOnly))
		{
			return;
		}
		QTextStream out(&file);
		for (int i = 0; i < vec.size(); ++i)
		{
			const GaussProbModel& gaussModel = vec[i];
			out << gaussModel.getMean() << " " << gaussModel.getStddev() << "\n";
		}
	}

	static void loadFromFile(vector<GaussProbModel>& vec, std::string fileName)
	{
		vec.clear();
        QFile file(fileName.data());
		if (!file.open(QIODevice::ReadOnly))
		{
			return;
		}
		QTextStream in(&file);
		while (!in.atEnd())
		{
			double mean, stddev;
			in >> mean;
			in >> stddev;
			vec.push_back(GaussProbModel(mean, stddev));
		}
	}

	double getMean() const { return mMean; }
	void setMean(double iVal) { mMean = iVal; }
	double getStddev() const { return mStddev; }
	void setStddev(double iVal) { mStddev = iVal; }

protected:
	double mMean;
	
	double mStddev;
	
private:
};

#endif // __GaussProbModel_h_