140 lines
3.7 KiB
Python
140 lines
3.7 KiB
Python
#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!"
|
|
|