/*
 * Chain.cpp
 *
 *  Created on: Jan 16, 2012
 *      Author: Anastasia Karavdina
 */

#include "Chain.h"

Chain::Chain(list<TwoParticleState> ls, Particle ip, list<Particle> lf) {
	_lstates = ls;
	_ip = ip;
	_lfp = lf;
	_Jc = _ip.Spin();
	list<TwoParticleState>::iterator it = _lstates.begin();
	vector<double> lya1 = (*it).Lya1();
	vector<double> lya2 = (*it).Lya2();
	for(int i=0;i<lya1.size();i++)
		for(int k=0;k<lya2.size();k++)
			_lyafp.push_back(lya1[i]+lya2[k]);

}
//Chain::Chain() {}
Chain::Chain(const Chain& t) {
	_lstates = t._lstates;
	_ip = t._ip;
	_lfp = t._lfp;
	_Jc = _ip.Spin();
	_lyafp = t._lyafp;
}
Chain::~Chain() {
}

void Chain::Print(){
	cout<<"Initial Particle: "<<_ip.Code()<<endl;
	cout<<"Final particles: ";
	for (list<Particle>::iterator it = _lfp.begin(); it != _lfp.end(); it++){
		Particle cur = (*it);
		cout<<cur.Code()<<", ";
	}
	cout<<" "<<endl;

}
//Add previous state to last one
//-> change initial particle
//-> change list of final particles
void Chain::AddState(TwoParticleState add2st){
	//_lstates.push_front(add2st);
	_lstates.push_back(add2st);
	Particle newinp = add2st.GetIP();

	pair<Particle,Particle> newfpp = add2st.GetFP();
	int codepreinit = _ip.Code();
	int codenewfin1 = newfpp.first.Code();
	int codenewfin2 = newfpp.second.Code();

	if(codepreinit==codenewfin1){
		_lfp.push_back(newfpp.second);//new and previous state connect by particle #1 -> add particle#2 in list of f.p
		cout<<"Add Particle with code="<<codenewfin2<<" to list of f.p"<<endl;
	}
	else{
		if(codepreinit==codenewfin2){
			_lfp.push_back(newfpp.first);//new and previous state connect by particle #2 -> add particle#1 in list of f.p
			cout<<"Add Particle with code="<<codenewfin1<<" to list of f.p"<<endl;
		}
		else
			cout<<"Connection for new 2Particle state is not found!"<<endl;
	}

	_ip = newinp;
}

//set up spin and helicity vector
void Chain::Update(){
	_Jc = _ip.Spin();
	//int _fJc = _ip.Spin();
	for (list<TwoParticleState>::iterator it1 = _lstates.begin(); it1 != _lstates.end(); it1++){
		TwoParticleState curst=(*it1);
		if(curst.J()>_Jc) _Jc = curst.J();
	}
	list<TwoParticleState>::iterator it = _lstates.begin();
//	TwoParticleState* lastchain = &(*it);
	vector<double> lya1 = (*it).Lya1();
	vector<double> lya2 = (*it).Lya2();
	_lyafp.erase(_lyafp.begin(),_lyafp.end());
	for(int i=0;i<lya1.size();i++)
		for(int k=0;k<lya2.size();k++)
			_lyafp.push_back(lya1[i]+lya2[k]);
}