//************************************************************************//
//									  //
//  Copyright 2017 Bertram Kopf (bertram@ep1.rub.de)			  //
//          	   - Ruhr-Universit??t Bochum 				  //
//									  //
//  This file is part of Pawian.					  //
//									  //
//  Pawian is free software: you can redistribute it and/or modify	  //
//  it under the terms of the GNU General Public License as published by  //
//  the Free Software Foundation, either version 3 of the License, or 	  //
//  (at your option) any later version.	 	      	  	   	  //
//									  //
//  Pawian 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	  //
//  GNU General Public License for more details.	      		  //
//									  //
//  You should have received a copy of the GNU General Public License     //
//  along with Pawian.  If not, see <http://www.gnu.org/licenses/>.	  //
//									  //
//************************************************************************//

// TMatrixDynamics class definition file. -*- C++ -*-
// Copyright 2017 Bertram Kopf

#include <getopt.h>
#include <fstream>
#include <string>

#include "Utils/PawianIOUtils.hh"
#include "PwaUtils/TMatrixDynamics.hh"
#include "PwaUtils/XdecAmpRegistry.hh"
#include "PwaUtils/AbsDecay.hh"
#include "PwaUtils/AbsXdecAmp.hh"
#include "PwaUtils/GlobalEnv.hh"
#include "PwaUtils/EvtDataScatteringList.hh"
#include "ErrLogger/ErrLogger.hh"
#include "Particle/Particle.hh"
#include "Particle/ParticleTable.hh"
#include "ConfigParser/KMatrixParser.hh"
#include "PwaDynamics/KMatrixRel.hh"
#include "PwaDynamics/KMatrixRelBg.hh"
#include "PwaDynamics/TMatrixRel.hh"
#include "PwaDynamics/KPole.hh"
#include "PwaDynamics/KPoleBarrier.hh"
#include "PwaDynamics/PPole.hh"
#include "PwaDynamics/PPoleBarrier.hh"
#include "PwaDynamics/AbsPhaseSpace.hh"
#include "PwaDynamics/PhaseSpaceFactory.hh"
#include "PwaDynamics/KMatrixFunctions.hh"
#include "FitParams/AbsPawianParameters.hh"
#include "Utils/IdStringMapRegistry.hh"
#include "Utils/PawianConstants.hh"

TMatrixDynamics::TMatrixDynamics(std::string& name, std::vector<Particle*>& fsParticles, 
				 Particle* mother, std::string& pathToConfigParser, 
				 std::string dataType, std::string projectionParticleNames) :
  AbsDynamics(name, fsParticles, mother)
  ,_kMatName("")
  ,_prodProjectionIndex(0)
  ,_decProjectionIndex(0)
  , _orderKMatBg(-1)
  ,_withKMatAdler(false)
  ,_dataTypeID(0)
  ,_prodIsNotDecChannel(false)
  ,_orbitalL(0)
  ,_currentAdler0(0.)
  ,_projectionParticleNames(projectionParticleNames)
{
  _pathToKMatCfgParser=PawianIOUtils::getFileName(GlobalEnv::instance()->KMatrixStorePath(), pathToConfigParser);
  _kMatrixParser = std::shared_ptr<KMatrixParser>(new KMatrixParser(_pathToKMatCfgParser));
  
  if(dataType=="Elasticity") _dataTypeID=1;
  else if(dataType=="Phase") _dataTypeID=2;
  else if(dataType=="ArgandUnits") _dataTypeID=3;
  else if(dataType=="Data") _dataTypeID=4;
  else if(dataType=="PhaseDiff") _dataTypeID=5; //in TMatrixCompareDynamics and FVectorCompareDynamics only
  else if(dataType=="PVecIntensity") _dataTypeID=6; //in FVectorIntensityDynamics only
  else if(dataType=="Treal") _dataTypeID=7;
  else if(dataType=="Timag") _dataTypeID=8;
  else if(dataType=="Tcheck") _dataTypeID=9;
  else {
    Alert << "production formalism/data type with the name" << dataType 
	  << " is not supported for pi pi scattering fits! \n It is working for: "
	  << "Elasticity, ArgandUnits, Phase or PhaseDiff!!!" << endmsg;
    exit(1); 
  }

  init();
  if(_prodProjectionIndex != _decProjectionIndex) _prodIsNotDecChannel=true;
  _isLdependent=true;
}

TMatrixDynamics::TMatrixDynamics(std::shared_ptr<KMatrixParser> kMatrixParser) :
  AbsDynamics()
  ,_kMatrixParser(kMatrixParser) 
{
  init();

}
TMatrixDynamics::~TMatrixDynamics()
{
}

complex<double> TMatrixDynamics::eval(EvtData* theData, AbsXdecAmp* grandmaAmp, Spin OrbMom){

  vector<std::shared_ptr<AbsPhaseSpace> > thePhpVecs=_tMatr->kMatrix()->phaseSpaceVec();
  double currentMass =
    theData->DoubleMassId.at(IdStringMapRegistry::instance()->stringId(EvtDataScatteringList::M_PIPISCAT_NAME));
  _tMatr->evalMatrix(currentMass, OrbMom);

  if(_dataTypeID==1) evalElasticity(theData, currentMass, OrbMom);   
  else if(_dataTypeID==2){
    if(!_prodIsNotDecChannel) evalPhase(theData, currentMass, OrbMom);
    else evalRelativePhase(theData, currentMass, OrbMom);
  }
  else if(_dataTypeID==3){
    evalArgandUnits(theData, currentMass, OrbMom);
  }
  else if(_dataTypeID==7){
    evalTreal(theData, currentMass, OrbMom);
  }
  else if(_dataTypeID==8){
    evalTimag(theData, currentMass, OrbMom);
  }
  else{
    Alert << "_dataTypeID = " <<_dataTypeID << " is not supported!!!" << endmsg;
    exit(1); 
  }

  return (*_tMatr)(_prodProjectionIndex,_decProjectionIndex);
}

void TMatrixDynamics::fillDefaultParams(std::shared_ptr<AbsPawianParameters> fitPar){
  //pole positions
  for(unsigned int i=0; i<_currentPoleMasses.size(); ++i){
    double valMass=_currentPoleMasses.at(i);
    double errMass=0.02;
    double minMass=valMass-5.*errMass;
    if(minMass<0.) minMass=0.;
    double maxMass=valMass+5.*errMass;
    fitPar->Add(_poleNames.at(i)+"Mass", valMass, errMass);
    fitPar->SetLimits(_poleNames.at(i)+"Mass", minMass, maxMass);
  }

  //g-factors
  for(unsigned int i=0; i<_poleNames.size(); ++i){
    std::vector<double> currentgFactorVec=_currentgFactorMap.at(i);
    for(unsigned int j=0; j<currentgFactorVec.size(); ++j){
      std::string currentName=_poleNames.at(i)+_gFactorNames.at(j)+"gFactor";
      double currentError=currentgFactorVec.at(j)/3.;
      if (currentError<1.e-5) currentError=0.01;
      fitPar->Add(currentName, currentgFactorVec.at(j), currentError);
      fitPar->SetLimits(currentName, 0., 3.);
    }
  }

  //k-matrix bg-terms
  if(_orderKMatBg>=0){
    for(unsigned int i=0; i<=fabs(_orderKMatBg); ++i){
      for(unsigned int j=0; j<_phpVecs.size(); ++j){
  	for(unsigned int k=j; k<_phpVecs.size(); ++k){
  	  std::string currentName=_bgTermNames.at(i).at(j).at(k);
  	  fitPar->Add(currentName, _currentBgTerms.at(i).at(j).at(k), fabs(_currentBgTerms.at(i).at(j).at(k))+0.3);
	  fitPar->SetLimits(currentName, -20., 20.);
  	}
      }
    }

    //Adler-term
    if(_withKMatAdler){
      //      fitPar->Add("s0"+_kMatName, _currentAdler0, 0.005, 0., 0.036437);
      fitPar->Add("s0"+_kMatName, _currentAdler0, 0.005, 0., 0.3);
    }
  }

  if(_dataTypeID==2 && _prodIsNotDecChannel){
    _paramNameRelPhase="deltaRel"+_kMatName+"to"+_gFactorNames.at(_decProjectionIndex);
    fitPar->Add(_paramNameRelPhase, 90., 10.);
    fitPar->SetLimits(_paramNameRelPhase, 0., 360.);
    _currentRelPhase = fitPar->Value(_paramNameRelPhase);
  }
}

void TMatrixDynamics::fillParamNameList(){
  _paramNameList.clear();

  std::map<std::string, std::vector<std::string> >::iterator itNameMap;  
  //pole positions
  for(unsigned int i=0; i<_currentPoleMasses.size(); ++i){
    _paramNameList.push_back(_poleNames.at(i)+"Mass");
    for(itNameMap=_paramNameListMap.begin(); itNameMap!=_paramNameListMap.end(); ++itNameMap){
      itNameMap->second.push_back(_poleNames.at(i)+"Mass");
    }
  }
  
  //g-factors
  for(unsigned int i=0; i<_poleNames.size(); ++i){
    std::vector<double> currentgFactorVec=_currentgFactorMap.at(i);
    for(unsigned int j=0; j<currentgFactorVec.size(); ++j){
      std::string currentName=_poleNames.at(i)+_gFactorNames.at(j)+"gFactor";
      _paramNameList.push_back(currentName);
      for(itNameMap=_paramNameListMap.begin(); itNameMap!=_paramNameListMap.end(); ++itNameMap){
	itNameMap->second.push_back(currentName);
      }
    }
  }
  
  //k-matrix bg-terms
  if(_orderKMatBg>=0){
    for(unsigned int i=0; i<=fabs(_orderKMatBg); ++i){
      for(unsigned int j=0; j<_phpVecs.size(); ++j){
  	for(unsigned int k=j; k<_phpVecs.size(); ++k){
  	  std::string currentName=_bgTermNames.at(i).at(j).at(k);
	  _paramNameList.push_back(currentName);
          for(itNameMap=_paramNameListMap.begin(); itNameMap!=_paramNameListMap.end(); ++itNameMap){
	    itNameMap->second.push_back(currentName);
	  }
  	}
      }
    }
    //Adler-term
    if(_withKMatAdler){
      _paramNameList.push_back("s0"+_kMatName);
      for(itNameMap=_paramNameListMap.begin(); itNameMap!=_paramNameListMap.end(); ++itNameMap){
	itNameMap->second.push_back("s0"+_kMatName);
      }
    }
  }
}

bool TMatrixDynamics::checkRecalculation(std::shared_ptr<AbsPawianParameters> fitParNew, 
					 std::shared_ptr<AbsPawianParameters> fitParOld){
  return true;
}

void TMatrixDynamics::updateFitParams(std::shared_ptr<AbsPawianParameters> fitPar){

  //pole positions
  for(unsigned int i=0; i<_currentPoleMasses.size(); ++i){
    std::string currentPoleName=_poleNames.at(i)+"Mass";
    double currentPoleMass=fitPar->Value(currentPoleName);
    _currentPoleMasses.at(i)=currentPoleMass;
    _kPoles.at(i)->updatePoleMass(currentPoleMass);
  }
 
 //g-factors
  for(unsigned int i=0; i<_poleNames.size(); ++i){
    std::vector<double>& currentgFactorVec=_currentgFactorMap.at(i);
    for(unsigned int j=0; j<currentgFactorVec.size(); ++j){
      std::string currentName=_poleNames.at(i)+_gFactorNames.at(j)+"gFactor";
      currentgFactorVec.at(j)=fitPar->Value(currentName);
    }
    _kPoles.at(i)->updategFactors(currentgFactorVec);
  }
    
  //k-matrix bg-terms
  if(_orderKMatBg>=0){
    for(unsigned int i=0; i<=fabs(_orderKMatBg); ++i){
      for(unsigned int j=0; j<_phpVecs.size(); ++j){
	for(unsigned int k=j; k<_phpVecs.size(); ++k){
	  double newVal=fitPar->Value(_bgTermNames.at(i).at(j).at(k));
	  _currentBgTerms.at(i).at(j).at(k)=newVal;
	  _kMatr->updateBgTerms(i,j,k,newVal);
	}
      }
    }

    //Adler-term
    if(_withKMatAdler){
      _currentAdler0=fitPar->Value("s0"+_kMatName);
      _kMatr->updates0Adler(_currentAdler0);
    }
  }

  if(_dataTypeID==2 && _prodIsNotDecChannel){
    _currentRelPhase=fitPar->Value(_paramNameRelPhase);
  }

}

void TMatrixDynamics::init(){
  _kMatName=_kMatrixParser->keyName();
  _orderKMatBg=_kMatrixParser->orderBg();
  _withKMatAdler=_kMatrixParser->useAdler();  
  _orbitalL=_kMatrixParser->orbitalMom(); 

  std::vector<std::string> poleNameAndMassVecs=_kMatrixParser->poles();
  std::vector<std::string>::iterator itString;
  for (itString=poleNameAndMassVecs.begin(); itString!=poleNameAndMassVecs.end(); ++itString){
    std::istringstream poleIString(*itString);
    std::string currentPoleName;
    std::string currentPoleMassStr;
    poleIString >> currentPoleName >> currentPoleMassStr;

    std::istringstream currentPoleMassiStr(currentPoleMassStr);
    double currentValue;
    if(!(currentPoleMassiStr >> currentValue)){
      Alert << "cannot convert " << currentPoleMassStr << " to a double value" << endmsg;
      exit(0);
    }

    _currentPoleMasses.push_back(currentValue);
    _poleNames.push_back(currentPoleName);
  }

  if(_currentPoleMasses.size()!= _kMatrixParser->noOfPoles()){
    Alert << "number of poles != number of pole masses" << endmsg;
    exit(0);
  }

  const std::vector<std::string> gFacStringVec=_kMatrixParser->gFactors();
  std::map<std::pair<std::string, std::string>, std::string> phpDescriptionVec=_kMatrixParser->phpDescriptionMap();

  std::vector<std::string>::const_iterator itStrConst;
  for(itStrConst=gFacStringVec.begin(); itStrConst!=gFacStringVec.end(); ++itStrConst){
    std::istringstream particles(*itStrConst);
    std::string firstParticleName;
    std::string secondParticleName;
    particles >> firstParticleName >> secondParticleName;

    Particle* firstParticle = GlobalEnv::instance()->particleTable()->particle(firstParticleName);
    Particle* secondParticle = GlobalEnv::instance()->particleTable()->particle(secondParticleName);
    if(0==firstParticle || 0==secondParticle){
      Alert << "particle with name: " << firstParticleName <<" or " << secondParticleName 
	    << " doesn't exist in pdg-table" << endmsg;
      exit(0);
    }

  std::pair<std::string, std::string> currentParticlePair=make_pair(firstParticleName, secondParticleName);
    std::string currentPhpDescription = phpDescriptionVec.at(currentParticlePair);

    std::vector<double> fsParticleMasses;
    fsParticleMasses.push_back(firstParticle->mass());
    fsParticleMasses.push_back(secondParticle->mass()); 

    std::shared_ptr<AbsPhaseSpace> currentPhp=PhaseSpaceFactory::instance()->getPhpPointer(currentPhpDescription, 
											   fsParticleMasses);
    _phpVecs.push_back(currentPhp);

    std::string gFactorKey=firstParticleName+secondParticleName;    
    _gFactorNames.push_back(gFactorKey);   

    for (int i=0; i<int(_kMatrixParser->noOfPoles()); ++i){
      std::string currentPoleName=_poleNames[i];
      std::string currentgValueStr;
      if(!(particles >> currentgValueStr)){
	Alert << "g-factors for pole " << currentPoleName << " does not exist!" << endmsg;
	exit(0);
      }
      std::istringstream currentgValueiStr(currentgValueStr);
      double currentGValue;
      if (!(currentgValueiStr >> currentGValue)){
	Alert << "cannot convert " << currentgValueStr << " to a double value" << endmsg;
	exit(0);
      }
      _currentgFactorMap[i].push_back(currentGValue);
    }
  }


  std::map<int, std::vector<double> >::iterator itgFac;
  for (itgFac=_currentgFactorMap.begin(); itgFac!=_currentgFactorMap.end(); ++itgFac){
    std::vector<double> currentgVector=itgFac->second;
    std::shared_ptr<KPole> currentPole;
    if (_kMatrixParser->useBarrierFactors()){
       currentPole=std::shared_ptr<KPole>(new KPoleBarrier(currentgVector, _currentPoleMasses.at(itgFac->first), 
							  _phpVecs, _kMatrixParser->orbitalMom(), 
							  _kMatrixParser->useTruncatedBarrierFactors()));
    }
    else{
       currentPole=std::shared_ptr<KPole>(new KPole(currentgVector, _currentPoleMasses.at(itgFac->first)));
    }
    _kPoles.push_back(currentPole);
  }

  if(_orderKMatBg<0) _kMatr=std::shared_ptr<KMatrixRel>(new KMatrixRel(_kPoles,_phpVecs ));
  else {
    _currentBgTerms.resize(_orderKMatBg+1);
    _bgTermNames.resize(_orderKMatBg+1);
    for(unsigned int i=0; i<= fabs(_orderKMatBg); ++i){
      _currentBgTerms.at(i).resize(_phpVecs.size());
      _bgTermNames.at(i).resize(_phpVecs.size());
      for(unsigned int j=0; j<_phpVecs.size(); ++j){
	_currentBgTerms.at(i).at(j).resize(_phpVecs.size());
	_bgTermNames.at(i).at(j).resize(_phpVecs.size());
	for(unsigned int k=j; k<_phpVecs.size(); ++k){
	  _currentBgTerms.at(i).at(j).at(k)=0.;

	  std::stringstream keyOrderStrStr;
	  keyOrderStrStr << i << j << k;
	  std::string keyOrder=keyOrderStrStr.str();
	  std::string currentName="bg"+keyOrder+_kMatName;
	  _bgTermNames.at(i).at(j).at(k)=currentName;
	}
      }
    }

    _kMatr=std::shared_ptr<KMatrixRel>(new KMatrixRelBg(_kPoles,_phpVecs, _orderKMatBg, _withKMatAdler ));
    if(_withKMatAdler){
      _currentAdler0=_kMatrixParser->s0Adler();
      _kMatr->updates0Adler(_currentAdler0);
      _kMatr->updatesnormAdler(_kMatrixParser->snormAdler());
    }
  }

  if (_projectionParticleNames == "") {
    _projectionParticleNames = _kMatrixParser->projection();
  } else {
    InfoMsg << _kMatName << " K-matrix projection changed!!!" 
	    << "\nfrom k-Matrix parser configuration: " << _kMatrixParser->projection()
	    << "\nto: " << _projectionParticleNames << "\n"<< endmsg;
  }
   
  //  const std::string porjectionParticleNames=_kMatrixParser->projection();
  std::istringstream projParticles(_projectionParticleNames);
  std::string firstProjParticleName;
  std::string secondProjParticleName;
  projParticles >> firstProjParticleName >> secondProjParticleName;
  std::string projKey=firstProjParticleName+secondProjParticleName;    

  bool found=false;
  for(unsigned int i=0; i<_gFactorNames.size();++i){
    if(projKey==_gFactorNames[i]){
      _decProjectionIndex=i;
      found=true;
    }
  }
  if (!found){
    Alert << "decay projection index for key " << projKey << " not found" << endmsg;
    exit(0);
  }
  // InfoMsg << _kMatName << ": decay projection index: " << _decProjectionIndex << endmsg;
  _tMatr=std::shared_ptr<TMatrixRel>(new TMatrixRel(_kMatr));
  _tMatr->SetOrbMom(_kMatrixParser->orbitalMom());

}

void TMatrixDynamics::evalElasticity(EvtData* theData, double currentMass, Spin OrbMom){
  vector<std::shared_ptr<AbsPhaseSpace> > thePhpVecs=_tMatr->kMatrix()->phaseSpaceVec();
  complex<double> currentTijRel=(*_tMatr)(_prodProjectionIndex,_decProjectionIndex);
  complex<double> SijRel;

  //note: this is a workaround
  if(_prodProjectionIndex == _decProjectionIndex){
    SijRel=complex<double>(1.,0.)+2.*PawianConstants::i *
      sqrt(thePhpVecs[_prodProjectionIndex]->factor(currentMass, OrbMom).real() *
	   thePhpVecs[_decProjectionIndex]->factor(currentMass, OrbMom).real()) * currentTijRel;
  }
  else{
    SijRel=2.*PawianConstants::i *
      sqrt(thePhpVecs[_prodProjectionIndex]->factor(currentMass, OrbMom).real() *
	   thePhpVecs[_decProjectionIndex]->factor(currentMass, OrbMom).real()) * currentTijRel;
  }
  
  //protection against numerical instabilities
  if(norm(SijRel) != norm(SijRel)) {
    WarningMsg << "numerical instability of norm(SijRel) yields to NAN ... redo calculation for mass+0.0001"
	       << endmsg;
    double newCurrentMass=currentMass+0.0001;
    _tMatr->evalMatrix(newCurrentMass, OrbMom);
    evalElasticity(theData, newCurrentMass, OrbMom);
    return;
  } 

  theData->DoubleId.at(IdStringMapRegistry::instance()->stringId(EvtDataScatteringList::FIT_PIPISCAT_NAME))=sqrt(norm(SijRel));
}


void TMatrixDynamics::evalPhase(EvtData* theData, double currentMass, Spin OrbMom=0){

  double deltaRel = KMatrixFunctions::deltaArgand(_tMatr, _prodProjectionIndex, currentMass, OrbMom);
  while(deltaRel>180.) deltaRel-=180.;
  while(deltaRel<0.) deltaRel+=180.;

  double phiData=theData->DoubleId.at(IdStringMapRegistry::instance()->stringId(EvtDataScatteringList::DATA_PIPISCAT_NAME));

  while( (phiData-deltaRel) > 90.) deltaRel+=180.;
  while( (deltaRel-phiData) > 90.) deltaRel-=180.;

  theData->DoubleId.at(IdStringMapRegistry::instance()->stringId(EvtDataScatteringList::FIT_PIPISCAT_NAME))=deltaRel;

}

void TMatrixDynamics::evalRelativePhase(EvtData* theData, double currentMass, Spin OrbMom){
  //double deltaiiRel = KMatrixFunctions::deltaArgand(_tMatr, _prodProjectionIndex, currentMass, OrbMom);
  //double deltajjRel = KMatrixFunctions::deltaArgand(_tMatr, _decProjectionIndex, currentMass, OrbMom); 
  //double phiRelFit=deltaiiRel+deltajjRel+_currentRelPhase;
  vector<std::shared_ptr<AbsPhaseSpace> > thePhpVecs=_tMatr->kMatrix()->phaseSpaceVec();
  complex<double> currentTijRel=(*_tMatr)(_prodProjectionIndex,_decProjectionIndex);
  complex<double> SijRel;
  SijRel=2.*PawianConstants::i *
    sqrt(thePhpVecs[_prodProjectionIndex]->factor(currentMass, OrbMom).real() *
	 thePhpVecs[_decProjectionIndex]->factor(currentMass, OrbMom).real()) * currentTijRel;
 
  double phiRelFit=KMatrixFunctions::deltaArgand(_tMatr, _prodProjectionIndex, currentMass, OrbMom);
  if(thePhpVecs[_decProjectionIndex]->factor(currentMass, OrbMom).real() > 1.e-5) phiRelFit=std::arg(SijRel)*PawianConstants::radToDeg;
  
  double phiRelData=theData->DoubleId.at(IdStringMapRegistry::instance()->stringId(EvtDataScatteringList::DATA_PIPISCAT_NAME));

  while( (phiRelData-phiRelFit) >  180.) phiRelFit+=360.;
  while( (phiRelFit-phiRelData) >  180.) phiRelFit-=360.;

  theData->DoubleId.at(IdStringMapRegistry::instance()->stringId(EvtDataScatteringList::FIT_PIPISCAT_NAME))=phiRelFit;
}

void TMatrixDynamics::evalArgandUnits(EvtData* theData, double currentMass, Spin OrbMom){
  vector<std::shared_ptr<AbsPhaseSpace> > thePhpVecs=_tMatr->kMatrix()->phaseSpaceVec();
  complex<double> currentTijRel=(*_tMatr)(_prodProjectionIndex,_decProjectionIndex);

  double sqrTij=0.;
  if( thePhpVecs[_prodProjectionIndex]->factor(currentMass, OrbMom).real() > 
      1.e-10 && thePhpVecs[_decProjectionIndex]->factor(currentMass, OrbMom).real() > 1.e-10) {
    sqrTij=(2.*_orbitalL+1.) * norm(sqrt(thePhpVecs[_prodProjectionIndex]->factor(currentMass, OrbMom).real()) *
				    currentTijRel*sqrt(thePhpVecs[_decProjectionIndex]->factor(currentMass, OrbMom).real()));
  }

  theData->DoubleId.at(IdStringMapRegistry::instance()->stringId(EvtDataScatteringList::FIT_PIPISCAT_NAME))=sqrTij;
}

void TMatrixDynamics::evalTreal(EvtData* theData, double currentMass, Spin OrbMom){
  complex<double> currentTijRel=(*_tMatr)(_prodProjectionIndex,_decProjectionIndex);
  theData->DoubleId.at(IdStringMapRegistry::instance()->stringId(EvtDataScatteringList::FIT_PIPISCAT_NAME))=currentTijRel.real();
}

void TMatrixDynamics::evalTimag(EvtData* theData, double currentMass, Spin OrbMom){
  complex<double> currentTijRel=(*_tMatr)(_prodProjectionIndex,_decProjectionIndex);
  theData->DoubleId.at(IdStringMapRegistry::instance()->stringId(EvtDataScatteringList::FIT_PIPISCAT_NAME))=currentTijRel.imag();
}

unsigned int TMatrixDynamics::noOfRotations(double currentMass){
  unsigned int noOfLoops=0;
  std::map<unsigned int, double >::iterator it;
  for(it=_noRotationMap.begin(); it!=_noRotationMap.end(); ++it){
    if(it->second > currentMass) break;
    noOfLoops=it->first; 
  } 
  return noOfLoops;
}

void TMatrixDynamics::fillMasses(EvtData* theData){
  Vector4<double> mass4Vec(0.,0.,0.,0.);
  std::vector<Particle*>::iterator it;
  for (it=_fsParticles.begin(); it != _fsParticles.end(); ++it){
    std::string currentName=(*it)->name();
    mass4Vec+=theData->FourVecsId.at(IdStringMapRegistry::instance()->stringId(currentName));
  }
  
  double sqrMass=mass4Vec.Mass2();
  if (sqrMass>0.) theData->DoubleMassId[_dynId]=mass4Vec.Mass();
  else if( sqrMass > -1.e-6) theData->DoubleMassId[_dynId]=0.;
  else{
    Alert << "mass4Vec.Mass2() is " << mass4Vec.Mass2() << " and thus < -1e-6 !!!" 
          << "\nexit !!!" << endmsg;
    exit(0); 
  }

  std::vector<std::shared_ptr<AbsPhaseSpace> >::iterator phpIt;
  for(phpIt=_phpVecs.begin(); phpIt!=_phpVecs.end(); ++phpIt){
    (*phpIt)->cacheFactors(mass4Vec.Mass());
  }

}

void TMatrixDynamics::setProdProjectionIndex(int idx){
  if(idx>=0) _prodProjectionIndex=idx;
  if( _prodProjectionIndex > (int) _phpVecs.size()){
    ErrMsg << "projection index for the production = " << _prodProjectionIndex << " is too large!!!"
	   << "\nmust be less than number of channels: " << _phpVecs.size() << endmsg;
    exit(1);
      }
  InfoMsg << " set projection index for the production: " << _prodProjectionIndex << endmsg;  
}


void TMatrixDynamics::doTcheck(Spin OrbMom){
  //complex<double> theMass(1.8, 0.);
  double theMass=1.8;
  InfoMsg << "\n\ndoTcheck at m = " << theMass << endmsg;
  _tMatr->evalMatrix(theMass, OrbMom);
  InfoMsg << "\nT-matrix:\n" << (*_tMatr) << endmsg;

  InfoMsg << "\nK-matrix:\n" << (*_tMatr->kMatrix()) << endmsg;
  
  vector<std::shared_ptr<AbsPhaseSpace> > thePhpVecs=_tMatr->kMatrix()->phaseSpaceVec();
  InfoMsg << "\nPhp-vec:";
  for (unsigned int i=0; i<thePhpVecs.size(); ++i){
    InfoMsg << "\nchannel " << i << ": " << thePhpVecs[i]->factor(theMass, OrbMom);
  }
  InfoMsg << endmsg;

  vector<std::shared_ptr<KPole> > thePoles= _tMatr->kMatrix()->kpoles();
  for(unsigned int i=0; i<thePoles.size(); ++i){
    std::shared_ptr<KPole> currentPole=thePoles.at(i);
    InfoMsg << "\npole : " << i << endmsg;
    std::vector< complex<double> > thebarrierFactors=currentPole->barrierFactors();
    for(unsigned int j=0; j< thebarrierFactors.size(); ++j){
      InfoMsg<< "\nbarrier " << j << ": " << thebarrierFactors.at(j);
    }
  }
  
  Matrix< complex<double> > SRel=Matrix< complex<double> > (_tMatr->NumRows(), _tMatr->NumCols());
  for(int i=0; i<_tMatr->NumRows(); ++i){
    for(int j=0; j<_tMatr->NumCols(); ++j){
       if (i==j) SRel(i,j)= complex<double>(1.,0.) + 2.*PawianConstants::i * sqrt(thePhpVecs[i]->factor(theMass, OrbMom).real()) * (*_tMatr)(i,j) * sqrt(thePhpVecs[j]->factor(theMass, OrbMom).real());
      else SRel(i,j)= sqrt(thePhpVecs[i]->factor(theMass, OrbMom).real()) * 2.*PawianConstants::i * (*_tMatr)(i,j) * sqrt(thePhpVecs[j]->factor(theMass, OrbMom).real());       
    }
  }

  InfoMsg << "\nSRel:\n" << SRel << endmsg;
  InfoMsg << "\nSRel*SRel.Adjoint():\n" << SRel*SRel.Adjoint() << endmsg;
}
