filesystem.cpp

 
#include "filesystem.h"
 
#include <cstddef>
#include <filesystem>
 
#include "dlinear/util/logging.h"
 
namespace dlinear {
 
std::string GetExtension(const std::string &name) {
  const std::size_t idx = name.rfind('.');
  if (idx == std::string::npos) return "";
  return name.substr(idx + 1);
}
 
std::vector<std::string> SplitStringByWhitespace(const char *in) {
  std::vector<std::string> r;
  for (const char *p = in; *p; ++p) {
    while (*p == ' ' || *p == '\t' || *p == '\r') ++p;
    if (*p == '\0') break;
    int length = 0;
    while (p[length] != ' ' && p[length] != '\t' && p[length] != '\r' && p[length]) ++length;
    r.emplace_back(p, length);
    p += length - 1;
  }
  return r;
}
 
std::vector<std::string> GetFiles(const std::string &path, const std::string &extension) {
  std::vector<std::string> files;
  for (const std::filesystem::directory_entry &entry : std::filesystem::directory_iterator(path)) {
    if (!extension.empty() && entry.path().extension() != extension) continue;
    files.emplace_back(entry.path());
  }
  return files;
}
 
}  // namespace dlinear