//*-- Author :    Ole Hansen   29 March 2001

//////////////////////////////////////////////////////////////////////////
//
// THaTrack
//
// A generic track.
//
//////////////////////////////////////////////////////////////////////////

#include "THaTrack.h"
#include "THaCluster.h"
#include "THaTrackID.h"
#include <iostream>

using namespace std;

const Double_t THaTrack::kBig = 1e38;

//_____________________________________________________________________________
THaTrack::~THaTrack()
{
  // Destructor. Delete the track ID associated with this track.

  delete fID;
}

//_____________________________________________________________________________
void THaTrack::Clear( const Option_t* opt )
{
  // If *opt == 'F' then reset all track quantities, else just
  // delete memory managed by this track.
  // (We need this behavior so we can Clear("C") the track TClonesArray
  // without the overhead of clearing everything.)
 
  //FIXME: too complicated. Do we really need to reallocate the trackID?

  if( opt && (*opt == 'F') ) {
    fTheta = fPhi = fX = fY = fP = fDp = kBig;
    fRX = fRY = fRTheta = fRPhi = kBig;
    fTX = fTY = fTTheta = fTPhi = kBig;
    fDX = fDY = fDTheta = fDPhi = kBig;
    fNclusters = fFlag = fType = 0;
    if( fPIDinfo ) fPIDinfo->Clear( opt );
    fPvect.SetXYZ( kBig, kBig, kBig );
    fVertex.SetXYZ( kBig, kBig, kBig );
    fVertexError.SetXYZ( kBig, kBig, kBig );
    fPathl = fTime = fdTime = fBeta = fdBeta = kBig;
    fChi2 = kBig; fNDoF = 0;
    memset( fClusters, 0, kMAXCL*sizeof(THaCluster*) );
  }
  delete fID; fID = 0;
}

//_____________________________________________________________________________
Int_t THaTrack::AddCluster( THaCluster* c )
{
  // Add a cluster to the internal list of pointers.
  // (Clusters are memory-managed by tracking detectors, not by tracks.)

  if( fNclusters >= kMAXCL )
    return 1;

  fClusters[ fNclusters++ ] = c;
  return 0;
}
   
//_____________________________________________________________________________
void THaTrack::Print( Option_t* opt ) const
{
  // Print track parameters
  TObject::Print( opt );
  cout << "Momentum = " << fP << " GeV/c\n";
  cout << "x_fp     = " << fX   << " m\n";
  cout << "y_fp     = " << fY   << " m\n";
  cout << "theta_fp = " << fTheta << " rad\n";
  cout << "phi_fp   = " << fPhi << " rad\n";

  for( int i=0; i<fNclusters; i++ )
    fClusters[i]->Print();
  if( fPIDinfo ) fPIDinfo->Print( opt );
}

//_____________________________________________________________________________
static Double_t SafeNDoF( Int_t dof )
{
  if( dof <= 0 )
    return 1e-10;
  return static_cast<Double_t>(dof);
}

//_____________________________________________________________________________
Int_t THaTrack::Compare(const TObject * obj) const
{
  // compare two tracks by chi2/ndof
  // for track array sorting

  const THaTrack* tr = dynamic_cast<const THaTrack*>(obj);
  if (!tr) return 0;

  Double_t v1 = GetChi2() / SafeNDoF( GetNDoF() );
  Double_t v2 = tr->GetChi2()/ SafeNDoF( tr->GetNDoF() );

  if( v1<v2 ) return -1;
  else if( v1==v2 ) return 0;
  else return 1;
}


//_____________________________________________________________________________

ClassImp(THaTrack)