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    : [email protected]
//-----------------------------------------------------------------------------
// 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/TMLink.h"
#include "TrkReco/THistogram.h"
#include "TrkReco/TMDCUtil.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);
    }
}