quickdev16/scripts/conv_zip.py
2009-10-19 11:52:20 +02:00

110 lines
2.3 KiB
Python

import binascii
import os
import sys
import time
import shutil
LEN = 2**16
huffman = False
TARGET=os.getcwd()
SOURCE=sys.argv[1]
DEFLATE=os.path.basename(sys.argv[1]) + ".deflate"
if os.path.isfile("rom.zip"):
os.unlink("rom.zip")
os.system("wine kzip.exe rom /s1 %s" % SOURCE)
os.system("wine DeflOpt.exe /a rom.zip")
os.system("ruby zip2raw.rb rom.zip")
if os.path.isfile("rom.zip"):
os.unlink("rom.zip")
data = open(DEFLATE).read()
os.unlink(DEFLATE)
zip_size = len(data)
cfile = open("/tmp/loader.c","w")
hfile = open("/tmp/loader.h","w")
parts = []
cnt = len(data) / ((2**15) -1 )
r = len(data) - (cnt * ((2**15) -1))
for i in range(0, cnt):
parts.append(((2**15) -1 ))
parts.append(r)
hfile.write('''/*
File: %s
Time: %s
*/
#ifndef __FIFO_H__
#define __FIFO_H__
#define ROM_ZIP_SIZE %i
#define ROM_BUFFER_CNT %i
''' % (os.path.basename(SOURCE),time.strftime("%a, %d %b %Y %H:%M:%S",
time.localtime()),zip_size,len(parts)))
for idx,val in enumerate(parts):
hfile.write('#define ROM_BUFFER_SIZE%02i %i\n' % (idx+1,val))
hfile.write('\n#endif\n')
hfile.close()
cfile.write('''/*
File: %s
Time: %s
*/
#include <avr/pgmspace.h>
#include <loader.h>
''')
addr = 0
for idx,val in enumerate(parts):
cfile.write('''
const char _rom%02i[ROM_BUFFER_SIZE%02i] PROGMEM = {
''' % (idx+1,idx+1))
l = addr
h = addr + parts[idx]
addr+= parts[idx]
for idx,c in enumerate(data[l:h]):
c = ord(c)
if idx<len(data)-1:
cfile.write("0x%02x," % c)
else:
cfile.write("0x%02x" % c)
if idx and idx%16==0:
cfile.write("\n")
cfile.write('''
};
''')
cfile.write('PGM_VOID_P _rom[ROM_BUFFER_CNT]= {')
for idx,val in enumerate(parts):
if idx<len(parts)-1:
cfile.write('''&_rom%02i,''' % (idx+1))
else:
cfile.write('''&_rom%02i''' % (idx+1))
cfile.write('''};
''')
cfile.write('const int _rom_size[ROM_BUFFER_CNT] = {')
for idx,val in enumerate(parts):
if idx<len(parts)-1:
cfile.write('''%i,''' % (val))
else:
cfile.write('''%i''' % (val))
cfile.write('''};
''')
cfile.close()
shutil.copy("/tmp/loader.c", os.path.join(TARGET,"loader.c"))
shutil.copy("/tmp/loader.h", os.path.join(TARGET,"loader.h"))
print "wrote loader.c"
print "wrote loader.h"