1

/* $Id$ */

     2

     3

#include "stdafx.h"

     4

#include "openttd.h"

     5

#include "fileio.h"

     6

#include "functions.h"

     7

#include "string.h"

     8

#include "macros.h"

     9

#include "variables.h"

    11

    12

/*************************************************/

    13

/* FILE IO ROUTINES ******************************/

    14

/*************************************************/

    15

    16

#define FIO_BUFFER_SIZE 512

    18

    19

typedef struct {

    20

	byte *buffer, *buffer_end;          ///< position pointer in local buffer and last valid byte of buffer

    21

	uint32 pos;                         ///< current (system) position in file

    22

	FILE *cur_fh;                       ///< current file handle

    24

	byte buffer_start[FIO_BUFFER_SIZE]; ///< local buffer when read from file

    30

} Fio;

    31

    32

static Fio _fio;

    33

    34

// Get current position in file

    35

uint32 FioGetPos(void)

    36

{

    37

	return _fio.pos + (_fio.buffer - _fio.buffer_start) - FIO_BUFFER_SIZE;

    38

}

    39

    40

void FioSeekTo(uint32 pos, int mode)

    41

{

    42

	if (mode == SEEK_CUR) pos += FioGetPos();

    43

	_fio.buffer = _fio.buffer_end = _fio.buffer_start + FIO_BUFFER_SIZE;

    44

	_fio.pos = pos;

    45

	fseek(_fio.cur_fh, _fio.pos, SEEK_SET);

    46

}

    47

    60

// Seek to a file and a position

    61

void FioSeekToFile(uint32 pos)

    62

{

    69

	assert(f != NULL);

    70

	_fio.cur_fh = f;

    71

	FioSeekTo(GB(pos, 0, 24), SEEK_SET);

    72

}

    73

    74

byte FioReadByte(void)

    75

{

    76

	if (_fio.buffer == _fio.buffer_end) {

    77

		_fio.pos += FIO_BUFFER_SIZE;

    78

		fread(_fio.buffer = _fio.buffer_start, 1, FIO_BUFFER_SIZE, _fio.cur_fh);

    79

	}

    80

	return *_fio.buffer++;

    81

}

    82

    83

void FioSkipBytes(int n)

    84

{

    85

	for (;;) {

    86

		int m = min(_fio.buffer_end - _fio.buffer, n);

    87

		_fio.buffer += m;

    88

		n -= m;

    89

		if (n == 0) break;

    90

		FioReadByte();

    91

		n--;

    92

	}

    93

}

    94

    95

uint16 FioReadWord(void)

    96

{

    97

	byte b = FioReadByte();

    98

	return (FioReadByte() << 8) | b;

    99

}

   100

   101

uint32 FioReadDword(void)

   102

{

   103

	uint b = FioReadWord();

   104

	return (FioReadWord() << 16) | b;

   105

}

   106

   107

void FioReadBlock(void *ptr, uint size)

   108

{

   109

	FioSeekTo(FioGetPos(), SEEK_SET);

   110

	_fio.pos += size;

   111

	fread(ptr, 1, size, _fio.cur_fh);

   112

}

   113

   114

static inline void FioCloseFile(int slot)

   115

{

   116

	if (_fio.handles[slot] != NULL) {

   117

		fclose(_fio.handles[slot]);

   118

		_fio.handles[slot] = NULL;

   122

	}

   123

}

   124

   125

void FioCloseAll(void)

   126

{

   127

	int i;

   128

   129

	for (i = 0; i != lengthof(_fio.handles); i++)

   130

		FioCloseFile(i);

   131

}

   132

   133

bool FioCheckFileExists(const char *filename)

   134

{

   135

	FILE *f = FioFOpenFile(filename);

   136

	if (f == NULL) return false;

   137

   138

	fclose(f);

   139

	return true;

   140

}

   141

   166

FILE *FioFOpenFile(const char *filename)

   167

{

   168

	FILE *f;

   169

	char buf[MAX_PATH];

   170

   171

	snprintf(buf, lengthof(buf), "%s%s", _paths.data_dir, filename);

   172

   173

	f = fopen(buf, "rb");

   174

#if !defined(WIN32)

   175

	if (f == NULL) {

   176

		strtolower(buf + strlen(_paths.data_dir) - 1);

   177

		f = fopen(buf, "rb");

   178

   179

#if defined SECOND_DATA_DIR

   180

		// tries in the 2nd data directory

   181

		if (f == NULL) {

   182

			snprintf(buf, lengthof(buf), "%s%s", _paths.second_data_dir, filename);

   183

			strtolower(buf + strlen(_paths.second_data_dir) - 1);

   184

			f = fopen(buf, "rb");

   185

		}

   186

#endif

   187

	}

   188

#endif

   189

   190

	return f;

   191

}

   192

   193

void FioOpenFile(int slot, const char *filename)

   194

{

   196

   201

	if (f == NULL) error("Cannot open file '%s%s'", _paths.data_dir, filename);

   202

   203

	FioCloseFile(slot); // if file was opened before, close it

   204

	_fio.handles[slot] = f;

   210

	FioSeekToFile(slot << 24);

   211

}