CheckExtension.h

Go to the documentation of this file.

#pragma once

#include <algorithm>
#include <string>

namespace CortidQCT {

namespace IO {

inline std::string extension(std::string const &filename,
                             bool toLower = false) {

  std::string extension = "";
  if (auto const dotPos = filename.find_last_of('.');
      dotPos != std::string::npos) {
    extension = filename.substr(dotPos + 1);
  }

  if (toLower) {
    std::transform(begin(extension), end(extension), begin(extension), tolower);
  }
  return extension;
}

template <class Extensions>
bool checkExtensions(std::string const &filename, Extensions &&extensions) {
  using std::begin;
  using std::end;
  using std::find;

  auto const ext = extension(filename, true);

  return find_if(begin(extensions), end(extensions), [&ext](auto const &rhs) {
           std::string query = rhs;
           std::transform(begin(query), end(query), begin(query), tolower);
           return query == ext;
         }) != end(extensions);
}

} // namespace IO

} // namespace CortidQCT