Test Image Bitmap Unswizzle and Output

import png, struct, os, math, pdb
import sys

xy = False  #Set to false for yx swizzle order

def chunks(l, n):
    """Yield successive n-sized chunks from l."""
    for i in range(0, len(l), n):
        yield l[i:i + n]

def unswizzle(writebuf, readbuf, writeoffset, segwidth, segheight, datawidth):
    global readoffset, buffer
##    print(segwidth, segheight)
    if segwidth == 2 and segheight == 2:
        if xy:
            writebuf[writeoffset:writeoffset+2] = readbuf[readoffset:readoffset+2]
            writebuf[writeoffset + datawidth:writeoffset + datawidth + 2] = readbuf[readoffset + 2:readoffset + 4]
        else:
            writebuf[writeoffset] = readbuf[readoffset]
            writebuf[writeoffset + datawidth] = readbuf[readoffset + 1]
            writebuf[writeoffset + 1] = readbuf[readoffset + 2]
            writebuf[writeoffset + datawidth + 1] = readbuf[readoffset + 3]
        readoffset += 4
    else:
        if xy:
            unswizzle(writebuf, readbuf, writeoffset, segwidth // 2, segheight // 2, datawidth)
            unswizzle(writebuf, readbuf, writeoffset + segwidth // 2, segwidth // 2, segheight // 2, datawidth)
            unswizzle(writebuf, readbuf, writeoffset + datawidth * (segheight // 2), segwidth // 2, segheight // 2, datawidth)
            unswizzle(writebuf, readbuf, writeoffset + datawidth * (segheight // 2) + segwidth // 2, segwidth // 2, segheight // 2, datawidth)
        else:
            unswizzle(writebuf, readbuf, writeoffset, segwidth // 2, segheight // 2, datawidth)
            unswizzle(writebuf, readbuf, writeoffset + datawidth * (segheight // 2), segwidth // 2, segheight // 2, datawidth)
            unswizzle(writebuf, readbuf, writeoffset + segwidth // 2, segwidth // 2, segheight // 2, datawidth)
            unswizzle(writebuf, readbuf, writeoffset + datawidth * (segheight // 2) + segwidth // 2, segwidth // 2, segheight // 2, datawidth)

with open('1ch6030.ykcmp.dec', 'rb') as f:
    palette = f.read(0x400)
    bitmap = f.read(0x40000)
width = 512
height = 512
buffer = bytearray(len(bitmap))         #Output buffer
readoffset = 0                          #Input position counter
if width == height:
    unswizzle(buffer, bitmap, 0, width, height, width)
elif width > height:
    for w in range(0, width, height):
        unswizzle(buffer, bitmap, w, height, height, width)
elif height > width:
    for h in range(0, height * width, width ** 2):
        unswizzle(buffer, bitmap, h, width, width, width)
else:
    print('error')
p = list(chunks(buffer, width))
with open('output.png', 'wb') as f:
    w = png.Writer(len(p[0]), len(p), palette=list(chunks(palette, 4)))
    w.write(f, p)