IO.cpp

   1 #include <sys/stat.h>
   2 #include <unistd.h>
   3 #include <sys/stat.h>
   4 #include "IO.hpp"
   5
   6
   7 #ifdef _WIN32
   8
   9 bool IO::MkDir( const std::string& dir )
  10 {
  11         return CreateDirectory( dir.c_str(), NULL ) != 0;
  12 }
  13
  14 bool IO::RmDir( const std::string& dir )
  15 {
  16         return RemoveDirectory( dir.c_str() ) != 0;
  17 }
  18
  19 Directory::Directory()
  20         : m_handle( INVALID_HANDLE_VALUE )
  21 {
  22 }
  23
  24 Directory::~Directory()
  25 {
  26         if( m_handle != INVALID_HANDLE_VALUE )
  27         {
  28                 FindClose( m_handle );
  29         }
  30 }
  31
  32 bool Directory::Open( const std::string& dir )
  33 {
  34         m_path = dir + "\\*";
  35         m_handle = FindFirstFile( m_path.c_str(), &m_ffd );
  36
  37         return m_handle != INVALID_HANDLE_VALUE;
  38 }
  39
  40 std::string Directory::GetName()
  41 {
  42         return m_ffd.cFileName;
  43 }
  44
  45 Directory& Directory::operator++()
  46 {
  47         if( FindNextFile( m_handle, &m_ffd ) == 0 )
  48         {
  49                 FindClose( m_handle );
  50                 m_handle = INVALID_HANDLE_VALUE;
  51         }
  52         return *this;
  53 }
  54
  55 Directory::operator bool()
  56 {
  57         return m_handle != INVALID_HANDLE_VALUE;
  58 }
  59
  60 #else
  61
  62 bool IO::MkDir( const std::string& dir )
  63 {
  64         return mkdir( dir.c_str(), S_IRWXU | S_IRGRP | S_IXGRP | S_IROTH | S_IXOTH ) == 0;
  65 }
  66
  67 bool IO::RmDir( const std::string& dir )
  68 {
  69         return rmdir( dir.c_str() ) == 0;
  70 }
  71
  72 Directory::Directory()
  73         : m_dir( NULL )
  74 {
  75 }
  76
  77 Directory::~Directory()
  78 {
  79         closedir( m_dir );
  80 }
  81
  82 bool Directory::Open( const std::string& dir )
  83 {
  84         m_dir = opendir( dir.c_str() );
  85         m_dirent = readdir( m_dir );
  86
  87         return m_dir != NULL;
  88 }
  89
  90 std::string Directory::GetName()
  91 {
  92         return m_dirent->d_name;
  93 }
  94
  95 Directory& Directory::operator++()
  96 {
  97         m_dirent = readdir( m_dir );
  98         return *this;
  99 }
 100
 101 Directory::operator bool()
 102 {
 103         return m_dirent != NULL;
 104 }
 105
 106 #endif