utils.hpp

Go to the documentation of this file.

#ifndef MNNCORRECT_UTILS_HPP
#define MNNCORRECT_UTILS_HPP
 
#include <vector>
#include <algorithm>
#include <memory>
#include <cstddef>
 
#include "knncolle/knncolle.hpp"
 
#ifndef MNNCORRECT_CUSTOM_PARALLEL
#include "subpar/subpar.hpp"
#endif
 
namespace mnncorrect {
 
typedef std::size_t BatchIndex;
 
enum class MergePolicy : char { INPUT, SIZE, VARIANCE, RSS };
 
template<typename Task_, class Run_>
void parallelize(int num_workers, Task_ num_tasks, Run_ run_task_range) {
#ifndef MNNCORRECT_CUSTOM_PARALLEL
    // Methods could allocate or throw, so nothrow_ = false is safest.
    subpar::parallelize_range<false>(num_workers, num_tasks, std::move(run_task_range));
#else
    MNNCORRECT_CUSTOM_PARALLEL(num_workers, num_tasks, run_task_range);
#endif
}
 
namespace internal {
 
template<typename Index_, typename Distance_>
using NeighborSet = std::vector<std::vector<std::pair<Index_, Distance_> > >;
 
template<typename Index_, typename Float_>
struct Corrected {
    Corrected() = default;
    Corrected(std::unique_ptr<knncolle::Prebuilt<Index_, Float_, Float_> > index, std::vector<Index_> ids) : index(std::move(index)), ids(std::move(ids)) {}
    std::unique_ptr<knncolle::Prebuilt<Index_, Float_, Float_> > index;
    std::vector<Index_> ids;
};
 
template<typename Index_, typename Float_>
struct BatchInfo {
    Index_ offset, num_obs;
    std::unique_ptr<knncolle::Prebuilt<Index_, Float_, Float_> > index;
    std::vector<Corrected<Index_, Float_> > extras;
};
 
}
}
 
#endif