tron@2186: /* $Id$ */
tron@2186: 
truelight@0: #include "stdafx.h"
Darkvater@1891: #include "openttd.h"
tron@1093: #include "fileio.h"
tron@2163: #include "functions.h"
tron@2159: #include "macros.h"
tron@2153: #include "variables.h"
truelight@810: #if defined(UNIX) || defined(__OS2__)
truelight@0: #include <ctype.h> // required for tolower()
truelight@0: #endif
truelight@0: 
truelight@0: /*************************************************/
truelight@0: /* FILE IO ROUTINES ******************************/
truelight@0: /*************************************************/
truelight@0: 
truelight@0: #define FIO_BUFFER_SIZE 512
truelight@0: 
truelight@0: typedef struct {
truelight@0: 	byte *buffer, *buffer_end;
truelight@0: 	uint32 pos;
truelight@0: 	FILE *cur_fh;
truelight@0: 	FILE *handles[32];
truelight@0: 	byte buffer_start[512];
truelight@0: } Fio;
truelight@0: 
truelight@0: static Fio _fio;
truelight@0: 
truelight@0: // Get current position in file
tron@1093: uint32 FioGetPos(void)
truelight@0: {
truelight@0: 	return _fio.pos + (_fio.buffer - _fio.buffer_start) - FIO_BUFFER_SIZE;
truelight@0: }
truelight@0: 
truelight@0: void FioSeekTo(uint32 pos, int mode)
truelight@0: {
truelight@0: 	if (mode == SEEK_CUR) pos += FioGetPos();
truelight@0: 	_fio.buffer = _fio.buffer_end = _fio.buffer_start + FIO_BUFFER_SIZE;
truelight@0: 	fseek(_fio.cur_fh, (_fio.pos=pos), SEEK_SET);
truelight@0: }
truelight@0: 
truelight@0: // Seek to a file and a position
truelight@0: void FioSeekToFile(uint32 pos)
truelight@0: {
truelight@0: 	FILE *f = _fio.handles[pos >> 24];
truelight@0: 	assert(f != NULL);
truelight@0: 	_fio.cur_fh = f;
truelight@0: 	FioSeekTo(pos & 0xFFFFFF, SEEK_SET);
truelight@0: }
truelight@0: 
tron@1093: byte FioReadByte(void)
truelight@0: {
truelight@0: 	if (_fio.buffer == _fio.buffer_end) {
truelight@0: 		_fio.pos += FIO_BUFFER_SIZE;
truelight@0: 		fread(_fio.buffer = _fio.buffer_start, 1, FIO_BUFFER_SIZE, _fio.cur_fh);
truelight@0: 	}
truelight@0: 	return *_fio.buffer++;
truelight@0: }
truelight@0: 
truelight@0: void FioSkipBytes(int n)
truelight@0: {
truelight@0: 	for(;;) {
truelight@0: 		int m = min(_fio.buffer_end - _fio.buffer, n);
truelight@0: 		_fio.buffer += m;
truelight@0: 		n -= m;
truelight@0: 		if (n == 0) break;
truelight@0: 		FioReadByte();
truelight@0: 		n--;
truelight@0: 	}
truelight@0: }
truelight@0: 
truelight@0: 
tron@1093: uint16 FioReadWord(void)
truelight@0: {
truelight@0: 	byte b = FioReadByte();
truelight@0: 	return (FioReadByte() << 8) | b;
truelight@0: }
truelight@0: 
tron@1093: uint32 FioReadDword(void)
truelight@0: {
truelight@0: 	uint b = FioReadWord();
truelight@0: 	return (FioReadWord() << 16) | b;
truelight@0: }
truelight@0: 
truelight@0: void FioReadBlock(void *ptr, uint size)
truelight@0: {
truelight@0: 	FioSeekTo(FioGetPos(), SEEK_SET);
truelight@0: 	_fio.pos += size;
truelight@0: 	fread(ptr, 1, size, _fio.cur_fh);
truelight@0: }
truelight@0: 
darkvater@1039: static inline void FioCloseFile(int slot)
darkvater@1039: {
darkvater@1039: 	if (_fio.handles[slot] != NULL) {
tron@1109: 		fclose(_fio.handles[slot]);
darkvater@1039: 		_fio.handles[slot] = NULL;
darkvater@1039: 	}
darkvater@1039: }
darkvater@1039: 
darkvater@1036: void FioCloseAll(void)
truelight@0: {
truelight@0: 	int i;
truelight@0: 
darkvater@1039: 	for (i = 0; i != lengthof(_fio.handles); i++)
darkvater@1039: 		FioCloseFile(i);
truelight@0: }
truelight@0: 
dominik@1198: bool FiosCheckFileExists(const char *filename)
dominik@1198: {
dominik@1198: 	FILE *f;
dominik@1198: 	char buf[MAX_PATH];
dominik@1198: 
dominik@1198: 	sprintf(buf, "%s%s", _path.data_dir, filename);
dominik@1198: 
dominik@1198: 	f = fopen(buf, "rb");
dominik@1198: #if !defined(WIN32)
dominik@1198: 	if (f == NULL) {
dominik@1198: 		char *s;
dominik@1198: 		// Make lower case and try again
dominik@1198: 		for(s=buf + strlen(_path.data_dir) - 1; *s != 0; s++)
dominik@1198: 			*s = tolower(*s);
dominik@1198: 		f = fopen(buf, "rb");
dominik@1198: 
dominik@1198: #if defined SECOND_DATA_DIR
dominik@1198: 	// tries in the 2nd data directory
dominik@1198: 		if (f == NULL) {
dominik@1198: 			sprintf(buf, "%s%s", _path.second_data_dir, filename);
dominik@1198: 			for(s=buf + strlen(_path.second_data_dir) - 1; *s != 0; s++)
dominik@1198: 			*s = tolower(*s);
dominik@1198: 		f = fopen(buf, "rb");
dominik@1198: 		}
dominik@1198: #endif
dominik@1198: 	}
dominik@1198: #endif
dominik@1198: 
dominik@1198: 	if (f == NULL)
dominik@1198: 		return false;
dominik@1198: 	else {
dominik@1198: 		fclose(f);
dominik@1198: 		return true;
dominik@1198: 	}
dominik@1198: }
dominik@1198: 
bjarni@2736: FILE *FioFOpenFile(const char *filename)
bjarni@2736: {
bjarni@2736: 	FILE *f;
bjarni@2736: 	char buf[MAX_PATH];
bjarni@2736: 
bjarni@2736: 	sprintf(buf, "%s%s", _path.data_dir, filename);
bjarni@2736: 
bjarni@2736: 	f = fopen(buf, "rb");
bjarni@2736: #if !defined(WIN32)
bjarni@2736: 	if (f == NULL) {
bjarni@2736: 		char *s;
bjarni@2736: 		// Make lower case and try again
bjarni@2736: 		for(s=buf + strlen(_path.data_dir) - 1; *s != 0; s++)
bjarni@2736: 			*s = tolower(*s);
bjarni@2736: 		f = fopen(buf, "rb");
bjarni@2736: 
bjarni@2736: #if defined SECOND_DATA_DIR
bjarni@2736: 		// tries in the 2nd data directory
bjarni@2736: 		if (f == NULL) {
bjarni@2736: 			sprintf(buf, "%s%s", _path.second_data_dir, filename);
bjarni@2736: 			for(s=buf + strlen(_path.second_data_dir) - 1; *s != 0; s++)
bjarni@2736: 				*s = tolower(*s);
bjarni@2736: 			f = fopen(buf, "rb");
bjarni@2736: 		}
bjarni@2736: #endif
bjarni@2736: 	}
bjarni@2736: #endif
bjarni@2736: 
bjarni@2736: 	return f;
bjarni@2736: }
bjarni@2736: 
truelight@0: void FioOpenFile(int slot, const char *filename)
truelight@0: {
truelight@0: 	FILE *f;
truelight@0: 	char buf[MAX_PATH];
truelight@0: 
truelight@0: 	sprintf(buf, "%s%s", _path.data_dir, filename);
truelight@193: 
truelight@0: 	f = fopen(buf, "rb");
truelight@0: #if !defined(WIN32)
truelight@0: 	if (f == NULL) {
truelight@0: 		char *s;
truelight@0: 		// Make lower case and try again
truelight@0: 		for(s=buf + strlen(_path.data_dir) - 1; *s != 0; s++)
truelight@0: 			*s = tolower(*s);
truelight@0: 		f = fopen(buf, "rb");
tron@915: 
bjarni@561: #if defined SECOND_DATA_DIR
bjarni@561: 	// tries in the 2nd data directory
bjarni@561: 		if (f == NULL) {
bjarni@561: 			sprintf(buf, "%s%s", _path.second_data_dir, filename);
bjarni@561: 			for(s=buf + strlen(_path.second_data_dir) - 1; *s != 0; s++)
bjarni@561: 			*s = tolower(*s);
bjarni@561: 		f = fopen(buf, "rb");
bjarni@561: 		}
bjarni@2188: 
bjarni@2188: 	if (f == NULL)
bjarni@2188: 		sprintf(buf, "%s%s", _path.data_dir, filename);	//makes it print the primary datadir path instead of the secundary one
bjarni@2188: 
bjarni@561: #endif
truelight@0: 	}
truelight@0: #endif
truelight@0: 
truelight@0: 	if (f == NULL)
truelight@0: 		error("Cannot open file '%s'", buf);
truelight@193: 
darkvater@1039: 	FioCloseFile(slot); // if file was opened before, close it
truelight@0: 	_fio.handles[slot] = f;
truelight@0: 	FioSeekToFile(slot << 24);
truelight@0: }