material bugfixes, plus the beginning of U8 archive code, and a nice little API for handling archive filesystems

5 files changed, 100 insertions, 264 deletions

diff --git a/lyt/common.cpp b/lyt/common.cpp
deleted file mode 100644
index ba86246..0000000
--- a/lyt/common.cpp
+++ /dev/null
@@ -1,109 +0,0 @@
-/*******************************************************************************
-  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(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;
-}
-
-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, QString str, int length) {
-	QByteArray paddedStr = PadByteArray(str.toAscii(), length);
-	out.writeRawData(paddedStr.constData(), paddedStr.length());
-}
-
diff --git a/lyt/common.h b/lyt/common.h
index c75813b..9c39339 100644
--- a/lyt/common.h
+++ b/lyt/common.h
@@ -19,13 +19,8 @@
 #define LYTCOMMON_H
 
 #include <QtGlobal>
-#include <QtCore/QByteArray>
-#include <QtGui/QColor>
-#include <QtCore/QPointF>
-#include <QtCore/QDataStream>
-#include <QtCore/QStringList>
-#include <QtCore/QDebug>
-#include <QtCore/QVector>
+
+#include "../wii/common.h"
 
 union Magic {
 	char str[4];
@@ -39,97 +34,9 @@ union Version {
 
 
 
-inline quint32 BitExtract(quint32 value, int count, int start) {
-	// this function relies on heavy compiler optimisation to be efficient :p
-	quint32 mask = 0;
-	for (int i = start; i < start+count; i++) {
-		mask |= (0x80000000 >> i);
-	}
-
-	return (value & mask) >> (32 - (start + count));
-}
-
-inline quint32 BitInsert(quint32 value, int newValue, int count, int start) {
-	quint32 mask = 0;
-	for (int i = start; i < start+count; i++) {
-		mask |= (0x80000000 >> i);
-	}
-
-	value &= ~mask;
-	value |= (newValue << (32 - (start + count))) & mask;
-	return value;
-}
-
-
-
-
-
-QByteArray PadByteArray(QByteArray original, int newLength, char padWith='\0');
-
-inline quint32 ColorToRGBA(QColor col) {
-	return (col.red() << 24) | (col.green() << 16) | (col.blue() << 8) | (col.alpha());
-}
-
-inline QColor RGBAToColor(quint32 col) {
-	return QColor(col >> 24, (col >> 16) & 0xFF, (col >> 8) & 0xFF, col & 0xFF);
-}
-
-inline void ReadRGBA8Color(QColor &out, QDataStream &in) {
-	quint32 col;
-	in >> (quint32&)col;
-	out = RGBAToColor(col);
-}
-
-inline void WriteRGBA8Color(QColor &in, QDataStream &out) {
-	out << (quint32)ColorToRGBA(in);
-}
-
-inline void ReadS10Color(QColor &out, QDataStream &in) {
-	quint16 r, g, b, a;
-	in >> (quint16&)r;
-	in >> (quint16&)g;
-	in >> (quint16&)b;
-	in >> (quint16&)a;
-	out.setRgb(r, g, b, a);
-}
-
-inline void WriteS10Color(QColor &in, QDataStream &out) {
-	out << (quint16)in.red();
-	out << (quint16)in.green();
-	out << (quint16)in.blue();
-	out << (quint16)in.alpha();
-}
-
-
-
-
 struct LYTTexCoords {
 	QPointF coord[4];
 };
 
-inline void ReadPointF(QDataStream &stream, QPointF &point) {
-	float x, y;
-	stream >> x;
-	stream >> y;
-	point.setX(x);
-	point.setY(y);
-}
-
-inline void WritePointF(QDataStream &stream, const QPointF &point) {
-	stream << (float)point.x();
-	stream << (float)point.y();
-}
-
-inline void InitDataStream(QDataStream &stream) {
-	stream.setByteOrder(QDataStream::BigEndian);
-	stream.setVersion(QDataStream::Qt_4_5);
-}
-
-QStringList ReadStringList(QDataStream &in);
-
-QString ReadFixedLengthASCII(QDataStream &in, int length);
-void WriteFixedLengthASCII(QDataStream &out, QString str, int length);
-
-
 
 #endif // LYTCOMMON_H
diff --git a/lyt/materials/indirectstage.h b/lyt/materials/indirectstage.h
index 30c6f9d..8f381bf 100644
--- a/lyt/materials/indirectstage.h
+++ b/lyt/materials/indirectstage.h
@@ -30,10 +30,10 @@ public:
 
 	void dumpToDebug();
 
-	int texCoord;
-	int texMap;
-	int wrap_s;
-	int wrap_t;
+	quint8 texCoord;
+	quint8 texMap;
+	quint8 wrap_s;
+	quint8 wrap_t;
 };
 
 
diff --git a/lyt/materials/material.cpp b/lyt/materials/material.cpp
index f79fd08..a4498c5 100644
--- a/lyt/materials/material.cpp
+++ b/lyt/materials/material.cpp
@@ -31,8 +31,61 @@ LYTLayout &LYTMaterial::layout() const {
 
 void LYTMaterial::dumpToDebug() {
 	qDebug() << "LYTMaterial" << name << "@" << (void*)this;
-	// todo: move dumpToDebug calls for the various Material structs out of
-	// LYTMaterial::read...() and put them here, since it makes more sense
+
+	for (int i = 0; i < 3; i++)
+		qDebug() << "Colour" << i << ":" << colours[i];
+
+	for (int i = 0; i < 4; i++)
+		qDebug() << "TEV Constant Colour" << i << ":" << tevKColour[i];
+
+	qDebug() << "TexMaps:" << texMaps.count();
+	foreach (LYTTexMap texMap, texMaps)
+		texMap.dumpToDebug();
+
+	qDebug() << "TexSRTs:" << texSRTs.count();
+	foreach (LYTTexSRT texSRT, texSRTs)
+		texSRT.dumpToDebug();
+
+	qDebug() << "TexCoordGens:" << texCoordGens.count();
+	foreach (LYTTexCoordGen texCoordGen, texCoordGens)
+		texCoordGen.dumpToDebug();
+
+	if (hasChanCtrl)
+		chanCtrl.dumpToDebug();
+	else
+		qDebug() << "ChanCtrl: none";
+
+	if (hasMatCol)
+		qDebug() << "MatCol:" << matCol;
+	else
+		qDebug() << "MatCol: none";
+
+	if (hasTevSwapTable)
+		tevSwapTable.dumpToDebug();
+	else
+		qDebug() << "TevSwapTable: none";
+
+	qDebug() << "IndTexSRTs:" << indTexSRTs.count();
+	foreach (LYTTexSRT indTexSRT, indTexSRTs)
+		indTexSRT.dumpToDebug();
+
+	qDebug() << "IndirectStages:" << indTexStages.count();
+	foreach (LYTIndirectStage indTexStage, indTexStages)
+		indTexStage.dumpToDebug();
+
+	qDebug() << "TevStages:" << tevStages.count();
+	foreach (LYTTevStage tevStage, tevStages)
+		tevStage.dumpToDebug();
+
+	if (hasAlphaCompare)
+		alphaCompare.dumpToDebug();
+	else
+		qDebug() << "AlphaCompare: none";
+
+	if (hasBlendMode)
+		blendMode.dumpToDebug();
+	else
+		qDebug() << "BlendMode: none";
 }
 
 
@@ -51,8 +104,9 @@ void LYTMaterial::readFromDataStream(QDataStream &in) {
 		ReadRGBA8Color(this->tevKColour[i], in);
 
 
-	LYTMaterialResourceNum resourceNum;
-	in >> (quint32&)resourceNum.value;
+	quint32 resNumValue;
+	in >> (quint32&)resNumValue;
+	LYTMaterialResourceNum resourceNum(resNumValue);
 
 	// this is really complicated -_-
 	// first off: TexMap
@@ -79,7 +133,7 @@ void LYTMaterial::readFromDataStream(QDataStream &in) {
 	// ChanCtrl
 	if (resourceNum.hasChanCtrl()) {
 		this->hasChanCtrl = true;
-		this->readChanCtrl(in);
+		this->chanCtrl.readFromDataStream(in);
 	} else {
 		this->hasChanCtrl = false;
 	}
@@ -87,7 +141,7 @@ void LYTMaterial::readFromDataStream(QDataStream &in) {
 	// MatCol
 	if (resourceNum.hasMatCol()) {
 		this->hasMatCol = true;
-		this->readMatCol(in);
+		ReadRGBA8Color(this->matCol, in);
 	} else {
 		this->hasMatCol = false;
 	}
@@ -95,7 +149,7 @@ void LYTMaterial::readFromDataStream(QDataStream &in) {
 	// TevSwapTable
 	if (resourceNum.hasTevSwapTable()) {
 		this->hasTevSwapTable = true;
-		this->readTevSwapTable(in);
+		this->tevSwapTable.readFromDataStream(in);
 	} else {
 		this->hasTevSwapTable = false;
 	}
@@ -124,7 +178,7 @@ void LYTMaterial::readFromDataStream(QDataStream &in) {
 	// AlphaCompare
 	if (resourceNum.hasAlphaCompare()) {
 		this->hasAlphaCompare = true;
-		this->readAlphaCompare(in);
+		this->alphaCompare.readFromDataStream(in);
 	} else {
 		this->hasAlphaCompare = false;
 	}
@@ -132,7 +186,7 @@ void LYTMaterial::readFromDataStream(QDataStream &in) {
 	// BlendMode
 	if (resourceNum.hasBlendMode()) {
 		this->hasBlendMode = true;
-		this->readBlendMode(in);
+		this->blendMode.readFromDataStream(in);
 	} else {
 		this->hasBlendMode = false;
 	}
@@ -143,60 +197,31 @@ void LYTMaterial::readFromDataStream(QDataStream &in) {
 void LYTMaterial::readTexMap(QDataStream &in) {
 	this->texMaps.append(LYTTexMap());
 	this->texMaps.last().readFromDataStream(in, m_layout);
-	this->texMaps.last().dumpToDebug();
 }
 
 void LYTMaterial::readTexSRT(QDataStream &in) {
 	this->texSRTs.append(LYTTexSRT());
 	this->texSRTs.last().readFromDataStream(in);
-	this->texSRTs.last().dumpToDebug();
 }
 
 void LYTMaterial::readTexCoordGen(QDataStream &in) {
 	this->texCoordGens.append(LYTTexCoordGen());
 	this->texCoordGens.last().readFromDataStream(in);
-	this->texCoordGens.last().dumpToDebug();
-}
-
-void LYTMaterial::readChanCtrl(QDataStream &in) {
-	this->chanCtrl.readFromDataStream(in);
-	this->chanCtrl.dumpToDebug();
-}
-
-void LYTMaterial::readMatCol(QDataStream &in) {
-	ReadRGBA8Color(this->matCol, in);
-}
-
-void LYTMaterial::readTevSwapTable(QDataStream &in) {
-	this->tevSwapTable.readFromDataStream(in);
-	this->tevSwapTable.dumpToDebug();
 }
 
 void LYTMaterial::readIndTexSRT(QDataStream &in) {
 	this->indTexSRTs.append(LYTTexSRT());
 	this->indTexSRTs.last().readFromDataStream(in);
-	this->indTexSRTs.last().dumpToDebug();
 }
 
 void LYTMaterial::readIndirectStage(QDataStream &in) {
 	this->indTexStages.append(LYTIndirectStage());
 	this->indTexStages.last().readFromDataStream(in);
-	this->indTexStages.last().dumpToDebug();
 }
 
 void LYTMaterial::readTevStage(QDataStream &in) {
 	this->tevStages.append(LYTTevStage());
 	this->tevStages.last().readFromDataStream(in);
-	this->tevStages.last().dumpToDebug();
 }
 
-void LYTMaterial::readAlphaCompare(QDataStream &in) {
-	this->alphaCompare.readFromDataStream(in);
-	this->alphaCompare.dumpToDebug();
-}
-
-void LYTMaterial::readBlendMode(QDataStream &in) {
-	this->blendMode.readFromDataStream(in);
-	this->blendMode.dumpToDebug();
-}
 
diff --git a/lyt/materials/material.h b/lyt/materials/material.h
index bfdf80b..d605244 100644
--- a/lyt/materials/material.h
+++ b/lyt/materials/material.h
@@ -40,19 +40,37 @@ class LYTLayout;
 
 class LYTMaterialResourceNum {
 public:
-	quint32 value;
-
-	inline  int getTexMapNum()		{ return BitExtract(value, 4, 28); }
-	inline  int getTexSRTNum()		{ return BitExtract(value, 4, 24); }
-	inline  int getTexCoordGenNum()	{ return BitExtract(value, 4, 20); }
-	inline bool hasChanCtrl()		{ return BitExtract(value, 1, 6);  }
-	inline bool hasMatCol()			{ return BitExtract(value, 1, 4);  }
-	inline bool hasTevSwapTable()	{ return BitExtract(value, 1, 19); }
-	inline bool hasAlphaCompare()	{ return BitExtract(value, 1, 8);  }
-	inline bool hasBlendMode()		{ return BitExtract(value, 1, 7);  }
-	inline  int getIndTexSRTNum()	{ return BitExtract(value, 2, 17); }
-	inline  int getIndTexStageNum()	{ return BitExtract(value, 3, 14); }
-	inline  int getTevStageNum()	{ return BitExtract(value, 5, 9);  }
+	LYTMaterialResourceNum() : m_value(0) { }
+	LYTMaterialResourceNum(int initValue) : m_value(initValue) { }
+
+	quint32 value() { return m_value; }
+
+	int getTexMapNum()		{ return BitExtract(m_value, 4, 28); }
+	int getTexSRTNum()		{ return BitExtract(m_value, 4, 24); }
+	int getTexCoordGenNum()	{ return BitExtract(m_value, 4, 20); }
+	bool hasChanCtrl()		{ return BitExtract(m_value, 1, 6);  }
+	bool hasMatCol()		{ return BitExtract(m_value, 1, 4);  }
+	bool hasTevSwapTable()	{ return BitExtract(m_value, 1, 19); }
+	bool hasAlphaCompare()	{ return BitExtract(m_value, 1, 8);  }
+	bool hasBlendMode()		{ return BitExtract(m_value, 1, 7);  }
+	int getIndTexSRTNum()	{ return BitExtract(m_value, 2, 17); }
+	int getIndTexStageNum()	{ return BitExtract(m_value, 3, 14); }
+	int getTevStageNum()	{ return BitExtract(m_value, 5, 9);  }
+
+	void setTexMapNum(int v)		{ m_value = BitInsert(m_value, v, 4, 28); }
+	void setTexSRTNum(int v)		{ m_value = BitInsert(m_value, v, 4, 24); }
+	void setTexCoordGenNum(int v)	{ m_value = BitInsert(m_value, v, 4, 20); }
+	void setHasChanCtrl(bool v)		{ m_value = BitInsert(m_value, v, 1, 6); }
+	void setHasMatCol(bool v)		{ m_value = BitInsert(m_value, v, 1, 4); }
+	void setHasTevSwapTable(bool v)	{ m_value = BitInsert(m_value, v, 1, 19); }
+	void setHasAlphaCompare(bool v)	{ m_value = BitInsert(m_value, v, 1, 8); }
+	void setHasBlendMode(bool v)	{ m_value = BitInsert(m_value, v, 1, 7); }
+	void setIndTexSRTNum(int v)		{ m_value = BitInsert(m_value, v, 2, 17); }
+	void setIndTexStageNum(int v)	{ m_value = BitInsert(m_value, v, 3, 14); }
+	void setTevStageNum(int v)		{ m_value = BitInsert(m_value, v, 5, 9); }
+
+private:
+	quint32 m_value;
 };
 
 
@@ -109,11 +127,6 @@ protected:
 	void readTexMap(QDataStream &in);
 	void readTexSRT(QDataStream &in);
 	void readTexCoordGen(QDataStream &in);
-	void readChanCtrl(QDataStream &in);
-	void readMatCol(QDataStream &in);
-	void readTevSwapTable(QDataStream &in);
-	void readAlphaCompare(QDataStream &in);
-	void readBlendMode(QDataStream &in);
 	void readIndirectStage(QDataStream &in);
 	void readIndTexSRT(QDataStream &in);
 	void readTevStage(QDataStream &in);