DirectoryIterator_unix.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 "ar_track_alvar/DirectoryIterator_private.h"

#include <dirent.h>

namespace alvar {

class DirectoryIteratorPrivateData
{
public:
    DirectoryIteratorPrivateData()
        : mHandle(NULL)
        , mData(NULL)
    {
    }
    
    DIR *mHandle;
    dirent *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()
{
    closedir(d->mHandle);
    delete d;
}

bool DirectoryIteratorPrivate::hasNext()
{
    if (d->mHandle == NULL) {
        d->mHandle = opendir(mDirectory.data());

        if (d->mHandle != NULL) {
            d->mData = readdir(d->mHandle);
            
            if (d->mData != NULL) {
                mValid = true;
                skip();
            }
        }
    }

    return mValid;
}

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

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

    d->mData = readdir(d->mHandle);
    if (d->mData == NULL) {
        mValid = false;
    }
    else {
        skip();
    }

    return mEntry;
}

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

        d->mData = readdir(d->mHandle);
        if (d->mData == NULL) {
            mValid = false;
            return;
        }
    }
}

} // namespace alvar