标题: 哈夫曼py程序 [打印本页]
作者: 航123    时间: 2018-6-24 10:59
标题: 哈夫曼py程序
程序可参考

py源程序如下:
  1. import hufcode
  2. import statistic
  3. import six
  4. def decompress():
  5.     print("please input youer filename")
  6.     inputfilename=input()
  7.     f = open(inputfilename,'rb')
  8.     filedata1 = f.read()
  9.     filedata=bytearray(filedata1)
  10.     filesize = f.tell()
  11.     #print(filedata)


  14.     a1 = filedata[0]
  15.     a2 = filedata[1]
  16.     a3 = filedata[2]
  17.     a4 = filedata[3]
  18.     j = 0
  19.     j = j|a1
  20.     j = j<<8
  21.     j = j|a2
  22.     j = j<<8
  23.     j = j|a3
  24.     j = j<<8
  25.     j = j|a4

  27.     leaf_node_size = j

  29.    
  30.    
  31.     char_freq = {}
  32.     for i in range(leaf_node_size):
  33.         c = filedata[4+i*5+0]

  35.         
  36.         a1 = filedata[4+i*5+1]
  37.         a2 = filedata[4+i*5+2]
  38.         a3 = filedata[4+i*5+3]
  39.         a4 = filedata[4+i*5+4]
  40.         j=0
  41.         j = j|a1
  42.         j = j<<8
  43.         j = j|a2
  44.         j = j<<8
  45.         j = j|a3
  46.         j = j<<8
  47.         j = j|a4
  48.         #print (c, j)
  49.         char_freq[c] = j
  50.         j=0

  52.     list1=[]
  53.     list_all=[]
  54.     for x in char_freq.keys():
  55.         list1.append((x,char_freq[x]))

  57.     list1.sort()
  58.     list_all=[item[0]for item in list1]
  59.     #print(list1)
  60.     #print(list_all)

  62.     nodes = hufcode.createNodes([item[1] for item in list1])   #出现次数
  63.     root = hufcode.createHuffmanTree(nodes)                #叶子节点
  64.     codes = hufcode.huffmanEncoding(nodes,root)            #编码


  67.     #print(codes)
  68.    

  70.     code1=''
  71.     for x in range(leaf_node_size*5+4,filesize):
  72.         c = filedata[x]
  73.         for i in range(8):
  74.             if c&128:
  75.                 code1 = code1 +'1'
  76.             else:
  77.                 code1 = code1 + '0'
  78.             c = c<<1
  79.     #print(code1)

  81.     newcode=dict(zip(codes,list_all))
  82.     print(newcode)
  83.    
  84.     #print(newcode.get('1000'))
  85.     #print(bytes((int(newcode.get('1000')),)).decode("ascii"))
  86.     i=0
  87.     res=''
  88.     while i < len(code1)-24:
  89.         j = i+1
  90.         while j < len(code1): #获取某一字符的code
  91.             if code1[i:j] in newcode.keys():
  92.                 break
  93.             j += 1
  94.         #print(code1[i:j])
  95.         res =res + bytes((int(newcode.get(str(code1[i:j]))),)).decode('ascii')
  96.         #print(res)
  97.         i = j
  98.    
  99.     sub_code = code1[-16:-8]
  100.     last_length = 0
  101.     for i in range(8):
  102.         last_length = last_length<<1
  103.         if sub_code[i] == '1':
  104.             last_length = last_length|1
  105.    
  106.     code1=code1[-24:-16]+code1[-8:-8+last_length]

  108.     i=0
  109.     while i < len(code1):
  110.         j = i+1
  111.         while j < len(code1): #获取某一字符的code
  112.             if code1[i:j] in newcode.keys():
  113.                 break
  114.             j += 1
  115.         #print(code1[i:j])
  116.         res =res + bytes((int(newcode.get(str(code1[i:j]))),)).decode('ascii')
  117.         #print(res)
  118.         i = j

  120.     print(res)
  121.    
  122.     print("\ndo you want save it? 1/yes,2/no")
  123.     ans=input()
  124.     ans=int(ans)
  125.     if ans==1:
  126.         print('please input the filename')
  127.         outfilename=input()
  128.         out = open(outfilename,"w")
  129.         out.write(res)
  130.         out.close
  131.     else:
  132.         pass
复制代码

所有资料51hei提供下载:
霍夫曼.rar (14.57 KB, 下载次数: 7)






欢迎光临 (http://www.51hei.com/bbs/) Powered by Discuz! X3.1