Community Core Audio (Alpha)

#include "ofxDirList.h"

#include <algorithm>

#include <string>

// Handy string functions

static std::string::size_type idx;

static string getExt(string filename){

    idx = filename.rfind('.');

    if(idx != std::string::npos){

        return filename.substr(idx+1);

    }

    else{

        return "";

    }

}

static string strToLower(string myStr){

    transform( myStr.begin(), myStr.end(), myStr.begin(), ::tolower );

    return myStr;

}

//----------------------------------------------------------

ofxDirList::ofxDirList(){

    reset();

}

//----------------------------------------------------------

void ofxDirList::reset(){

        allowedFileExt.clear();

        nameArray.clear();

        pathArray.clear();

}

//----------------------------------------------------------

void ofxDirList::setVerbose(bool _verbose){

    ofLog(OF_LOG_WARNING, "ofxDirList setVerbose is depreciated use ofSetLogLevel instead");

}

//----------------------------------------------------------

bool ofxDirList::allowExt(string ext){

        allowedFileExt.push_back( strToLower(ext) );

        return true;

}

//----------------------------------------------------------

string ofxDirList::getName(int pos){

        if(pos >= (int)nameArray.size()) return 0;

        return nameArray[pos];

}

//----------------------------------------------------------

string ofxDirList::getPath(int pos){

        if(pos >= (int)pathArray.size()) return 0;

        return pathArray[pos];

}

//----------------------------------------------------------

int ofxDirList::listDir(string directory){

    directory = ofToDataPath(directory);

        nameArray.clear();

        pathArray.clear();

    if(directory.length() <= 0)return 0;

    //if the trailing slash was not added - then add it

        if( directory[directory.length()-1] != '/'){

        directory = directory + "/";

        }

        DIR *dir = NULL;

        struct dirent *entry;

    //open the directory

    ofLog(OF_LOG_VERBOSE, "ofxDirList - attempting to open %s", directory.c_str());

    dir = opendir(directory.c_str());

        if(dir == NULL){

                ofLog(OF_LOG_ERROR, "ofxDirList - error opening directory");

                return 0;

        }else{

                ofLog(OF_LOG_VERBOSE, "ofxDirList - success opening directory");

        }

    string entry_name = "";

    string ext = "";

        bool skip = false;

        while ((entry = readdir(dir)) != NULL){

        //turn it into a C++ string

        entry_name = entry->d_name;

        //lets get the length of the string here as we query it again

        int fileLen = entry_name.length();

                if(fileLen <= 0)continue; //if the name is not existant

                if(entry_name[0] == '.')continue; //ignore invisible files, ./ and ../

                //by default we don't skip files unless we are checking extensions

                skip = false;

                if(allowedFileExt.size() > 0){

                        //we will skip this files unless it has an allowed extension

                        skip = true;

                        for(int i = 0; i < (int)allowedFileExt.size(); i++){

                                //if the wildecard * has been entered for an ext type then don't check any extensions

                                if( allowedFileExt[i] == "*"){ skip = false; break; }

                                int extLen = allowedFileExt[i].length();

                                //the extension has to be shorter than the filename - simple check

                                if(extLen >= fileLen) continue;

                //lets get the ext as lowercase

                ext = strToLower( getExt(entry_name) );

                //if no ext - then skip this ext check

                if( ext == "" )continue;

                //if we find a match then stop checking and approve this file

                if(ext == allowedFileExt[i]){

                    skip = false;

                    break;

                }

                        }

                }

                if(skip) continue;

                //finally we store the result

        pathArray.push_back(directory + entry_name);

        nameArray.push_back(entry_name);

                ofLog(OF_LOG_VERBOSE, "ofxDirList - listing %s ", nameArray.back().c_str());

        }

        ofLog(OF_LOG_VERBOSE, "ofxDirList - listed %i files in %s", nameArray.size(), directory.c_str());

        return nameArray.size();

}