1

#include "stdafx.h"

     2

#include "ttd.h"

     3

#include "fileio.h"

     4

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

     5

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

     6

#endif

     7

     8

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

     9

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

    10

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

    11

    12

#define FIO_BUFFER_SIZE 512

    13

    14

typedef struct {

    15

	byte *buffer, *buffer_end;

    16

	uint32 pos;

    17

	FILE *cur_fh;

    18

	FILE *handles[32];

    19

	byte buffer_start[512];

    20

} Fio;

    21

    22

static Fio _fio;

    23

    24

// Get current position in file

    25

uint32 FioGetPos(void)

    26

{

    27

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

    28

}

    29

    30

void FioSeekTo(uint32 pos, int mode)

    31

{

    32

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

    33

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

    34

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

    35

}

    36

    37

// Seek to a file and a position

    38

void FioSeekToFile(uint32 pos)

    39

{

    40

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

    41

	assert(f != NULL);

    42

	_fio.cur_fh = f;

    43

	FioSeekTo(pos & 0xFFFFFF, SEEK_SET);

    44

}

    45

    46

byte FioReadByte(void)

    47

{

    48

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

    49

		_fio.pos += FIO_BUFFER_SIZE;

    50

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

    51

	}

    52

	return *_fio.buffer++;

    53

}

    54

    55

void FioSkipBytes(int n)

    56

{

    57

	for(;;) {

    58

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

    59

		_fio.buffer += m;

    60

		n -= m;

    61

		if (n == 0) break;

    62

		FioReadByte();

    63

		n--;

    64

	}

    65

}

    66

    67

    68

uint16 FioReadWord(void)

    69

{

    70

	byte b = FioReadByte();

    71

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

    72

}

    73

    74

uint32 FioReadDword(void)

    75

{

    76

	uint b = FioReadWord();

    77

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

    78

}

    79

    80

void FioReadBlock(void *ptr, uint size)

    81

{

    82

	FioSeekTo(FioGetPos(), SEEK_SET);

    83

	_fio.pos += size;

    84

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

    85

}

    86

    87

static inline void FioCloseFile(int slot)

    88

{

    89

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

    90

		fclose(_fio.handles[slot]);

    91

		_fio.handles[slot] = NULL;

    92

	}

    93

}

    94

    95

void FioCloseAll(void)

    96

{

    97

	int i;

    98

    99

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

   100

		FioCloseFile(i);

   101

}

   102

   103

bool FiosCheckFileExists(const char *filename)

   104

{

   105

	FILE *f;

   106

	char buf[MAX_PATH];

   107

   108

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

   109

   110

	f = fopen(buf, "rb");

   111

#if !defined(WIN32)

   112

	if (f == NULL) {

   113

		char *s;

   114

		// Make lower case and try again

   115

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

   116

			*s = tolower(*s);

   117

		f = fopen(buf, "rb");

   118

   119

#if defined SECOND_DATA_DIR

   120

	// tries in the 2nd data directory

   121

		if (f == NULL) {

   122

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

   123

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

   124

			*s = tolower(*s);

   125

		f = fopen(buf, "rb");

   126

		}

   127

#endif

   128

	}

   129

#endif

   130

   131

	if (f == NULL)

   132

		return false;

   133

	else {

   134

		fclose(f);

   135

		return true;

   136

	}

   137

}

   138

   139

void FioOpenFile(int slot, const char *filename)

   140

{

   141

	FILE *f;

   142

	char buf[MAX_PATH];

   143

   144

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

   145

   146

	f = fopen(buf, "rb");

   147

#if !defined(WIN32)

   148

	if (f == NULL) {

   149

		char *s;

   150

		// Make lower case and try again

   151

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

   152

			*s = tolower(*s);

   153

		f = fopen(buf, "rb");

   154

   155

#if defined SECOND_DATA_DIR

   156

	// tries in the 2nd data directory

   157

		if (f == NULL) {

   158

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

   159

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

   160

			*s = tolower(*s);

   161

		f = fopen(buf, "rb");

   162

		}

   163

#endif

   164

	}

   165

#endif

   166

   167

	if (f == NULL)

   168

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

   169

   170

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

   171

	_fio.handles[slot] = f;

   172

	FioSeekToFile(slot << 24);

   173

}

   174

   175