DirTraverser.h

Go to the documentation of this file.

// -*- c-basic-offset: 4 -*-
 
#include <wx/dir.h>
#include <wx/filename.h>
 
class DirTraverser : public wxDirTraverser
{
public:
    DirTraverser():wxDirTraverser() { }
 
    //Called when directory traverser evaluates a file
    virtual wxDirTraverseResult OnFile(const wxString& file)
    {
        wxFileName fileName(file);
        wxString ext = fileName.GetExt();
        //we add all project files to array
        if (ext.CmpNoCase("pto") == 0 || ext.CmpNoCase("ptp") == 0||
                ext.CmpNoCase("pts") == 0|| ext.CmpNoCase("oto") == 0)
        {
            projectFiles.Add(file);
        }
 
        //TO-DO: include image file heuristics to detect potential projects
        return wxDIR_CONTINUE;
    }
 
    //Called when directory traverser evaluates a directory
    virtual wxDirTraverseResult OnDir(const wxString& WXUNUSED(dir))
    {
        return wxDIR_CONTINUE;
    }
 
    //Returns an array with all project files found
    wxArrayString GetProjectFiles()
    {
        return projectFiles;
    }
private:
    wxArrayString projectFiles;
};