Builder.hpp

Go to the documentation of this file.

#ifndef KNNCOLLE_BUILDER_HPP
#define KNNCOLLE_BUILDER_HPP
 
#include "Prebuilt.hpp"
#include <memory>
 
namespace knncolle {
 
template<class Matrix_, typename Float_>
class Builder {
public:
    virtual ~Builder() = default;
    virtual Prebuilt<typename Matrix_::dimension_type, typename Matrix_::index_type, Float_>* build_raw(const Matrix_& data) const = 0;
 
    std::shared_ptr<Prebuilt<typename Matrix_::dimension_type, typename Matrix_::index_type, Float_> > build_shared(const Matrix_& data) const {
        return std::shared_ptr<Prebuilt<typename Matrix_::dimension_type, typename Matrix_::index_type, Float_> >(build_raw(data));
    }
 
    std::unique_ptr<Prebuilt<typename Matrix_::dimension_type, typename Matrix_::index_type, Float_> > build_unique(const Matrix_& data) const {
        return std::unique_ptr<Prebuilt<typename Matrix_::dimension_type, typename Matrix_::index_type, Float_> >(build_raw(data));
    }
};
 
}
 
#endif