filecounter.cpp

/////////////////////////////////////////////////////////////////////////////////
// Author:      Steven Lamerton
// Copyright:   Copyright (C) 2009 Steven Lamerton
// License:     GNU GPL 2 (See readme for more info)
/////////////////////////////////////////////////////////////////////////////////

#include "filecounter.h"
#include "path.h"
#include "toucan.h"
#include <wx/dir.h>
#include <wx/log.h>

FileCounter::FileCounter(){
	m_Count = 0;
}

void FileCounter::AddPath(const wxString &path){
    m_Paths.Add(Path::Normalise(path));
}

void FileCounter::AddPaths(const wxArrayString &paths){
	for(unsigned int i = 0; i < paths.Count(); i++){
        m_Paths.Add(Path::Normalise(paths.Item(i)));
	}
}

long FileCounter::GetCount() const{
	return m_Count;
}

//ATTN : Potentially replace some of this with a wxDirTraverser
bool FileCounter::Count(){
	for(unsigned int i = 0; i < m_Paths.Count(); i++){
		if(wxDirExists(m_Paths.Item(i))){
			CountFolder(m_Paths.Item(i));		
		}
		else{
			m_Count++;
		}

	}
	return true;
}

bool FileCounter::CountFolder(wxString path){
	if(wxGetApp().GetAbort()){
		return true;
	}
	wxGetApp().Yield();
	if (path[path.length()-1] != wxFILE_SEP_PATH) {
		path += wxFILE_SEP_PATH;       
	}
	wxDir dir(path);
	wxString strFilename;
	//Suppress warnings about folders we cannot enumerate
	wxLogNull null;
	bool blDir = dir.GetFirst(&strFilename);
	if(blDir){
		do {
			if(wxDirExists(path + strFilename)){
				CountFolder(path + strFilename);
			}
			else{
				m_Count++;
			}
		}
		while(dir.GetNext(&strFilename));
	} 
	return true;
}