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"

    10

    11

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

    12

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

    13

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

    14

    15

#define FIO_BUFFER_SIZE 512

    16

    17

typedef struct {

    23

} Fio;

    24

    25

static Fio _fio;

    26

    27

// Get current position in file

    28

uint32 FioGetPos(void)

    29

{

    30

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

    31

}

    32

    33

void FioSeekTo(uint32 pos, int mode)

    34

{

    35

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

    36

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

    39

}

    40

    41

// Seek to a file and a position

    42

void FioSeekToFile(uint32 pos)

    43

{

    44

	FILE *f = _fio.handles[pos >> 24];

    45

	assert(f != NULL);

    46

	_fio.cur_fh = f;

    48

}

    49

    50

byte FioReadByte(void)

    51

{

    52

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

    53

		_fio.pos += FIO_BUFFER_SIZE;

    54

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

    55

	}

    56

	return *_fio.buffer++;

    57

}

    58

    59

void FioSkipBytes(int n)

    60

{

    61

	for (;;) {

    62

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

    63

		_fio.buffer += m;

    64

		n -= m;

    65

		if (n == 0) break;

    66

		FioReadByte();

    67

		n--;

    68

	}

    69

}

    70

    71

    72

uint16 FioReadWord(void)

    73

{

    74

	byte b = FioReadByte();

    75

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

    76

}

    77

    78

uint32 FioReadDword(void)

    79

{

    80

	uint b = FioReadWord();

    81

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

    82

}

    83

    84

void FioReadBlock(void *ptr, uint size)

    85

{

    86

	FioSeekTo(FioGetPos(), SEEK_SET);

    87

	_fio.pos += size;

    88

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

    89

}

    90

    91

static inline void FioCloseFile(int slot)

    92

{

    93

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

    94

		fclose(_fio.handles[slot]);

    95

		_fio.handles[slot] = NULL;

    96

	}

    97

}

    98

    99

void FioCloseAll(void)

   100

{

   101

	int i;

   102

   103

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

   104

		FioCloseFile(i);

   105

}

   106

   107

bool FioCheckFileExists(const char *filename)

   108

{

   109

	FILE *f = FioFOpenFile(filename);

   110

	if (f == NULL) return false;

   111

   112

	fclose(f);

   113

	return true;

   114

}

   115

   116

FILE *FioFOpenFile(const char *filename)

   117

{

   118

	FILE *f;

   119

	char buf[MAX_PATH];

   120

   121

	sprintf(buf, "%s%s", _path.data_dir, filename);

   122

   123

	f = fopen(buf, "rb");

   124

#if !defined(WIN32)

   125

	if (f == NULL) {

   126

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

   127

		f = fopen(buf, "rb");

   128

   129

#if defined SECOND_DATA_DIR

   130

		// tries in the 2nd data directory

   131

		if (f == NULL) {

   132

			sprintf(buf, "%s%s", _path.second_data_dir, filename);

   133

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

   134

			f = fopen(buf, "rb");

   135

		}

   136

#endif

   137

	}

   138

#endif

   139

   140

	return f;

   141

}

   142

   143

void FioOpenFile(int slot, const char *filename)

   144

{

   145

	FILE *f = FioFOpenFile(filename);

   146

   147

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

   148

   149

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

   150

	_fio.handles[slot] = f;

   151

	FioSeekToFile(slot << 24);

   152

}