1

#include "stdafx.h"

     2

#include "ttd.h"

     3

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

     4

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

     5

#endif

     6

     7

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

     8

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

     9

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

    10

    11

#define FIO_BUFFER_SIZE 512

    12

    13

typedef struct {

    14

	byte *buffer, *buffer_end;

    15

	uint32 pos;

    16

	FILE *cur_fh;

    17

	FILE *handles[32];

    18

	byte buffer_start[512];

    19

} Fio;

    20

    21

static Fio _fio;

    22

    23

// Get current position in file

    24

uint32 FioGetPos()

    25

{

    26

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

    27

}

    28

    29

void FioSeekTo(uint32 pos, int mode)

    30

{

    31

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

    32

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

    33

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

    34

}

    35

    36

// Seek to a file and a position

    37

void FioSeekToFile(uint32 pos)

    38

{

    39

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

    40

	assert(f != NULL);

    41

	_fio.cur_fh = f;

    42

	FioSeekTo(pos & 0xFFFFFF, SEEK_SET);

    43

}

    44

    45

byte FioReadByte()

    46

{

    47

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

    48

		_fio.pos += FIO_BUFFER_SIZE;

    49

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

    50

	}

    51

	return *_fio.buffer++;

    52

}

    53

    54

void FioSkipBytes(int n)

    55

{

    56

	for(;;) {

    57

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

    58

		_fio.buffer += m;

    59

		n -= m;

    60

		if (n == 0) break;

    61

		FioReadByte();

    62

		n--;

    63

	}

    64

}

    65

    66

    67

uint16 FioReadWord()

    68

{

    69

	byte b = FioReadByte();

    70

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

    71

}

    72

    73

uint32 FioReadDword()

    74

{

    75

	uint b = FioReadWord();

    76

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

    77

}

    78

    79

void FioReadBlock(void *ptr, uint size)

    80

{

    81

	FioSeekTo(FioGetPos(), SEEK_SET);

    82

	_fio.pos += size;

    83

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

    84

}

    85

    94

void FioCloseAll(void)

    95

{

    96

	int i;

    97

   100

}

   101

   102

void FioOpenFile(int slot, const char *filename)

   103

{

   104

	FILE *f;

   105

	char buf[MAX_PATH];

   106

   107

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

   108

   109

	f = fopen(buf, "rb");

   110

#if !defined(WIN32)

   111

	if (f == NULL) {

   112

		char *s;

   113

		// Make lower case and try again

   114

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

   115

			*s = tolower(*s);

   116

		f = fopen(buf, "rb");

   117

   118

#if defined SECOND_DATA_DIR

   119

	// tries in the 2nd data directory

   120

		if (f == NULL) {

   121

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

   122

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

   123

			*s = tolower(*s);

   124

		f = fopen(buf, "rb");

   125

		}

   126

#endif

   127

	}

   128

#endif

   129

   130

	if (f == NULL)

   131

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

   132

   134

	_fio.handles[slot] = f;

   135

	FioSeekToFile(slot << 24);

   136

}

   137

   138