PtPartitioningIndex.h

#ifndef PtPartitioningIndex_h
#define PtPartitioningIndex_h

#include <vector>

#include "ClusteringComparisonIndex.h"

namespace stab {
    class PtPartitioningIndex : public ClusteringComparisonIndex
    {
    public:
        inline virtual ~PtPartitioningIndex() {}

        double index(const PseudoJet* inputParticles,
                     const unsigned* clustering1,
                     const unsigned* clustering2,
                     unsigned size) const;
    private:
        virtual double processInterClusterMatrix(const double* data,
                                                 unsigned nrows,
                                                 unsigned ncols,
                                                 double ptSum,
                                                 double ptSumsq) const = 0;

        unsigned max_element( const unsigned * clu, unsigned size ) const;

        mutable std::vector<double> storage_;
    };

    class BestPtMatchIndex : public PtPartitioningIndex
    {
        double processInterClusterMatrix(const double* data,
                                         unsigned nrows,
                                         unsigned ncols,
                                         double ptSum,
                                         double ptSumsq) const;
        struct ElementSort
        {
            double ptsum;
            unsigned row;
            unsigned col;

            inline bool operator<(const ElementSort& r) const
            {
                return ptsum > r.ptsum;
            }
        };

        mutable std::vector<ElementSort> storage_;
        mutable std::vector<int> intstorage_;
    };


    class TransverseRandIndex2 : public PtPartitioningIndex
    {
        double processInterClusterMatrix(const double* data,
                                         unsigned nrows,
                                         unsigned ncols,
                                         double ptSum,
                                         double ptSumsq) const;
        mutable std::vector<long double> storage_;
    };
}

#endif // PtPartitioningIndex_h