HoughTrack Class Reference

#include <HoughTrack.h>

Public Member Functions
	HoughTrack ()
	~HoughTrack ()
	HoughTrack (const HoughTrack &other)
	HoughTrack (const HoughPeak &centerPeak, std::vector< const HoughHit * > trackHitList, double, double)
TrkRecoTrk *	getTrk () const
TrkRecoTrk *	getTrk2D () const
vector< MdcHit * > &	getVecForClean3D ()
HoughPeak	getCenterPeak () const
Hough2D	get2D () const
Hough3D	get3D () const
recHitCol &	getHoughHitList ()
void	setCharge (int charge)
int	getCharge () const
double	getPt_least () const
double	getPt2D () const
double	getPt3D () const
double	getP () const
double	getPz () const
double	getRho () const
double	getTheta () const
double	getD0 () const
double	getOmega () const
double	getPhi0 () const
double	getTanl () const
double	getZ0 () const
double	getTanl_zs () const
double	getZ0_zs () const
int	getHitNum (int) const
int	getHitNumA (int) const
int	getHitNumS (int) const
int	getNfit2D () const
int	getNfit3D () const
int	getStat2D () const
int	getStat3D () const
double	getCirR () const
double	getCirX () const
double	getCirY () const
double	getChi2_3D () const
double	getChi2_2D () const
double	getNfit () const
void	setHoughHitList (vector< HoughHit > vec_hit)
void	printRecHit () const
void	print ()
void	sortHit ()
void	fit (double bunchtime)
int	calzs ()
void	fitzs ()
int	judgeTrack ()
int	find_stereo_hit ()
int	find_pair_hit ()
int	cutNoise_inner ()
int	cut_axial_inner ()
HoughTrack &	operator= (const HoughTrack &other)
HoughTrack &	add (const HoughTrack &other)
int	fit2D (double bunchtime)
int	fit3D ()
int	fit3D_inner ()
int	trackCharge2D ()
int	trackCharge3D ()
void	outerHit ()
void	cald_layer ()
double	getMapRho () const
double	getMapTheta () const
double	getPro () const
void	Leastfit (vector< double >, vector< double >, double &, double &)
void	setMdcHit (const vector< MdcHit * > *mdchit)

Static Public Attributes
static bool	m_debug = 0

Detailed Description

Definition at line 14 of file HoughTrack.h.

Constructor & Destructor Documentation

HoughTrack() [1/3]

HoughTrack::HoughTrack

(

)

Definition at line 12 of file HoughTrack.cxx.

                       {
  p_trk   = NULL;
  p_trk2D = NULL;
}

Referenced by add(), HoughTrack(), and operator=().

~HoughTrack()

HoughTrack::~HoughTrack

(

)

Definition at line 16 of file HoughTrack.cxx.

{}

HoughTrack() [2/3]

HoughTrack::HoughTrack

(

const HoughTrack &

other

)

Definition at line 71 of file HoughTrack.cxx.

    : _dist( other._dist )
    , _charge( other._charge )
    , _ptLeast( other._ptLeast )
    , _pt2D( other._pt2D )
    , _pt3D( other._pt3D )
    , _pz( other._pz )
    , _p( other._p )
    , _d0( other._d0 )
    , _omega( other._omega )
    , _phi0( other._phi0 )
    , _z0( other._z0 )
    , _tanl( other._tanl )
    , _z0_zs( other._z0_zs )
    , _tanl_zs( other._tanl_zs )
    , _centerPeak( other._centerPeak )
    , _Hough2D( other._Hough2D )
    , _Hough3D( other._Hough3D )
    , _bunchTime( other._bunchTime )
    , _centerX( other._centerX )
    , _centerY( other._centerY )
    , _centerR( other._centerR )
    , _chi2_aver( other._chi2_aver )
    , _nfit( other._nfit )
    , _chi2_aver2D( other._chi2_aver2D )
    , _nfit2D( other._nfit2D )
    , _stat2D( other._stat2D )
    , _stat3D( other._stat3D )
    , _recHitVec( other._recHitVec )
    , p_trk( other.p_trk )
    , p_trk2D( other.p_trk2D )
    , _maprho( other._maprho )
    , _maptheta( other._maptheta )
    , t_pro_correct( other.t_pro_correct )
    , _houghList( other._houghList )
    , vec_mdcHit( other.vec_mdcHit ) {
  // _recHitVec.clear();
  // if(_recHitVec.size() != 0 ){
  //  //cout<<"clear first "<<endl;
  //  for(int i =0;i<_recHitVec.size();i++){
  //    delete _recHitVec[i];
  //    _recHitVec[i]=NULL;
  //  }
  //  _recHitVec.clear();
  //}
  // for(int i =0;i<other._recHitVec.size();i++){
  //  //const HoughHit* p_hit= new HoughHit( *(other._recHitVec[i]));
  //  HoughRecHit p_hit ( (trackHitList[i])->digi(),0.,0. );
  //  _recHitVec.push_back(p_hit);
  //}
}

HoughTrack() [3/3]

HoughTrack::HoughTrack	(	const HoughPeak &	centerPeak,
		std::vector< const HoughHit * >	trackHitList,
		double	rho,
		double	theta )

Definition at line 143 of file HoughTrack.cxx.

                                                   {
  _centerPeak = centerPeak;
  _recHitVec.clear();
  _stat2D    = 0;
  _stat3D    = 0;
  _tanl      = -999;
  _z0        = -999;
  _ptLeast   = -999;
  _pt2D      = -999;
  _pt3D      = -999;
  _p         = -999;
  _d0        = -999;
  _omega     = -999;
  _phi0      = -999;
  _z0        = -999;
  _tanl      = -999;
  _chi2_aver = -999;
  _nfit      = -999;
  p_trk      = NULL;
  p_trk2D    = NULL;
  _charge    = 0; // undeter mine
  _maprho    = rho;
  _maptheta  = theta;
  // if(_recHitVec.size()!= 0 ){
  //   cout<<" Mind  trackhitlist size !=0 "<<endl;
  //   for(int i =0;i<_recHitVec.size();i++){
  //     delete _recHitVec[i];
  //     _recHitVec[i]=NULL;
  //   }
  //   _recHitVec.clear();
  // }
  int t_size = trackHitList.size();
  for ( int i = 0; i < t_size; i++ )
  {
    HoughRecHit p_hit( *( trackHitList[i] ), 0., 0., 1 );
    p_hit.setPtr2D( &_Hough2D );
    p_hit.setflag( 0 );
    _recHitVec.push_back( p_hit );
  }
}

Member Function Documentation

add()

HoughTrack & HoughTrack::add

(

const HoughTrack &

other

)

Definition at line 123 of file HoughTrack.cxx.

                                                     {
  for ( int i = 0; i < other._recHitVec.size(); i++ )
  {
    int same = 0;
    for ( int j = 0; j < _recHitVec.size(); j++ )
    {
      if ( _recHitVec[j].digi() == other._recHitVec[i].digi() ) { same = 1; }
    }
    if ( same == 0 )
    {
      // const HoughHit* p_hit= new HoughHit( *(other._recHitVec[i]));
      // HoughRecHit p_hit ( (other._recHitVec[i]).digi(),0.,1);
      HoughRecHit p_hit( ( other._recHitVec[i] ) );
      _recHitVec.push_back( p_hit );
    }
    // cout<<" after copy: "<<i<<"  new: "<<p_hit<<" old:  "<<other._recHitVec[i]<<endl;
  }
  return *this;
}

cald_layer()

void HoughTrack::cald_layer

(

)

Definition at line 982 of file HoughTrack.cxx.

                            {
  // in truth
  double                             k, b, theta, rho, x_cross, y_cross;
  vector<double>                     vtemp, utemp;
  std::vector<HoughRecHit>::iterator iter = _recHitVec.begin();
  for ( int iHit = 0; iter != _recHitVec.end(); iter++, iHit++ )
  {
    const HoughRecHit h = ( *iter );
    //  if( h.getCirList()!=0 ) continue;
    if ( h.digi()->getTrackIndex() >= 0 && h.getStyle() == 0 && h.getSlayerType() == 0 &&
         h.getCirList() == 0 && utemp.size() < 10 ) // ??use 2nd half
    {
      utemp.push_back( h.getUTruth() );
      vtemp.push_back( h.getVTruth() );
    }
  }
  Leastfit( utemp, vtemp, k, b );
  // calcu truth
  // k,b from truth
  x_cross = -b / ( k + 1 / k );
  y_cross = b / ( 1 + k * k );
  rho     = sqrt( x_cross * x_cross + y_cross * y_cross );
  theta   = atan2( y_cross, x_cross );
  //
  // cout<<"track center position "<<_centerX<<" "<<_centerY<<endl;
  std::vector<HoughRecHit>::iterator iter0 = _recHitVec.begin();
  for ( ; iter0 != _recHitVec.end(); iter0++ )
  {
    HoughRecHit* hit = &( *iter0 );
    if ( hit->getSlayerType() != 0 ) continue;
    //  if( hit->getCirList()!=0 ) continue;        // use in learn distribute
    // double cirr_hit = hit->getDriftDistTruth();
    double cirx_hit = hit->getMidX();
    double ciry_hit = hit->getMidY();
    double cirr_hit = hit->getDriftDist();
    double l1l2     = sqrt( ( cirx_hit - _centerX ) * ( cirx_hit - _centerX ) +
                            ( ciry_hit - _centerY ) * ( ciry_hit - _centerY ) );
    double deltaD = 1.e9; // add initialize 25-05-15
    if ( l1l2 > _centerR ) deltaD = l1l2 - _centerR - cirr_hit;
    if ( l1l2 <= _centerR ) deltaD = l1l2 - _centerR + cirr_hit;
    hit->setDeltaD( deltaD );
    // cal flight length
 
    double theta_temp;
    double l_temp = 1.e9; // add initialize 25-05-15
    if ( _centerX == 0 || cirx_hit - _centerX == 0 ) { theta_temp = 0; }
    else
    {
      theta_temp = M_PI - atan2( ciry_hit - _centerY, cirx_hit - _centerX ) +
                   atan2( _centerY, _centerX );
      if ( theta_temp > 2 * M_PI ) { theta_temp = theta_temp - 2 * M_PI; }
      if ( theta_temp < 0 ) { theta_temp = theta_temp + 2 * M_PI; }
    }
    // cout<<" charge "<<_charge <<" "<<theta_temp<<endl;
    if ( _charge == -1 ) { l_temp = _centerR * theta_temp; }
    if ( _charge == 1 )
    {
      theta_temp = 2 * M_PI - theta_temp;
      l_temp     = _centerR * ( theta_temp );
    }
    //  cout<<"("<<hit->getLayerId()<<","<<hit->getWireId()<<") "<<l_temp<<endl;
    hit->setFltLen( l_temp );
 
    //    cout<<"int map deltaD: ("<<hit->getLayerId()<<","<<hit->getWireId()<<")"<<
    // hit->getDeltaD()<<endl;
  }
}

calzs()

int HoughTrack::calzs

(

)

Definition at line 601 of file HoughTrack.cxx.

                      {
  int n_zs = 0;
  for ( int i = 0; i < _recHitVec.size(); i++ )
  {
    if ( _recHitVec[i].getSlayerType() == 0 || _recHitVec[i].getflag() != 0 ) continue;
    //  if ( _recHitVec[i].getLayerId()>=8) continue;
    HoughStereo zs( _bunchTime, &_Hough2D, &( _recHitVec[i] ) );
    //  cout<<"("<<_recHitVec[i].getLayerId()<<","<<_recHitVec[i].getWireId()<<")  stat
    //"<<stat<<" style
    //"<<_recHitVec[i].getStyle() <<endl;   cout<<endl;
 
    // zs.setAmb(-1);   //right
    // int stat_left = zs.cald();
 
    // zs.setAmb(1);    //left
    // int stat_right= zs.cald();
    // if( stat_left==-1  &&  stat_right==-1 )  _recHitVec[i].setflag(-999);
 
    int stat = zs.cald();
    _recHitVec[i].setnsol( stat );
    if ( stat == 0 ) _recHitVec[i].setflag( -999 );
 
    zs.setRecHit();
    _recHitVec[i].setAmb( -999 ); // check
 
    if ( _recHitVec[i].getLayerId() < 8 ) n_zs++;
    if ( m_debug > 0 ) zs.print();
  }
  //    int naddStero = cutNoise_inner();
  // return naddStero;
  return n_zs;
}

Referenced by fit3D(), and fit3D_inner().

cut_axial_inner()

int HoughTrack::cut_axial_inner

(

)

Definition at line 797 of file HoughTrack.cxx.

                                {
  for ( int ilay = 0; ilay < 43; ilay++ )
  {
    // if( (ilay<7&&ilay<20) || ilay>35 ) continue;
    if ( ( ilay >= 0 && ilay <= 7 ) || ( ilay >= 20 && ilay <= 35 ) ) continue;
    //  for( int ilay=0;ilay<8;ilay++)
    double disToCir  = 9999;
    int    max       = -999;
    int    count     = 0;
    int    count_cut = 0;
    for ( int ihit = 0; ihit < _recHitVec.size(); ihit++ )
    {
      if ( _recHitVec[ihit].getLayerId() == ilay && _recHitVec[ihit].getflag() == 0 )
      {
        count++;
        if ( ( fabs( _recHitVec[ihit].getDisToCir() ) < disToCir ) )
        {
          disToCir  = fabs( _recHitVec[ihit].getDisToCir() );
          max       = ihit;
          count_cut = 1;
        }
      }
    }
    if ( m_debug > 0 )
      cout << "ilay count count_cut " << ilay << " " << count << " " << count_cut << endl;
    if ( count_cut != 0 && count > 1 ) _recHitVec[max].setflag( -999 );
    if ( m_debug > 0 && count_cut != 0 && count > 1 )
      cout << "delete (" << _recHitVec[max].getLayerId() << "," << _recHitVec[max].getWireId()
           << ")" << endl;
  }
  int size_of_axial = 0; // add initialize 25-05-15
  for ( int ihit = 0; ihit < _recHitVec.size(); ihit++ )
  {
    if ( _recHitVec[ihit].getSlayerType() == 0 && _recHitVec[ihit].getflag() == 0 )
      size_of_axial++;
  }
  return size_of_axial;
}

cutNoise_inner()

int HoughTrack::cutNoise_inner

(

)

Definition at line 836 of file HoughTrack.cxx.

                               {
  //  for( int ilay=0;ilay<43;ilay++)
  //    if( (ilay>7&&ilay<20) || ilay>35 ) continue;
  for ( int ilay = 0; ilay < 8; ilay++ )
  {
    double z_0       = 0;
    double z_1       = 0;
    int    min       = -999;
    int    count     = 0;
    int    count_cut = 0;
    //  cout<<" size "<<_recHitVec.size()<<endl;
    for ( int ihit = 0; ihit < _recHitVec.size(); ihit++ )
    {
      if ( _recHitVec[ihit].getLayerId() == ilay && _recHitVec[ihit].getflag() == 0 )
      {
        count++;
        if ( ( ( fabs( _recHitVec[ihit].getzAmb( 0 ) ) > z_0 ) &&
               ( fabs( _recHitVec[ihit].getzAmb( 1 ) ) > z_1 ) ) )
        {
          if ( z_0 != 0 && z_1 != 0 ) count_cut++;
          z_0 = fabs( _recHitVec[ihit].getzAmb( 0 ) );
          z_1 = fabs( _recHitVec[ihit].getzAmb( 1 ) );
          min = ihit;
        }
        else if ( ( ( fabs( _recHitVec[ihit].getzAmb( 0 ) ) < z_0 ) &&
                    ( fabs( _recHitVec[ihit].getzAmb( 1 ) ) < z_1 ) ) )
        { count_cut++; }
      }
    }
    if ( m_debug > 0 )
      cout << "ilay count count_cut " << ilay << " " << count << " " << count_cut << endl;
    if ( count_cut > 0 && count > 1 ) _recHitVec[min].setflag( -999 );
    if ( m_debug > 0 && count_cut > 0 && count > 1 )
      cout << "delete (" << _recHitVec[min].getLayerId() << "," << _recHitVec[min].getWireId()
           << ")" << endl;
  }
  int size_of_stereo = 0; // add initialize 25-05-15
  for ( int ihit = 0; ihit < _recHitVec.size(); ihit++ )
  {
    if ( _recHitVec[ihit].getSlayerType() != 0 && _recHitVec[ihit].getflag() == 0 )
      size_of_stereo++;
  }
  return size_of_stereo;
}

find_pair_hit()

int HoughTrack::find_pair_hit

(

)

Definition at line 1049 of file HoughTrack.cxx.

                              {
  int nster1  = 0;
  int nster2  = 0;
  int nster3  = 0;
  int naxial1 = 0;
  int naxial2 = 0;
  int naxial3 = 0;
  for ( int ihit = 0; ihit < _houghList.size(); ihit++ )
  {
    const HoughHit hit = _houghList[ihit];
    if ( hit.driftTime( _bunchTime, 0 ) > 1000 ) continue;
    if ( hit.getLayerId() >= 24 ) continue;
    HoughRecHit p_hit( hit, 0, 0, 1 );
    p_hit.setPtr2D( &_Hough2D );
    std::pair<double, double> theta_l   = calcuArcTrack( p_hit );
    double                    dist      = calcuDistToTrack( p_hit );
    double                    distToCir = calcuDistToCir( p_hit );
    int                       flag      = judge_half( p_hit );
    int                       layer     = p_hit.getLayerId();
    int                       wire      = p_hit.getWireId();
    int                       slayer    = p_hit.getSlayerType();
    double                    disCut;
    if ( slayer == 0 ) disCut = 6;
    else
    {
      if ( layer < 8 ) disCut = 6;
      else disCut = 6;
    }
    if ( m_debug > 0 )
      cout << "(" << layer << "," << wire << ") "
           << " pair dist flag " << dist << "  " << flag << endl;
    if ( fabs( dist ) < disCut )
    {
      if ( flag != 1 ) continue; // find hit in the 2nd half
      if ( layer < 4 ) nster1++;
      else if ( layer < 8 ) nster2++;
      else if ( layer < 12 ) naxial1++;
      else if ( layer < 16 ) naxial2++;
      else if ( layer < 20 ) naxial3++;
      else nster3++;
    }
  }
  if ( m_debug > 0 )
  {
    cout << "naxial_1  " << naxial1 << endl;
    cout << "naxial_2  " << naxial2 << endl;
    cout << "naxial_3  " << naxial3 << endl;
    cout << "stereo_1  " << nster1 << endl;
    cout << "stereo_2  " << nster2 << endl;
    cout << "stereo_3  " << nster3 << endl;
  }
  if ( nster1 >= 2 && nster2 >= 2 && nster3 >= 2 && naxial1 >= 2 && naxial2 >= 2 &&
       naxial3 >= 2 )
    return 1;
  else return 0;
}

Referenced by MdcHoughFinder::execute().

find_stereo_hit()

int HoughTrack::find_stereo_hit

(

)

Definition at line 757 of file HoughTrack.cxx.

                                {
  vector<HoughRecHit> vec_rec; // store hit temp
  int                 naddStero = 0;
  for ( int ihit = 0; ihit < _houghList.size(); ihit++ )
  {
    const HoughHit hit = _houghList[ihit];
    if ( hit.getSlayerType() == 0 ) continue;
    if ( hit.driftTime( _bunchTime, 0 ) > 800 ) continue;
    HoughRecHit p_hit( hit, 0, 0, 1 );
    p_hit.setPtr2D( &_Hough2D );
    std::pair<double, double> theta_l   = calcuArcTrack( p_hit );
    double                    dist      = calcuDistToTrack( p_hit );
    double                    distToCir = calcuDistToCir( p_hit );
    int                       flag      = judge_half( p_hit );
    int                       layer     = p_hit.getLayerId();
    int                       wire      = p_hit.getWireId();
    double                    disCut;
    if ( layer < 8 ) disCut = 4;
    else disCut = 6;
    //  double dist=sqrt( pow( (xhit-x_cir),2)+pow( (yhit-y_cir),2) )-r_cir;
    //  cout<<"find stereo hit dist?("<<layer<<" ,"<<wire<<" ) "<<dist<<endl;
    if ( m_debug > 0 )
      cout << "(" << layer << "," << wire << ") "
           << " rec hit dist theta " << dist << "  " << theta_l.first << endl;
    if ( fabs( dist ) < disCut )
    {
      p_hit.setDisToTrack( dist );
      p_hit.setDisToCir( dist );
      p_hit.setRet( theta_l );
      p_hit.setflag( flag );
      //  if( p_hit.getflag()==0 ) vec_rec.push_back(p_hit);
      //      vec_rec.push_back(p_hit);
      if ( p_hit.getflag() != 0 ) continue;
      _recHitVec.push_back( p_hit );
      naddStero++;
    }
  }
  return naddStero;
}

Referenced by MdcHoughFinder::execute().

fit()

void HoughTrack::fit

(

double

bunchtime

)

fit2D()

int HoughTrack::fit2D

(

double

bunchtime

)

Definition at line 192 of file HoughTrack.cxx.

                                        {
  sortHit();
  //  cout<<"in fit2d"<<endl;
  _bunchTime = bunchtime;
  _stat2D    = fitLeast();
  //  outerHit();
  _stat2D = fit_global2D( _recHitVec );
  // printRecHit();
  // collectAxialHit();
  //_Hough2D.print();
  // cald_layer();
  return _stat2D;
}

Referenced by MdcHoughFinder::execute().

fit3D()

int HoughTrack::fit3D

(

)

Definition at line 317 of file HoughTrack.cxx.

                      {
  sortHit();
  //  cout<<"in fit3d"<<endl;
  int nhit_zs = calzs();
  if ( nhit_zs >= 3 ) fitzs();
  else
  {
    _tanl = 0.;
    _z0   = 0.;
  }
  outerHit();
  if ( m_debug > 0 )
  {
    cout << " before 3d fit " << endl;
    this->printRecHit();
  }
  _stat3D = fit_global3D( 0 );
  return _stat3D;
}

Referenced by MdcHoughFinder::execute().

fit3D_inner()

int HoughTrack::fit3D_inner

(

)

Definition at line 337 of file HoughTrack.cxx.

                            { // for multi turn track
  sortHit();
  int nhit_zs = calzs();
  cutMultiCirHit();
  if ( nhit_zs >= 3 ) fitzs();
  else
  {
    //  cout<<" not enough ster hit in zs fit"<<endl;
    _tanl = 0.;
    _z0   = 0.;
  }
  cutMultiCirHit_after_zs();
  outerHit();
  if ( m_debug > 0 )
  {
    cout << " before 3d fit " << endl;
    this->printRecHit();
  }
  _stat3D = fit_global3D( 0 );
  return _stat3D;
}

Referenced by MdcHoughFinder::execute().

fitzs()

void HoughTrack::fitzs

(

)

Definition at line 633 of file HoughTrack.cxx.

                       {
  HoughZsFit zsfit( &_recHitVec );
  _tanl         = zsfit.getTanl();
  _z0           = zsfit.getZ0();
  _tanl_zs      = zsfit.getTanl();
  _z0_zs        = zsfit.getZ0();
  t_pro_correct = zsfit.getPro();
  if ( m_debug > 0 ) printRecHit();
}

Referenced by fit3D(), and fit3D_inner().

get2D()

Hough2D HoughTrack::get2D ( ) const

inline

Definition at line 30 of file HoughTrack.h.

{ return _Hough2D; }

get3D()

Hough3D HoughTrack::get3D ( ) const

inline

Definition at line 31 of file HoughTrack.h.

{ return _Hough3D; }

getCenterPeak()

HoughPeak HoughTrack::getCenterPeak ( ) const

inline

Definition at line 29 of file HoughTrack.h.

{ return _centerPeak; }

getCharge()

int HoughTrack::getCharge ( ) const

inline

Definition at line 35 of file HoughTrack.h.

{ return _charge; }

getChi2_2D()

double HoughTrack::getChi2_2D ( ) const

inline

Definition at line 63 of file HoughTrack.h.

{ return _chi2_aver2D; }

getChi2_3D()

double HoughTrack::getChi2_3D ( ) const

inline

Definition at line 62 of file HoughTrack.h.

{ return _chi2_aver; }

getCirR()

double HoughTrack::getCirR ( ) const

inline

Definition at line 59 of file HoughTrack.h.

{ return _centerR; }

getCirX()

double HoughTrack::getCirX ( ) const

inline

Definition at line 60 of file HoughTrack.h.

{ return _centerX; }

getCirY()

double HoughTrack::getCirY ( ) const

inline

Definition at line 61 of file HoughTrack.h.

{ return _centerY; }

getD0()

double HoughTrack::getD0 ( ) const

inline

Definition at line 43 of file HoughTrack.h.

{ return _d0; }

getHitNum()

int HoughTrack::getHitNum

(

int

select

)

const

Definition at line 643 of file HoughTrack.cxx.

                                            {
  int size = _recHitVec.size();
  int n0, n1, n2, n3, n4, n5, n6;
  n0 = n1 = n2 = n3 = n4 = n5 = n6 = 0;
  for ( int i = 0; i < size; i++ )
  {
    int cir   = _recHitVec[i].getCirList();
    int index = _recHitVec[i].digi()->getTrackIndex();
    int style = _recHitVec[i].getStyle();
    n0++;
    if ( style == 1 ) n1++;
    if ( style == 2 ) n2++;
    if ( index < 0 ) n3++;
    if ( index >= 0 && cir <= 1 && style == 0 ) n4++;
    if ( index < 0 || cir > 1 ) n5++;
    if ( index >= 0 && cir == 0 && style == 0 ) n6++;
  }
  if ( select == 0 ) return n0;
  if ( select == 1 ) return n1;
  if ( select == 2 ) return n2;
  if ( select == 3 ) return n3;
  if ( select == 4 ) return n4;
  if ( select == 5 ) return n5;
  if ( select == 6 ) return n6;
  std::cerr << __FILE__ << ":" << __LINE__ << ": should not reach here" << std::endl;
  exit( 1 );
}

getHitNumA()

int HoughTrack::getHitNumA

(

int

select

)

const

Definition at line 671 of file HoughTrack.cxx.

                                             {
  int size = _recHitVec.size();
  int n0, n1, n2, n3, n4, n5, n6;
  n0 = n1 = n2 = n3 = n4 = n5 = n6 = 0;
  for ( int i = 0; i < size; i++ )
  {
    int cir   = _recHitVec[i].getCirList();
    int index = _recHitVec[i].digi()->getTrackIndex();
    int type  = _recHitVec[i].getSlayerType();
    int style = _recHitVec[i].getStyle();
    if ( type == 0 ) n0++;
    if ( type == 0 && style == 1 ) n1++;
    if ( type == 0 && style == 2 ) n2++;
    if ( type == 0 && index < 0 ) n3++;
    if ( type == 0 && index >= 0 && cir <= 1 && style == 0 ) n4++;
    if ( type == 0 && ( index < 0 || cir > 1 ) ) n5++;
    if ( type == 0 && index >= 0 && cir == 0 && style == 0 ) n6++;
  }
  if ( select == 0 ) return n0;
  if ( select == 1 ) return n1;
  if ( select == 2 ) return n2;
  if ( select == 3 ) return n3;
  if ( select == 4 ) return n4;
  if ( select == 5 ) return n5;
  if ( select == 6 ) return n6;
  std::cerr << __FILE__ << ":" << __LINE__ << ": should not reach here" << std::endl;
  exit( 1 );
}

getHitNumS()

int HoughTrack::getHitNumS

(

int

select

)

const

Definition at line 699 of file HoughTrack.cxx.

                                             {
  int size = _recHitVec.size();
  int n0, n1, n2, n3, n4, n5, n6;
  n0 = n1 = n2 = n3 = n4 = n5 = n6 = 0;
  for ( int i = 0; i < size; i++ )
  {
    int cir   = _recHitVec[i].getCirList();
    int index = _recHitVec[i].digi()->getTrackIndex();
    int type  = _recHitVec[i].getSlayerType();
    int style = _recHitVec[i].getStyle();
    if ( type != 0 ) n0++;
    if ( type != 0 && style == 1 ) n1++;
    if ( type != 0 && style == 2 ) n2++;
    if ( type != 0 && index < 0 ) n3++;
    if ( type != 0 && index >= 0 && cir <= 1 && style == 0 ) n4++;
    if ( type != 0 && ( index < 0 || cir > 1 ) ) n5++;
    if ( type != 0 && index >= 0 && cir == 0 && style == 0 ) n6++;
  }
  if ( select == 0 ) return n0;
  if ( select == 1 ) return n1;
  if ( select == 2 ) return n2;
  if ( select == 3 ) return n3;
  if ( select == 4 ) return n4;
  if ( select == 5 ) return n5;
  if ( select == 6 ) return n6;
  std::cerr << __FILE__ << ":" << __LINE__ << ": should not reach here" << std::endl;
  exit( 1 );
}

getHoughHitList()

recHitCol & HoughTrack::getHoughHitList ( )

inline

Definition at line 32 of file HoughTrack.h.

{ return _recHitVec; }

getMapRho()

double HoughTrack::getMapRho ( ) const

inline

Definition at line 90 of file HoughTrack.h.

{ return _maprho; }

getMapTheta()

double HoughTrack::getMapTheta ( ) const

inline

Definition at line 91 of file HoughTrack.h.

{ return _maptheta; }

getNfit()

double HoughTrack::getNfit ( ) const

inline

Definition at line 64 of file HoughTrack.h.

{ return _nfit; }

getNfit2D()

int HoughTrack::getNfit2D ( ) const

inline

Definition at line 54 of file HoughTrack.h.

{ return _Hough2D.getNfit(); }

getNfit3D()

int HoughTrack::getNfit3D ( ) const

inline

Definition at line 55 of file HoughTrack.h.

{ return _Hough3D.getNfit(); }

getOmega()

double HoughTrack::getOmega ( ) const

inline

Definition at line 44 of file HoughTrack.h.

{ return _omega; }

getP()

double HoughTrack::getP ( ) const

inline

Definition at line 39 of file HoughTrack.h.

{ return _p; }

getPhi0()

double HoughTrack::getPhi0 ( ) const

inline

Definition at line 45 of file HoughTrack.h.

{ return _phi0; }

getPro()

double HoughTrack::getPro ( ) const

inline

Definition at line 92 of file HoughTrack.h.

{ return t_pro_correct; }

getPt2D()

double HoughTrack::getPt2D ( ) const

inline

Definition at line 37 of file HoughTrack.h.

{ return _pt2D; }

getPt3D()

double HoughTrack::getPt3D ( ) const

inline

Definition at line 38 of file HoughTrack.h.

{ return _pt3D; }

Referenced by more_pt().

getPt_least()

double HoughTrack::getPt_least ( ) const

inline

Definition at line 36 of file HoughTrack.h.

{ return _ptLeast; }

getPz()

double HoughTrack::getPz ( ) const

inline

Definition at line 40 of file HoughTrack.h.

{ return _pz; }

getRho()

double HoughTrack::getRho ( ) const

inline

Definition at line 41 of file HoughTrack.h.

{ return _centerPeak.getRho(); }

getStat2D()

int HoughTrack::getStat2D ( ) const

inline

Definition at line 56 of file HoughTrack.h.

{ return _stat2D; }

getStat3D()

int HoughTrack::getStat3D ( ) const

inline

Definition at line 57 of file HoughTrack.h.

{ return _stat3D; }

getTanl()

double HoughTrack::getTanl ( ) const

inline

Definition at line 46 of file HoughTrack.h.

{ return _tanl; }

getTanl_zs()

double HoughTrack::getTanl_zs ( ) const

inline

Definition at line 48 of file HoughTrack.h.

{ return _tanl_zs; }

getTheta()

double HoughTrack::getTheta ( ) const

inline

Definition at line 42 of file HoughTrack.h.

{ return _centerPeak.getTheta(); }

getTrk()

TrkRecoTrk * HoughTrack::getTrk ( ) const

inline

Definition at line 25 of file HoughTrack.h.

{ return p_trk; }

getTrk2D()

TrkRecoTrk * HoughTrack::getTrk2D ( ) const

inline

Definition at line 26 of file HoughTrack.h.

{ return p_trk2D; }

getVecForClean3D()

vector< MdcHit * > & HoughTrack::getVecForClean3D ( )

inline

Definition at line 28 of file HoughTrack.h.

{ return _Hough3D.getVecForClean(); }

getZ0()

double HoughTrack::getZ0 ( ) const

inline

Definition at line 47 of file HoughTrack.h.

{ return _z0; }

getZ0_zs()

double HoughTrack::getZ0_zs ( ) const

inline

Definition at line 49 of file HoughTrack.h.

{ return _z0_zs; }

judgeTrack()

int HoughTrack::judgeTrack

(

)

Definition at line 728 of file HoughTrack.cxx.

                           {
  // hit number test
  int axialHit = 0;
  //  int stereoHit=0;
  //  int innerHit=0;
  vector<int> vec_layer;
  for ( int i = 0; i < _recHitVec.size(); i++ )
  {
    int layer = _recHitVec[i].getLayerId();
    int slant = _recHitVec[i].getSlayerType();
    if ( 0 == slant ) axialHit++;
    //  if( 0!=slant ) stereoHit++;
    //  if( layer<8 ) innerHit++;
  }
  //  if ( axialHit <5 || stereoHit <3 || innerHit ==0 ) return 0;
  if ( axialHit < 5 ) return 0;
  else return 1;
  // // continues hit test
  // bool continues=false;
  // for(int i =0;i<vec_layer.size();i++){
  //   int layer = vec_layer[i];
  //   vector<int>::iterator iter = find( vec_layer.begin(),vec_layer.end(),layer+1 );
  //   if( iter != vec_layer.end() ){
  //     vector<int>::iterator iter2 = find( vec_layer.begin(),vec_layer.end(),layer-1 );
  //     if( iter2 != vec_layer.end() )  continues = true;
  //   }
  // }
}

Leastfit()

void HoughTrack::Leastfit	(	vector< double >	u,
		vector< double >	v,
		double &	k,
		double &	b )

Definition at line 959 of file HoughTrack.cxx.

                                                                                    {
  int N = u.size();
  if ( N <= 2 ) return;
  double* x = new double[N];
  double* y = new double[N];
  for ( int i = 0; i < N; i++ )
  {
    x[i] = u[i];
    y[i] = v[i];
  }
 
  TF1*    fpol1 = new TF1( "fpol1", "pol1", -50, 50 );
  TGraph* tg    = new TGraph( N, x, y );
  tg->Fit( "fpol1", "QN" );
  double ktemp = fpol1->GetParameter( 1 );
  double btemp = fpol1->GetParameter( 0 );
  k            = ktemp;
  b            = btemp;
  delete[] x;
  delete[] y;
  delete fpol1;
  delete tg;
}

Referenced by cald_layer().

operator=()

HoughTrack & HoughTrack::operator=

(

const HoughTrack &

other

)

Definition at line 18 of file HoughTrack.cxx.

                                                         {
  _dist    = ( other._dist );
  _charge  = ( other._charge );
  _ptLeast = ( other._ptLeast );
  _pt2D    = ( other._pt2D );
  _pt3D    = ( other._pt3D );
  _pz      = ( other._pz );
  _p       = ( other._p );
 
  _d0      = other._d0;
  _omega   = other._omega;
  _phi0    = other._phi0;
  _tanl    = ( other._tanl );
  _z0      = ( other._z0 );
  _tanl_zs = ( other._tanl_zs );
  _z0_zs   = ( other._z0_zs );
 
  _centerPeak   = ( other._centerPeak );
  _Hough2D      = ( other._Hough2D );
  _Hough3D      = ( other._Hough3D );
  _bunchTime    = ( other._bunchTime );
  _centerX      = ( other._centerX );
  _centerY      = ( other._centerY );
  _centerR      = ( other._centerR );
  _chi2_aver    = ( other._chi2_aver );
  _nfit         = ( other._nfit );
  _chi2_aver2D  = ( other._chi2_aver2D );
  _nfit2D       = ( other._nfit2D );
  _recHitVec    = ( other._recHitVec );
  _stat2D       = ( other._stat2D );
  _stat3D       = ( other._stat3D );
  p_trk         = other.p_trk;
  p_trk2D       = other.p_trk2D;
  _maprho       = other._maprho;
  _maptheta     = other._maptheta;
  t_pro_correct = other.t_pro_correct;
  _houghList    = other._houghList;
  vec_mdcHit    = other.vec_mdcHit;
  //  if(_recHitVec.size() != 0 ){
  //    for(int i =0;i<_recHitVec.size();i++){
  //      delete _recHitVec[i];
  //      _recHitVec[i]=NULL;
  //    }
  //    _recHitVec.clear();
  //  }
  // for(int i =0;i<other._recHitVec.size();i++){
  //   //const HoughHit* p_hit= new HoughHit( *(other._recHitVec[i]));
  //   HoughRecHit p_hit ( (trackHitList[i])->digi(),0.,0. );
  //   _recHitVec.push_back(p_hit);
  // }
  return *this;
}

outerHit()

void HoughTrack::outerHit

(

)

Definition at line 207 of file HoughTrack.cxx.

                          {
  vector<int> vec_layer_num[43];
  for ( int ilay = 0; ilay < 43; ilay++ )
  {
    for ( int ihit = 0; ihit < _recHitVec.size(); ihit++ )
    {
      if ( _recHitVec[ihit].getLayerId() == ilay && _recHitVec[ihit].getflag() == 0 )
      { vec_layer_num[ilay].push_back( _recHitVec[ihit].getWireId() ); }
    }
    std::sort( vec_layer_num[ilay].begin(), vec_layer_num[ilay].end(), less_layer );
    // for(int j=0;j<vec_layer_num[ilay].size();j++)
    // cout<<"("<<ilay<<","<<vec_layer_num[ilay][j]<<")"<<endl;
    if ( vec_layer_num[ilay].size() < 4 ) continue;
    for ( int j = 0; j < vec_layer_num[ilay].size(); j++ )
    {
      //      if( (vec_layer_num[ilay][j]+1 == vec_layer_num[ilay][j+1]) &&
      //(vec_layer_num[ilay][j+1]+1 ==
      // vec_layer_num[ilay][j+2]) &&  (vec_layer_num[ilay][j+2]+1 == vec_layer_num[ilay][j+3])
      // ) {
      for ( int jhit = 0; jhit < _recHitVec.size(); jhit++ )
      {
        if ( ( ilay == _recHitVec[jhit].getLayerId() ) &&
             ( vec_layer_num[ilay][j] == _recHitVec[jhit].getWireId() ) )
        { _recHitVec[jhit].setflag( -1 ); }
      }
      //}
    }
  }
}

Referenced by fit3D(), and fit3D_inner().

print()

void HoughTrack::print

(

)

Definition at line 1105 of file HoughTrack.cxx.

                       {
  cout << "print HoughTrack : " << p_trk->id() << endl;
  cout << "par : " << _d0 << "," << _omega << "," << _phi0 << "," << _z0 << "," << _tanl
       << ", pt: " << _pt3D << endl;
  TrkHitList*              qhits   = p_trk->hits();
  TrkHotList::hot_iterator hotIter = qhits->hotList().begin();
  int                      lay     = ( (MdcHit*)( hotIter->hit() ) )->layernumber();
  int                      wir     = ( (MdcHit*)( hotIter->hit() ) )->wirenumber();
  while ( hotIter != qhits->hotList().end() )
  {
    cout << "(" << ( (MdcHit*)( hotIter->hit() ) )->layernumber() << ","
         << ( (MdcHit*)( hotIter->hit() ) )->wirenumber() << ":" << hotIter->isActive()
         << ")  ";
    cout << "nuse " << hotIter->hit()->nUsedHits() << endl;
 
    hotIter++;
  }
}

printRecHit()

void HoughTrack::printRecHit

(

)

const

Definition at line 881 of file HoughTrack.cxx.

                                   {
  cout << "print rec hit" << endl;
  double rho   = _centerPeak.getRho();
  double theta = _centerPeak.getTheta();
  int    size  = _recHitVec.size();
  for ( int i = 0; i < size; i++ )
  {
    int layer   = _recHitVec[i].getLayerId();
    int wire    = _recHitVec[i].getWireId();
    int slant   = _recHitVec[i].getSlayerType();
    int flag    = _recHitVec[i].getflag();
    int style   = _recHitVec[i].getStyle();
    int cirlist = _recHitVec[i].getCirList();
    if ( slant == 0 )
      std::cout << "axial hit (" << layer << "," << wire << ") "
                << _recHitVec[i].getDisToTrack() << " " << _recHitVec[i].getDisToCir() << " "
                << flag << " " << style << " " << cirlist << std::endl;
  }
  for ( int i = 0; i < size; i++ )
  {
    int layer   = _recHitVec[i].getLayerId();
    int wire    = _recHitVec[i].getWireId();
    int slant   = _recHitVec[i].getSlayerType();
    int flag    = _recHitVec[i].getflag();
    int style   = _recHitVec[i].getStyle();
    int cirlist = _recHitVec[i].getCirList();
    if ( slant != 0 )
      std::cout << "stereo hit (" << layer << "," << wire << ") "
                << _recHitVec[i].getDisToTrack() << " " << flag << " " << style << " "
                << cirlist << std::endl;
  }
}

Referenced by fit3D(), fit3D_inner(), and fitzs().

setCharge()

void HoughTrack::setCharge ( int charge )

inline

Definition at line 34 of file HoughTrack.h.

{ _charge = charge; }

Referenced by MdcHoughFinder::execute().

setHoughHitList()

void HoughTrack::setHoughHitList ( vector< HoughHit > vec_hit )

inline

Definition at line 66 of file HoughTrack.h.

{ _houghList = vec_hit; }

Referenced by MdcHoughFinder::execute().

setMdcHit()

void HoughTrack::setMdcHit ( const vector< MdcHit * > * mdchit )

inline

Definition at line 94 of file HoughTrack.h.

{ vec_mdcHit = mdchit; }

Referenced by MdcHoughFinder::execute().

sortHit()

void HoughTrack::sortHit

(

)

Definition at line 188 of file HoughTrack.cxx.

                         {
  std::sort( _recHitVec.begin(), _recHitVec.end(), digi_in_track );
}

Referenced by fit2D(), fit3D(), and fit3D_inner().

trackCharge2D()

int HoughTrack::trackCharge2D

(

)

Definition at line 914 of file HoughTrack.cxx.

                              {
  int n_neg( 0 );
  int n_pos( 0 );
  for ( int ihit = 0; ihit < _recHitVec.size(); ihit++ )
  {
    double xhit  = _recHitVec[ihit].getMidX();
    double yhit  = _recHitVec[ihit].getMidY();
    double x_cir = _Hough2D.getCirX();
    double y_cir = _Hough2D.getCirY();
    double r_cir = _Hough2D.getCirR();
    if ( ( x_cir * yhit - y_cir * xhit >= 0 ) ) n_neg++;
    if ( ( x_cir * yhit - y_cir * xhit <= 0 ) ) n_pos++;
  }
  if ( m_debug > 0 ) cout << " in track charge 2d " << n_neg << " " << n_pos << endl;
  if ( ( _charge == -1 && n_neg < 3 ) || ( _charge == 1 && n_pos < 3 ) ) return 0;
  else if ( _charge == 0 ) { cout << " wrong ! in track charge 3D not set charge " << endl; }
  else return 1;
 
  std::cerr << __FILE__ << ":" << __LINE__ << ": should not reach here" << std::endl;
  exit( 1 );
}

Referenced by MdcHoughFinder::execute().

trackCharge3D()

int HoughTrack::trackCharge3D

(

)

Definition at line 936 of file HoughTrack.cxx.

                              {
  int n_neg( 0 );
  int n_pos( 0 );
  for ( int ihit = 0; ihit < _recHitVec.size(); ihit++ )
  {
    if ( _recHitVec[ihit].getSlayerType() == 0 ) continue;
    if ( _recHitVec[ihit].getLayerId() > 7 ) continue;
    double xhit  = _recHitVec[ihit].getMidX();
    double yhit  = _recHitVec[ihit].getMidY();
    double x_cir = _Hough2D.getCirX();
    double y_cir = _Hough2D.getCirY();
    double r_cir = _Hough2D.getCirR();
    if ( ( x_cir * yhit - y_cir * xhit >= 0 ) ) n_neg++;
    if ( ( x_cir * yhit - y_cir * xhit <= 0 ) ) n_pos++;
  }
  if ( m_debug > 0 ) cout << " in track charge 3d " << n_neg << " " << n_pos << endl;
  if ( ( _charge == -1 && n_neg < 2 ) || ( _charge == 1 && n_pos < 2 ) ) return 0;
  else if ( _charge == 0 ) { cout << " wrong ! in track charge 3D not set charge " << endl; }
  else return 1;
 
  std::cerr << __FILE__ << ":" << __LINE__ << ": should not reach here" << std::endl;
  exit( 1 );
}

Referenced by MdcHoughFinder::execute().

Member Data Documentation

m_debug

bool HoughTrack::m_debug = 0

static

Definition at line 79 of file HoughTrack.h.

Referenced by calzs(), cut_axial_inner(), cutNoise_inner(), find_pair_hit(), find_stereo_hit(), fit3D(), fit3D_inner(), fitzs(), MdcHoughFinder::initialize(), trackCharge2D(), and trackCharge3D().

---

The documentation for this class was generated from the following files:

• HoughTrack.h
• HoughTrack.cxx