98 lines
2.4 KiB
Python
98 lines
2.4 KiB
Python
|
#!/usr/bin/env python
|
||
|
|
import re
|
||
|
|
import sys
|
||
|
|
|
||
|
|
class Node:
|
||
|
|
def __init__(self, depth):
|
||
|
|
self.depth = depth
|
||
|
|
self.children = {}
|
||
|
|
|
||
|
|
def to_bin(s):
|
||
|
|
res = []
|
||
|
|
for i in range(0, len(s), 8):
|
||
|
|
x = s[i:i+8]
|
||
|
|
x += '0'*(8 - len(x))
|
||
|
|
a = 0
|
||
|
|
for j in range(8):
|
||
|
|
a *= 2
|
||
|
|
a += ord(x[j]) - ord('0')
|
||
|
|
res.append(a) #chr(a))
|
||
|
|
return res
|
||
|
|
|
||
|
|
nodes = []
|
||
|
|
|
||
|
|
def insert(node, sym, binpat, nbits, pidx):
|
||
|
|
if pidx == len(binpat) - 1:
|
||
|
|
#assert(binpat[pidx] not in node.children)
|
||
|
|
mx = (8 - (nbits & 0x7)) & 0x7;
|
||
|
|
#print "last", bin(binpat[pidx]), mx
|
||
|
|
for i in range(1 << mx):
|
||
|
|
node.children[binpat[pidx] + i] = sym
|
||
|
|
else:
|
||
|
|
if binpat[pidx] not in node.children:
|
||
|
|
node.children[binpat[pidx]] = -len(nodes)
|
||
|
|
nextnode = Node(pidx + 1)
|
||
|
|
nodes.append(nextnode)
|
||
|
|
else:
|
||
|
|
nextnode = nodes[-node.children[binpat[pidx]]]
|
||
|
|
insert(nextnode, sym, binpat, nbits, pidx + 1)
|
||
|
|
|
||
|
|
symbol_tbl = [(None, 0) for i in range(257)]
|
||
|
|
tables = {}
|
||
|
|
|
||
|
|
root = Node(0)
|
||
|
|
nodes.append(root)
|
||
|
|
|
||
|
|
for line in sys.stdin:
|
||
|
|
m = re.match(r'.*\(\s*(\d+)\) ([|01]+) \[(\d+)\] .*', line)
|
||
|
|
if m:
|
||
|
|
#print m.group(1), m.group(2), m.group(3)
|
||
|
|
sym = int(m.group(1))
|
||
|
|
pat = re.sub(r'\|', '', m.group(2))
|
||
|
|
nbits = int(m.group(3))
|
||
|
|
assert(len(pat) == nbits)
|
||
|
|
binpat = to_bin(pat)
|
||
|
|
assert(len(binpat) == (nbits+7)/8)
|
||
|
|
symbol_tbl[sym] = (binpat, nbits)
|
||
|
|
#print "Inserting", sym
|
||
|
|
insert(root, sym, binpat, nbits, 0)
|
||
|
|
|
||
|
|
print '''\
|
||
|
|
typedef struct {
|
||
|
|
size_t nbits;
|
||
|
|
uint8_t code[4];
|
||
|
|
} nghttp2_huff_sym;
|
||
|
|
'''
|
||
|
|
|
||
|
|
print '''\
|
||
|
|
nghttp2_huff_sym huff_sym_table[] = {'''
|
||
|
|
for i in range(257):
|
||
|
|
pat = list(symbol_tbl[i][0])
|
||
|
|
pat += [0]*(4 - len(pat))
|
||
|
|
print '''\
|
||
|
|
{{ {}, {{ {} }} }}{}\
|
||
|
|
'''.format(symbol_tbl[i][1], ', '.join([str(k) for k in pat]),
|
||
|
|
',' if i < 256 else '')
|
||
|
|
print '};'
|
||
|
|
print ''
|
||
|
|
|
||
|
|
print '''int16_t huff_decode_table[][256] = {'''
|
||
|
|
for j in range(len(nodes)):
|
||
|
|
node = nodes[j]
|
||
|
|
print '/* {} */'.format(j)
|
||
|
|
print '{'
|
||
|
|
for i in range(256):
|
||
|
|
if i in node.children:
|
||
|
|
sys.stdout.write('''\
|
||
|
|
{}{}'''.format(node.children[i], ',' if i < 255 else ''))
|
||
|
|
else:
|
||
|
|
sys.stdout.write(''' NGHTTP2_HD_HUFF_NO_ENT,''')
|
||
|
|
if (i+1)&0x7 == 0:
|
||
|
|
print ''
|
||
|
|
sys.stdout.write('}')
|
||
|
|
if j == len(nodes) - 1:
|
||
|
|
print ''
|
||
|
|
else:
|
||
|
|
print ','
|
||
|
|
print '};'
|