framework/delibs/decpp/deDirectoryIterator.cpp

/*-------------------------------------------------------------------------
 * drawElements Quality Program Tester Core
 * ----------------------------------------
 *
 * Copyright 2014 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *
 *//*!
 * \file
 * \brief Directory iterator.
 *//*--------------------------------------------------------------------*/

#include "deDirectoryIterator.hpp"
#include "deString.h"

#if (DE_DIRITER == DE_DIRITER_WIN32)
#	include <direct.h> /* _chdir() */
#	include <io.h> /* _findfirst(), _findnext() */
#endif

namespace de
{

#if (DE_DIRITER == DE_DIRITER_WIN32)

DirectoryIterator::DirectoryIterator (const FilePath& path)
	: m_path(FilePath::normalize(path))
{
	DE_CHECK_RUNTIME_ERR(m_path.exists());
	DE_CHECK_RUNTIME_ERR(m_path.getType() == FilePath::TYPE_DIRECTORY);

	m_handle	= _findfirst32((std::string(m_path.getPath()) + "/*").c_str(), &m_fileInfo);
	m_hasItem	= m_handle != -1;

	skipCurAndParent();
}

DirectoryIterator::~DirectoryIterator (void)
{
	if (m_handle != -1)
		_findclose(m_handle);
}

bool DirectoryIterator::hasItem (void) const
{
	return m_hasItem;
}

FilePath DirectoryIterator::getItem (void) const
{
	DE_ASSERT(hasItem());
	return FilePath::join(m_path, m_fileInfo.name);
}

void DirectoryIterator::next (void)
{
	m_hasItem = (_findnext32(m_handle, &m_fileInfo) == 0);
	skipCurAndParent();
}

void DirectoryIterator::skipCurAndParent (void)
{
	while (m_hasItem && (deStringEqual(m_fileInfo.name, "..") || deStringEqual(m_fileInfo.name, ".")))
		m_hasItem = (_findnext32(m_handle, &m_fileInfo) == 0);
}

#elif (DE_DIRITER == DE_DIRITER_POSIX)

DirectoryIterator::DirectoryIterator (const FilePath& path)
	: m_path	(FilePath::normalize(path))
	, m_handle	(DE_NULL)
	, m_curEntry(DE_NULL)
{
	DE_CHECK_RUNTIME_ERR(m_path.exists());
	DE_CHECK_RUNTIME_ERR(m_path.getType() == FilePath::TYPE_DIRECTORY);

	m_handle = opendir(m_path.getPath());
	DE_CHECK_RUNTIME_ERR(m_handle);

	// Find first entry
	next();
}

DirectoryIterator::~DirectoryIterator (void)
{
	closedir(m_handle);
}

bool DirectoryIterator::hasItem (void) const
{
	return (m_curEntry != DE_NULL);
}

FilePath DirectoryIterator::getItem (void) const
{
	DE_ASSERT(hasItem());
	return FilePath::join(m_path, m_curEntry->d_name);
}

void DirectoryIterator::next (void)
{
	do
	{
		m_curEntry = readdir(m_handle);
	} while (m_curEntry && (deStringEqual(m_curEntry->d_name, "..") || deStringEqual(m_curEntry->d_name, ".")));
}

#endif

} // de