/*
    Copyright (C) 2012 Bartosz Broda, Jacek Bochenek, Łukasz Biłeńkij, 
    Damian Jaworski, Wojciech Mazur, Dominik Piasecki, Maciej Piasecki, 
    Radosław Ramocki
    
    This program is free software; you can redistribute it and/or modify it
under the terms of the GNU Lesser General Public License as published by the Free
Software Foundation; either version 3 of the License, or (at your option)
any later version.

    This program is distributed in the hope that it will be useful, but
WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
or FITNESS FOR A PARTICULAR PURPOSE. 

    See the LICENSE and COPYING files for more details.
*/

#ifndef ZSCORES_H
#define ZSCORES_H

#include "comparator_all.h"
#include <vector>
#include <map>

using namespace smartcomparator::compare;

namespace clustering
{
	namespace zscores
	{
		typedef std::map<int, std::vector<size_t> > ClusterMap;

		class ZScores
		{
		public:
			ZScores(std::vector<int> cluster_ids, CompareFunctionPtr cmp_fun);

			ClusterMap & GetClusterMap();

			double GetInternalZScore(size_t row_index);

			double GetExternalZScore(size_t row_index);

		private:
			void CreateClusterMap(std::vector<int> cluster_ids);

			void CountInternalZScores();

			void CountExternalZScores();

			ClusterMap _clusterMap;

			CompareFunctionPtr cmp_fun;

			/**
			  * (s_j - \mu_j) / sigma_j for every row
			  */
			std::map<size_t, double> internals, externals;
		};
	}
}

#endif //ZSCORES_H
