1

#include "stdafx.h"

     2

#include "openttd.h"

     3

#include "fileio.h"

     5

#include "macros.h"

     6

#include "variables.h"

     7

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

     8

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

     9

#endif

    10

    11

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

    12

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

    13

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

    14

    15

#define FIO_BUFFER_SIZE 512

    16

    17

typedef struct {

    18

	byte *buffer, *buffer_end;

    19

	uint32 pos;

    20

	FILE *cur_fh;

    21

	FILE *handles[32];

    22

	byte buffer_start[512];

    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;

    37

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

    38

}

    39

    40

// Seek to a file and a position

    41

void FioSeekToFile(uint32 pos)

    42

{

    43

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

    44

	assert(f != NULL);

    45

	_fio.cur_fh = f;

    46

	FioSeekTo(pos & 0xFFFFFF, SEEK_SET);

    47

}

    48

    49

byte FioReadByte(void)

    50

{

    51

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

    52

		_fio.pos += FIO_BUFFER_SIZE;

    53

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

    54

	}

    55

	return *_fio.buffer++;

    56

}

    57

    58

void FioSkipBytes(int n)

    59

{

    60

	for(;;) {

    61

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

    62

		_fio.buffer += m;

    63

		n -= m;

    64

		if (n == 0) break;

    65

		FioReadByte();

    66

		n--;

    67

	}

    68

}

    69

    70

    71

uint16 FioReadWord(void)

    72

{

    73

	byte b = FioReadByte();

    74

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

    75

}

    76

    77

uint32 FioReadDword(void)

    78

{

    79

	uint b = FioReadWord();

    80

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

    81

}

    82

    83

void FioReadBlock(void *ptr, uint size)

    84

{

    85

	FioSeekTo(FioGetPos(), SEEK_SET);

    86

	_fio.pos += size;

    87

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

    88

}

    89

    90

static inline void FioCloseFile(int slot)

    91

{

    92

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

    93

		fclose(_fio.handles[slot]);

    94

		_fio.handles[slot] = NULL;

    95

	}

    96

}

    97

    98

void FioCloseAll(void)

    99

{

   100

	int i;

   101

   102

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

   103

		FioCloseFile(i);

   104

}

   105

   106

bool FiosCheckFileExists(const char *filename)

   107

{

   108

	FILE *f;

   109

	char buf[MAX_PATH];

   110

   111

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

   112

   113

	f = fopen(buf, "rb");

   114

#if !defined(WIN32)

   115

	if (f == NULL) {

   116

		char *s;

   117

		// Make lower case and try again

   118

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

   119

			*s = tolower(*s);

   120

		f = fopen(buf, "rb");

   121

   122

#if defined SECOND_DATA_DIR

   123

	// tries in the 2nd data directory

   124

		if (f == NULL) {

   125

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

   126

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

   127

			*s = tolower(*s);

   128

		f = fopen(buf, "rb");

   129

		}

   130

#endif

   131

	}

   132

#endif

   133

   134

	if (f == NULL)

   135

		return false;

   136

	else {

   137

		fclose(f);

   138

		return true;

   139

	}

   140

}

   141

   142

void FioOpenFile(int slot, const char *filename)

   143

{

   144

	FILE *f;

   145

	char buf[MAX_PATH];

   146

   147

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

   148

   149

	f = fopen(buf, "rb");

   150

#if !defined(WIN32)

   151

	if (f == NULL) {

   152

		char *s;

   153

		// Make lower case and try again

   154

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

   155

			*s = tolower(*s);

   156

		f = fopen(buf, "rb");

   157

   158

#if defined SECOND_DATA_DIR

   159

	// tries in the 2nd data directory

   160

		if (f == NULL) {

   161

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

   162

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

   163

			*s = tolower(*s);

   164

		f = fopen(buf, "rb");

   165

		}

   166

#endif

   167

	}

   168

#endif

   169

   170

	if (f == NULL)

   171

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

   172

   173

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

   174

	_fio.handles[slot] = f;

   175

	FioSeekToFile(slot << 24);

   176

}

   177

   178