mevaluation_answers.hpp

/*
 * Copyright (C) 2012-2015, Juan Manuel Barrios <juanmanuel@barrios.cl>
 * All rights reserved.
 *
 * This file is part of P-VCD. https://github.com/juanbarrios/p-vcd
 * P-VCD is made available under the terms of the BSD 2-Clause License.
 */

#ifndef MKNN_EVALUATION_ANSWERS_HPP
#define MKNN_EVALUATION_ANSWERS_HPP

#include <map>
#include <string>
#include <set>
#include <vector>
#include "../metricknn_impl.hpp"

namespace mknn {
template<typename TypeObject> class MknnRankedAnswer {
public:
    MknnRankedAnswer();
    MknnRankedAnswer(TypeObject id_object, long long rank_position);
    ~MknnRankedAnswer();
    const TypeObject& getIdObject() const;
    long long getRank() const;
    bool operator <(const MknnRankedAnswer& other) const;

private:
    TypeObject id_object;
    //rank starting from 0
    long long rank_position;
};

template<typename TypeQuery, typename TypeObject> class MknnAnswers {
public:
    MknnAnswers();
    ~MknnAnswers();
    void addAnswer(TypeQuery id_query, TypeObject id_object,
            long long rank_position);
    long long getNumRanks();
    std::set<TypeQuery> getQueryNames();
    std::vector<MknnRankedAnswer<TypeObject>> &getQueryAnswers(
            TypeQuery id_query);
private:
    std::map<TypeQuery, std::vector<MknnRankedAnswer<TypeObject>>>answers;
};

typedef MknnRankedAnswer<std::string> MknnRankedAnswer_s;
typedef MknnAnswers<std::string, std::string> MknnAnswers_s;

}
#endif