1

#include "stdafx.h"

     2

#include "openttd.h"

     3

#include "fileio.h"

     5

#if defined(UNIX) || defined(__OS2__)

     6

#include <ctype.h> // required for tolower()

     7

#endif

     8

     9

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

    10

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

    11

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

    12

    13

#define FIO_BUFFER_SIZE 512

    14

    15

typedef struct {

    16

	byte *buffer, *buffer_end;

    17

	uint32 pos;

    18

	FILE *cur_fh;

    19

	FILE *handles[32];

    20

	byte buffer_start[512];

    21

} Fio;

    22

    23

static Fio _fio;

    24

    25

// Get current position in file

    26

uint32 FioGetPos(void)

    27

{

    28

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

    29

}

    30

    31

void FioSeekTo(uint32 pos, int mode)

    32

{

    33

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

    34

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

    35

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

    36

}

    37

    38

// Seek to a file and a position

    39

void FioSeekToFile(uint32 pos)

    40

{

    41

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

    42

	assert(f != NULL);

    43

	_fio.cur_fh = f;

    44

	FioSeekTo(pos & 0xFFFFFF, SEEK_SET);

    45

}

    46

    47

byte FioReadByte(void)

    48

{

    49

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

    50

		_fio.pos += FIO_BUFFER_SIZE;

    51

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

    52

	}

    53

	return *_fio.buffer++;

    54

}

    55

    56

void FioSkipBytes(int n)

    57

{

    58

	for(;;) {

    59

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

    60

		_fio.buffer += m;

    61

		n -= m;

    62

		if (n == 0) break;

    63

		FioReadByte();

    64

		n--;

    65

	}

    66

}

    67

    68

    69

uint16 FioReadWord(void)

    70

{

    71

	byte b = FioReadByte();

    72

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

    73

}

    74

    75

uint32 FioReadDword(void)

    76

{

    77

	uint b = FioReadWord();

    78

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

    79

}

    80

    81

void FioReadBlock(void *ptr, uint size)

    82

{

    83

	FioSeekTo(FioGetPos(), SEEK_SET);

    84

	_fio.pos += size;

    85

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

    86

}

    87

    88

static inline void FioCloseFile(int slot)

    89

{

    90

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

    91

		fclose(_fio.handles[slot]);

    92

		_fio.handles[slot] = NULL;

    93

	}

    94

}

    95

    96

void FioCloseAll(void)

    97

{

    98

	int i;

    99

   100

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

   101

		FioCloseFile(i);

   102

}

   103

   104

bool FiosCheckFileExists(const char *filename)

   105

{

   106

	FILE *f;

   107

	char buf[MAX_PATH];

   108

   109

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

   110

   111

	f = fopen(buf, "rb");

   112

#if !defined(WIN32)

   113

	if (f == NULL) {

   114

		char *s;

   115

		// Make lower case and try again

   116

		for(s=buf + strlen(_path.data_dir) - 1; *s != 0; s++)

   117

			*s = tolower(*s);

   118

		f = fopen(buf, "rb");

   119

   120

#if defined SECOND_DATA_DIR

   121

	// tries in the 2nd data directory

   122

		if (f == NULL) {

   123

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

   124

			for(s=buf + strlen(_path.second_data_dir) - 1; *s != 0; s++)

   125

			*s = tolower(*s);

   126

		f = fopen(buf, "rb");

   127

		}

   128

#endif

   129

	}

   130

#endif

   131

   132

	if (f == NULL)

   133

		return false;

   134

	else {

   135

		fclose(f);

   136

		return true;

   137

	}

   138

}

   139

   140

void FioOpenFile(int slot, const char *filename)

   141

{

   142

	FILE *f;

   143

	char buf[MAX_PATH];

   144

   145

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

   146

   147

	f = fopen(buf, "rb");

   148

#if !defined(WIN32)

   149

	if (f == NULL) {

   150

		char *s;

   151

		// Make lower case and try again

   152

		for(s=buf + strlen(_path.data_dir) - 1; *s != 0; s++)

   153

			*s = tolower(*s);

   154

		f = fopen(buf, "rb");

   155

   156

#if defined SECOND_DATA_DIR

   157

	// tries in the 2nd data directory

   158

		if (f == NULL) {

   159

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

   160

			for(s=buf + strlen(_path.second_data_dir) - 1; *s != 0; s++)

   161

			*s = tolower(*s);

   162

		f = fopen(buf, "rb");

   163

		}

   164

#endif

   165

	}

   166

#endif

   167

   168

	if (f == NULL)

   169

		error("Cannot open file '%s'", buf);

   170

   171

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

   172

	_fio.handles[slot] = f;

   173

	FioSeekToFile(slot << 24);

   174

}

   175

   176