1

/* $Id$ */

     2

     3

#include "stdafx.h"

     4

#include "openttd.h"

     5

#include "fileio.h"

     6

#include "functions.h"

     7

#include "macros.h"

     8

#include "variables.h"

     9

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

    10

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

    11

#endif

    12

    13

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

    14

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

    15

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

    16

    17

#define FIO_BUFFER_SIZE 512

    18

    19

typedef struct {

    20

	byte *buffer, *buffer_end;

    21

	uint32 pos;

    22

	FILE *cur_fh;

    23

	FILE *handles[32];

    24

	byte buffer_start[512];

    25

} Fio;

    26

    27

static Fio _fio;

    28

    29

// Get current position in file

    30

uint32 FioGetPos(void)

    31

{

    32

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

    33

}

    34

    35

void FioSeekTo(uint32 pos, int mode)

    36

{

    37

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

    38

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

    39

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

    40

}

    41

    42

// Seek to a file and a position

    43

void FioSeekToFile(uint32 pos)

    44

{

    45

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

    46

	assert(f != NULL);

    47

	_fio.cur_fh = f;

    48

	FioSeekTo(pos & 0xFFFFFF, SEEK_SET);

    49

}

    50

    51

byte FioReadByte(void)

    52

{

    53

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

    54

		_fio.pos += FIO_BUFFER_SIZE;

    55

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

    56

	}

    57

	return *_fio.buffer++;

    58

}

    59

    60

void FioSkipBytes(int n)

    61

{

    62

	for(;;) {

    63

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

    64

		_fio.buffer += m;

    65

		n -= m;

    66

		if (n == 0) break;

    67

		FioReadByte();

    68

		n--;

    69

	}

    70

}

    71

    72

    73

uint16 FioReadWord(void)

    74

{

    75

	byte b = FioReadByte();

    76

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

    77

}

    78

    79

uint32 FioReadDword(void)

    80

{

    81

	uint b = FioReadWord();

    82

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

    83

}

    84

    85

void FioReadBlock(void *ptr, uint size)

    86

{

    87

	FioSeekTo(FioGetPos(), SEEK_SET);

    88

	_fio.pos += size;

    89

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

    90

}

    91

    92

static inline void FioCloseFile(int slot)

    93

{

    94

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

    95

		fclose(_fio.handles[slot]);

    96

		_fio.handles[slot] = NULL;

    97

	}

    98

}

    99

   100

void FioCloseAll(void)

   101

{

   102

	int i;

   103

   104

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

   105

		FioCloseFile(i);

   106

}

   107

   108

bool FiosCheckFileExists(const char *filename)

   109

{

   110

	FILE *f;

   111

	char buf[MAX_PATH];

   112

   113

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

   114

   115

	f = fopen(buf, "rb");

   116

#if !defined(WIN32)

   117

	if (f == NULL) {

   118

		char *s;

   119

		// Make lower case and try again

   120

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

   121

			*s = tolower(*s);

   122

		f = fopen(buf, "rb");

   123

   124

#if defined SECOND_DATA_DIR

   125

	// tries in the 2nd data directory

   126

		if (f == NULL) {

   127

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

   128

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

   129

			*s = tolower(*s);

   130

		f = fopen(buf, "rb");

   131

		}

   132

#endif

   133

	}

   134

#endif

   135

   136

	if (f == NULL)

   137

		return false;

   138

	else {

   139

		fclose(f);

   140

		return true;

   141

	}

   142

}

   143

   144

void FioOpenFile(int slot, const char *filename)

   145

{

   146

	FILE *f;

   147

	char buf[MAX_PATH];

   148

   149

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

   150

   151

	f = fopen(buf, "rb");

   152

#if !defined(WIN32)

   153

	if (f == NULL) {

   154

		char *s;

   155

		// Make lower case and try again

   156

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

   157

			*s = tolower(*s);

   158

		f = fopen(buf, "rb");

   159

   160

#if defined SECOND_DATA_DIR

   161

	// tries in the 2nd data directory

   162

		if (f == NULL) {

   163

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

   164

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

   165

			*s = tolower(*s);

   166

		f = fopen(buf, "rb");

   167

		}

   168

   169

	if (f == NULL)

   170

		sprintf(buf, "%s%s", _path.data_dir, filename);	//makes it print the primary datadir path instead of the secundary one

   171

   172

#endif

   173

	}

   174

#endif

   175

   176

	if (f == NULL)

   177

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

   178

   179

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

   180

	_fio.handles[slot] = f;

   181

	FioSeekToFile(slot << 24);

   182

}