1

#include "stdafx.h"

     2

#include "openttd.h"

     3

#include "fileio.h"

     5

#include "variables.h"

     6

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

     7

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

     8

#endif

     9

    10

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

    11

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

    12

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

    13

    14

#define FIO_BUFFER_SIZE 512

    15

    16

typedef struct {

    17

	byte *buffer, *buffer_end;

    18

	uint32 pos;

    19

	FILE *cur_fh;

    20

	FILE *handles[32];

    21

	byte buffer_start[512];

    22

} Fio;

    23

    24

static Fio _fio;

    25

    26

// Get current position in file

    27

uint32 FioGetPos(void)

    28

{

    29

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

    30

}

    31

    32

void FioSeekTo(uint32 pos, int mode)

    33

{

    34

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

    35

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

    36

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

    37

}

    38

    39

// Seek to a file and a position

    40

void FioSeekToFile(uint32 pos)

    41

{

    42

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

    43

	assert(f != NULL);

    44

	_fio.cur_fh = f;

    45

	FioSeekTo(pos & 0xFFFFFF, SEEK_SET);

    46

}

    47

    48

byte FioReadByte(void)

    49

{

    50

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

    51

		_fio.pos += FIO_BUFFER_SIZE;

    52

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

    53

	}

    54

	return *_fio.buffer++;

    55

}

    56

    57

void FioSkipBytes(int n)

    58

{

    59

	for(;;) {

    60

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

    61

		_fio.buffer += m;

    62

		n -= m;

    63

		if (n == 0) break;

    64

		FioReadByte();

    65

		n--;

    66

	}

    67

}

    68

    69

    70

uint16 FioReadWord(void)

    71

{

    72

	byte b = FioReadByte();

    73

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

    74

}

    75

    76

uint32 FioReadDword(void)

    77

{

    78

	uint b = FioReadWord();

    79

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

    80

}

    81

    82

void FioReadBlock(void *ptr, uint size)

    83

{

    84

	FioSeekTo(FioGetPos(), SEEK_SET);

    85

	_fio.pos += size;

    86

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

    87

}

    88

    89

static inline void FioCloseFile(int slot)

    90

{

    91

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

    92

		fclose(_fio.handles[slot]);

    93

		_fio.handles[slot] = NULL;

    94

	}

    95

}

    96

    97

void FioCloseAll(void)

    98

{

    99

	int i;

   100

   101

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

   102

		FioCloseFile(i);

   103

}

   104

   105

bool FiosCheckFileExists(const char *filename)

   106

{

   107

	FILE *f;

   108

	char buf[MAX_PATH];

   109

   110

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

   111

   112

	f = fopen(buf, "rb");

   113

#if !defined(WIN32)

   114

	if (f == NULL) {

   115

		char *s;

   116

		// Make lower case and try again

   117

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

   118

			*s = tolower(*s);

   119

		f = fopen(buf, "rb");

   120

   121

#if defined SECOND_DATA_DIR

   122

	// tries in the 2nd data directory

   123

		if (f == NULL) {

   124

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

   125

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

   126

			*s = tolower(*s);

   127

		f = fopen(buf, "rb");

   128

		}

   129

#endif

   130

	}

   131

#endif

   132

   133

	if (f == NULL)

   134

		return false;

   135

	else {

   136

		fclose(f);

   137

		return true;

   138

	}

   139

}

   140

   141

void FioOpenFile(int slot, const char *filename)

   142

{

   143

	FILE *f;

   144

	char buf[MAX_PATH];

   145

   146

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

   147

   148

	f = fopen(buf, "rb");

   149

#if !defined(WIN32)

   150

	if (f == NULL) {

   151

		char *s;

   152

		// Make lower case and try again

   153

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

   154

			*s = tolower(*s);

   155

		f = fopen(buf, "rb");

   156

   157

#if defined SECOND_DATA_DIR

   158

	// tries in the 2nd data directory

   159

		if (f == NULL) {

   160

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

   161

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

   162

			*s = tolower(*s);

   163

		f = fopen(buf, "rb");

   164

		}

   165

#endif

   166

	}

   167

#endif

   168

   169

	if (f == NULL)

   170

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

   171

   172

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

   173

	_fio.handles[slot] = f;

   174

	FioSeekToFile(slot << 24);

   175

}

   176

   177