Source/cmDirectory.cxx

   1 /*=========================================================================
   2
   3   Program:   Insight Segmentation & Registration Toolkit
   4   Module:    $RCSfile: cmDirectory.cxx,v $
   5   Language:  C++
   6   Date:      $Date: 2002-03-13 15:25:07 $
   7   Version:   $Revision: 1.6 $
   8
   9   Copyright (c) 2002 Insight Consortium. All rights reserved.
  10   See ITKCopyright.txt or http://www.itk.org/HTML/Copyright.htm for details.
  11
  12      This software is distributed WITHOUT ANY WARRANTY; without even
  13      the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
  14      PURPOSE.  See the above copyright notices for more information.
  15
  16 =========================================================================*/
  17 #include "cmDirectory.h"
  18
  19
  20
  21 // First microsoft compilers
  22
  23 #ifdef _MSC_VER
  24 #include <windows.h>
  25 #include <io.h>
  26 #include <ctype.h>
  27 #include <fcntl.h>
  28 #include <stdio.h>
  29 #include <stdlib.h>
  30 #include <string.h>
  31 #include <sys/stat.h>
  32 #include <sys/types.h>
  33
  34 /**
  35  *
  36  */
  37 bool
  38 cmDirectory
  39 ::Load(const char* name)
  40 {
  41   char* buf;
  42   size_t n = strlen(name);
  43   if ( name[n - 1] == '/' )
  44     {
  45     buf = new char[n + 1 + 1];
  46     sprintf(buf, "%s*", name);
  47     }
  48   else
  49     {
  50     buf = new char[n + 2 + 1];
  51     sprintf(buf, "%s/*", name);
  52     }
  53   struct _finddata_t data;      // data of current file
  54
  55   // Now put them into the file array
  56   size_t srchHandle = _findfirst(buf, &data);
  57   delete [] buf;
  58
  59   if ( srchHandle == -1 )
  60     {
  61     return 0;
  62     }
  63
  64   // Loop through names
  65   do
  66     {
  67     m_Files.push_back(data.name);
  68     }
  69   while ( _findnext(srchHandle, &data) != -1 );
  70   m_Path = name;
  71   return _findclose(srchHandle) != -1;
  72 }
  73
  74 #else
  75
  76 // Now the POSIX style directory access
  77
  78 #include <sys/types.h>
  79 #include <dirent.h>
  80
  81 /**
  82  *
  83  */
  84 bool
  85 cmDirectory
  86 ::Load(const char* name)
  87 {
  88   DIR* dir = opendir(name);
  89
  90   if (!dir)
  91     {
  92     return 0;
  93     }
  94
  95   for (dirent* d = readdir(dir); d; d = readdir(dir) )
  96     {
  97     m_Files.push_back(d->d_name);
  98     }
  99   m_Path = name;
 100   closedir(dir);
 101   return 1;
 102 }
 103
 104 #endif
 105
 106
 107 /**
 108  *
 109  */
 110 const char*
 111 cmDirectory
 112 ::GetFile(size_t index)
 113 {
 114   if ( index >= m_Files.size() )
 115     {
 116     cmSystemTools::Error("Bad index for GetFile on cmDirectory\n", 0);
 117     return 0;
 118     }
 119   return m_Files[index].c_str();
 120 }
 121