summaryrefslogtreecommitdiff
path: root/MapTool/wii/common.cpp
blob: 8f552b3425677a28c899ed3224f737316a77f19b (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
/*******************************************************************************
  This file is part of LayoutStudio (http://github.com/Treeki/LayoutStudio)
  Copyright (c) 2010 Treeki (treeki@gmail.com)

  This program is free software: you can redistribute it and/or modify
  it under the terms of the GNU General Public License as published by
  the Free Software Foundation, version 2.0.

  This program is distributed in the hope that it will be useful,
  but WITHOUT ANY WARRANTY; without even the implied warranty of
  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  GNU General Public License 2.0 for more details.

  You should have received a copy of the GNU General Public License 2.0
  along with this program.  If not, see <http://www.gnu.org/licenses/>.
*******************************************************************************/

#include "common.h"

QByteArray PadByteArray(const QByteArray original, int newLength, char padWith) {
	QByteArray newArray = original;

	if (original.length() > newLength) {
		// the original array is longer than the length desired, so truncate it
		newArray.truncate(newLength);

	} else if (original.length() < newLength) {
		// the original array is shorter, so pad it
		int oldLength = original.length();
		newArray.resize(newLength);

		for (int i = oldLength; i < newLength; i++) {
			newArray[i] = '\0';
		}
	}

	return newArray;
}

QStringList ReadStringList(QDataStream &in) {
	QStringList output;

	quint16 count;
	in >> (quint16&)count;
	in.skipRawData(2); // padding

	QVector<quint32> stringOffsets(count);

	// save the initial offset so we can get the strings later
	// string offsets are based on the first offset entry (after the count)
	// NOT on the section offset
	qint64 savedPos = in.device()->pos();

	for (int i = 0; i < count; i++) {
		quint32 offset;
		in >> (quint32&)offset;
		in.skipRawData(4); // unused?

		stringOffsets[i] = offset;
	}

	// ok, now we can get the strings
	for (int i = 0; i < count; i++) {
		in.device()->seek(savedPos + stringOffsets[i]);

		// how fun: no length is stored for each string, they're just zero
		// terminated. so let's try to figure it out!
		int stringLength = 0;
		char check;

		in >> (quint8&)check;
		while (check != 0) {
			stringLength += 1;
			in >> (quint8&)check;
		}

		// now read the string
		char *buffer = new char[stringLength];

		in.device()->seek(savedPos + stringOffsets[i]);
		in.readRawData(buffer, stringLength);

		output.append(QString::fromAscii(buffer, stringLength));

		delete[] buffer;


		qDebug() << "Read string:" << output.last();
	}

	return output;
}

void WriteStringList(QDataStream &out, const QStringList list) {
	out << (quint16)list.count();
	WritePadding(2, out);

	// calculate offsets for every string, and write them
	// offset 0 points to the first entry in the offset list, etc, so
	// take that into account for the string offset calculation
	quint32 currentOffset = list.count() * 8;

	foreach (QString str, list) {
		out << (quint32)currentOffset;
		WritePadding(4, out); // unused?

		currentOffset += str.length() + 1;
	}

	// now write the strings
	foreach (QString str, list) {
		QByteArray rawStr = str.toAscii();
		rawStr.append('\0');
		out.writeRawData(rawStr.constData(), rawStr.length());
	}
}

QString ReadFixedLengthASCII(QDataStream &in, int length) {
	QByteArray readStr(length, '\0');
	in.readRawData(readStr.data(), readStr.length());

	QString str = QString::fromAscii(readStr.data(), readStr.length());
	if (str.contains(QChar('\0')))
		str.truncate(str.indexOf(QChar('\0')));

	return str;
}

void WriteFixedLengthASCII(QDataStream &out, const QString str, int length) {
	QByteArray paddedStr = PadByteArray(str.toAscii(), length);
	out.writeRawData(paddedStr.constData(), paddedStr.length());
}