vil_image_list.cxx

Go to the documentation of this file.

#include <cstdlib>
#include "vil_image_list.h"
//:
// \file
#include <sys/stat.h>
#ifdef _MSC_VER
#  include <vcl_msvc_warnings.h>
#endif
#include <vil/vil_image_resource.h>
#include <vil/vil_blocked_image_resource.h>
#include <vil/vil_pyramid_image_resource.h>
#include <vil/vil_load.h>

static bool il_verbose = false;
bool vil_image_list::vil_is_directory(char const* fn)
{
  struct stat fs;
  return stat(fn, &fs) == 0 && (fs.st_mode & S_IFMT) == S_IFDIR;
}

#if defined(_WIN32) && !defined(__CYGWIN__)
#include <io.h>
std::vector<std::string> vil_image_list::files()
{
  std::vector<std::string> temp;
  if (!this->vil_is_directory(directory_.c_str()))
    return temp;
  // This mess should go away soon.
# if defined __MINGW32__
  typedef long handle_type;      // works with msvc6
# else
  typedef intptr_t handle_type;  // not found by msvc6
#endif

  handle_type handle;
  struct _finddata_t data;
  handle = _findfirst((directory_+"\\*").c_str(), &data);
  if (handle<0)
    return temp;
  std::string s = data.name;
  std::string filename = directory_+ "\\" + s;
  vil_image_resource_sptr resc;
  if (!this->vil_is_directory(filename.c_str()))
    temp.push_back(filename);
  while ( true )
  {
    if (_findnext(handle, &data) != 0) {
      _findclose(handle);
      return temp;
    }
    s = data.name;
    filename = directory_+ "\\" + s;
    if (!this->vil_is_directory(filename.c_str()))
      temp.push_back(filename);
  }

  return temp;
}
#else // !defined(_WIN32) || defined(__CYGWIN__)

#include <dirent.h>
std::vector<std::string> vil_image_list::files()
{
  std::vector<std::string> temp;
  if (!this->vil_is_directory(directory_.c_str()))
    return temp;
  DIR* dir_handle = opendir(directory_.c_str());
  dirent* de;
  de = readdir(dir_handle);
  if (de==nullptr)
    return temp;
  std::string s = de->d_name;
  std::string filename = directory_+ "/" + s;
  if (!this->vil_is_directory(filename.c_str()))
  {
#ifdef IL_DEBUG
    std::cout << "Found File(0) " << filename << '\n';
#endif
    temp.push_back(filename);
  }
  while ( true )
  {
    de = readdir(dir_handle);
    if (de == nullptr) {
      closedir(dir_handle);
      return temp;
    }
    s = de->d_name;
    filename = directory_+ "/" + s;
    if (!this->vil_is_directory(filename.c_str()))
    {
#ifdef IL_DEBUG
      std::cout << "Found File " << filename << '\n';
#endif
      temp.push_back(filename);
    }
  }
  return temp;
}

#endif // !defined(_WIN32) || defined(__CYGWIN__)
std::vector<vil_image_resource_sptr> vil_image_list::resources()
{
  std::vector<vil_image_resource_sptr>  temp;
  std::vector<std::string> fs = this->files();
  for (auto & f : fs)
  {
    vil_image_resource_sptr resc = vil_load_image_resource(f.c_str(), il_verbose);
    if (resc)
      temp.push_back(resc);
  }
  return temp;
}

std::vector<vil_image_resource_sptr> vil_image_list::blocked_resources()
{
  std::vector<vil_image_resource_sptr>  temp;
  std::vector<std::string> fs = this->files();
  for (auto & f : fs)
  {
    vil_image_resource_sptr resc = vil_load_image_resource(f.c_str(), il_verbose);
    vil_image_resource_sptr bir = blocked_image_resource(resc).ptr();
    if (bir)
      temp.push_back(bir);
  }
  return temp;
}

std::vector<vil_image_resource_sptr> vil_image_list::pyramids()
{
  std::vector<vil_image_resource_sptr>  temp;
  std::vector<std::string> fs = this->files();
  for (auto & f : fs)
  {
    vil_pyramid_image_resource_sptr pyr =
      vil_load_pyramid_resource(f.c_str(), il_verbose);
    if (pyr)
      temp.emplace_back(pyr.ptr());
  }
  return temp;
}
//:remove a file
bool vil_image_list::remove_file(std::string& filename)
{
#if defined(_WIN32) && !defined(__CYGWIN__)
  std::string command = "del " + filename;
#else
  std::string command = "rm " + filename;
#endif
  return std::system(command.c_str())==0;
}

//:removes all files from the directory. sub-directories are not touched
bool vil_image_list::clean_directory()
{
  std::vector<std::string> filez = this->files();
  bool good = true;
  std::cout << "starting to remove ..\n";
  for (auto & fit : filez)
    if (!this->remove_file(fit))
      good = false;
  std::cout << "finished remove ..\n";
  return good;
}