TFastFinder.cxx

Go to the documentation of this file.

//-----------------------------------------------------------------------------
// $Id: TFastFinder.cxx,v 1.7 2010/03/31 09:58:59 liucy Exp $
//-----------------------------------------------------------------------------
// Filename : TFastFinder.cc
// Section  : Tracking
// Owner    : Yoshi Iwasaki
// Email    : yoshihito.iwasaki@kek.jp
//-----------------------------------------------------------------------------
// Description : A class to find tracks with the conformal method.
//               See http://bsunsrv1.kek.jp/~yiwasaki/tracking/
//-----------------------------------------------------------------------------
 
#include "TrkReco/TFastFinder.h"
#include "TrkReco/TConformalFinder0.h"
#include "TrkReco/THistogram.h"
#include "TrkReco/TMDCUtil.h"
#include "TrkReco/TMLink.h"
#include "TrkReco/TTrack.h"
 
TFastFinder::TFastFinder() : _builder( "fast find builder", 30 ) {
  _selector.nLinks( 4 );
  _selector.nSuperLayers( 2 );
  _selector.minPt( 0.05 );
  _selector.maxImpact( 100. );
  _selector.maxSigma( 30. );
  _selector.nLinksStereo( 3 );
  _selector.maxDistance( 30. );
  _builder.trackSelector( _selector );
}
 
TFastFinder::~TFastFinder() {}
 
std::string TFastFinder::version( void ) const { return "0.00"; }
 
void TFastFinder::dump( const std::string& msg, const std::string& pre ) const {
  std::cout << pre;
  TFinderBase::dump( msg );
}
 
void TFastFinder::clear( void ) {
  _axialHits.removeAll();
  _stereoHits.removeAll();
  HepAListDeleteAll( _axialLinks );
  HepAListDeleteAll( _stereoLinks );
}
 
int TFastFinder::doit( const AList<TMDCWireHit>& axialHits,
                       const AList<TMDCWireHit>& stereoHits, AList<TTrack>& tracks ) {
 
  //...Select good hits...
  selectHits2( axialHits, stereoHits );
 
  //...Conformal transformation with IP constraint...
  TConformalFinder0::conformalTransformationRphi( ORIGIN, _axialHits, _axialLinks );
  TConformalFinder0::conformalTransformationRphi( ORIGIN, _stereoHits, _stereoLinks );
 
  //...Make a histogram...
  THistogram hist( 288 );
  hist.fillX( _axialLinks );
  AList<TSegment0> clusters = hist.clusters0();
 
  //...Cluster loop...
  unsigned n = clusters.length();
  for ( unsigned i = 0; i < n; i++ )
  {
 
    //...2D track...
    TTrack* t = _builder.buildRphi( clusters[i]->links() );
    if ( t == NULL ) continue;
#ifdef TRKRECO_DEBUG_DETAIL
    std::cout << "TFastFinder::doit ... 2D track found" << std::endl;
#endif
 
    //...Make it 3D...
    TTrack* ts = t;
    ts         = _builder.buildStereo( *t, findCloseHits( _stereoLinks, *t ) );
    if ( ts == NULL ) continue;
#ifdef TRKRECO_DEBUG_DETAIL
    std::cout << "TFastFinder::doit ... 3D track found" << std::endl;
#endif
 
    //...OK...
    t->assign( WireHitFastFinder );
    t->finder( TrackFastFinder );
    //  t->assign(WireHitFastFinder, TrackFastFinder);
    tracks.append( t );
    _stereoLinks.remove( t->links() );
  }
 
  //...Termination...
  HepAListDeleteAll( clusters );
  return 0;
}
 
void TFastFinder::selectHits( const AList<TMDCWireHit>& axialHits,
                              const AList<TMDCWireHit>& stereoHits ) {
  unsigned n = axialHits.length();
  for ( unsigned i = 0; i < n; i++ )
  {
    const TMDCWireHit& h = *axialHits[i];
    if ( ( h.state() & WireHitIsolated ) && ( h.state() & WireHitContinuous ) )
      _axialHits.append( (TMDCWireHit&)h );
  }
  n = stereoHits.length();
  for ( unsigned i = 0; i < n; i++ )
  {
    const TMDCWireHit& h = *stereoHits[i];
    if ( ( h.state() & WireHitIsolated ) && ( h.state() & WireHitContinuous ) )
      _stereoHits.append( (TMDCWireHit&)h );
  }
}
 
void TFastFinder::selectHits2( const AList<TMDCWireHit>& axialHits,
                               const AList<TMDCWireHit>& stereoHits ) {
  selectSimpleSegments( axialHits, _axialHits );
  selectSimpleSegments( stereoHits, _stereoHits );
}
 
AList<TMLink> TFastFinder::findCloseHits( const AList<TMLink>& links,
                                          const TTrack&        track ) const {
  //
  // Coded by J.Suzuki
  //
  AList<TMLink> list;
 
  //...Check condition...
  if ( track.links().length() == 0 )
  {
#ifdef TRKRECO_DEBUG_DETAIL
    std::cout << "TConformalFinder::findCloseHits !!! ";
    std::cout << " no links found in a track : This should not be happened";
    std::cout << std::endl;
#endif
 
    return list;
  }
 
  //...Parameters...
  //    float dRcut[11] = {0, 3.5, 0., 5.5, 0., 6.5, 0., 7.5, 0., 9.5, 0.};
  float dRcut[11] = { 0., 4.3, 0., 6.5, 0., 7.5, 0., 8.0, 0., 9.5, 0. };
 
  //...Select Stereo hits associated to the current r-phi curve...
  double   R0         = track.helix().curv();
  double   xInnerWire = track.links()[0]->wire()->xyPosition().x();
  double   yInnerWire = track.links()[0]->wire()->xyPosition().y();
  unsigned nall       = links.length();
  for ( unsigned j = 0; j < nall; j++ )
  {
    TMLink&         t     = *links[j];
    const TMDCWire& w     = *t.wire();
    HepVector3D     X     = w.xyPosition() - track.helix().center();
    double          Rmag2 = X.mag2();
    double          DR    = fabs( sqrt( Rmag2 ) - fabs( R0 ) );
    t.zStatus( -10 );
    t.zPair( 0 );
    if ( DR < dRcut[w.superLayerId()] &&
         ( xInnerWire * w.xyPosition().x() + yInnerWire * w.xyPosition().y() ) > 0. )
    { list.append( t ); }
  }
 
  return list;
}
 
void TFastFinder::selectSimpleSegments( const AList<TMDCWireHit>& in,
                                        AList<TMDCWireHit>&       out ) const {
  AList<TMDCWireHit> hits = in;
  while ( hits.last() )
  {
    TMDCWireHit& h = *hits.last();
 
    //...Start clustering...
    // AList<TMDCWireHit> & cluster = * new AList<TMDCWireHit>();
    AList<TMDCWireHit> cluster;
    AList<TMDCWireHit> toBeChecked;
    bool               ok = true;
    toBeChecked.append( h );
    while ( toBeChecked.length() )
    {
      TMDCWireHit& a = *toBeChecked.last();
      toBeChecked.remove( a );
      if ( cluster.hasMember( a ) ) continue;
 
      //...Check hit...
      unsigned state = a.state();
      if ( !( state & WireHitIsolated ) ) ok = false;
      if ( !( state & WireHitContinuous ) ) ok = false;
 
      //...Append...
      cluster.append( a );
 
      //...Neighbor hit...
      unsigned ptn = ( state & WireHitNeighborPattern ) >> 12;
      for ( unsigned i = 0; i < 6; i++ )
      {
        if ( ( ptn >> i ) % 2 )
        {
          const TMDCWireHit& b = *a.wire()->neighbor( i )->hit();
          toBeChecked.append( (TMDCWireHit&)b );
        }
      }
    }
 
    //...Check cluster size...
    if ( ( cluster.length() < 4 ) || ( cluster.length() > 8 ) ) ok = false;
 
    //...OK. Good hits...
    if ( ok ) out.append( cluster );
 
    //...Remove cluster...
    hits.remove( cluster );
 
    //...For debug...
#ifdef TRKRECO_DEBUG_DETAIL
    std::cout << "TFastFinder::selectSimpleSegment ... cluster : ok = ";
    std::cout << ok << " : ";
#endif
    for ( unsigned i = 0; i < cluster.length(); i++ )
    {
      TMDCWireHit& h = *cluster[i];
#ifdef TRKRECO_DEBUG_DETAIL
      std::cout << h.wire()->name() << ",";
#endif
      if ( !ok )
      {
        unsigned state = h.state();
        if ( state & WireHitIsolated ) state ^= WireHitIsolated;
        if ( state & WireHitContinuous ) state ^= WireHitContinuous;
        h.state( state );
      }
    }
#ifdef TRKRECO_DEBUG_DETAIL
    std::cout << std::endl;
#endif
 
  remove:
    hits.remove( h );
  }
}