CatIDSearch.h

Go to the documentation of this file.

#pragma once
 
namespace krotos
{
 
class CatIDSearch
{
 
  public:
    CatIDSearch(std::string root);
    ~CatIDSearch();
 
    std::string extractCatID(std::string path);
    std::string findSimilar(std::string path);
    std::vector<std::string> findAllSimilar(std::string path);
 
  private:
    bool addDatasetItem(std::string path);
 
    std::mt19937 generator;
    std::map<std::string, std::vector<std::string>> dataset;
    std::set<std::string> filenames;
};
 
} // namespace krotos