#!/usr/env/env python 
#-*- coding: cp936 -*- 
''''' 
add Head Infomation for pcm file 
''' 
import sys 
import struct 
import os 
__author__ = 'bob_hu, hewitt924@gmail.com' 
__date__ = 'Dec 19,2011' 
__update__ = 'Dec 19,2011' 
def geneHeadInfo(sampleRate,bits,sampleNum): 
  ''''' 
  生成頭信息，需要采樣率，每個(gè)采樣的位數(shù)，和整個(gè)wav的采樣的字節(jié)數(shù) 
  ''' 
  rHeadInfo = '\x52\x49\x46\x46' 
  fileLength = struct.pack('i',sampleNum + 36) 
  rHeadInfo += fileLength 
  rHeadInfo += '\x57\x41\x56\x45\x66\x6D\x74\x20\x10\x00\x00\x00\x01\x00\x01\x00' 
  rHeadInfo += struct.pack('i',sampleRate) 
  rHeadInfo += struct.pack('i',sampleRate * bits / 8) 
  rHeadInfo += '\x02\x00' 
  rHeadInfo += struct.pack('H',bits) 
  rHeadInfo += '\x64\x61\x74\x61' 
  rHeadInfo += struct.pack('i',sampleNum) 
  return rHeadInfo 
if __name__ == '__main__': 
  if len(sys.argv) != 5: 
    print "usage: python %s inFile sampleRate bits outFile" % sys.argv[0] 
    sys.exit(1) 
  fout = open(sys.argv[4],'wb') #用二進(jìn)制的寫入模式 
  #fout.write(struct.pack('4s','\x66\x6D\x74\x20'))
  #寫入一個(gè)長(zhǎng)度為4的串，這個(gè)串的二進(jìn)制內(nèi)容為 66 6D 74 20 
  #Riff_flag,afd,fad,afdd, = struct.unpack('4c',fin.read(4))
  #讀入四個(gè)字節(jié)，每一個(gè)都解析成一個(gè)字母 
  #open(sys.argv[4],'wb').write(struct.pack('4s','fmt '))
  #將字符串解析成二進(jìn)制后再寫入 
  #open(sys.argv[4],'wb').write('\x3C\x9C\x00\x00\x57')
  #直接寫入二進(jìn)制內(nèi)容：3C 9C 00 00 57 
  #fout.write(struct.pack('i',6000)) #寫入6000的二進(jìn)制形式 
  #check whether inFile has head-Info 
  fin = open(sys.argv[1],'rb') 
  Riff_flag, = struct.unpack('4s',fin.read(4)) 
  if Riff_flag == 'RIFF': 
    print "%s 有頭信息" % sys.argv[1] 
    fin.close() 
    sys.exit(0) 
  else: 
    print "%s 沒(méi)有頭信息" % sys.argv[1] 
    fin.close() 
    #采樣率 
    sampleRate = int(sys.argv[2]) 
    #bit位 
    bits = int(sys.argv[3]) 
    fin = open(sys.argv[1],'rb') 
    startPos = fin.tell() 
    fin.seek(0,os.SEEK_END) 
    endPos = fin.tell() 
    sampleNum = (endPos - startPos) 
    print sampleNum 
    headInfo = geneHeadInfo(sampleRate,bits,sampleNum) 
    fout.write(headInfo) 
    fin.seek(os.SEEK_SET) 
    fout.write(fin.read()) 
    fin.close() 
    fout.close()