TrackerLevel2/src/TrkLevel2.cpp

/**
 * \file TrkLevel2.cpp
 * \author Elena Vannuccini
 */
#include <TrkLevel2.h>
#include <iostream>
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*);
    int  readb_(const char*); 
}
//--------------------------------------
//
//
//--------------------------------------
TrkTrack::TrkTrack(){
    seqno = -1; 
    image = -1; 
    chi2  = 0;
    for(int it1=0;it1<5;it1++){
        al[it1] = 0;
        for(int it2=0;it2<5;it2++)
            coval[it1][it2] = 0;
    };
    for(int ip=0;ip<6;ip++){
        xgood[ip]  = 0;
        ygood[ip]  = 0;
        xm[ip]     = 0;
        ym[ip]     = 0;
        zm[ip]     = 0;
        resx[ip]   = 0;
        resy[ip]   = 0;
        xv[ip]     = 0;
        yv[ip]     = 0;
        zv[ip]     = 0;
        axv[ip]    = 0;
        ayv[ip]    = 0;
        dedx_x[ip] = 0;
        dedx_y[ip] = 0;
    };    
};
//--------------------------------------
//
//
//--------------------------------------
TrkTrack::TrkTrack(const TrkTrack& t){
    seqno = t.seqno; 
    image = t.image; 
    chi2  = t.chi2;
    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];
    };
    for(int ip=0;ip<6;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];
    };    
};
//--------------------------------------
//
//
//--------------------------------------
/**
 * Evaluates the trajectory in the apparatus associated to the track. 
 * It integrates the equations of motion in the magnetic field. The magnetic field should be previously loaded ( by calling  TrkLevel2::LoadField() ), otherwise an error message is returned.  
 * @param t pointer to an object of the class Trajectory, 
 * which z coordinates should be previously initialized by calling the proper constructor ( Trajectory::Trajectory(int n, float* zin) ).
 * @return error flag.
 */
int TrkTrack::DoTrack(Trajectory* t){

    double *dxout = new double[t->npoint];
    double *dyout = new double[t->npoint];
    double *dzin = new double[t->npoint];
    double dal[5];

    int ifail = 0;

    for (int i=0; i<5; i++)         dal[i]  = (double)al[i];
    for (int i=0; i<t->npoint; i++) dzin[i] = (double)t->z[i];

    dotrack_(&(t->npoint),dzin,dxout,dyout,dal,&ifail);
    
    for (int i=0; i<t->npoint; i++){
        t->x[i] = (float)*dxout++;
        t->y[i] = (float)*dyout++;
    }

//    delete [] dxout;
//    delete [] dyout;
//    delete [] dzin;

    return ifail;
};
//--------------------------------------
//
//
//--------------------------------------
/**
 * Evaluates the trajectory in the apparatus associated to the track. 
 * It integrates the equations of motion in the magnetic field. The magnetic field should be previously loaded ( by calling  TrkLevel2::LoadField() ), otherwise an error message is returned.  
 * @param t pointer to an object of the class Trajectory, 
 * which z coordinates should be previously initialized by calling the proper constructor ( Trajectory::Trajectory(int n, float* zin) ).
 * @return error flag.
 */
int TrkTrack::DoTrack2(Trajectory* t){

    double *dxout   = new double[t->npoint];
    double *dyout   = new double[t->npoint];
    double *dthxout = new double[t->npoint];
    double *dthyout = new double[t->npoint];
    double *dtlout  = new double[t->npoint];
    double *dzin    = new double[t->npoint];
    double dal[5];

    int ifail = 0;

    for (int i=0; i<5; i++)         dal[i]  = (double)al[i];
    for (int i=0; i<t->npoint; i++) dzin[i] = (double)t->z[i];

    dotrack2_(&(t->npoint),dzin,dxout,dyout,dthxout,dthyout,dtlout,dal,&ifail);
    
    for (int i=0; i<t->npoint; i++){
        t->x[i]   = (float)*dxout++;
        t->y[i]   = (float)*dyout++;
        t->thx[i] = (float)*dthxout++;
        t->thy[i] = (float)*dthyout++;
        t->tl[i]  = (float)*dtlout++;
    }

//    delete [] dxout;
//    delete [] dyout;
//    delete [] dzin;

    return ifail;
};
//--------------------------------------
//
//
//--------------------------------------
//float TrkTrack::BdL(){
//};
//--------------------------------------
//
//
//--------------------------------------
Float_t TrkTrack::GetRigidity(){ 
        Float_t rig=0;
        if(chi2>0)rig=1./al[4];
        if(rig<0) rig=-rig;
        return rig;
};
//
Float_t TrkTrack::GetDeflection(){
        Float_t def=0;
        if(chi2>0)def=al[4];
        return def;
};
//
Float_t TrkTrack::GetDEDX(){
        Float_t dedx=0; 
        for(Int_t i=0; i<6; i++)dedx+=dedx_x[i]*xgood[i]+dedx_y[i]*ygood[i]; 
        dedx = dedx/(this->GetNX()+this->GetNY());
        return dedx;
};

//--------------------------------------
//
//
//--------------------------------------
void TrkTrack::Dump(){
    cout << endl << "========== Track " ;
    cout << endl << "al       : "; for(int i=0; i<5; i++)cout << al[i] << " ";
    cout << endl << "chi^2    : "<< chi2;
    cout << endl << "xgood    : "; for(int i=0; i<6; i++)cout << xgood[i] ;
    cout << endl << "ygood    : "; for(int i=0; i<6; i++)cout << ygood[i] ;
    cout << endl << "xm       : "; for(int i=0; i<6; i++)cout << xm[i] << " ";
    cout << endl << "ym       : "; for(int i=0; i<6; i++)cout << ym[i] << " ";
    cout << endl << "zm       : "; for(int i=0; i<6; i++)cout << zm[i] << " ";
    cout << endl << "dedx_x   : "; for(int i=0; i<6; i++)cout << dedx_x[i] << " ";
    cout << endl << "dedx_y   : "; for(int i=0; i<6; i++)cout << dedx_y[i] << " ";
}
//--------------------------------------
//
//
//--------------------------------------
TrkSinglet::TrkSinglet(){
    plane    = 0;
    coord[0] = 0; 
    coord[1] = 0;
    sgnl     = 0;
};
//--------------------------------------
//
//
//--------------------------------------
TrkSinglet::TrkSinglet(const TrkSinglet& s){
    plane    = s.plane;
    coord[0] = s.coord[0]; 
    coord[1] = s.coord[1];
    sgnl     = s.sgnl;
};
//--------------------------------------
//
//
//--------------------------------------
void TrkSinglet::Dump(){
    int i=0;
    cout << endl << "========== Singlet " ;
    cout << endl << "plane    : " << plane;
    cout << endl << "coord[2] : "; while( i<2 && cout << coord[i] << " ") i++;
    cout << endl << "sgnl     : " << sgnl;
}
//--------------------------------------
//
//
//--------------------------------------
TrkLevel2::TrkLevel2(){
    good2    = -1;
    for(Int_t i=0; i<12 ; i++){
        crc[i] = -1;
    };
    Track    = new TClonesArray("TrkTrack");
    SingletX = new TClonesArray("TrkSinglet");
    SingletY = new TClonesArray("TrkSinglet");
//    Track    = 0;
//    Singlet = 0;
//    SingletY = 0;
}
//--------------------------------------
//
//
//--------------------------------------
void TrkLevel2::Dump(){
    TClonesArray &t  = *Track;
    TClonesArray &sx = *SingletX;
    TClonesArray &sy = *SingletY;

    cout << endl << endl << "=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-";
    cout << endl << "good2    : " << good2;
    cout << endl << "crc      : "; for(int i=0; i<12; i++) cout << crc[i];
    cout << endl << "ntrk()   : " << this->ntrk() ;
    cout << endl << "nclsx()  : " << this->nclsx();
    cout << endl << "nclsy()  : " << this->nclsy();
    for(int i=0; i<this->ntrk(); i++)     ((TrkTrack *)t[i])->Dump();
    for(int i=0; i<this->nclsx(); i++) ((TrkSinglet *)sx[i])->Dump();
    for(int i=0; i<this->nclsy(); i++) ((TrkSinglet *)sy[i])->Dump();
}
//--------------------------------------
//
//
//--------------------------------------
/**
 * Fills a TrkLevel2 object with values from a struct cTrkLevel2 (to get data from F77 common).
 */
void TrkLevel2::FillCommonVar(cTrkLevel2 *l2){
    //
//    Track    = new TClonesArray("TrkTrack");
//    SingletX = new TClonesArray("TrkSinglet");
//    SingletY = new TClonesArray("TrkSinglet");
//  temporary objects:
    TrkSinglet* t_singlet = new TrkSinglet();
    TrkTrack*   t_track   = new TrkTrack();
//  general variables
    good2 = l2->good2;
    for(Int_t i=0; i<12 ; i++){
        crc[i] = l2->crc[i];
    };
//  *** TRACKS ***
    TClonesArray &t = *Track;
    for(int i=0; i<l2->ntrk; i++){
        t_track->seqno = i;
        t_track->image = l2->image[i]-1;
//      cout << "track "<<i<<t_track->seqno << t_track->image<<endl;
        t_track->chi2  = l2->chi2_nt[i];
        for(int it1=0;it1<5;it1++){
            t_track->al[it1] = l2->al_nt[i][it1];
            for(int it2=0;it2<5;it2++)
                t_track->coval[it1][it2] = l2->coval[i][it2][it1];
        };
        for(int ip=0;ip<6;ip++){
            t_track->xgood[ip]  = l2->xgood_nt[i][ip];
            t_track->ygood[ip]  = l2->ygood_nt[i][ip];
            t_track->xm[ip]     = l2->xm_nt[i][ip];
            t_track->ym[ip]     = l2->ym_nt[i][ip];
            t_track->zm[ip]     = l2->zm_nt[i][ip];
            t_track->resx[ip]   = l2->resx_nt[i][ip];
            t_track->resy[ip]   = l2->resy_nt[i][ip];
            t_track->xv[ip]     = l2->xv_nt[i][ip];
            t_track->yv[ip]     = l2->yv_nt[i][ip];
            t_track->zv[ip]     = l2->zv_nt[i][ip];
            t_track->axv[ip]    = l2->axv_nt[i][ip];
            t_track->ayv[ip]    = l2->ayv_nt[i][ip];
            t_track->dedx_x[ip] = l2->dedx_x[i][ip];
            t_track->dedx_y[ip] = l2->dedx_y[i][ip];
        };
        new(t[i]) TrkTrack(*t_track);
        t_track->Clear();
    };
//  *** SINGLETS ***
    TClonesArray &sx = *SingletX;
    for(int i=0; i<l2->nclsx; i++){
        t_singlet->plane    = l2->planex[i];
        t_singlet->coord[0] = l2->xs[i][0];
        t_singlet->coord[1] = l2->xs[i][1];
        t_singlet->sgnl     = l2->signlxs[i];
        new(sx[i]) TrkSinglet(*t_singlet);
        t_singlet->Clear();
    }
    TClonesArray &sy = *SingletY;
    for(int i=0; i<l2->nclsy; i++){
        t_singlet->plane    = l2->planey[i];
        t_singlet->coord[0] = l2->ys[i][0];
        t_singlet->coord[1] = l2->ys[i][1];
        t_singlet->sgnl     = l2->signlys[i];
        new(sy[i]) TrkSinglet(*t_singlet);
        t_singlet->Clear();
        };
        
        delete t_track;
        delete t_singlet;
}
//--------------------------------------
//
//
//--------------------------------------
void TrkLevel2::Clear(){
    good2    = -1;
    for(Int_t i=0; i<12 ; i++){
        crc[i] = -1;
    };
/*    Track->RemoveAll();
    SingletX->RemoveAll();
    SingletY->RemoveAll();*/
        // modify to avoid memory leakage
        Track->Clear();
        SingletX->Clear();
        SingletY->Clear();
}
//--------------------------------------
//
//
//--------------------------------------
/**
 * Sort physical tracks and stores them in a TObjectArray, ordering by increasing chi**2 value (in case of track image, it selects the one with lower chi**2). The total number of physical tracks is given by GetNTracks() and the it-th physical track can be retrieved by means of the method GetTrack(int it).
 * This method is overridden by PamLevel2::GetTracks(), where calorimeter and TOF information is used.
 */
TClonesArray *TrkLevel2::GetTracks(){
    TClonesArray *sorted = GetTracks_NFitSorted();
    return sorted;
};

TClonesArray *TrkLevel2::GetTracks_Chi2Sorted(){

    TClonesArray *sorted = new TClonesArray("TrkTrack");
    TClonesArray &t = *Track;
    TClonesArray &ts = *sorted;
    int N=this->ntrk();
    vector<int> m(N); for(int i=0; i<N; i++)m[i]=1;

    int indo=0;
    int indi=0;
    while(N != 0){
        float chi2ref=1000000;
        for(int i=0; i<this->ntrk(); i++){
            if(((TrkTrack *)t[i])->chi2 < chi2ref && m[i]==1){
                chi2ref = ((TrkTrack *)t[i])->chi2;
                indi = i;
            }
        }
        if( ((TrkTrack *)t[indi])->image != -1 ){
            m[((TrkTrack *)t[indi])->image] = 0;
            N--;
        }
        new(ts[indo]) TrkTrack(*(TrkTrack*)t[indi]);
        m[indi] = 0;
        N--;    
        indo++; 
    }
    return sorted;
}
TClonesArray *TrkLevel2::GetTracks_NFitSorted(){

    TClonesArray *sorted = new TClonesArray("TrkTrack");
    TClonesArray &t = *Track;
    TClonesArray &ts = *sorted;
    int N=this->ntrk();
    vector<int> m(N); for(int i=0; i<N; i++)m[i]=1;

    int indo=0;
    int indi=0;
    while(N != 0){
        int nfit =0;
        float chi2ref=1000000;
        // first loop to search maximum num. of fit points
        for(int i=0; i<this->ntrk(); i++){
            if( ((TrkTrack *)t[i])->GetNtot() >= nfit && m[i]==1){
                nfit =    ((TrkTrack *)t[i])->GetNtot();
//              cout << "1** "<<i<< " " << nfit<<endl;
            }
        }
        //second loop to search minimum chi2 among selected
        for(int i=0; i<this->ntrk(); i++){
            if(    ((TrkTrack *)t[i])->chi2 < chi2ref 
                && ((TrkTrack *)t[i])->GetNtot()== nfit 
                && m[i]==1){
                chi2ref = ((TrkTrack *)t[i])->chi2;
                indi = i;
//              cout << "2** "<<i<< " " << nfit <<" "<<chi2ref<<endl;
            }
        }
        if( ((TrkTrack *)t[indi])->HasImage() ){
            m[((TrkTrack *)t[indi])->image] = 0;
            N--;

//          Int_t nfiti=((TrkTrack *)t[((TrkTrack *)t[indi])->image  ])->GetNtot();
//          Float_t chi2i=((TrkTrack *)t[((TrkTrack *)t[indi])->image  ])->chi2;
                
//          cout << "i** "<< ((TrkTrack *)t[indi])->image << " " << nfiti <<" "<<chi2i<<endl;
        }
        new(ts[indo]) TrkTrack(*(TrkTrack*)t[indi]);
        m[indi] = 0;
        N--;    
        indo++; 
    }
    return sorted;
}
//--------------------------------------
//
//
//--------------------------------------
/**
 * Retrieves the is-th stored track. 
 * @param it Track number, ranging from 0 to ntrk().
 * Fitted tracks ( images included ) are stored in a TObjectArray ( TrkLevel2::Track ) in the same order they are returned by the F77 fitting routine. 
 */
TrkTrack *TrkLevel2::GetStoredTrack(int is){

    if(is >= this->ntrk()){
        cout << "** TrkLevel2 ** Track "<< is << "doen not exits! " << endl; 
        cout << "                Stored tracks ntrk() = "<< this->ntrk() << endl;
        return 0;
    }
    TClonesArray &t = *(Track);
    TrkTrack *track = (TrkTrack*)t[is];
    return track;
}
//--------------------------------------
//
//
//--------------------------------------
/**
 * Retrieves the it-th "physical" track, sorted by the method GetNTracks(). 
 * @param it Track number, ranging from 0 to GetNTracks().
 */
TrkTrack *TrkLevel2::GetTrack(int it){
    
    if(it >= this->GetNTracks()){
        cout << "** TrkLevel2 ** Track "<< it << "does not exits! " << endl; 
        cout << "                Physical tracks GetNTracks() = "<< this->ntrk() << endl;
        return 0;
    }
    TrkTrack *track = (TrkTrack*)(*(this->GetTracks()))[it];
        GetTracks()->Delete();////TEMPORANEO
    return track;
}
Int_t TrkLevel2::GetNTracks(){
        Int_t ntot=0;
        ntot = GetTracks()->GetEntries();
        GetTracks()->Delete();////TEMPORANEO
        return ntot;
};
//--------------------------------------
//
//
//--------------------------------------
/**
 * Retrieves (if present) the image of the it-th "physical" track, sorted by the method GetNTracks(). 
 * @param it Track number, ranging from 0 to GetNTracks().
 */
TrkTrack *TrkLevel2::GetTrackImage(int it){

    if(it >= this->GetNTracks()){
        cout << "** TrkLevel2 ** Track "<< it << "does not exits! " << endl; 
        cout << "                Physical tracks GetNTracks() = "<< this->ntrk() << endl;
        return 0;
    }
    TrkTrack *track = (TrkTrack*)(*(this->GetTracks()))[it];
    if(!track->HasImage()){
        cout << "** TrkLevel2 ** Track "<< it << "does not have image! " << endl; 
        return 0;
    }
    TrkTrack *image = (TrkTrack*)(*Track)[track->image];
    return image;
    
}
//--------------------------------------
//
//
//--------------------------------------
/**
 * Loads the magnetic field. 
 * @param s Path of the magnetic-field files.
 */
void TrkLevel2::LoadField(TString s){
    readb_(s.Data());
};
//--------------------------------------
//
//
//--------------------------------------
/**
 * Trajectory default constructor. 
 * (By default is created with z-coordinates inside the tracking volume)
  */
Trajectory::Trajectory(){
    npoint = 10;
    x = new float[npoint];
    y = new float[npoint];
    z = new float[npoint];
    thx = new float[npoint];
    thy = new float[npoint];
    tl = new float[npoint];
    float dz = ((ZTRKUP)-(ZTRKDW))/(npoint-1);
    for(int i=0; i<npoint; i++){
        x[i] = 0;
        y[i] = 0;
        z[i] = (ZTRKUP) - i*dz;
        thx[i] = 0;
        thy[i] = 0;
        tl[i] = 0;
    }
}
//--------------------------------------
//
//
//--------------------------------------
/**
 * Trajectory constructor.
 * (By default is created with z-coordinates inside the tracking volume)
 * \param n Number of points
 */
Trajectory::Trajectory(int n){
    if(n<=0){
        cout << "NB! Trajectory must have at least 1 point >>> created with 10 points" << endl;
        n=10;
    }
    npoint = n;
    x = new float[npoint];
    y = new float[npoint];
    z = new float[npoint];
    thx = new float[npoint];
    thy = new float[npoint];
    tl = new float[npoint];
    float dz = ((ZTRKUP)-(ZTRKDW))/(npoint-1);
    for(int i=0; i<npoint; i++){
        x[i] = 0;
        y[i] = 0;
        z[i] = (ZTRKUP) - i*dz;
        thx[i] = 0;
        thy[i] = 0;
        tl[i] = 0;
    }
}
//--------------------------------------
//
//
//--------------------------------------
/**
 * Trajectory constructor.
 * \param n Number of points
 * \param pz Pointer to float array, defining z coordinates
 */
Trajectory::Trajectory(int n, float* zin){
    npoint = 10;
    if(n>0)npoint = n;
    x = new float[npoint];
    y = new float[npoint];
    z = new float[npoint];
    thx = new float[npoint];
    thy = new float[npoint];
    tl = new float[npoint];
    int i=0;
    do{
        x[i] = 0;
        y[i] = 0;
        z[i] = zin[i];
        thx[i] = 0;
        thy[i] = 0;
        tl[i] = 0;
        i++;            
    }while(zin[i-1] > zin[i] && i < npoint);
    npoint=i;
    if(npoint != n)cout << "NB! Trajectory created with "<<npoint<<" points"<<endl;
}
//--------------------------------------
//
//
//--------------------------------------
/**
 * Dump the trajectory coordinates.
 */
void Trajectory::Dump(){
    cout <<endl<< "Trajectory ========== "<<endl;
    for (int i=0; i<npoint; i++){
        cout << i <<" >> " << x[i] <<" "<< y[i] <<" "<< z[i] ;
        cout <<" -- " << thx[i] <<" "<< thy[i] ;
        cout <<" -- " << tl[i] << endl;
    };
}
//--------------------------------------
//
//
//--------------------------------------
/**
 * Get trajectory length between two points
 * @param ifirst first point (default 0)
 * @param ilast last point (default npoint)
 */
float Trajectory::GetLength(int ifirst, int ilast){
    if( ifirst<0    ) ifirst = 0;
    if( ilast>=npoint) ilast  = npoint-1;
    float l=0;
    for(int i=ifirst;i<=ilast;i++){
        l=l+tl[i]; 
    };
    if(z[ilast] > ZINI)l=l-tl[ilast];
    if(z[ifirst] < ZINI)   l=l-tl[ifirst];
    
    return l;

}
ClassImp(TrkLevel2);
ClassImp(TrkSinglet);
ClassImp(TrkTrack);
ClassImp(Trajectory);
1	mocchiut	1.1	/**
2			* \file TrkLevel2.cpp
3			* \author Elena Vannuccini
4			*/
5			#include <TrkLevel2.h>
6			#include <iostream>
7			using namespace std;
8			//......................................
9			// F77 routines
10			//......................................
11			extern "C" {
12			void dotrack_(int, double, double, double, double, int);
13	pam-fi	1.2	void dotrack2_(int, double, double, double, double,double, double, double,int*);
14	mocchiut	1.1	int readb_(const char*);
15			}
16			//--------------------------------------
17			//
18			//
19			//--------------------------------------
20			TrkTrack::TrkTrack(){
21	pam-fi	1.3	seqno = -1;
22			image = -1;
23	mocchiut	1.1	chi2 = 0;
24			for(int it1=0;it1<5;it1++){
25			al[it1] = 0;
26			for(int it2=0;it2<5;it2++)
27			coval[it1][it2] = 0;
28			};
29			for(int ip=0;ip<6;ip++){
30			xgood[ip] = 0;
31			ygood[ip] = 0;
32			xm[ip] = 0;
33			ym[ip] = 0;
34			zm[ip] = 0;
35			resx[ip] = 0;
36			resy[ip] = 0;
37			xv[ip] = 0;
38			yv[ip] = 0;
39			zv[ip] = 0;
40			axv[ip] = 0;
41			ayv[ip] = 0;
42			dedx_x[ip] = 0;
43			dedx_y[ip] = 0;
44			};
45			};
46			//--------------------------------------
47			//
48			//
49			//--------------------------------------
50			TrkTrack::TrkTrack(const TrkTrack& t){
51	pam-fi	1.3	seqno = t.seqno;
52	mocchiut	1.1	image = t.image;
53			chi2 = t.chi2;
54			for(int it1=0;it1<5;it1++){
55			al[it1] = t.al[it1];
56			for(int it2=0;it2<5;it2++)
57			coval[it1][it2] = t.coval[it1][it2];
58			};
59			for(int ip=0;ip<6;ip++){
60			xgood[ip] = t.xgood[ip];
61			ygood[ip] = t.ygood[ip];
62			xm[ip] = t.xm[ip];
63			ym[ip] = t.ym[ip];
64			zm[ip] = t.zm[ip];
65			resx[ip] = t.resx[ip];
66			resy[ip] = t.resy[ip];
67			xv[ip] = t.xv[ip];
68			yv[ip] = t.yv[ip];
69			zv[ip] = t.zv[ip];
70			axv[ip] = t.axv[ip];
71			ayv[ip] = t.ayv[ip];
72			dedx_x[ip] = t.dedx_x[ip];
73			dedx_y[ip] = t.dedx_y[ip];
74			};
75			};
76			//--------------------------------------
77			//
78			//
79			//--------------------------------------
80			/**
81			* Evaluates the trajectory in the apparatus associated to the track.
82			* It integrates the equations of motion in the magnetic field. The magnetic field should be previously loaded ( by calling TrkLevel2::LoadField() ), otherwise an error message is returned.
83			* @param t pointer to an object of the class Trajectory,
84			* which z coordinates should be previously initialized by calling the proper constructor ( Trajectory::Trajectory(int n, float* zin) ).
85			* @return error flag.
86			*/
87			int TrkTrack::DoTrack(Trajectory* t){
88
89			double *dxout = new double[t->npoint];
90			double *dyout = new double[t->npoint];
91			double *dzin = new double[t->npoint];
92			double dal[5];
93
94			int ifail = 0;
95
96			for (int i=0; i<5; i++) dal[i] = (double)al[i];
97			for (int i=0; i<t->npoint; i++) dzin[i] = (double)t->z[i];
98
99			dotrack_(&(t->npoint),dzin,dxout,dyout,dal,&ifail);
100
101			for (int i=0; i<t->npoint; i++){
102			t->x[i] = (float)*dxout++;
103			t->y[i] = (float)*dyout++;
104			}
105
106			// delete [] dxout;
107			// delete [] dyout;
108			// delete [] dzin;
109
110			return ifail;
111			};
112			//--------------------------------------
113			//
114			//
115			//--------------------------------------
116	pam-fi	1.2	/**
117			* Evaluates the trajectory in the apparatus associated to the track.
118			* It integrates the equations of motion in the magnetic field. The magnetic field should be previously loaded ( by calling TrkLevel2::LoadField() ), otherwise an error message is returned.
119			* @param t pointer to an object of the class Trajectory,
120			* which z coordinates should be previously initialized by calling the proper constructor ( Trajectory::Trajectory(int n, float* zin) ).
121			* @return error flag.
122			*/
123			int TrkTrack::DoTrack2(Trajectory* t){
124
125			double *dxout = new double[t->npoint];
126			double *dyout = new double[t->npoint];
127			double *dthxout = new double[t->npoint];
128			double *dthyout = new double[t->npoint];
129			double *dtlout = new double[t->npoint];
130			double *dzin = new double[t->npoint];
131			double dal[5];
132
133			int ifail = 0;
134
135			for (int i=0; i<5; i++) dal[i] = (double)al[i];
136			for (int i=0; i<t->npoint; i++) dzin[i] = (double)t->z[i];
137
138			dotrack2_(&(t->npoint),dzin,dxout,dyout,dthxout,dthyout,dtlout,dal,&ifail);
139
140			for (int i=0; i<t->npoint; i++){
141			t->x[i] = (float)*dxout++;
142			t->y[i] = (float)*dyout++;
143			t->thx[i] = (float)*dthxout++;
144			t->thy[i] = (float)*dthyout++;
145			t->tl[i] = (float)*dtlout++;
146			}
147
148			// delete [] dxout;
149			// delete [] dyout;
150			// delete [] dzin;
151
152			return ifail;
153			};
154			//--------------------------------------
155			//
156			//
157			//--------------------------------------
158	mocchiut	1.1	//float TrkTrack::BdL(){
159			//};
160			//--------------------------------------
161			//
162			//
163			//--------------------------------------
164			Float_t TrkTrack::GetRigidity(){
165			Float_t rig=0;
166			if(chi2>0)rig=1./al[4];
167			if(rig<0) rig=-rig;
168			return rig;
169			};
170			//
171			Float_t TrkTrack::GetDeflection(){
172			Float_t def=0;
173			if(chi2>0)def=al[4];
174			return def;
175			};
176			//
177			Float_t TrkTrack::GetDEDX(){
178			Float_t dedx=0;
179			for(Int_t i=0; i<6; i++)dedx+=dedx_x[i]xgood[i]+dedx_y[i]ygood[i];
180			dedx = dedx/(this->GetNX()+this->GetNY());
181			return dedx;
182			};
183
184			//--------------------------------------
185			//
186			//
187			//--------------------------------------
188			void TrkTrack::Dump(){
189			cout << endl << "========== Track " ;
190			cout << endl << "al : "; for(int i=0; i<5; i++)cout << al[i] << " ";
191			cout << endl << "chi^2 : "<< chi2;
192			cout << endl << "xgood : "; for(int i=0; i<6; i++)cout << xgood[i] ;
193			cout << endl << "ygood : "; for(int i=0; i<6; i++)cout << ygood[i] ;
194			cout << endl << "xm : "; for(int i=0; i<6; i++)cout << xm[i] << " ";
195			cout << endl << "ym : "; for(int i=0; i<6; i++)cout << ym[i] << " ";
196			cout << endl << "zm : "; for(int i=0; i<6; i++)cout << zm[i] << " ";
197			cout << endl << "dedx_x : "; for(int i=0; i<6; i++)cout << dedx_x[i] << " ";
198			cout << endl << "dedx_y : "; for(int i=0; i<6; i++)cout << dedx_y[i] << " ";
199			}
200			//--------------------------------------
201			//
202			//
203			//--------------------------------------
204			TrkSinglet::TrkSinglet(){
205			plane = 0;
206			coord[0] = 0;
207			coord[1] = 0;
208			sgnl = 0;
209			};
210			//--------------------------------------
211			//
212			//
213			//--------------------------------------
214			TrkSinglet::TrkSinglet(const TrkSinglet& s){
215			plane = s.plane;
216			coord[0] = s.coord[0];
217			coord[1] = s.coord[1];
218			sgnl = s.sgnl;
219			};
220			//--------------------------------------
221			//
222			//
223			//--------------------------------------
224			void TrkSinglet::Dump(){
225			int i=0;
226			cout << endl << "========== Singlet " ;
227			cout << endl << "plane : " << plane;
228			cout << endl << "coord[2] : "; while( i<2 && cout << coord[i] << " ") i++;
229			cout << endl << "sgnl : " << sgnl;
230			}
231			//--------------------------------------
232			//
233			//
234			//--------------------------------------
235			TrkLevel2::TrkLevel2(){
236			good2 = -1;
237			for(Int_t i=0; i<12 ; i++){
238			crc[i] = -1;
239			};
240			Track = new TClonesArray("TrkTrack");
241			SingletX = new TClonesArray("TrkSinglet");
242			SingletY = new TClonesArray("TrkSinglet");
243	pam-fi	1.3	// Track = 0;
244			// Singlet = 0;
245			// SingletY = 0;
246	mocchiut	1.1	}
247			//--------------------------------------
248			//
249			//
250			//--------------------------------------
251			void TrkLevel2::Dump(){
252			TClonesArray &t = *Track;
253			TClonesArray &sx = *SingletX;
254			TClonesArray &sy = *SingletY;
255
256			cout << endl << endl << "=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-";
257			cout << endl << "good2 : " << good2;
258			cout << endl << "crc : "; for(int i=0; i<12; i++) cout << crc[i];
259			cout << endl << "ntrk() : " << this->ntrk() ;
260			cout << endl << "nclsx() : " << this->nclsx();
261			cout << endl << "nclsy() : " << this->nclsy();
262			for(int i=0; i<this->ntrk(); i++) ((TrkTrack *)t[i])->Dump();
263			for(int i=0; i<this->nclsx(); i++) ((TrkSinglet *)sx[i])->Dump();
264			for(int i=0; i<this->nclsy(); i++) ((TrkSinglet *)sy[i])->Dump();
265			}
266			//--------------------------------------
267			//
268			//
269			//--------------------------------------
270			/**
271			* Fills a TrkLevel2 object with values from a struct cTrkLevel2 (to get data from F77 common).
272			*/
273			void TrkLevel2::FillCommonVar(cTrkLevel2 *l2){
274	pam-fi	1.3	//
275			// Track = new TClonesArray("TrkTrack");
276			// SingletX = new TClonesArray("TrkSinglet");
277			// SingletY = new TClonesArray("TrkSinglet");
278	mocchiut	1.1	// temporary objects:
279			TrkSinglet* t_singlet = new TrkSinglet();
280			TrkTrack* t_track = new TrkTrack();
281			// general variables
282			good2 = l2->good2;
283			for(Int_t i=0; i<12 ; i++){
284			crc[i] = l2->crc[i];
285			};
286			// * TRACKS *
287			TClonesArray &t = *Track;
288			for(int i=0; i<l2->ntrk; i++){
289	pam-fi	1.3	t_track->seqno = i;
290	mocchiut	1.1	t_track->image = l2->image[i]-1;
291	pam-fi	1.3	// cout << "track "<<i<<t_track->seqno << t_track->image<<endl;
292	mocchiut	1.1	t_track->chi2 = l2->chi2_nt[i];
293			for(int it1=0;it1<5;it1++){
294			t_track->al[it1] = l2->al_nt[i][it1];
295			for(int it2=0;it2<5;it2++)
296			t_track->coval[it1][it2] = l2->coval[i][it2][it1];
297			};
298			for(int ip=0;ip<6;ip++){
299			t_track->xgood[ip] = l2->xgood_nt[i][ip];
300			t_track->ygood[ip] = l2->ygood_nt[i][ip];
301			t_track->xm[ip] = l2->xm_nt[i][ip];
302			t_track->ym[ip] = l2->ym_nt[i][ip];
303			t_track->zm[ip] = l2->zm_nt[i][ip];
304			t_track->resx[ip] = l2->resx_nt[i][ip];
305			t_track->resy[ip] = l2->resy_nt[i][ip];
306			t_track->xv[ip] = l2->xv_nt[i][ip];
307			t_track->yv[ip] = l2->yv_nt[i][ip];
308			t_track->zv[ip] = l2->zv_nt[i][ip];
309			t_track->axv[ip] = l2->axv_nt[i][ip];
310			t_track->ayv[ip] = l2->ayv_nt[i][ip];
311			t_track->dedx_x[ip] = l2->dedx_x[i][ip];
312			t_track->dedx_y[ip] = l2->dedx_y[i][ip];
313			};
314			new(t[i]) TrkTrack(*t_track);
315			t_track->Clear();
316			};
317			// * SINGLETS *
318			TClonesArray &sx = *SingletX;
319			for(int i=0; i<l2->nclsx; i++){
320			t_singlet->plane = l2->planex[i];
321			t_singlet->coord[0] = l2->xs[i][0];
322			t_singlet->coord[1] = l2->xs[i][1];
323			t_singlet->sgnl = l2->signlxs[i];
324			new(sx[i]) TrkSinglet(*t_singlet);
325			t_singlet->Clear();
326			}
327			TClonesArray &sy = *SingletY;
328			for(int i=0; i<l2->nclsy; i++){
329			t_singlet->plane = l2->planey[i];
330			t_singlet->coord[0] = l2->ys[i][0];
331			t_singlet->coord[1] = l2->ys[i][1];
332			t_singlet->sgnl = l2->signlys[i];
333			new(sy[i]) TrkSinglet(*t_singlet);
334			t_singlet->Clear();
335			};
336	pam-fi	1.5
337			delete t_track;
338			delete t_singlet;
339	mocchiut	1.1	}
340			//--------------------------------------
341			//
342			//
343			//--------------------------------------
344			void TrkLevel2::Clear(){
345			good2 = -1;
346			for(Int_t i=0; i<12 ; i++){
347			crc[i] = -1;
348			};
349	pam-fi	1.5	/* Track->RemoveAll();
350	mocchiut	1.1	SingletX->RemoveAll();
351	pam-fi	1.5	SingletY->RemoveAll();*/
352			// modify to avoid memory leakage
353			Track->Clear();
354			SingletX->Clear();
355			SingletY->Clear();
356	mocchiut	1.1	}
357			//--------------------------------------
358			//
359			//
360			//--------------------------------------
361			/**
362			* Sort physical tracks and stores them in a TObjectArray, ordering by increasing chi2 value (in case of track image, it selects the one with lower chi2). The total number of physical tracks is given by GetNTracks() and the it-th physical track can be retrieved by means of the method GetTrack(int it).
363			* This method is overridden by PamLevel2::GetTracks(), where calorimeter and TOF information is used.
364			*/
365			TClonesArray *TrkLevel2::GetTracks(){
366	pam-fi	1.3	TClonesArray *sorted = GetTracks_NFitSorted();
367			return sorted;
368			};
369	mocchiut	1.4
370	pam-fi	1.3	TClonesArray *TrkLevel2::GetTracks_Chi2Sorted(){
371	mocchiut	1.1
372			TClonesArray *sorted = new TClonesArray("TrkTrack");
373			TClonesArray &t = *Track;
374			TClonesArray &ts = *sorted;
375			int N=this->ntrk();
376			vector<int> m(N); for(int i=0; i<N; i++)m[i]=1;
377
378			int indo=0;
379			int indi=0;
380			while(N != 0){
381			float chi2ref=1000000;
382			for(int i=0; i<this->ntrk(); i++){
383			if(((TrkTrack *)t[i])->chi2 < chi2ref && m[i]==1){
384			chi2ref = ((TrkTrack *)t[i])->chi2;
385			indi = i;
386			}
387			}
388			if( ((TrkTrack *)t[indi])->image != -1 ){
389			m[((TrkTrack *)t[indi])->image] = 0;
390			N--;
391			}
392			new(ts[indo]) TrkTrack((TrkTrack)t[indi]);
393			m[indi] = 0;
394			N--;
395			indo++;
396			}
397			return sorted;
398			}
399	pam-fi	1.3	TClonesArray *TrkLevel2::GetTracks_NFitSorted(){
400
401			TClonesArray *sorted = new TClonesArray("TrkTrack");
402			TClonesArray &t = *Track;
403			TClonesArray &ts = *sorted;
404			int N=this->ntrk();
405			vector<int> m(N); for(int i=0; i<N; i++)m[i]=1;
406
407			int indo=0;
408			int indi=0;
409			while(N != 0){
410			int nfit =0;
411			float chi2ref=1000000;
412			// first loop to search maximum num. of fit points
413			for(int i=0; i<this->ntrk(); i++){
414			if( ((TrkTrack *)t[i])->GetNtot() >= nfit && m[i]==1){
415			nfit = ((TrkTrack *)t[i])->GetNtot();
416			// cout << "1** "<<i<< " " << nfit<<endl;
417			}
418			}
419			//second loop to search minimum chi2 among selected
420			for(int i=0; i<this->ntrk(); i++){
421			if( ((TrkTrack *)t[i])->chi2 < chi2ref
422			&& ((TrkTrack *)t[i])->GetNtot()== nfit
423			&& m[i]==1){
424			chi2ref = ((TrkTrack *)t[i])->chi2;
425			indi = i;
426			// cout << "2** "<<i<< " " << nfit <<" "<<chi2ref<<endl;
427			}
428			}
429			if( ((TrkTrack *)t[indi])->HasImage() ){
430			m[((TrkTrack *)t[indi])->image] = 0;
431			N--;
432
433			// Int_t nfiti=((TrkTrack )t[((TrkTrack )t[indi])->image ])->GetNtot();
434			// Float_t chi2i=((TrkTrack )t[((TrkTrack )t[indi])->image ])->chi2;
435
436			// cout << "i** "<< ((TrkTrack *)t[indi])->image << " " << nfiti <<" "<<chi2i<<endl;
437			}
438			new(ts[indo]) TrkTrack((TrkTrack)t[indi]);
439			m[indi] = 0;
440			N--;
441			indo++;
442			}
443			return sorted;
444			}
445	mocchiut	1.1	//--------------------------------------
446			//
447			//
448			//--------------------------------------
449			/**
450			* Retrieves the is-th stored track.
451			* @param it Track number, ranging from 0 to ntrk().
452			* Fitted tracks ( images included ) are stored in a TObjectArray ( TrkLevel2::Track ) in the same order they are returned by the F77 fitting routine.
453			*/
454			TrkTrack *TrkLevel2::GetStoredTrack(int is){
455
456			if(is >= this->ntrk()){
457			cout << " TrkLevel2 Track "<< is << "doen not exits! " << endl;
458			cout << " Stored tracks ntrk() = "<< this->ntrk() << endl;
459			return 0;
460			}
461			TClonesArray &t = *(Track);
462			TrkTrack track = (TrkTrack)t[is];
463			return track;
464			}
465			//--------------------------------------
466			//
467			//
468			//--------------------------------------
469			/**
470			* Retrieves the it-th "physical" track, sorted by the method GetNTracks().
471			* @param it Track number, ranging from 0 to GetNTracks().
472			*/
473			TrkTrack *TrkLevel2::GetTrack(int it){
474
475			if(it >= this->GetNTracks()){
476			cout << " TrkLevel2 Track "<< it << "does not exits! " << endl;
477			cout << " Physical tracks GetNTracks() = "<< this->ntrk() << endl;
478			return 0;
479			}
480			TrkTrack track = (TrkTrack)(*(this->GetTracks()))[it];
481	pam-fi	1.5	GetTracks()->Delete();////TEMPORANEO
482	mocchiut	1.1	return track;
483			}
484	pam-fi	1.5	Int_t TrkLevel2::GetNTracks(){
485			Int_t ntot=0;
486			ntot = GetTracks()->GetEntries();
487			GetTracks()->Delete();////TEMPORANEO
488			return ntot;
489			};
490	mocchiut	1.1	//--------------------------------------
491			//
492			//
493			//--------------------------------------
494			/**
495			* Retrieves (if present) the image of the it-th "physical" track, sorted by the method GetNTracks().
496			* @param it Track number, ranging from 0 to GetNTracks().
497			*/
498			TrkTrack *TrkLevel2::GetTrackImage(int it){
499
500			if(it >= this->GetNTracks()){
501			cout << " TrkLevel2 Track "<< it << "does not exits! " << endl;
502			cout << " Physical tracks GetNTracks() = "<< this->ntrk() << endl;
503			return 0;
504			}
505			TrkTrack track = (TrkTrack)(*(this->GetTracks()))[it];
506			if(!track->HasImage()){
507			cout << " TrkLevel2 Track "<< it << "does not have image! " << endl;
508			return 0;
509			}
510			TrkTrack image = (TrkTrack)(*Track)[track->image];
511			return image;
512
513			}
514			//--------------------------------------
515			//
516			//
517			//--------------------------------------
518			/**
519			* Loads the magnetic field.
520			* @param s Path of the magnetic-field files.
521			*/
522			void TrkLevel2::LoadField(TString s){
523			readb_(s.Data());
524			};
525			//--------------------------------------
526			//
527			//
528			//--------------------------------------
529			/**
530	pam-fi	1.2	* Trajectory default constructor.
531			* (By default is created with z-coordinates inside the tracking volume)
532			*/
533			Trajectory::Trajectory(){
534			npoint = 10;
535			x = new float[npoint];
536			y = new float[npoint];
537			z = new float[npoint];
538			thx = new float[npoint];
539			thy = new float[npoint];
540			tl = new float[npoint];
541			float dz = ((ZTRKUP)-(ZTRKDW))/(npoint-1);
542			for(int i=0; i<npoint; i++){
543			x[i] = 0;
544			y[i] = 0;
545			z[i] = (ZTRKUP) - i*dz;
546			thx[i] = 0;
547			thy[i] = 0;
548			tl[i] = 0;
549			}
550			}
551			//--------------------------------------
552			//
553			//
554			//--------------------------------------
555			/**
556	mocchiut	1.1	* Trajectory constructor.
557	pam-fi	1.2	* (By default is created with z-coordinates inside the tracking volume)
558	mocchiut	1.1	* \param n Number of points
559			*/
560			Trajectory::Trajectory(int n){
561	pam-fi	1.2	if(n<=0){
562			cout << "NB! Trajectory must have at least 1 point >>> created with 10 points" << endl;
563			n=10;
564			}
565	mocchiut	1.1	npoint = n;
566			x = new float[npoint];
567			y = new float[npoint];
568			z = new float[npoint];
569	pam-fi	1.2	thx = new float[npoint];
570			thy = new float[npoint];
571			tl = new float[npoint];
572			float dz = ((ZTRKUP)-(ZTRKDW))/(npoint-1);
573	mocchiut	1.1	for(int i=0; i<npoint; i++){
574	pam-fi	1.2	x[i] = 0;
575	mocchiut	1.1	y[i] = 0;
576	pam-fi	1.2	z[i] = (ZTRKUP) - i*dz;
577			thx[i] = 0;
578			thy[i] = 0;
579			tl[i] = 0;
580	mocchiut	1.1	}
581			}
582			//--------------------------------------
583			//
584			//
585			//--------------------------------------
586			/**
587			* Trajectory constructor.
588			* \param n Number of points
589			* \param pz Pointer to float array, defining z coordinates
590			*/
591			Trajectory::Trajectory(int n, float* zin){
592	pam-fi	1.2	npoint = 10;
593			if(n>0)npoint = n;
594	mocchiut	1.1	x = new float[npoint];
595			y = new float[npoint];
596			z = new float[npoint];
597	pam-fi	1.2	thx = new float[npoint];
598			thy = new float[npoint];
599			tl = new float[npoint];
600			int i=0;
601			do{
602	mocchiut	1.1	x[i] = 0;
603			y[i] = 0;
604			z[i] = zin[i];
605	pam-fi	1.2	thx[i] = 0;
606			thy[i] = 0;
607			tl[i] = 0;
608			i++;
609			}while(zin[i-1] > zin[i] && i < npoint);
610			npoint=i;
611			if(npoint != n)cout << "NB! Trajectory created with "<<npoint<<" points"<<endl;
612	mocchiut	1.1	}
613			//--------------------------------------
614			//
615			//
616			//--------------------------------------
617			/**
618			* Dump the trajectory coordinates.
619			*/
620			void Trajectory::Dump(){
621			cout <<endl<< "Trajectory ========== "<<endl;
622			for (int i=0; i<npoint; i++){
623	pam-fi	1.2	cout << i <<" >> " << x[i] <<" "<< y[i] <<" "<< z[i] ;
624			cout <<" -- " << thx[i] <<" "<< thy[i] ;
625			cout <<" -- " << tl[i] << endl;
626	mocchiut	1.1	};
627			}
628	pam-fi	1.2	//--------------------------------------
629			//
630			//
631			//--------------------------------------
632			/**
633			* Get trajectory length between two points
634			* @param ifirst first point (default 0)
635			* @param ilast last point (default npoint)
636			*/
637			float Trajectory::GetLength(int ifirst, int ilast){
638			if( ifirst<0 ) ifirst = 0;
639			if( ilast>=npoint) ilast = npoint-1;
640			float l=0;
641			for(int i=ifirst;i<=ilast;i++){
642			l=l+tl[i];
643			};
644			if(z[ilast] > ZINI)l=l-tl[ilast];
645			if(z[ifirst] < ZINI) l=l-tl[ifirst];
646
647			return l;
648	mocchiut	1.1
649	pam-fi	1.2	}
650	mocchiut	1.1	ClassImp(TrkLevel2);
651			ClassImp(TrkSinglet);
652			ClassImp(TrkTrack);
653			ClassImp(Trajectory);