/*
 * NicaUnigenSource.cxx
 *
 *  Created on: 27 mar 2017
 *      Author: Daniel Wielanek
 *		E-mail: daniel.wielanek@gmail.com
 *		Warsaw University of Technology, Faculty of Physics
 */
#include "NicaUnigenSource.h"
#include "FairRootManager.h"
#include "FairLogger.h"
#include <TBranch.h>
#include <fstream>
#include "UEvent.h"
#include "UParticle.h"

NicaUnigenSource::NicaUnigenSource():fUnigenChain(NULL),fFileName("data.root"),fEvent(NULL){
}

NicaUnigenSource::NicaUnigenSource(TString inFile): fUnigenChain(NULL),fFileName(inFile),fEvent(NULL){
}

NicaUnigenSource::~NicaUnigenSource() {
	if(fUnigenChain) delete fUnigenChain;
}

Bool_t NicaUnigenSource::Init() {
	FairRootManager *mngr = FairRootManager::Instance();
	fUnigenChain= new TChain("events");
	if(fFileName.EndsWith(".root")){
		LOG(debug)<<"NicaUnigenSource: opening single file"<<fFileName;
		fUnigenChain->Add(fFileName);
	}else{// this is long list
		std::ifstream list;
		list.open(fFileName);
		do{
			TString temp;
			list>>temp;
			if(temp.Length()>1){
				fUnigenChain->Add(temp);
			}else{
				break;
			}
			LOG(debug)<<"Adding file "<<temp<<" to chain";
		}while(!list.eof());
		list.close();
	}

	fEvent= new UEvent();
	if(fUnigenChain->GetBranch("event")){
		fUnigenChain->Print();
		fUnigenChain->SetBranchStatus("event",1);
		fUnigenChain->SetBranchAddress("event",&fEvent);
	}else{
		std::cout<<"Event read II"<<std::endl;
		fUnigenChain->SetBranchStatus("UEvent.",1);
		fUnigenChain->SetBranchAddress("UEvent.",&fEvent);
	}
	mngr->SetInChain(fUnigenChain,-1);\
	mngr->Register("UEvent.","UEvent",(TNamed*)fEvent,kFALSE);
	  return kTRUE;
}

Int_t NicaUnigenSource::ReadEvent(UInt_t unsignedInt) {
	//std::cout<<"READING EVENT " <<unsignedInt<<std::endl;
	 fUnigenChain->GetEntry(unsignedInt);
	// std::cout<<"xxx"<<std::endl;
	 return 0;
}

void NicaUnigenSource::Close() {
}

void NicaUnigenSource::Boost(Double_t vx, Double_t vy, Double_t vz) {
	for(int i=0;i<fEvent->GetNpa();i++){
		UParticle *p = fEvent->GetParticle(i);
		TLorentzVector mom = p->GetMomentum();
		TLorentzVector  pos = p->GetPosition();
		mom.Boost(vx,vy,vz);
		pos.Boost(vx,vy,vz);
		p->SetMomentum(mom);
		p->SetPosition(pos);
	}
}

Int_t NicaUnigenSource::CheckMaxEventNo(Int_t /*int1*/) {
	return fUnigenChain->GetEntries();
}