Builder.hpp

Go to the documentation of this file.

#ifndef KNNCOLLE_BUILDER_HPP
#define KNNCOLLE_BUILDER_HPP
 
#include "Prebuilt.hpp"
#include "Matrix.hpp"
#include <memory>
#include <utility>
#include <type_traits>
 
namespace knncolle {
 
template<typename Index_, typename Data_, typename Distance_, class Matrix_ = Matrix<Index_, Data_> >
class Builder {
public:
    // Rule of 5 all of this.
    Builder() = default;
    Builder(Builder&&) = default;
    Builder(const Builder&) = default;
    Builder& operator=(Builder&&) = default;
    Builder& operator=(const Builder&) = default;
    virtual ~Builder() = default;
 
    static_assert(std::is_same<decltype(std::declval<Matrix_>().num_observations()), Index_>::value);
    static_assert(std::is_same<typename std::remove_pointer<decltype(std::declval<Matrix_>().new_extractor()->next())>::type, const Data_>::value);
public:
    virtual Prebuilt<Index_, Data_, Distance_>* build_raw(const Matrix_& data) const = 0;
 
    std::shared_ptr<Prebuilt<Index_, Data_, Distance_> > build_shared(const Matrix_& data) const {
        return std::shared_ptr<Prebuilt<Index_, Data_, Distance_> >(build_raw(data));
    }
 
    std::unique_ptr<Prebuilt<Index_, Data_, Distance_> > build_unique(const Matrix_& data) const {
        return std::unique_ptr<Prebuilt<Index_, Data_, Distance_> >(build_raw(data));
    }
};
 
}
 
#endif