PEP Cleanup and remove abs paths

This commit is contained in:
optixx 2016-02-09 22:16:01 +01:00
parent 69616740eb
commit ce8042c926

View File

@ -8,8 +8,8 @@ LEN = 2**16
huffman = False
TARGET = os.getcwd()
SOURCE = sys.argv[1]
HUFFMAN_ENCODER="/Users/david/Devel/arch/avr/code/quickdev16/packages/huffman/huffman-encode"
LOC = os.path.dirname(os.path.realpath(__file__))
HUFFMAN_ENCODER = os.path.join(LOC, "..", "tools", "huffman", "huffman-encode")
data = open(SOURCE, "r").read()
print "Load %s, %i bytes" % (SOURCE, len(data))
@ -21,7 +21,7 @@ print "RLE crunch (%i) bytes" % (len(data))
rle_size = len(data)
huffman_size = 0
if huffman == True:
if huffman:
binfile = open("/tmp/loader.rle", "w")
binfile.write(data)
binfile.close()
@ -35,17 +35,14 @@ if huffman == True:
os.unlink("/tmp/loader.rle")
os.unlink("/tmp/loader.rle.hfm")
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
@ -57,14 +54,16 @@ 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)))
''' % (
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('''/*
@ -75,7 +74,6 @@ Time: %s
#include <loader.h>
''')
addr = 0
for idx, val in enumerate(parts):
cfile.write('''
@ -113,9 +111,6 @@ for idx,val in enumerate(parts):
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"))