python pcm音频添加头转成Wav格式文件的方法_Python

python pcm音频添加头转成Wav格式文件的方法

2021-05-13 01:00benhuo931115 Python

今天小编就为大家分享一篇python pcm音频添加头转成Wav格式文件的方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧

如下所示：

				?

									''''' 

									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): 

									  ''''' 

									  生成头信息，需要采样率，每个采样的位数，和整个wav的采样的字节数 

									  '''

									  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') #用二进制的写入模式 

									  #fout.write(struct.pack('4s','\x66\x6D\x74\x20'))

									  #写入一个长度为4的串，这个串的二进制内容为 66 6D 74 20 

									  #Riff_flag,afd,fad,afdd, = struct.unpack('4c',fin.read(4))

									  #读入四个字节，每一个都解析成一个字母 

									  #open(sys.argv[4],'wb').write(struct.pack('4s','fmt '))

									  #将字符串解析成二进制后再写入 

									  #open(sys.argv[4],'wb').write('\x3C\x9C\x00\x00\x57')

									  #直接写入二进制内容：3C 9C 00 00 57 

									  #fout.write(struct.pack('i',6000)) #写入6000的二进制形式 

									  #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 没有头信息" % 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()