TrackConnection.h

#ifndef TRACKCONNECTION_H
#define TRACKCONNECTION_H 1

#include <lcio.h>
#include <string>
#include <map>

#include <EVENT/TrackerHit.h>
#include <IMPL/TrackerHitImpl.h>

namespace marlintpc
{

class TrackConnection
{

    public:

        TrackConnection();

        ~TrackConnection();

        struct UsedFlag : lcrtrel::LCIntExtension<UsedFlag> {} ;

        void mergeTracks(IMPL::TrackImpl*& mergedTrack, std::multimap<int, IMPL::TrackImpl*> &tracksByModuleID);

        void setConeAngle(double angle);

    private:

        std::multimap<int, IMPL::TrackImpl*> *_tracksByModuleID;

        bool connectSegment(IMPL::TrackImpl* mergedTrack, std::multimap<int, IMPL::TrackImpl*>::iterator nextSegment);

        void connect(IMPL::TrackImpl* mergedTrack, std::multimap<int, IMPL::TrackImpl*>::iterator nextSegment);

#if 0   // 2D
        class Compare
        {
            public:
                bool operator()(EVENT::TrackerHit* left,  EVENT::TrackerHit* right)
                {
                    const double* xv1 = left->getPosition();
                    const double* xv2 = right->getPosition();
                    double r12 = xv1[0] * xv1[0] + xv1[1] * xv1[1];
                    double r22 = xv2[0] * xv2[0] + xv2[1] * xv2[1];
                    return r12 < r22;
                }
        };

#else   // 3D
        class Compare
        {
            public:
                bool operator()(EVENT::TrackerHit* left,  EVENT::TrackerHit* right)
                {
                    const double* xv1 = left->getPosition();
                    const double* xv2 = right->getPosition();
                    double r12 = xv1[0] * xv1[0] + xv1[1] * xv1[1] + xv1[2] * xv1[2];
                    double r22 = xv2[0] * xv2[0] + xv2[1] * xv2[1] + xv2[2] * xv2[2];
                    return r12 < r22;
                }
        };
#endif

        const double* _x1;
        const double* _x3;

        double _coneAngle;

};

}// namespace marlintpc

#endif // TRACKCONNECTION_H