#include "CbmFieldCreator.h"
#include "PndConstField.h"
#include "CbmFieldMap.h"
#include "PndSolenoidMap.h"
#include "PndDipoleMap.h"
#include "PndTransMap.h"
#include "CbmRunAna.h"
#include "CbmRuntimeDb.h"
#include "PndMultiField.h"
#include "PndMultiFieldPar.h"
#include "PndConstPar.h"
#include "PndDipoleMap.h"
#include "PndSolenoidMap.h"
#include "PndDipolePar.h"
#include "PndSolenoidPar.h"
#include "PndTransPar.h"
#include "PndTransMap.h"
#include "PndMapPar.h"
#include "TObjArray.h"


static CbmFieldCreator gCbmFieldCreator;


CbmFieldCreator::CbmFieldCreator()
	:CbmFieldFactory()
{
	fCreator=this;
  fSPar=0;
  fDPar=0;
  fTPar=0;
  fCPar=0;
  fMPar=0;
  fFieldPar=0;

}
CbmFieldCreator::~CbmFieldCreator()
{
}
void CbmFieldCreator::SetParm()
{
 
  cout << "CbmFieldCreator::SetParm()" <<endl;
  CbmRunAna *Run = CbmRunAna::Instance();
  CbmRuntimeDb *RunDB = Run->GetRuntimeDb();
  
  fFieldPar = (CbmFieldPar*) RunDB->getContainer("CbmFieldPar");
  fSPar = (PndSolenoidPar  *)RunDB->getContainer("PndSolenoidPar");
  fDPar = (PndDipolePar    *)RunDB->getContainer("PndDipolePar");
  fTPar = (PndTransPar     *)RunDB->getContainer("PndTransPar");
  fCPar = (PndConstPar     *)RunDB->getContainer("PndConstPar");
  fMPar = (PndMultiFieldPar*)RunDB->getContainer("PndMultiFieldPar");
 
}

CbmField* CbmFieldCreator::createCbmField()
{       
        cout << "CbmFieldCreator::createCbmField()" <<endl;
        CbmField *fMagneticField=0;
	PndMultiField *MField=0;
	Int_t Type=-1;
        Bool_t multi=kFALSE;
	MField = new PndMultiField(fMPar);
	Type= fMPar->GetType();
	if (Type==-1) {delete MField; MField=0;}
        if(MField) {
		multi=kTRUE;
		TObjArray *fParArray=fMPar->GetParArray();
		TIterator *Iter=fParArray->MakeIterator();
   		Iter->Reset();
   		PndMapPar* fPar = NULL;
   		while( (fPar = (PndMapPar*)Iter->Next() ) ) {
			fPar->Print();
			if (fPar->GetType()==0){
				CbmField *fField1 = new PndConstField((PndConstPar*)fPar);
				MField->AddField(fField1);
			}
			if (fPar->GetType()==1){
				CbmField *fField2 = new CbmFieldMap((CbmFieldPar*)fPar);
				MField->AddField(fField2);
			}
			if (fPar->GetType()==2){
				CbmField *fField3 = new PndSolenoidMap((PndSolenoidPar*)fPar);
				MField->AddField(fField3);
			}
			if (fPar->GetType()==3){
				CbmField *fField4 = new PndDipoleMap((PndDipolePar*)fPar);
				MField->AddField(fField4);
			}
			if (fPar->GetType()==4){
				CbmField *fField5 = new PndTransMap((PndTransPar*)fPar);
				MField->AddField(fField5);
			}
		}
	}

	CbmField *fField1 = new PndConstField(fCPar);
	Type= fCPar->GetType();
	if (Type==-1) {delete fField1; fField1=0;}
	if(fField1)fMagneticField=fField1;	

	CbmField *fField2 = new CbmFieldMap(fFieldPar);
	Type= fFieldPar->GetType();
	if (Type==-1){delete fField2; fField2=0;}
	if(fField2)fMagneticField=fField2;	

	CbmField *fField3 = new PndSolenoidMap(fSPar);
	Type= fSPar->GetType();
	if (Type==-1) {delete fField3; fField3=0;}
	if(fField3)fMagneticField=fField3;
	
	CbmField *fField4 = new PndDipoleMap(fDPar);
	Type= fDPar->GetType();
	if (Type==-1) {delete fField4; fField4=0;}
	if(fField4)fMagneticField=fField4;	

	CbmField *fField5 = new PndTransMap(fTPar);
	Type= fTPar->GetType();
	if (Type==-1){delete fField5; fField5=0;}
	if(fField5)fMagneticField=fField5;	

        
	if (multi) {
		MField->Init();
		return MField;
	}else{
                if(fMagneticField){
			fMagneticField->Init();	 
		}
		return fMagneticField;
	}
}


ClassImp(CbmFieldCreator)