image_info_impl.hpp

Go to the documentation of this file.

#ifndef MLPACK_CORE_DATA_IMAGE_INFO_IMPL_HPP
#define MLPACK_CORE_DATA_IMAGE_INFO_IMPL_HPP

#ifdef HAS_STB // Compile this only if stb is present.

// In case it hasn't been included yet.
#include "image_info.hpp"

namespace mlpack {
namespace data {

static const std::vector<std::string> loadFileTypes({"jpg", "png", "tga",
    "bmp", "psd", "gif", "hdr", "pic", "pnm", "jpeg"});

static const std::vector<std::string> saveFileTypes({"jpg", "png", "tga",
    "bmp", "hdr"});

inline bool ImageFormatSupported(const std::string& fileName, const bool save)
{
  if (save)
  {
    // Iterate over all supported file types that can be saved.
    for (auto extension : saveFileTypes)
    {
      if (extension == Extension(fileName))
        return true;
    }
  }
  else
  {
    // Iterate over all supported file types that can be loaded.
    for (auto extension : loadFileTypes)
    {
      if (extension == Extension(fileName))
        return true;
    }
  }

  return false;
}

} // namespace data
} // namespace mlpack

#endif // HAS_STB.

namespace mlpack {
namespace data {

inline ImageInfo::ImageInfo(const size_t width,
                            const size_t height,
                            const size_t channels,
                            const size_t quality) :
    width(width),
    height(height),
    channels(channels),
    quality(quality)
{
  // Do nothing.
}

} // namespace data
} // namespace mlpack

#endif