/*********************************************************************

  Class for getting sorted correlation.

     Written by k-fleak 

**********************************************************************/

#include <iostream>
#include <iterator>
#include <vector>

#ifndef __GETSORTCORR_CPP
#define __GETSORTCORR_CPP


class GetSortCorr{
private:
  std::vector<std::vector<double> > SortCorrSet;
public:
  GetSortCorr(std::vector<std::vector<int> > &sortIndex, std::vector<std::vector<std::pair<double, double> > > &CorrSet);
  ~GetSortCorr();
  void  prints();
};


GetSortCorr::GetSortCorr(std::vector<std::vector<int> > &sortIndex, std::vector<std::vector<std::pair<double, double> > > &CorrSet){
  std::vector<std::vector<int> >::iterator it;
  it = sortIndex.begin();
  while ( it != sortIndex.end()){
    std::vector<int>::iterator it2;
    it2 = (*it).begin();
    std::vector<double> tmpCorr;
    std::vector<double> tmpVecSigma;
    std::vector<double> tmpVecNCluster;
    while ( it2 != (*it).end()){
      tmpVecSigma.resize(CorrSet[*it2].size());
      tmpVecNCluster.resize(CorrSet[*it2].size());
      for ( int j = 0; j < CorrSet[*it2].size(); ++j ){
	tmpVecSigma[j] += (CorrSet[*it2][j]).first;
	tmpVecNCluster[j] += (CorrSet[*it2][j]).second;
      }
      ++it2;
    }

    for ( int i = 0; i < tmpVecSigma.size(); ++i){
      tmpCorr.push_back(tmpVecSigma[i] / tmpVecNCluster[i]);
    }
    SortCorrSet.push_back(tmpCorr);
    tmpCorr.clear();
    tmpVecSigma.clear();
    tmpVecNCluster.clear();
    ++it;
  }
}

GetSortCorr::~GetSortCorr(){}

void GetSortCorr::prints(){
  std::vector<std::vector<double> >::iterator it;
  std::vector<double>::iterator it2;
  it = SortCorrSet.begin();
  while ( it != SortCorrSet.end()){
    it2 = (*it).begin();
    while ( it2 != (*it).end()){
      std::cout << *it2 << " ";
      ++it2;
    }
    std::cout << std::endl;
    ++it;
  }

}
  
    
#endif