DirectoryIterator_win.cpp

Go to the documentation of this file.

/*
 * This file is part of ALVAR, A Library for Virtual and Augmented Reality.
 *
 * Copyright 2007-2012 VTT Technical Research Centre of Finland
 *
 * Contact: VTT Augmented Reality Team <alvar.info@vtt.fi>
 *          <http://www.vtt.fi/multimedia/alvar.html>
 *
 * ALVAR is free software; you can redistribute it and/or modify it under the
 * terms of the GNU Lesser General Public License as published by the Free
 * Software Foundation; either version 2.1 of the License, or (at your option)
 * any later version.
 *
 * This library is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License
 * for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with ALVAR; if not, see
 * <http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html>.
 */

#include "DirectoryIterator_private.h"

#include <windows.h>

namespace alvar {

class DirectoryIteratorPrivateData
{
public:
    DirectoryIteratorPrivateData()
        : mHandle(INVALID_HANDLE_VALUE)
        , mData()
    {
    }
    
    HANDLE mHandle;
    WIN32_FIND_DATA mData;
};

DirectoryIteratorPrivate::DirectoryIteratorPrivate(const std::string &path)
    : d(new DirectoryIteratorPrivateData())
    , mDirectory(path)
    , mEntry()
    , mValid(false)
{
    if (mDirectory.at(mDirectory.length() - 1) != '\\') {
        mDirectory.append("\\");
    }
}

DirectoryIteratorPrivate::~DirectoryIteratorPrivate()
{
    FindClose(d->mHandle);
    delete d;
}

bool DirectoryIteratorPrivate::hasNext()
{
    if (d->mHandle == INVALID_HANDLE_VALUE) {
        std::string search = mDirectory + "*";
        d->mHandle = FindFirstFile(search.data(), &d->mData);

        if (d->mHandle != INVALID_HANDLE_VALUE) {
            mValid = true;
            skip();
        }
    }

    return mValid;
}

std::string DirectoryIteratorPrivate::next()
{
    if (!hasNext()) {
        return "";
    }

    mEntry = std::string(d->mData.cFileName);

    if (!FindNextFile(d->mHandle, &d->mData)) {
        mValid = false;
    }
    else {
        skip();
    }

    return mEntry;
}

void DirectoryIteratorPrivate::skip()
{
    while (true) {
        if (std::string(d->mData.cFileName) != "." && std::string(d->mData.cFileName) != "..") {
            return;
        }

        if (!FindNextFile(d->mHandle, &d->mData)) {
            mValid = false;
            return;
        }
    }
}

} // namespace alvar