include all compressed and decompressed graphics

* frontpics & animation tiles
* backpics
* trainer pics
* attack animation gfx
* intro gfx
* intro tilemaps
* title gfx
* tilesets
* player backpic
* dude backpic
* pokegear/town map

fixes #33, #34 and #35

original-commit-id: 2cdf7a3688c43766663028cf6e644f13fae4bdf3

1 files changed, 98 insertions, 43 deletions

diff --git a/gfx.py b/gfx.py
index 6b895aa..f36b944 100644
--- a/gfx.py
+++ b/gfx.py
@@ -144,16 +144,16 @@ def to_file(filename, data):
 # and the count (bits 0-4)
 # followed by additional params
 
-cpr_lit = 0
+lz_lit = 0
 # print literal for [count] bytes
 
-cpr_iter = 1
+lz_iter = 1
 # print one byte [count] times
 
-cpr_alt = 2
+lz_alt = 2
 # print alternating bytes (2 params) for [count] bytes
 
-cpr_zeros = 3
+lz_zeros = 3
 # print 00 for [count] bytes
 
 # repeater control commands have a signed parameter used to determine the start point
@@ -161,21 +161,21 @@ cpr_zeros = 3
 # positive values are added to the start address of the decompressed data
 # and negative values are subtracted from the current position
 
-cpr_repeat = 4
+lz_repeat = 4
 # print [count] bytes from decompressed data
 
-cpr_flip = 5
+lz_flip = 5
 # print [count] bytes from decompressed data in bit order 01234567
 
-cpr_reverse = 6
+lz_reverse = 6
 # print [count] bytes from decompressed data backwards
 
-cpr_hi = 7
+lz_hi = 7
 # -used when the count exceeds 5 bits. uses a 10-bit count instead
 # -bits 2-4 now contain the control code, bits 0-1 are bits 8-9 of the count
 # -the following byte contains bits 0-7 of the count
 
-cpr_end = 0xff
+lz_end = 0xff
 # if 0xff is encountered the decompression ends
 
 # since frontpics have animation tiles lumped onto them,
@@ -272,7 +272,7 @@ class Compressed:
 		self.doLiterals()
 		
 		# done
-		self.output.append(cpr_end)
+		self.output.append(lz_end)
 	
 	
 	def getCurByte(self):
@@ -294,10 +294,10 @@ class Compressed:
 	
 	def doLiterals(self):
 		if len(self.literals) > lowmax:
-			self.output.append( (cpr_hi << 5) | (cpr_lit << 2) | ((len(self.literals) - 1) >> 8) )
+			self.output.append( (lz_hi << 5) | (lz_lit << 2) | ((len(self.literals) - 1) >> 8) )
 			self.output.append( (len(self.literals) - 1) & 0xff )
 		elif len(self.literals) > 0:
-			self.output.append( (cpr_lit << 5) | (len(self.literals) - 1) )
+			self.output.append( (lz_lit << 5) | (len(self.literals) - 1) )
 		for byte in self.literals:
 			self.output.append(byte)
 		self.literals = []	
@@ -416,14 +416,14 @@ class Compressed:
 				# decide which side we're copying from
 				if (self.address - repeat[1]) <= 0x80:
 					self.doLiterals()
-					self.stream.append( (cpr_repeat << 5) | length - 1 )
+					self.stream.append( (lz_repeat << 5) | length - 1 )
 					
 					# wrong?
 					self.stream.append( (((self.address - repeat[1])^0xff)+1)&0xff )
 
 				else:
 					self.doLiterals()
-					self.stream.append( (cpr_repeat << 5) | length - 1 )
+					self.stream.append( (lz_repeat << 5) | length - 1 )
 					
 					# wrong?
 					self.stream.append(repeat[1]>>8)
@@ -453,10 +453,10 @@ class Compressed:
 	
 	def doWhitespace(self):
 		if (len(self.zeros) + 1) >= lowmax:
-			self.stream.append( (cpr_hi << 5) | (cpr_zeros << 2) | ((len(self.zeros) - 1) >> 8) )
+			self.stream.append( (lz_hi << 5) | (lz_zeros << 2) | ((len(self.zeros) - 1) >> 8) )
 			self.stream.append( (len(self.zeros) - 1) & 0xff )
 		elif len(self.zeros) > 1:
-			self.stream.append( cpr_zeros << 5 | (len(self.zeros) - 1) )
+			self.stream.append( lz_zeros << 5 | (len(self.zeros) - 1) )
 		else:
 			raise Exception, "checkWhitespace() should prevent this from happening"
 	
@@ -509,12 +509,12 @@ class Compressed:
 		num_alts = len(self.iters) + 1
 		
 		if num_alts > lowmax:
-			self.stream.append( (cpr_hi << 5) | (cpr_alt << 2) | ((num_alts - 1) >> 8) )
+			self.stream.append( (lz_hi << 5) | (lz_alt << 2) | ((num_alts - 1) >> 8) )
 			self.stream.append( num_alts & 0xff )
 			self.stream.append( self.alts[0] )
 			self.stream.append( self.alts[1] )
 		elif num_alts > 2:
-			self.stream.append( (cpr_alt << 5) | (num_alts - 1) )
+			self.stream.append( (lz_alt << 5) | (num_alts - 1) )
 			self.stream.append( self.alts[0] )
 			self.stream.append( self.alts[1] )
 		else:
@@ -551,13 +551,13 @@ class Compressed:
 			self.next()
 		
 		if (len(self.iters) - 1) >= lowmax:
-			self.stream.append( (cpr_hi << 5) | (cpr_iter << 2) | ((len(self.iters)-1) >> 8) )
+			self.stream.append( (lz_hi << 5) | (lz_iter << 2) | ((len(self.iters)-1) >> 8) )
 			self.stream.append( (len(self.iters) - 1) & 0xff )
 			self.stream.append( iter )
 		elif len(self.iters) > 3:
 			# 3 or fewer isn't worth the trouble and actually longer
 			# if part of a larger literal set
-			self.stream.append( (cpr_iter << 5) | (len(self.iters) - 1) )
+			self.stream.append( (lz_iter << 5) | (len(self.iters) - 1) )
 			self.stream.append( iter )
 		else:
 			self.address = original_address
@@ -579,11 +579,11 @@ class Decompressed:
 	splits output into pic [size] and animation tiles if applicable
 	data can be fed in from rom if [start] is specified"""
 	
-	def __init__(self, cpr = None, mode = None, size = None, start = 0):
+	def __init__(self, lz = None, mode = None, size = None, start = 0):
 		# todo: play nice with Compressed
 	
-		assert cpr, 'need something to compress!'
-		self.cpr = cpr
+		assert lz, 'need something to compress!'
+		self.lz = lz
 		
 		self.byte = None
 		self.address = 0
@@ -621,12 +621,12 @@ class Decompressed:
 		while True:
 			self.getCurByte()
 			
-			if (self.byte == cpr_end):
+			if (self.byte == lz_end):
 				break
 			
 			self.cmd = (self.byte & 0b11100000) >> 5
 			
-			if self.cmd == cpr_hi: # 10-bit param
+			if self.cmd == lz_hi: # 10-bit param
 				self.cmd = (self.byte & 0b00011100) >> 2
 				self.length = (self.byte & 0b00000011) << 8
 				self.next()
@@ -635,13 +635,13 @@ class Decompressed:
 				self.length = (self.byte & 0b00011111) + 1
 			
 			# literals
-			if self.cmd == cpr_lit:
+			if self.cmd == lz_lit:
 				self.doLiteral()
-			elif self.cmd == cpr_iter:
+			elif self.cmd == lz_iter:
 				self.doIter()
-			elif self.cmd == cpr_alt:
+			elif self.cmd == lz_alt:
 				self.doAlt()
-			elif self.cmd == cpr_zeros:
+			elif self.cmd == lz_zeros:
 				self.doZeros()
 				
 			else: # repeaters
@@ -654,11 +654,11 @@ class Decompressed:
 					self.next()
 					self.displacement += self.byte
 				
-				if self.cmd == cpr_flip:
+				if self.cmd == lz_flip:
 					self.doFlip()
-				elif self.cmd == cpr_reverse:
+				elif self.cmd == lz_reverse:
 					self.doReverse()
-				else: # cpr_repeat
+				else: # lz_repeat
 					self.doRepeat()
 			
 			self.address += 1
@@ -666,7 +666,7 @@ class Decompressed:
 	
 	
 	def getCurByte(self):
-		self.byte = ord(self.cpr[self.start+self.address])
+		self.byte = ord(self.lz[self.start+self.address])
 	
 	def next(self):
 		self.address += 1
@@ -1155,18 +1155,44 @@ def decompress_title():
 		gfx = Decompressed( rom, 'horiz', None, address )
 		to_file(filename, gfx.output)
 
+def decompress_tilesets():
+	tileset_headers = 0x4d596
+	len_tileset = 15
+	num_tilesets = 0x25
+	for tileset in range(num_tilesets):
+		ptr = tileset*len_tileset + tileset_headers
+		address = (ord(rom[ptr])*0x4000) + (((ord(rom[ptr+1]))+ord(rom[ptr+2])*0x100)&0x3fff)
+		tiles = Decompressed( rom, 'horiz', None, address )
+		filename = '../gfx/tilesets/'+str(tileset).zfill(2)+'.2bpp'
+		to_file( filename, tiles.output )
+		#print '(' + hex(address) + ', '+ hex(address+tiles.address+1) + '),'
+
+misc = [
+	('player', 0x2BA1A, 'vert'),
+	('dude', 0x2BBAA, 'vert'),
+	('town_map', 0xF8BA0, 'horiz'),
+	('pokegear', 0x1DE2E4, 'horiz'),
+	('pokegear_sprites', 0x914DD, 'horiz'),
+]
+def decompress_misc():
+	for name, address, mode in misc:
+		filename = '../gfx/misc/' + name + '.2bpp'
+		gfx = Decompressed( rom, mode, None, address )
+		to_file(filename, gfx.output)
 
 def decompress_all(debug = False):
 	"""decompress all known compressed data in baserom"""
+	
 	#mkdir_p('../gfx/')
 	#mkdir_p('../gfx/frontpics/')
-	#mkdir_p('../gfx/backpics/')
 	#mkdir_p('../gfx/anim/')
+	#mkdir_p('../gfx/backpics/')
 	#mkdir_p('../gfx/trainers/')
 	#mkdir_p('../gfx/fx/')
-	#mkdir_p('../gfx/misc/')
 	#mkdir_p('../gfx/intro/')
 	#mkdir_p('../gfx/title/')
+	#mkdir_p('../gfx/tilesets/')
+	#mkdir_p('../gfx/misc/')
 	
 	if debug: print 'fronts'
 	decompress_monsters(front)
@@ -1185,9 +1211,16 @@ def decompress_all(debug = False):
 	
 	if debug: print 'intro'
 	decompress_intro()
+	
 	if debug: print 'title'
 	decompress_title()
 	
+	if debug: print 'tilesets'
+	decompress_tilesets()
+	
+	if debug: print 'misc'
+	decompress_misc()
+	
 	return
 
 
@@ -1212,9 +1245,9 @@ def compress_file(filein, fileout, mode = 'horiz'):
 	image = f.read()
 	f.close()
 	
-	cpr = Compressed(image, mode)
+	lz = Compressed(image, mode)
 	
-	to_file(fileout, cpr.output)
+	to_file(fileout, lz.output)
 
 
 
@@ -1229,11 +1262,24 @@ def compress_monster_frontpic(id, fileout):
 	anim = open(fanim, 'rb').read()
 	image = pic + anim
 	
-	cpr = Compressed(image, mode, 5)
+	lz = Compressed(image, mode, 5)
 	
-	out = '../gfx/frontpics/cpr/' + str(id).zfill(3) + '.cpr'
+	out = '../gfx/frontpics/lz/' + str(id).zfill(3) + '.lz'
 	
-	to_file(out, cpr.output)
+	to_file(out, lz.output)
+
+
+
+def get_uncompressed_gfx(start, num_tiles, filename):
+	"""grab tiles directly from rom and write to file"""
+	bytes_per_tile = 0x10
+	length = num_tiles*bytes_per_tile
+	end = start + length
+	rom = load_rom()
+	image = []
+	for address in range(start,end):
+		image.append(ord(rom[address]))
+	to_file(filename, image)
 
 
 
@@ -1255,14 +1301,23 @@ if __name__ == "__main__":
 		decompress_from_address(addr, fileout, mode)
 		if debug: print 'decompressed to ' + args.arg2 + ' from ' + hex(int(args.arg1,16)) + '!'
 		
-	elif args.cmd == 'cpr':
-		# python gfx.py cpr [filein] [fileout] [mode]
+	elif args.cmd == 'lz':
+		# python gfx.py lz [filein] [fileout] [mode]
 		filein = args.arg1
 		fileout = args.arg2
 		mode = args.arg3
 		compress_file(filein, fileout, mode)
 		if debug: print 'compressed ' + filein + ' to ' + fileout + '!'
-		
+	
+	elif args.cmd == 'lzf':
+		# python gfx.py lzf [id] [fileout]
+		compress_monster_frontpic(int(args.arg1), args.arg2)
+	
+	elif args.cmd == 'un':
+		# python gfx.py un [address] [num_tiles] [filename]
+		get_uncompressed_gfx(int(args.arg1,16), int(args.arg2), args.arg3)
+	
 	else:
+		# python gfx.py
 		decompress_all()
 		if debug: print 'decompressed known gfx to ../gfx/!'