122 lines
2.1 KiB
C++
122 lines
2.1 KiB
C++
/*=========================================================================
|
|
|
|
Program: Insight Segmentation & Registration Toolkit
|
|
Module: $RCSfile$
|
|
Language: C++
|
|
Date: $Date$
|
|
Version: $Revision$
|
|
|
|
Copyright (c) 2002 Insight Consortium. All rights reserved.
|
|
See ITKCopyright.txt or http://www.itk.org/HTML/Copyright.htm for details.
|
|
|
|
This software is distributed WITHOUT ANY WARRANTY; without even
|
|
the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
|
|
PURPOSE. See the above copyright notices for more information.
|
|
|
|
=========================================================================*/
|
|
#include "cmDirectory.h"
|
|
|
|
|
|
|
|
// First microsoft compilers
|
|
|
|
#ifdef _MSC_VER
|
|
#include <windows.h>
|
|
#include <io.h>
|
|
#include <ctype.h>
|
|
#include <fcntl.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <sys/stat.h>
|
|
#include <sys/types.h>
|
|
|
|
/**
|
|
*
|
|
*/
|
|
bool
|
|
cmDirectory
|
|
::Load(const char* name)
|
|
{
|
|
char* buf;
|
|
size_t n = strlen(name);
|
|
if ( name[n - 1] == '/' )
|
|
{
|
|
buf = new char[n + 1 + 1];
|
|
sprintf(buf, "%s*", name);
|
|
}
|
|
else
|
|
{
|
|
buf = new char[n + 2 + 1];
|
|
sprintf(buf, "%s/*", name);
|
|
}
|
|
struct _finddata_t data; // data of current file
|
|
|
|
// Now put them into the file array
|
|
size_t srchHandle = _findfirst(buf, &data);
|
|
delete [] buf;
|
|
|
|
if ( srchHandle == -1 )
|
|
{
|
|
return 0;
|
|
}
|
|
|
|
// Loop through names
|
|
do
|
|
{
|
|
m_Files.push_back(data.name);
|
|
}
|
|
while ( _findnext(srchHandle, &data) != -1 );
|
|
m_Path = name;
|
|
return _findclose(srchHandle) != -1;
|
|
}
|
|
|
|
#else
|
|
|
|
// Now the POSIX style directory access
|
|
|
|
#include <sys/types.h>
|
|
#include <dirent.h>
|
|
|
|
/**
|
|
*
|
|
*/
|
|
bool
|
|
cmDirectory
|
|
::Load(const char* name)
|
|
{
|
|
DIR* dir = opendir(name);
|
|
|
|
if (!dir)
|
|
{
|
|
return 0;
|
|
}
|
|
|
|
for (dirent* d = readdir(dir); d; d = readdir(dir) )
|
|
{
|
|
m_Files.push_back(d->d_name);
|
|
}
|
|
m_Path = name;
|
|
closedir(dir);
|
|
return 1;
|
|
}
|
|
|
|
#endif
|
|
|
|
|
|
/**
|
|
*
|
|
*/
|
|
const char*
|
|
cmDirectory
|
|
::GetFile(size_t index)
|
|
{
|
|
if ( index >= m_Files.size() )
|
|
{
|
|
cmSystemTools::Error("Bad index for GetFile on cmDirectory\n", 0);
|
|
return 0;
|
|
}
|
|
return m_Files[index].c_str();
|
|
}
|
|
|