openmw/components/file_finder/file_finder.hpp

#ifndef FILE_FINDER_MAIN_H
#define FILE_FINDER_MAIN_H

#include "search.hpp"
#include "filename_less.hpp"
#include <map>

namespace FileFinder
{

class FileFinder
{
  std::map<std::string, std::string, path_less> table;

  struct Inserter : ReturnPath
  {
    FileFinder *owner;
    int cut;

    void add(const boost::filesystem::path &pth)
    {
      std::string file = pth.file_string();
      std::string key = file.substr(cut);
      owner->table[key] = file;
    }
  };

  Inserter inserter;

public:
  FileFinder(const boost::filesystem::path &path, bool recurse=true)
  {
    inserter.owner = this;

    // Remember the original path length, so we can cut it away from
    // the relative paths used as keys
    std::string pstring = path.file_string();
    inserter.cut = pstring.size();

    // If the path does not end in a slash, then boost will add one
    // later, which means one more character we have to remove.
    char last = pstring[pstring.size()-1];
    if(last != '\\' && last != '/')
      inserter.cut++;

    // Fill the map
    find(path, inserter, recurse);
  }

  bool has(const std::string& file) const
  {
    return table.find(file) != table.end();
  }

  // Find the full path from a relative path.
  const std::string &lookup(const std::string& file) const
  {
    return table.find(file)->second;
  }
};
}
#endif