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

    86

void FioCloseAll()

    87

{

    88

	int i;

    89

    90

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

    91

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

    92

			fclose(_fio.handles[i]);

    93

			_fio.handles[i] = NULL;

    94

		}

    95

	}

    96

}

    97

    98

void FioOpenFile(int slot, const char *filename)

    99

{

   100

	FILE *f;

   101

	char buf[MAX_PATH];

   102

   103

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

   104

   105

	f = fopen(buf, "rb");

   106

#if !defined(WIN32)

   107

	if (f == NULL) {

   108

		char *s;

   109

		// Make lower case and try again

   110

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

   111

			*s = tolower(*s);

   112

		f = fopen(buf, "rb");

   113

   114

#if defined SECOND_DATA_DIR

   115

	// tries in the 2nd data directory

   116

		if (f == NULL) {

   117

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

   118

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

   119

			*s = tolower(*s);

   120

		f = fopen(buf, "rb");

   121

		}

   122

#endif

   123

	}

   124

#endif

   125

   126

	if (f == NULL)

   127

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

   128

   129

	_fio.handles[slot] = f;

   130

	FioSeekToFile(slot << 24);

   131

}

   132

   133