//-----------------------------------------------------------
// File and Version Information:
// $Id$
//
// Description:
//      Implementation of class PndTpcCluster
//      see PndTpcCluster.h for details
//
// Environment:
//      Software developed for the PANDA Detector at FAIR.
//
// Author List:
//      Sebastian Neubert    TUM            (original author)
//
//
//-----------------------------------------------------------

// Panda Headers ----------------------

// This Class' Header ------------------
#include "PndTpcCluster.h"


// C/C++ Headers ----------------------
#include <iostream>
#include <set>
using namespace std;

// Collaborating Class Headers --------
#include "TVectorD.h"

// Class Member definitions -----------


//TODO: get rid of these horrible const_casts!!!!!! 

PndTpcCluster::PndTpcCluster()
  : FairHit(),
    fpos(0,0,0), fsig(0.1,0.1,0.01),fcov(3,3),famp(0), fsize(0), 
    fnPad(0), fnPadX(0), fnPadY(0), findex(0), fsector(0), findexInTrack(-1)
{}

PndTpcCluster::PndTpcCluster(const PndTpcCluster& clust) 
  : FairHit(2, const_cast<TVector3&>(clust.fpos), const_cast<TVector3&>(clust.fsig), 0),
    fpos(clust.fpos), fsig(clust.fsig),fcov(clust.fcov), faxis(clust.faxis),
    famp(clust.famp), fhasaxis(clust.fhasaxis), fnPad(clust.fnPad), 
    fnPadX(clust.fnPadX),fnPadY(clust.fnPadY),
    findex(clust.findex), fsize(clust.fsize),fsector(clust.fsector),findexInTrack(clust.findexInTrack),
    fmcid(clust.fmcid), digis(clust.digis)
{ 
  ;
}

PndTpcCluster::PndTpcCluster(const TVector3& Pos, double Amp, 
			     unsigned int Index, unsigned int Size)
  : FairHit(),
    fpos(Pos), fsig(0.1,0.1,0.01),fcov(3,3), famp(Amp), findex(Index), 
    fsize(Size),fsector(0),findexInTrack(-1)
{
  SetPosition(Pos);
  SetDetectorID(2);
  SetPositionError(TVector3(0.1,0.1,0.01));
  SetRefIndex(0);
}

PndTpcCluster::PndTpcCluster(const TVector3& Pos, const TVector3& Sig, double Amp, 
			     unsigned int Index, unsigned int Size)
  : FairHit(2, const_cast<TVector3&>(Pos), const_cast<TVector3&>(Sig), 0),
    fpos(Pos), fsig(Sig) ,fcov(3,3), famp(Amp), findex(Index), fsize(Size),
    fsector(0),findexInTrack(-1)
{}



void
PndTpcCluster::addDigi(const PndTpcDigi* d){
  digis.push_back(*d); // makes a copy!
  //AddLink(FairLink("PndTpcDigi", d.index()));
}
void
PndTpcCluster::addDigi(const PndTpcDigi* d, double share){
  digis.push_back(*d); // makes a copy!
  double a = digis.back().amp()*share;
  digis.back().amp(a);
    //AddLink(FairLink("PndTpcDigi", d.index()));
}

//BEWARE: Cannot be used after reading Cluster from file (if digi is not loaded)
const PndTpcDigi*
PndTpcCluster::getDigi(int i) const{
  assert (i<digis.size());
  return &(digis[i]);
}




TVector3
PndTpcCluster::calcAxis(unsigned int i) const {
  assert(i<3);
  TVectorD eigenValues(3);
  TMatrixD eigenVec=fcov.EigenVectors(eigenValues);

  // eigenvalues are sorted according to their value
  // in descendign order -> first one is largest
  TVectorD a=TMatrixDColumn(eigenVec,i);
  TVector3 Axis(a[0],a[1],a[2]);
  Axis.SetMag(1);
  return Axis;
}

unsigned int 
PndTpcCluster::get2DSize() { //return the 2-dimensional size of the cluster
  std::set<unsigned int> ids;
  for(int k=0; k<digis.size(); k++) {
    unsigned int id = (digis[k]).padId();
    ids.insert(id);
  }
  return ids.size();
}


PndTpcCluster::~PndTpcCluster(){}

std::ostream& operator<< (std::ostream& s, const PndTpcCluster& me){
  s << "PndTpcCluster\n"
    << "pos=("<<me.fpos.X()<<","<<me.fpos.Y()<<","<<me.fpos.Z()<<")\n"
    << "amp=" << me.famp << "\n"
    << "size=" << me.fsize <<"\n"
    << "index="<< me.findex;
  return s;
} 

ClassImp(PndTpcCluster)