Apple-2-SW/asmgen
1
0
Fork 0
mirror of https://github.com/robmcmullen/asmgen.git synced 2024-06-17 04:29:46 +00:00
Code Issues Projects Releases Wiki Activity

Converted intentation to spaces instead of tabs

Browse Source
This commit is contained in:
Rob McMullen 2017-06-20 06:37:57 -07:00
parent d05bc7af85
commit 8a1de40e0e
1 changed files with 247 additions and 247 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

494
HiSprite.py
View File

@ -4,320 +4,320 @@ import sys,os,png
import argparse import argparse
class Colors: class Colors:
black,magenta,green,orange,blue,white,key = range(7) black,magenta,green,orange,blue,white,key = range(7)
def main(argv): def main(argv):
parser = argparse.ArgumentParser(description="Sprite compiler for 65C02/6502 to generate assembly code to render all shifts of the given sprite, optionally with exclusive-or drawing (if background will be non-black). Generated code has conditional compilation directives for the CC65 assembler to allow the same file to be compiled for either architecture.") parser = argparse.ArgumentParser(description="Sprite compiler for 65C02/6502 to generate assembly code to render all shifts of the given sprite, optionally with exclusive-or drawing (if background will be non-black). Generated code has conditional compilation directives for the CC65 assembler to allow the same file to be compiled for either architecture.")
parser.add_argument("-v", "--verbose", default=0, action="count") parser.add_argument("-v", "--verbose", default=0, action="count")
parser.add_argument("-t", "--tables", action="store_true", default=False, help="output only lookup tables for horizontal sprite shifts (division and modulus 7)") parser.add_argument("-t", "--tables", action="store_true", default=False, help="output only lookup tables for horizontal sprite shifts (division and modulus 7)")
parser.add_argument("-x", "--xdraw", action="store_true", default=False, help="use XOR for sprite drawing") parser.add_argument("-x", "--xdraw", action="store_true", default=False, help="use XOR for sprite drawing")
parser.add_argument("files", metavar="IMAGE", nargs="+", help="a PNG image [or a list of them]. PNG files must not have an alpha channel!") parser.add_argument("files", metavar="IMAGE", nargs="+", help="a PNG image [or a list of them]. PNG files must not have an alpha channel!")
options, extra_args = parser.parse_known_args() options, extra_args = parser.parse_known_args()
if options.tables: if options.tables:
printHorizontalLookup() printHorizontalLookup()
exit(0) exit(0)
for pngfile in options.files: for pngfile in options.files:
process(pngfile, options.xdraw) process(pngfile, options.xdraw)
def process(pngfile, xdraw=False): def process(pngfile, xdraw=False):
reader = png.Reader(pngfile) reader = png.Reader(pngfile)
try: try:
pngdata = reader.asRGB8() pngdata = reader.asRGB8()
except: except:
usage() usage()
width = pngdata[0] width = pngdata[0]
height = pngdata[1] height = pngdata[1]
pixelData = list(pngdata[2]) pixelData = list(pngdata[2])
byteWidth = width/2+1+1 # TODO: Calculate a power of two for this byteWidth = width/2+1+1 # TODO: Calculate a power of two for this
niceName = os.path.splitext(pngfile)[0].upper() niceName = os.path.splitext(pngfile)[0].upper()
disclaimer() disclaimer()
# Prologue # Prologue
print "%s: ;%d bytes per row" % (niceName,byteWidth) print "%s: ;%d bytes per row" % (niceName,byteWidth)
print "\tSAVE_AXY" print "\tSAVE_AXY"
print "\tldy PARAM0" print "\tldy PARAM0"
print "\tldx MOD7_2,y" print "\tldx MOD7_2,y"
print ".ifpC02" print ".ifpC02"
print "\tjmp (%s_JMP,x)\n" % (niceName) print "\tjmp (%s_JMP,x)\n" % (niceName)
offset_suffix = "" offset_suffix = ""
# Bit-shift jump table for 65C02 # Bit-shift jump table for 65C02
print "%s_JMP:" % (niceName) print "%s_JMP:" % (niceName)
for shift in range(0,7): for shift in range(0,7):
print "\t.addr %s_SHIFT%d" % (niceName,shift) print "\t.addr %s_SHIFT%d" % (niceName,shift)
print ".else" print ".else"
# Fast jump table routine; faster and smaller than self-modifying code # Fast jump table routine; faster and smaller than self-modifying code
print "\tlda %s_JMP+1,x" % (niceName) print "\tlda %s_JMP+1,x" % (niceName)
print "\tpha" print "\tpha"
print "\tlda %s_JMP,x" % (niceName) print "\tlda %s_JMP,x" % (niceName)
print "\tpha" print "\tpha"
print "\trts\n" print "\trts\n"
# Bit-shift jump table for generic 6502 # Bit-shift jump table for generic 6502
print "%s_JMP:" % (niceName) print "%s_JMP:" % (niceName)
for shift in range(0,7): for shift in range(0,7):
print "\t.addr %s_SHIFT%d-1" % (niceName,shift) print "\t.addr %s_SHIFT%d-1" % (niceName,shift)
print ".endif" print ".endif"
# Blitting functions # Blitting functions
print "\n" print "\n"
for shift in range(0,7): for shift in range(0,7):
# Track cycle count of the blitter. We start with fixed overhead: # Track cycle count of the blitter. We start with fixed overhead:
# SAVE_AXY + RESTORE_AXY + rts + sprite jump table # SAVE_AXY + RESTORE_AXY + rts + sprite jump table
cycleCount = 9 + 12 + 6 + 3 + 4 + 6 cycleCount = 9 + 12 + 6 + 3 + 4 + 6
print "%s_SHIFT%d:" % (niceName,shift) print "%s_SHIFT%d:" % (niceName,shift)
print "\tldx PARAM1" print "\tldx PARAM1"
cycleCount += 3 cycleCount += 3
rowStartCode,extraCycles = rowStartCalculatorCode(); rowStartCode,extraCycles = rowStartCalculatorCode();
print rowStartCode print rowStartCode
cycleCount += extraCycles cycleCount += extraCycles
spriteChunks = layoutSpriteChunk(pixelData,width,height,shift,xdraw,cycleCount) spriteChunks = layoutSpriteChunk(pixelData,width,height,shift,xdraw,cycleCount)
for row in range(height): for row in range(height):
for chunkIndex in range(len(spriteChunks)): for chunkIndex in range(len(spriteChunks)):
print spriteChunks[chunkIndex][row] print spriteChunks[chunkIndex][row]
print "\n" print "\n"
def layoutSpriteChunk(pixelData,width,height,shift,xdraw,cycleCount): def layoutSpriteChunk(pixelData,width,height,shift,xdraw,cycleCount):
colorStreams = byteStreamsFromPixels(pixelData,width,height,shift,bitsForColor,highBitForColor) colorStreams = byteStreamsFromPixels(pixelData,width,height,shift,bitsForColor,highBitForColor)
maskStreams = byteStreamsFromPixels(pixelData,width,height,shift,bitsForMask,highBitForMask) maskStreams = byteStreamsFromPixels(pixelData,width,height,shift,bitsForMask,highBitForMask)
code = generateBlitter(colorStreams,maskStreams,height,xdraw,cycleCount) code = generateBlitter(colorStreams,maskStreams,height,xdraw,cycleCount)
return code return code
def byteStreamsFromPixels(pixelData,width,height,shift,bitDelegate,highBitDelegate): def byteStreamsFromPixels(pixelData,width,height,shift,bitDelegate,highBitDelegate):
byteStreams = ["" for x in range(height)] byteStreams = ["" for x in range(height)]
byteWidth = width/2+1+1 byteWidth = width/2+1+1
for row in range(height): for row in range(height):
bitStream = "" bitStream = ""
# Compute raw bitstream for row from PNG pixels # Compute raw bitstream for row from PNG pixels
for pixelIndex in range(width): for pixelIndex in range(width):
pixel = pixelColor(pixelData,row,pixelIndex) pixel = pixelColor(pixelData,row,pixelIndex)
bitStream += bitDelegate(pixel) bitStream += bitDelegate(pixel)
# Shift bit stream as needed # Shift bit stream as needed
bitStream = shiftStringRight(bitStream,shift) bitStream = shiftStringRight(bitStream,shift)
bitStream = bitStream[:byteWidth*8] bitStream = bitStream[:byteWidth*8]
# Split bitstream into bytes # Split bitstream into bytes
bitPos = 0 bitPos = 0
byteSplits = [0 for x in range(byteWidth)] byteSplits = [0 for x in range(byteWidth)]
for byteIndex in range(byteWidth): for byteIndex in range(byteWidth):
remainingBits = len(bitStream) - bitPos remainingBits = len(bitStream) - bitPos
bitChunk = "" bitChunk = ""
if remainingBits < 0: if remainingBits < 0:
bitChunk = "0000000" bitChunk = "0000000"
else: else:
if remainingBits < 7: if remainingBits < 7:
bitChunk = bitStream[bitPos:] bitChunk = bitStream[bitPos:]
bitChunk += fillOutByte(7-remainingBits) bitChunk += fillOutByte(7-remainingBits)
else: else:
bitChunk = bitStream[bitPos:bitPos+7] bitChunk = bitStream[bitPos:bitPos+7]
bitChunk = bitChunk[::-1] bitChunk = bitChunk[::-1]
# Determine palette bit from first pixel on each row # Determine palette bit from first pixel on each row
highBit = highBitDelegate(pixelData[row][0]) highBit = highBitDelegate(pixelData[row][0])
byteSplits[byteIndex] = highBit + bitChunk byteSplits[byteIndex] = highBit + bitChunk
bitPos += 7 bitPos += 7
byteStreams[row] = byteSplits; byteStreams[row] = byteSplits;
return byteStreams return byteStreams
def generateBlitter(colorStreams,maskStreams,height,xdraw,baseCycleCount): def generateBlitter(colorStreams,maskStreams,height,xdraw,baseCycleCount):
byteWidth = len(colorStreams[0]) byteWidth = len(colorStreams[0])
spriteChunks = [["" for y in range(height)] for x in range(byteWidth)] spriteChunks = [["" for y in range(height)] for x in range(byteWidth)]
cycleCount = baseCycleCount cycleCount = baseCycleCount
optimizationCount = 0 optimizationCount = 0
for row in range(height): for row in range(height):
byteSplits = colorStreams[row] byteSplits = colorStreams[row]
# Generate blitting code # Generate blitting code
for chunkIndex in range(len(byteSplits)): for chunkIndex in range(len(byteSplits)):
# Optimization # Optimization
if byteSplits[chunkIndex] != "00000000" and \ if byteSplits[chunkIndex] != "00000000" and \
byteSplits[chunkIndex] != "10000000": byteSplits[chunkIndex] != "10000000":
# Store byte into video memory # Store byte into video memory
if xdraw: if xdraw:
spriteChunks[chunkIndex][row] = \ spriteChunks[chunkIndex][row] = \
"\tlda (SCRATCH0),y\n" + \ "\tlda (SCRATCH0),y\n" + \
"\teor #%%%s\n" % byteSplits[chunkIndex] + \ "\teor #%%%s\n" % byteSplits[chunkIndex] + \
"\tsta (SCRATCH0),y\n"; "\tsta (SCRATCH0),y\n";
cycleCount += 5 + 2 + 6 cycleCount += 5 + 2 + 6
else: else:
spriteChunks[chunkIndex][row] = \ spriteChunks[chunkIndex][row] = \
"\tlda #%%%s\n" % byteSplits[chunkIndex] + \ "\tlda #%%%s\n" % byteSplits[chunkIndex] + \
"\tsta (SCRATCH0),y\n"; "\tsta (SCRATCH0),y\n";
cycleCount += 2 + 6 cycleCount += 2 + 6
else: else:
optimizationCount += 1 optimizationCount += 1
# Increment indices # Increment indices
if chunkIndex == len(byteSplits)-1: if chunkIndex == len(byteSplits)-1:
spriteChunks[chunkIndex][row] += "\n" spriteChunks[chunkIndex][row] += "\n"
else: else:
spriteChunks[chunkIndex][row] += "\tiny" spriteChunks[chunkIndex][row] += "\tiny"
cycleCount += 2 cycleCount += 2
# Finish the row # Finish the row
if row<height-1: if row<height-1:
rowStartCode,extraCycles = rowStartCalculatorCode() rowStartCode,extraCycles = rowStartCalculatorCode()
spriteChunks[chunkIndex][row] += "\tinx\n" + rowStartCode; spriteChunks[chunkIndex][row] += "\tinx\n" + rowStartCode;
cycleCount += 2 + extraCycles cycleCount += 2 + extraCycles
else: else:
spriteChunks[chunkIndex][row] += "\tRESTORE_AXY\n" spriteChunks[chunkIndex][row] += "\tRESTORE_AXY\n"
spriteChunks[chunkIndex][row] += "\trts\t;Cycle count: %d, Optimized %d rows." % (cycleCount,optimizationCount) + "\n" spriteChunks[chunkIndex][row] += "\trts\t;Cycle count: %d, Optimized %d rows." % (cycleCount,optimizationCount) + "\n"
return spriteChunks return spriteChunks
def rowStartCalculatorCode(): def rowStartCalculatorCode():
return \ return \
"\tlda HGRROWS_H1,x\n" + \ "\tlda HGRROWS_H1,x\n" + \
"\tsta SCRATCH1\n" + \ "\tsta SCRATCH1\n" + \
"\tlda HGRROWS_L,x\n" + \ "\tlda HGRROWS_L,x\n" + \
"\tsta SCRATCH0\n" + \ "\tsta SCRATCH0\n" + \
"\tldy PARAM0\n" + \ "\tldy PARAM0\n" + \
"\tlda DIV7_2,y\n" + \ "\tlda DIV7_2,y\n" + \
"\ttay\n", 4 + 3 + 4 + 3 + 3 + 4 + 2; "\ttay\n", 4 + 3 + 4 + 3 + 3 + 4 + 2;
def fillOutByte(numBits): def fillOutByte(numBits):
filler = "" filler = ""
for bit in range(numBits): for bit in range(numBits):
filler += "0" filler += "0"
return filler return filler
def shiftStringRight(string,shift): def shiftStringRight(string,shift):
if shift==0: if shift==0:
return string return string
shift *=2 shift *=2
result = "" result = ""
for i in range(shift): for i in range(shift):
result += "0" result += "0"
result += string result += string
return result return result
def bitsForColor(pixel): def bitsForColor(pixel):
if pixel == Colors.black: if pixel == Colors.black:
return "00" return "00"
else: else:
if pixel == Colors.white: if pixel == Colors.white:
return "11" return "11"
else: else:
if pixel == Colors.green or pixel == Colors.orange: if pixel == Colors.green or pixel == Colors.orange:
return "01" return "01"
# blue or magenta # blue or magenta
return "10" return "10"
def bitsForMask(pixel): def bitsForMask(pixel):
if pixel == Colors.black: if pixel == Colors.black:
return "00" return "00"
return "11" return "11"
def highBitForColor(pixel): def highBitForColor(pixel):
# Note that we prefer high-bit white because blue fringe is less noticeable than magenta. # Note that we prefer high-bit white because blue fringe is less noticeable than magenta.
highBit = "0" highBit = "0"
if pixel == Colors.orange or pixel == Colors.blue or pixel == Colors.white: if pixel == Colors.orange or pixel == Colors.blue or pixel == Colors.white:
highBit = "1" highBit = "1"
return highBit return highBit
def highBitForMask(pixel): def highBitForMask(pixel):
return "1" return "1"
def pixelColor(pixelData,row,col): def pixelColor(pixelData,row,col):
r = pixelData[row][col*3] r = pixelData[row][col*3]
g = pixelData[row][col*3+1] g = pixelData[row][col*3+1]
b = pixelData[row][col*3+2] b = pixelData[row][col*3+2]
color = Colors.black color = Colors.black
if r==255 and g==0 and b==255: if r==255 and g==0 and b==255:
color = Colors.magenta color = Colors.magenta
else: else:
if r==0 and g==255 and b==0: if r==0 and g==255 and b==0:
color = Colors.green color = Colors.green
else: else:
if r==0 and g==0 and b==255: if r==0 and g==0 and b==255:
color = Colors.blue color = Colors.blue
else: else:
if r==255 and g>0 and b==0: if r==255 and g>0 and b==0:
color = Colors.orange color = Colors.orange
else: else:
if r==255 and g==255 and b==255: if r==255 and g==255 and b==255:
color = Colors.white color = Colors.white
else: else:
if r==g and r==b and r!=0 and r!=255: # Any gray is chroma key if r==g and r==b and r!=0 and r!=255: # Any gray is chroma key
color = Colors.key color = Colors.key
return color return color
def printHorizontalLookup(): def printHorizontalLookup():
disclaimer() disclaimer()
print "DIV7_2:" print "DIV7_2:"
for pixel in range(140): for pixel in range(140):
print "\t.byte $%02x" % ((pixel / 7)*2) print "\t.byte $%02x" % ((pixel / 7)*2)
print "\n\nMOD7_2:" print "\n\nMOD7_2:"
for pixel in range(140): for pixel in range(140):
print "\t.byte $%02x" % ((pixel % 7)*2) print "\t.byte $%02x" % ((pixel % 7)*2)
def disclaimer(): def disclaimer():
print ''' print '''
; This file was generated by HiSprite.py, a sprite compiler by Quinn Dunki. ; This file was generated by HiSprite.py, a sprite compiler by Quinn Dunki.
; If you feel the need to modify this file, you are probably doing it wrong. ; If you feel the need to modify this file, you are probably doing it wrong.
''' '''
return return
if __name__ == "__main__": if __name__ == "__main__":
main(sys.argv[1:]) main(sys.argv[1:])