Method batchSearchInTwoSteps

python/src/ngtpy.cpp:877–917

  }

  void batchSearchInTwoSteps(
    py::array_t<float> queries,
    BatchResults &results,
    size_t size
  ) {
    const py::buffer_info &qinfo = queries.request();
    const std::vector<long int> &qshape = qinfo.shape;
    auto nOfQueries = qshape[0];
    size_t dimension = qshape[1];
    auto pseudoDimension = QBG::Index::getQuantizer().property.dimension;
    auto *queryPtr = static_cast<float*>(qinfo.ptr);

    size	= size > 0 ? size : defaultNumOfSearchObjects;

    results.results.clear();
    results.resultList.clear();

    std::unique_ptr<float[]> qs(new float[queries.size() * pseudoDimension]);
#pragma omp parallel for
    for (int idx = 0; idx < nOfQueries; idx++) {
      float *qptr = queryPtr + idx * dimension;
      float *qsptr = &qs[idx * pseudoDimension];
      memset(qsptr + dimension, 0, sizeof(float) * (pseudoDimension - dimension));
      memcpy(qsptr, qptr, dimension * sizeof(float));
    }
    QBG::BatchSearchContainer sc;
    sc.setObjectVectors(&qs[0], nOfQueries, pseudoDimension);
    sc.setSize(size);
    sc.setRefinementExpansion(defaultResultExpansion);
    sc.setEpsilon(defaultEpsilon);
    sc.setBlobEpsilon(defaultBlobEpsilon);
    sc.setEdgeSize(defaultEdgeSize);
    sc.setNumOfProbes(defaultNumOfProbes);
#ifdef NGTQBG_FUNCTION_SELECTOR
    sc.functionSelector = defaultFunctionSelector;
#endif
    QBG::Index::searchInTwoSteps(sc);
    results.resultList = std::move(sc.getBatchResult());
    results.size = results.resultList.size();
    return;
  }

  void parallelSearchInOneStep(
    py::array_t<float> queries,