//------------------------------------------------------------------------------------
//----------------          PndSdsStripCorrelator.h          -------------------------
// Class to correlate top and bottom clusters on double sided strip sensors
//
// Authors: R. Kliemt (Uni Bonn)
//          H.G. Zaunick (Uni Bonn)
// Date:    June 2012
//
//------------------------------------------------------------------------------------

#ifndef PNDSDSSTRIPCLUSTECORRELATOR_H
#define PNDSDSSTRIPCLUSTECORRELATOR_H 1

#include <map>
#include <vector>
#include <utility>
#include <iostream>

using namespace std;

//Data types

struct PndSdsStripCorrelatorCand {
  PndSdsStripCorrelatorCand()
  : top(), bot(), q_top(), q_bot(), prob(1.)
  {}
  PndSdsStripCorrelatorCand(int t, int b, double qt, double qb, double p)
  : top(t), bot(b), q_top(qt), q_bot(qb), prob(p)
  {}
  int top,bot;
  double q_top, q_bot;
  double prob;
};


struct PndSdsStripCorrelatorCombi {
public:
  PndSdsStripCorrelatorCombi()
  : pairlist(),prob(1.)
  {}
  PndSdsStripCorrelatorCombi(const std::vector<PndSdsStripCorrelatorCand>& list, double p)
  : prob(p)
  {
    pairlist.insert(pairlist.end(),list.begin(),list.end());
  }
  PndSdsStripCorrelatorCombi(const PndSdsStripCorrelatorCombi& a_combi)
  : prob(a_combi.prob)
  {
    pairlist.insert(pairlist.end(),a_combi.pairlist.begin(),a_combi.pairlist.end());
  }
  std::vector<PndSdsStripCorrelatorCand> pairlist;
  double prob;
};  


// Now the "real" class

class PndSdsStripCorrelator {
  
public:
  PndSdsStripCorrelator(int mode=0, double cut=0., double noise=0., double threshold=0.);
  ~PndSdsStripCorrelator();
  
  void Reset();
  
  void Setup(int mode=0, double cut=0., double noise=0., double threshold=0.);
  
  void AddCluster(int moduleId, int side, int clusterIndex, double charge);
  
  vector<pair<int,int> > GetCorrelationList();
  
  vector<double> GetProbList();
  vector<double> GetSecondProbList();
  vector<int> GetMultProbList();
  
private:
  void CalcChargeDifferenceCut();
  void CalcLikelihoodAlgo();
  void CalcAll();
  std::vector<PndSdsStripCorrelatorCombi> getCombinations(std::map<int,std::map<int,PndSdsStripCorrelatorCand> > matrix, int cols, int rows);
  std::map<int,std::map<int,PndSdsStripCorrelatorCand> > getSubMatrix(std::map<int,std::map<int,PndSdsStripCorrelatorCand> > matrix, int cols, int rows, int pivotCol, int pivotRow);
  
  //      Module            Side                    clusterindex charge
  map<int,map<int,vector<pair<int,double> > > > fClusterList; //List of the cluster indices and their charges
  vector<pair<int,int> > fCorrelationList; //List of matching top/bottom clusters
  vector<double> fCorrelationProbList; //List of associated prob values for each module
  vector<double> fSecondProbList; //List of the second prob values for each module
  vector<int> fMultProbList; //List of number of correlations
  
  double fCut; //Cut parameter: ChargeDifference or minimum Likelihood
  double fNoise; //Noise sigma[e-]
  double fThreshold; //Discrimination threshold [e-]
  int fMode; //which mode to use 0:ChargeDifferenceCut 1:Liklelihood
  bool fCalculated; //flag for automatic calculation calls
  
  
};


#endif