//-----------------------------------------------------------
// File and Version Information:
// $Id$
//
// Description:
//      Context-aware storage of PID probabilities from
//      one momentum slice and one fopi detector
//      Input is created in analysis macros
//     
//
//
// Environment:
//      Software developed for GEM-TPC detector in FOPI.
//
// Author List:
//      Felix Boehmer    TUM            (original author)
//
//
//-----------------------------------------------------------



#include "FopiPidProbSet.h"

#include <cmath>
#include <utility>

ClassImp(FopiPidProbSet)

FopiPidProbSet::FopiPidProbSet()
  :
  fLow(0.),
  fHigh(0.),
  fRangeLow(0.),
  fRangeHigh(0.),
  fDetType(TString(""))
{;}

FopiPidProbSet::FopiPidProbSet(const TString& det, double min, double max,
			       double valRangeLow, double valRangeHigh)
  :
  fLow(min),
  fHigh(max),
  fRangeLow(valRangeLow),
  fRangeHigh(valRangeHigh),
  fDetType(det)
{;}




int 
FopiPidProbSet::addProb(const FopiPidProb& prob, double fraction, double err) {
  double pMin, pMax;
  prob.getMomRange(pMin, pMax);
  double diffL = std::fabs(pMin-fLow);
  double diffH = std::fabs(pMax-fHigh);
  
  //consistency check
  if(diffL+diffH > 1.e-5) {
    std::cerr<<"FopiPidProbSet::addProb() Conflict in momentum range; not adding anything!"
	     <<std::endl;
    return 1;
  }
  if(!prob.getDetType().EqualTo(fDetType)) {
    std::cerr<<"FopiPidProbSet::addProb() Conflict in detector type; not adding anything!"
	     <<std::endl;
    return 1;
  }

  //all OK:
  std::pair<TString,FopiPidProb> dP(prob.getPartType(),prob);
  std::pair<TString,double> dF(prob.getPartType(),fraction);
  std::pair<TString,double> dFErr(prob.getPartType(),err);
  std::pair<std::map<TString,FopiPidProb>::iterator,bool> insP = fProbMap.insert(dP);
  std::pair<std::map<TString,double>::iterator,bool> insF = fFracMap.insert(dF);
  std::pair<std::map<TString,double>::iterator,bool> insFErr = fFracErrMap.insert(dFErr);
  if(insP.second==false || insF.second==false || insFErr.second==false ) {
    std::cerr<<"FopiPidProbSet::addProb() Already had an entry for this PDF; not adding anything!"
	     <<std::endl;      
    return 1;
  }
  return 0;
}

void
FopiPidProbSet::getMomWindow(double& min, double& max) const {
  min=fLow;
  max=fHigh;
}

void
FopiPidProbSet::getValidityRange(double& rangeLow, double& rangeHigh) const {
  rangeLow = fRangeLow;
  rangeHigh = fRangeHigh;
}


double 
FopiPidProbSet::getFraction(const TString& part) const {
  std::map<TString, double>::const_iterator it = fFracMap.find(part);
  if(it!=fFracMap.end()) 
    return it->second;
  else {
    std::cout<<"FopiPidProbSet::getFraction() Lookup for particle "<<part
	     <<" failed.. returning -1."<<std::endl;
    return -1.;
  }
}

double 
FopiPidProbSet::getFractionError(const TString& part) const {
  std::map<TString, double>::const_iterator it = fFracErrMap.find(part);
  if(it!=fFracErrMap.end()) 
    return it->second;
  else 
    return -1.;
}

const FopiPidProb*
FopiPidProbSet::getProb(const TString& part) const {
  std::map<TString, FopiPidProb>::const_iterator it = fProbMap.find(part);
  if(it!=fProbMap.end()) 
    return &(it->second);
  else 
    return NULL;
}

void
FopiPidProbSet::clear() {
  fProbMap.clear();
  fFracMap.clear();
}