#ifndef CT_HOUGHTRANSFORM_H
#define CT_HOUGHTRANSFORM_H

#include "ct_shapedata/ct_linedata.h"
#include "ct_shape2ddata/ct_circle2ddata.h"

#include "eigen/Eigen/Core"
#include <QMap>
#include <QVector>

#include "pluginShared_global.h"


class PLUGINSHAREDSHARED_EXPORT CT_HoughTransform
{
public:

    struct HoughObject
    {
        HoughObject(const QVector<double> &coords, void* object)
        {
            _houghCoords.resize(coords.size());

            for (int i = 0 ; i < coords.size() ; i++)
            {
                _houghCoords[i] = coords.at(i);
            }
            _object = object;
        }

        double operator()(int idx) const
        {
            if (idx < 0 || idx >= _houghCoords.size()) {return 0;}
            return _houghCoords[idx];
        }

        double &operator[](int idx)
        {
            return _houghCoords[idx];
        }

        int size() const {return _houghCoords.size();}

        QVector<double> _houghCoords;
        void*           _object;
    };


    static void sortHoughtObjects(QList<CT_HoughTransform::HoughObject*> &objects, double minDist, QMultiMap<double, CT_HoughTransform::HoughObject*> &scoreMap);

    static void getHoughtDistances(QList<CT_HoughTransform::HoughObject *> &objects, double maxDist, QMultiMap<double, QPair<CT_HoughTransform::HoughObject*, CT_HoughTransform::HoughObject*>* > &outDistMap);

    static double findBestHoughtObjects(QList<CT_HoughTransform::HoughObject *> &objects, double minDist, CT_HoughTransform::HoughObject* bestObject);

    static void createHoughObjectsFromVerticalLines(QList<CT_LineData*> &lines, double referenceHeight, QList<CT_HoughTransform::HoughObject*> &outObjects);

    static void createHoughObjectsFromHorizontalCircles(QList<CT_Circle2DData*> &circles, double radiusWeight, QList<CT_HoughTransform::HoughObject*> &outObjects);

};

#endif // CT_HOUGHTRANSFORM_H