Initial commit

Hatenatools/NTFT.py

@@ -0,0 +1,140 @@
+#NTFT.py by pbsds
+#Free to use as long as credit is given to pbsds / Peder Bergebakken Sundt
+#PIL is required to write images to disk
+#
+#Credit:
+#
+#	-The guys behind TiledGGD at https://code.google.com/p/tiledggd/
+#	 This sped up the my work a lot
+import sys
+try:
+	import Image
+	hasPIL = True
+except ImportError:
+	hasPIL = False
+
+
+#helpers
+def AscDec(ascii, LittleEndian=False):#Converts a ascii string into a decimal
+	ret = 0
+	l = map(ord, ascii)
+	if LittleEndian: l.reverse()
+	for i in l:
+		ret = (ret<<8) | i
+	return ret
+def DecAsc(dec, length=None, LittleEndian=False):#Converts a decimal into an ascii string of chosen length
+	out = []
+	while dec <> 0:
+		out.insert(0, dec&0xFF)
+		dec >>= 8
+	#"".join(map(chr, out))
+	
+	if length:
+		if len(out) > length:
+			#return "".join(map(chr, out[-length:]))
+			out = out[-length:]
+		if len(out) < length:
+			#return "".join(map(chr, [0]*(length-len(out)) + out))
+			out = [0]*(length-len(out)) + out
+			
+	if LittleEndian: out.reverse()
+	return "".join(map(chr, out))
+
+#Class NTFT:
+#
+#	todo: write documentation
+class NTFT:
+	def __init__(self):
+		self.Loaded = False
+	def ReadFile(self, path, size):
+		f = open(path, "rb")
+		ret = self.Read(f.read(), size)
+		f.close()
+		return ret
+	def Read(self, data, (w, h)):
+		if w*h*2 <> len(data):
+			print "Invalid size"
+			return False
+		
+		self.Image = [[None for _ in xrange(h)] for _ in xrange(w)]
+		for y in xrange(h):
+			for x in xrange(w):
+				pos = (x + y*w)*2
+				#print pos
+				byte = AscDec(data[pos:pos+2], True)
+				byte = AscDec(data[pos:pos+2], True)
+				#ARGB1555 -> RGBA8:#require true
+				a = (byte >> 15       ) * 0xFF
+				b = (byte >> 10 & 0x1F) * 0xFF / 0x1F
+				g = (byte >> 5  & 0x1F) * 0xFF / 0x1F
+				r = (byte       & 0x1F) * 0xFF / 0x1F
+				# #RGB565 -> RGBA8:
+				# a = 0xFF
+				# r = (byte >> 11 & 0x1F) * 0xFF / 0x1F
+				# g = (byte >> 5  & 0x3F) * 0xFF / 0x3F
+				# b = (byte       & 0x1F) * 0xFF / 0x1F
+				#print x, y
+				self.Image[x][y] = (r<<24) | (g<<16) | (b<<8) | a#RGBA8
+		
+		self.Loaded = True
+		return self
+	def WriteFile(self, path):
+		pass
+	def Pack(self):
+		pass
+
+#Function WriteImage:
+#
+#	Writes a 2D list if uint32 RGBA values as a image files.
+#	Designed to work with PPM().Thumbnail or PPM().GetFrame(n)
+#
+#	This function requires the PIl imaging module
+def WriteImage(image, outputPath):
+	if not hasPIL or not image: return False
+	
+	out = []
+	for y in xrange(len(image[0])):
+		for x in xrange(len(image)):
+			out.append(DecAsc(image[x][y], 4))
+	
+	out = Image.fromstring("RGBA", (len(image), len(image[0])), "".join(out))
+	
+	filetype = outputPath[outputPath.rfind(".")+1:]
+	out.save(outputPath, filetype)
+	
+	return True
+
+#testing:
+# i = NTFT().ReadFile("NTFTtests/kaeru.ntft", (64, 32))
+# WriteImage(i.Image, "NTFTtests/kaeru.png")
+
+# i = NTFT().ReadFile("NTFTtests/News.ntft", (32, 32))
+# WriteImage(i.Image, "NTFTtests/News.png")
+
+# i = NTFT().ReadFile("NTFTtests/Special Room.ntft", (32, 32))
+# WriteImage(i.Image, "NTFTtests/Special Room.png")
+
+if __name__ == "__main__":
+	print "              ==      NTFT.py     =="
+	print "             ==      by pbsds      =="
+	print "              ==       v0.05      =="
+	print
+	
+	if len(sys.argv) < 4:
+		print "Usage:"
+		print "      NTFT.py <input> <output> <width> <height>"
+		print ""
+		print "The NTFT file contain only the colordata, so it's up to the user to find and"
+		print "store the resolution of the images."
+		print "32x32 is the normal size for button icons in UGO files."
+		sys.exit()
+	
+	input = sys.argv[1]
+	output = sys.argv[2]
+	width = int(sys.argv[3])
+	height = int(sys.argv[4])
+	
+	print "Converting..."
+	WriteImage(NTFT().ReadFile(input, (width, height)).Image, output)
+	print "Done!"
+