/*******************************************************************************
 * Copyright 上海赜睿信息科技有限公司(Zilliz) - All Rights Reserved
 * Unauthorized copying of this file, via any medium is strictly prohibited.
 * Proprietary and confidential.
 ******************************************************************************/
#pragma once

#include "ExecutionEngine.h"

#include <memory>
#include <string>

namespace faiss {
    class Index;
}

namespace zilliz {
namespace vecwise {
namespace engine {


class FaissExecutionEngine : public ExecutionEngine {
public:

    FaissExecutionEngine(uint16_t dimension,
            const std::string& location,
            const std::string& build_index_type,
            const std::string& raw_index_type);

    FaissExecutionEngine(std::shared_ptr<faiss::Index> index,
            const std::string& location,
            const std::string& build_index_type,
            const std::string& raw_index_type);

    Status AddWithIds(long n, const float *xdata, const long *xids) override;

    size_t Count() const override;

    size_t Size() const override;

    size_t Dimension() const override;

    size_t PhysicalSize() const override;

    Status Serialize() override;

    Status Load() override;

    Status Merge(const std::string& location) override;

    Status Search(long n,
                  const float *data,
                  long k,
                  float *distances,
                  long *labels) const override;

    ExecutionEnginePtr BuildIndex(const std::string&) override;

    Status Cache() override;

protected:
    std::shared_ptr<faiss::Index> pIndex_;
    std::string location_;

    std::string build_index_type_;
    std::string raw_index_type_;
};


} // namespace engine
} // namespace vecwise
} // namespace zilliz