PEP Cleanup and remove abs paths
This commit is contained in:
parent
69616740eb
commit
ce8042c926
@ -4,15 +4,15 @@ import sys
|
||||
import time
|
||||
import shutil
|
||||
|
||||
LEN = 2**16
|
||||
LEN = 2 ** 16
|
||||
huffman = False
|
||||
TARGET=os.getcwd()
|
||||
SOURCE=sys.argv[1]
|
||||
TARGET = os.getcwd()
|
||||
SOURCE = sys.argv[1]
|
||||
LOC = os.path.dirname(os.path.realpath(__file__))
|
||||
HUFFMAN_ENCODER = os.path.join(LOC, "..", "tools", "huffman", "huffman-encode")
|
||||
|
||||
HUFFMAN_ENCODER="/Users/david/Devel/arch/avr/code/quickdev16/packages/huffman/huffman-encode"
|
||||
|
||||
data = open(SOURCE,"r").read()
|
||||
print "Load %s, %i bytes" % (SOURCE,len(data))
|
||||
data = open(SOURCE, "r").read()
|
||||
print "Load %s, %i bytes" % (SOURCE, len(data))
|
||||
data = data[:LEN]
|
||||
print "Use %i bytes" % (len(data))
|
||||
data = binascii.rlecode_hqx(data)
|
||||
@ -21,33 +21,30 @@ print "RLE crunch (%i) bytes" % (len(data))
|
||||
rle_size = len(data)
|
||||
huffman_size = 0
|
||||
|
||||
if huffman == True:
|
||||
binfile = open("/tmp/loader.rle","w")
|
||||
if huffman:
|
||||
binfile = open("/tmp/loader.rle", "w")
|
||||
binfile.write(data)
|
||||
binfile.close()
|
||||
|
||||
cmd = "%s /tmp/loader.rle" % HUFFMAN_ENCODER
|
||||
print cmd
|
||||
os.system(cmd)
|
||||
data = open("/tmp/loader.rle.hfm","r").read()
|
||||
data = open("/tmp/loader.rle.hfm", "r").read()
|
||||
print "HUFFMAN crunch (%i) bytes" % (len(data))
|
||||
huffman_size = len(data)
|
||||
os.unlink("/tmp/loader.rle")
|
||||
os.unlink("/tmp/loader.rle.hfm")
|
||||
|
||||
|
||||
cfile = open("/tmp/loader.c","w")
|
||||
hfile = open("/tmp/loader.h","w")
|
||||
|
||||
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))
|
||||
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(((2 ** 15) - 1))
|
||||
parts.append(r)
|
||||
|
||||
hfile.write('''/*
|
||||
File: %s
|
||||
File: %s
|
||||
Time: %s
|
||||
*/
|
||||
#ifndef __FIFO_H__
|
||||
@ -57,65 +54,63 @@ Time: %s
|
||||
#define ROM_RLE_SIZE %i
|
||||
#define ROM_BUFFER_CNT %i
|
||||
|
||||
''' % (os.path.basename(SOURCE),time.strftime("%a, %d %b %Y %H:%M:%S",
|
||||
time.localtime()), huffman_size, rle_size,len(parts)))
|
||||
|
||||
|
||||
for idx,val in enumerate(parts):
|
||||
hfile.write('#define ROM_BUFFER_SIZE%02i %i\n' % (idx+1,val))
|
||||
|
||||
''' % (
|
||||
os.path.basename(SOURCE),
|
||||
time.strftime("%a, %d %b %Y %H:%M:%S", time.localtime()),
|
||||
huffman_size,
|
||||
rle_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
|
||||
File: %s
|
||||
Time: %s
|
||||
*/
|
||||
#include <avr/pgmspace.h>
|
||||
#include <loader.h>
|
||||
#include <avr/pgmspace.h>
|
||||
#include <loader.h>
|
||||
''')
|
||||
|
||||
|
||||
addr = 0
|
||||
for idx,val in enumerate(parts):
|
||||
for idx, val in enumerate(parts):
|
||||
cfile.write('''
|
||||
const char _rom%02i[ROM_BUFFER_SIZE%02i] PROGMEM = {
|
||||
''' % (idx+1,idx+1))
|
||||
l = addr
|
||||
''' % (idx + 1, idx + 1))
|
||||
l = addr
|
||||
h = addr + parts[idx]
|
||||
addr+= parts[idx]
|
||||
for idx,c in enumerate(data[l:h]):
|
||||
addr += parts[idx]
|
||||
for idx, c in enumerate(data[l:h]):
|
||||
c = ord(c)
|
||||
if idx<len(data)-1:
|
||||
if idx < len(data) - 1:
|
||||
cfile.write("0x%02x," % c)
|
||||
else:
|
||||
cfile.write("0x%02x" % c)
|
||||
if idx and idx%16==0:
|
||||
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))
|
||||
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('''&_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:
|
||||
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"))
|
||||
|
||||
shutil.copy("/tmp/loader.c", os.path.join(TARGET, "loader.c"))
|
||||
shutil.copy("/tmp/loader.h", os.path.join(TARGET, "loader.h"))
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user