blame: source/code/listFiles.cpp

/*

===========================================================================

blockattack - Block Attack - Rise of the Blocks

Copyright (C) 2005-2012 Poul Sander

This program is free software: you can redistribute it and/or modify

it under the terms of the GNU General Public License as published by

the Free Software Foundation, either version 2 of the License, or

(at your option) any later version.

This program 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 General Public License for more details.

You should have received a copy of the GNU General Public License

along with this program.  If not, see http://www.gnu.org/licenses/

Source information and contacts persons can be found at

http://blockattack.sf.net

===========================================================================

*/

/*

listFiles.cpp

*/

#include "listFiles.h"

using namespace std;

void ListFiles::setDirectory(const string &directory)

{

	for (int i=0; i<MAX_NR_OF_FILES; i++) {

		filenames[i]="";

	}

#if defined(_WIN32)

	DWORD dwError;

	string directory2 = directory+"/*";

	hFind = FindFirstFile(directory2.c_str(), &FindFileData);

	if (hFind == INVALID_HANDLE_VALUE)

	{

		cout << "Invalid file handle. Error is " << GetLastError() << endl;

	}

	else

	{

		nrOfFiles=0;

		filenames[nrOfFiles] = FindFileData.cFileName;

		cout << "File: " << FindFileData.cFileName << endl;

		while ((FindNextFile(hFind, &FindFileData) != 0) && FindFileData.cFileName[0]!='.' && (nrOfFiles<MAX_NR_OF_FILES-1))

		{

			nrOfFiles++;

			filenames[nrOfFiles] = FindFileData.cFileName;

			cout << "File: " << FindFileData.cFileName << endl;

		}

		dwError = GetLastError();

		FindClose(hFind);

		if (dwError != ERROR_NO_MORE_FILES)

		{

			cout << "FindNextFile error. Error is " << dwError << endl;

		}

	}

#elif defined(__unix__)

	DIR *DirectoryPointer;

	struct dirent *dp;

	nrOfFiles=0;

	//cout << "Will look in: " << directory << endl;

	DirectoryPointer = opendir(directory.c_str());

	if(!DirectoryPointer) {

		return;

	}

	while ((dp=readdir(DirectoryPointer))&&(nrOfFiles<MAX_NR_OF_FILES-1))

	{

		string name = (string)(char*)dp->d_name;

		if (!isInList(name) && name[0]!='.' )

		{

			nrOfFiles++;

			filenames[nrOfFiles] = name;

		}

	}

	closedir(DirectoryPointer);

#endif

	startFileNr=FIRST_FILE;

	//Put code here

}

bool ListFiles::isInList(const string &name)

{

	for (int i=0; (i<=nrOfFiles); i++)

	{

		if (0==strcmp(name.c_str(),filenames[i].c_str()))

		{

			return true;

		}

	}

	return false;

}

void ListFiles::setDirectory2(const string &dic)

{

#if defined(__unix__)

	DIR *DirectoryPointer;

	struct dirent *dp;

	//cout << "Will look in: " << dic << endl;

	DirectoryPointer = opendir(dic.c_str());

	if(!DirectoryPointer) {

		return;

	}

	while ((dp=readdir(DirectoryPointer))&&(nrOfFiles<MAX_NR_OF_FILES-1))

	{

		string name = (string)(char*)dp->d_name;

		if (!isInList(name) && name != "." && name != "..")

		{

			nrOfFiles++;

			filenames[nrOfFiles] = name;

		}

	}

	closedir(DirectoryPointer);

	startFileNr=FIRST_FILE;

#endif

}

string ListFiles::getFileName(int nr)

{

	if (startFileNr+nr<MAX_NR_OF_FILES) {

		return filenames[startFileNr+nr];

	}

	else

	{

		return "";

	}

}

bool ListFiles::fileExists(int nr)

{

	string emptyString="";

	if (startFileNr+nr<MAX_NR_OF_FILES)

	{

		if (filenames[startFileNr+nr]==emptyString) {

			return false;

		}

		else {

			return true;

		}

	}

	else {

		return false;

	}

}

void ListFiles::back()

{

	if (startFileNr>FIRST_FILE) {

		startFileNr = startFileNr-10;

	}

	if (startFileNr<FIRST_FILE) {

		startFileNr = FIRST_FILE;

	}

}

void ListFiles::forward()

{

	if (startFileNr<nrOfFiles-FIRST_FILE) {

		startFileNr = startFileNr+10;

	}

}

string ListFiles::getRandom()

{

	int numberOfFiles = nrOfFiles-FIRST_FILE+1;

	if(numberOfFiles<1) {

		return "";

	}

	int select = rand()%numberOfFiles;

	return filenames[FIRST_FILE+select];

}