/home/bes3soft/bes3soft/Boss/7.0.2/dist/7.0.2/Reconstruction/TrkReco/TrkReco-00-08-59-patch4-slc6tag/src/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/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);
    }
}

---