/**
 * \file ExtTrack.cpp
 * \author Elena Vannuccini
 */
#include <ExtTrack.h>
#include <iostream>
#include <math.h>
using namespace std;
//......................................
// F77 routines
//......................................
extern "C" {    
//     void dotrack_(int*, double*, double*, double*, double*, int*);
//     void dotrack2_(int*, double*, double*, double*, double*,double*, double*, double*,int*);
//     void dotrack3_(int*, double*, double*, double*, double*,double*, double*, double*,double*,int*);
//     void mini2_(int*,int*,int*); 
//     void guess_();
//     void gufld_(float*, float*);
//     float risxeta2_(float *);
//     float risxeta3_(float *);
//     float risxeta4_(float *);
//     float risyeta2_(float *);
}

//--------------------------------------
//
//
//--------------------------------------
ExtTrack::ExtTrack(){
    chi2  = 0;
    nstep = 0;
    for(int it1=0;it1<5;it1++){
	al[it1] = 0;
	for(int it2=0;it2<5;it2++)coval[it1][it2] = 0;
    };
    nplanes=0;

    xgood   = 0;
    ygood   = 0;
    xm      = 0;
    ym      = 0;
    zm      = 0;
    resx    = 0;
    resy    = 0;
    xv      = 0;
    yv      = 0;
    zv      = 0;
    axv     = 0;
    ayv     = 0;
    dedx_x  = 0;
    dedx_y  = 0;

    TrkParams::SetMiniDefault();

};
//--------------------------------------
//
//
//--------------------------------------
ExtTrack::ExtTrack(Int_t dim){
    chi2  = 0;
    nstep = 0;
    for(int it1=0;it1<5;it1++){
	al[it1] = 0;
	for(int it2=0;it2<5;it2++)coval[it1][it2] = 0;
    };
    
    SetDimension(dim);

    TrkParams::SetMiniDefault();
};
//--------------------------------------
//
//
//--------------------------------------
ExtTrack::ExtTrack(const ExtTrack& t){
    chi2  = t.chi2;
    nstep = t.nstep;
    for(int it1=0;it1<5;it1++){
	al[it1] = t.al[it1];
	for(int it2=0;it2<5;it2++)coval[it1][it2] = t.coval[it1][it2];
    };
    SetDimension(t.nplanes);
    for(int ip=0;ip<nplanes;ip++){
	xgood[ip]  = t.xgood[ip];
	ygood[ip]  = t.ygood[ip];
	xm[ip]     = t.xm[ip];
	ym[ip]     = t.ym[ip];
	zm[ip]     = t.zm[ip];
	resx[ip]   = t.resx[ip];
	resy[ip]   = t.resy[ip];
	xv[ip]     = t.xv[ip];
	yv[ip]     = t.yv[ip];
	zv[ip]     = t.zv[ip];
	axv[ip]    = t.axv[ip];
	ayv[ip]    = t.ayv[ip];
	dedx_x[ip] = t.dedx_x[ip];
	dedx_y[ip] = t.dedx_y[ip];
    };

    TrkParams::SetMiniDefault();
};
//--------------------------------------
//
//
//--------------------------------------
void ExtTrack::SetDimension(Int_t dim){

    nplanes = dim;

    xgood   = new Int_t[nplanes];
    ygood   = new Int_t[nplanes];
    xm      = new Float_t[nplanes];
    ym      = new Float_t[nplanes];
    zm      = new Float_t[nplanes];
    resx    = new Float_t[nplanes];
    resy    = new Float_t[nplanes];
    xv      = new Float_t[nplanes];
    yv      = new Float_t[nplanes];
    zv      = new Float_t[nplanes];
    axv     = new Float_t[nplanes];
    ayv     = new Float_t[nplanes];
    dedx_x  = new Float_t[nplanes];
    dedx_y  = new Float_t[nplanes];
    
    SetMeasure(0,0,0);
    SetResolution(0,0);
    SetGood(0,0);
    FitReset();
}
//--------------------------------------
//
//
//--------------------------------------
void ExtTrack::Clear(Option_t* option){

    FitReset();
    if(nplanes){
	delete []	xgood;
	delete []	ygood;
	delete []	xm;
	delete []	ym;
	delete []	zm;
	delete []	resx;
	delete []	resy;
	delete []	xv;
	delete []	yv;
	delete []	zv;
	delete []	axv;
	delete []	ayv;
	delete []	dedx_x;
	delete []	dedx_y;
    }

    nplanes = 0;
    xgood   = 0;
    ygood   = 0;
    xm      = 0;
    ym      = 0;
    zm      = 0;
    resx    = 0;
    resy    = 0;
    xv      = 0;
    yv      = 0;
    zv      = 0;
    axv     = 0;
    ayv     = 0;
    dedx_x  = 0;
    dedx_y  = 0;

};
//--------------------------------------
//
//
//--------------------------------------
void ExtTrack::Delete(){
    Clear();
};

/**
 * Set the  position measurements 
 */
void ExtTrack::SetMeasure(Double_t *xmeas, Double_t *ymeas, Double_t *zmeas){
    for(Int_t i=0; i<nplanes; i++) xm[i]= (xmeas? *xmeas++ : 0.);
    for(Int_t i=0; i<nplanes; i++) ym[i]= (ymeas? *ymeas++ : 0.);
    for(Int_t i=0; i<nplanes; i++) zm[i]= (ymeas? *zmeas++ : 0.);
}
/**
 * Set the TrkTrack position resolution 
 */
void ExtTrack::SetResolution(Double_t *rx, Double_t *ry){
    for(Int_t i=0; i<nplanes; i++) resx[i]= (rx? *rx++ : 0.);
    for(Int_t i=0; i<nplanes; i++) resy[i]= (ry? *ry++ : 0.);
}
/**
 * Set the TrkTrack good measurement 
 */
void ExtTrack::SetGood(Int_t *xg, Int_t *yg){

    for(Int_t i=0; i<nplanes; i++) xgood[i]=(xg? *xg++ : 0 );
    for(Int_t i=0; i<nplanes; i++) ygood[i]=(yg? *yg++ : 0 );
}

//--------------------------------------
//
//
//--------------------------------------
void ExtTrack::Dump(){
    cout << endl << "========== Track " ;
    cout << endl << "al       : "; for(int i=0; i<5; i++)cout << al[i] << " ";
    cout << endl << "chi^2    : "<< chi2;
    cout << endl << "n.step   : "<< nstep;
    cout << endl << "xgood    : "; for(int i=0; i<nplanes; i++)cout << XGood(i) ;
    cout << endl << "ygood    : "; for(int i=0; i<nplanes; i++)cout << YGood(i) ;
    cout << endl << "xm       : "; for(int i=0; i<nplanes; i++)cout << xm[i] << " ";
    cout << endl << "ym       : "; for(int i=0; i<nplanes; i++)cout << ym[i] << " ";
    cout << endl << "zm       : "; for(int i=0; i<nplanes; i++)cout << zm[i] << " ";
    cout << endl << "xv       : "; for(int i=0; i<nplanes; i++)cout << xv[i] << " ";
    cout << endl << "yv       : "; for(int i=0; i<nplanes; i++)cout << yv[i] << " ";
    cout << endl << "zv       : "; for(int i=0; i<nplanes; i++)cout << zv[i] << " ";
    cout << endl << "resx     : "; for(int i=0; i<nplanes; i++)cout << resx[i] << " ";
    cout << endl << "resy     : "; for(int i=0; i<nplanes; i++)cout << resy[i] << " ";
    cout << endl << "coval    : "; for(int i=0; i<5; i++)cout << coval[0][i]<<" ";
    cout << endl << "           "; for(int i=0; i<5; i++)cout << coval[1][i]<<" ";
    cout << endl << "           "; for(int i=0; i<5; i++)cout << coval[2][i]<<" ";
    cout << endl << "           "; for(int i=0; i<5; i++)cout << coval[3][i]<<" ";
    cout << endl << "           "; for(int i=0; i<5; i++)cout << coval[4][i]<<" ";
    cout << endl << "dedx_x   : "; for(int i=0; i<nplanes; i++)cout << dedx_x[i] << " ";
    cout << endl << "dedx_y   : "; for(int i=0; i<nplanes; i++)cout << dedx_y[i] << " ";

    cout << endl;
}
/**
 * Reset the fit parameters 
 */
void ExtTrack::FitReset(){
    for(int i=0; i<5; i++) al[i]=-9999.;
    chi2=0.;
    nstep=0;
    for(int i=0; i<nplanes; i++) xv[i]=0.;
    for(int i=0; i<nplanes; i++) yv[i]=0.;
    for(int i=0; i<nplanes; i++) zv[i]=0.;
    for(int i=0; i<nplanes; i++) axv[i]=0.;
    for(int i=0; i<nplanes; i++) ayv[i]=0.;
    for(int i=0; i<5; i++) {
	for(int j=0; j<5; j++) coval[i][j]=0.;
    }
}
/**
 * Method to fill minimization-routine common
 * NB (zini=23.5 is implicitelly set)
 */
void ExtTrack::FillMiniStruct(cMini2track& track){

    for(int i=0; i<nplanes; i++){ 

	track.xgood[i]=XGood(i);
	track.ygood[i]=YGood(i);
	
	track.xm[i]=xm[i];
	track.ym[i]=ym[i];
	track.zm[i]=zm[i];
	
	float segment = 100.; //temporaneo
	track.xm_a[i]=xm[i];
	track.xm_b[i]=xm[i];
	track.ym_a[i]=ym[i];
	track.ym_b[i]=ym[i];
	if(       XGood(i) && !YGood(i) ){
	    track.ym_a[i] = track.ym_a[i]+segment;
	    track.ym_b[i] = track.ym_b[i]-segment;
	}else if( !XGood(i) && YGood(i)){
	    track.xm_a[i] = track.xm_a[i]+segment;
	    track.xm_b[i] = track.xm_b[i]-segment;
	}

	
       	track.resx[i]=resx[i];
	track.resy[i]=resy[i];
    }

    for(int i=0; i<5; i++) track.al[i]=al[i]; 

    track.zini = 55.;

//    track.zini = 23.5; 
// ZINI = 23.5 !!! it should be the same parameter in all codes
    
}
/**
 * Method to set values from  minimization-routine common
 */
void ExtTrack::SetFromMiniStruct(cMini2track *track){

    for(int i=0; i<5; i++) {
	al[i]=track->al[i];
	for(int j=0; j<5; j++) coval[i][j]=track->cov[i][j];
    }
    chi2  = track->chi2;
    nstep = track->nstep;
    for(int i=0; i<6; i++){ 
	xv[i]  = track->xv[i];
	yv[i]  = track->yv[i];
	zv[i]  = track->zv[i];
	xm[i]  = track->xm[i];
	ym[i]  = track->ym[i];
	zm[i]  = track->zm[i];
	axv[i] = track->axv[i];
	ayv[i] = track->ayv[i];	
    }
    
}

ClassImp(ExtTrack);