ClusteringComparison.h

#ifndef ClusteringComparison_h
#define ClusteringComparison_h

#include "ClusteringComparisonIndex.h"

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

        double index(const PseudoJet* inputParticles,
                     const unsigned* clustering1,
                     const unsigned* clustering2,
                     unsigned size) const;
    private:
        // c1 -- weighted sum of pairs assigned to the same cluster
        //       in both clusterings.
        // c2 -- weighted sum of pairs assigned to different clusters
        //       in both clusterings.
        // c3 -- weighted sum of pairs assigned to the same cluster
        //       in clustering 1 but to different clusters in clustering 2.
        // c4 -- weighted sum of pairs assigned to the same cluster
        //       in clustering 2 but to different clusters in clustering 1.
        //
        virtual double combineCounts(double c1, double c2,
                                     double c3, double c4) const = 0;
    };

    class TransverseRandIndex : public TransversePairwiseIndex
    {
        inline double combineCounts(const double c1, const double c2,
                                    const double c3, const double c4) const
        {
            return (c1 + c2)/(c1 + c2 + c3 + c4);
        }
    };

    class TransverseJaccardIndex : public TransversePairwiseIndex
    {
        inline double combineCounts(const double c1, const double c2,
                                    const double c3, const double c4) const
        {
            return c1/(c1 + c3 + c4);
        }
    };    
}

#endif // ClusteringComparison_h