added RGBA8 tileset support

1 files changed, 55 insertions, 37 deletions

diff --git a/Koopuzzle/Koopuzzle.py b/Koopuzzle/Koopuzzle.py
index 8672d57..cb63412 100755
--- a/Koopuzzle/Koopuzzle.py
+++ b/Koopuzzle/Koopuzzle.py
@@ -838,7 +838,7 @@ class PiecesModel(QtCore.QAbstractListModel):
 ################## Python-based RGB5a3 Decoding code from my BRFNT program ##################
 
 
-def RGB4A3Decode(tex):
+def RGB5A3Decode(tex):
 	dest = QtGui.QImage(896,448,QtGui.QImage.Format_ARGB32)
 	dest.fill(QtCore.Qt.transparent)
 	
@@ -856,19 +856,16 @@ def RGB4A3Decode(tex):
 					# newpixel = (int(tex[i]) << 8) | int(tex[i+1])
 					
 	
-					# if(newpixel >= 0x8000): # Check if it's RGB555
-					# 	red = ((newpixel >> 10) & 0x1F) * 255 / 0x1F
-					# 	green = ((newpixel >> 5) & 0x1F) * 255 / 0x1F
-					# 	blue = (newpixel & 0x1F) * 255 / 0x1F
-					# 	alpha = 0xFF
-	
-					# else: # If not, it's RGB4A3
-
-					# IT'S ALWAYS FUCKING RGB4A3 - who makes a tileset without alpha?!
-					alpha = ((newpixel & 0x7000) >> 12) * 255 / 0x7
-					red = ((newpixel & 0xF00) >> 8) * 255 / 0xF
-					green = ((newpixel & 0xF0) >> 4) * 255 / 0xF
-					blue = (newpixel & 0xF) * 255 / 0xF
+					if(newpixel >= 0x8000): # Check if it's RGB555
+						red = ((newpixel >> 10) & 0x1F) * 255 / 0x1F
+						green = ((newpixel >> 5) & 0x1F) * 255 / 0x1F
+						blue = (newpixel & 0x1F) * 255 / 0x1F
+						alpha = 0xFF
+					else: # If not, it's RGB4A3
+						alpha = ((newpixel & 0x7000) >> 12) * 255 / 0x7
+						red = ((newpixel & 0xF00) >> 8) * 255 / 0xF
+						green = ((newpixel & 0xF0) >> 4) * 255 / 0xF
+						blue = (newpixel & 0xF) * 255 / 0xF
 	
 					argb = (blue) | (green << 8) | (red << 16) | (alpha << 24)
 					dest.setPixel(xpixel, ypixel, argb)
@@ -876,7 +873,7 @@ def RGB4A3Decode(tex):
 	return dest
 
 
-def RGB4A3Encode(tex):
+def RGB5A3Encode(tex, force4A3=True):
 	destBuffer = create_string_buffer(896*tex.height()*2)
 
 	shortstruct = struct.Struct('>H')
@@ -894,29 +891,45 @@ def RGB4A3Encode(tex):
 					g = (pixel >> 8) & 0xFF
 					b = pixel & 0xFF
 					
-					# if a < 245: #RGB4A3
-
-					# IT'S ALWAYS FUCKING RGB4A3 - who makes a tileset without alpha?!
-					alpha = a/32
-					red = r/16
-					green = g/16
-					blue = b/16
+					if a < 245 or force4A3: #RGB4A3
+						alpha = a/32
+						red = r/16
+						green = g/16
+						blue = b/16
+						rgbDAT = (blue) | (green << 4) | (red << 8) | (alpha << 12)
+					else: # RGB555
+						red = r/8
+						green = g/8
+						blue = b/8
+						rgbDAT = (blue) | (green << 5) | (red << 10) | (0x8000) # 0rrrrrgggggbbbbb
 
-					rgbDAT = (blue) | (green << 4) | (red << 8) | (alpha << 12)
-				
-					# else: # RGB555
-					#     red = r/8
-					#     green = g/8
-					#     blue = b/8
-						
-					#     rgbDAT = (blue) | (green << 5) | (red << 10) | (0x8000) # 0rrrrrgggggbbbbb
-																											
 					shortstruct.pack_into(destBuffer, offset, rgbDAT)
 					offset += 2
 					
 	return destBuffer.raw
 
 
+def RGBA8Encode(tex):
+	destBuffer = create_string_buffer(896*tex.height()*4)
+
+	shortstruct = struct.Struct('>H')
+	offset = 0
+
+	for ytile in xrange(0, tex.height(), 4):
+		for xtile in xrange(0, 896, 4):
+			for ypixel in xrange(ytile, ytile + 4):
+				for xpixel in xrange(xtile, xtile + 4):
+					
+					pixel = tex.pixel(xpixel, ypixel)
+					
+					shortstruct.pack_into(destBuffer, offset, pixel>>16)
+					shortstruct.pack_into(destBuffer, offset+32, pixel&0xFFFF)
+					offset += 2
+			offset += 32
+
+	return destBuffer.raw
+
+
 #############################################################################################
 ############ Main Window Class. Takes care of menu functions and widget creation ############
 
@@ -1023,7 +1036,7 @@ class MainWindow(QtGui.QMainWindow):
 				QtGui.QMessageBox.warning(None, 'Error',  'Error - the necessary files were not found.\n\nNot a valid Koopatlas tileset, sadly.')
 				return
 			
-			dest = RGB4A3Decode(Image)
+			dest = RGB5A3Decode(Image)
 			
 			self.tileImage = QtGui.QPixmap.fromImage(dest)
 			
@@ -1167,6 +1180,8 @@ class MainWindow(QtGui.QMainWindow):
 
 		# Prepare tiles, objects, object metadata, and textures and stuff into buffers.
 
+		self.usesRGBA8 = ('RGBA8' in name)
+
 		textureBuffer = self.PackTexture()
 		optTextureBuffer, optMappings = self.PackOptimisedTexture()
 		objectBuffers = self.PackObjects()
@@ -1209,11 +1224,11 @@ class MainWindow(QtGui.QMainWindow):
 				mapping[i] = len(sortedTiles)
 				sortedTiles.append(img)
 
-		tex = self.PackTexture(sortedTiles)
+		tex = self.PackTexture(sortedTiles, 'rgba8' if self.usesRGBA8 else 'real_rgb5a3')
 		maps = ','.join(map(str, mapping))
 		return tex, maps
 
-	def PackTexture(self, imageList=None):
+	def PackTexture(self, imageList=None, mode='standard'):
 		if imageList is None:
 			imageList = map(lambda x: x.image, Tileset.tiles)
 		usesPixmap = isinstance(imageList[0], QtGui.QPixmap)
@@ -1297,9 +1312,12 @@ class MainWindow(QtGui.QMainWindow):
 		painter.end()
 		tex.save('beef.png')
 
-		dest = RGB4A3Encode(tex)
-				
-		return dest
+		if mode == 'rgba8':
+			return RGBA8Encode(tex)
+		elif mode == 'real_rgb5a3':
+			return RGB5A3Encode(tex, False)
+		else:
+			return RGB5A3Encode(tex)