Python:PortAudio + Opus编码/解码

时间:2013-07-18 16:09:41

标签: python portaudio pyaudio opus

我正在使用Pyaudio从我的麦克风中捕获音频,并尝试使用opus编解码器对其进行编码/解码。我正在使用SvartalF(https://github.com/svartalf/python-opus)制作的libopus绑定。

这是我的代码:

import pyaudio
from opus import encoder, decoder

def streaming(p):
    chunk = 960
    FORMAT = pyaudio.paInt16
    CHANNELS = 1
    RATE = 48000
    streamin = p.open(format = FORMAT,
            channels = CHANNELS, 
            rate = RATE, 
            input = True,
            input_device_index = 7,
            frames_per_buffer = chunk)
    streamout = p.open(format = FORMAT,
            channels = CHANNELS, 
            rate = 48000, 
            output = True,
            output_device_index = p.get_default_input_device_info()["index"],
            frames_per_buffer = chunk)
    enc = encoder.Encoder(RATE,CHANNELS,'voip')
    dec = decoder.Decoder(RATE,CHANNELS)
    data = []
    for i in xrange(100):
        data.append(streamin.read(chunk*2))
    streamout.write(''.join(data))
    encdata = []
    for x in data:
        encdata.append(enc.encode(x,chunk))
    print "DATA LENGTH :", len(''.join(data))
    print "ENCDATA LENGTH :", len(''.join(encdata))
    decdata = ''
    for x in encdata:
        decdata += dec.decode(x,chunk)
    print "DECDATA LENGTH :", len(decdata)
    streamout.write(decdata)
    streamin.close()
    streamout.close()


p = pyaudio.PyAudio()
streaming(p)
p.terminate()

我必须在chunk*2chunk中放置data.append(streamin.read(chunk*2))而不是DECDATA LENGTH == DATA LENGTH*2,我不知道为什么。

输出:

DATA LENGTH :    384000  
ENCDATA LENGTH : 12865  
DECDATA LENGTH : 384000

没有编码/解码,第一个streamout.write(''.join(data))完美无缺。通过编码/解码,streamout.write(decdata)有点可行,但有很多混乱。

我在这里做错了什么?

2 个答案:

答案 0 :(得分:2)

这似乎是由解码方法中的python-opus中的错误引起的。

根据Opus API opus_decode 会返回已解码的样本数。 python绑定假设它将完全填充它传入的结果缓冲区,因此每组解码样本都会附加一个静音。这种沉默导致低框架尺寸的开裂和较高框架尺寸的断续。虽然文档没有说明任何内容,但似乎返回的数字是每个频道。

将行150 of opus/api/decoder.py更改为以下内容为我解决了这个问题:

    return array.array('h', pcm[:result*channels]).tostring()

如果您需要使用 decode_float 方法,可能需要进行相同的更改。

答案 1 :(得分:-1)

只需将输出切成两半并取得第一部分。 通过反复试验,我发现这个解决方案令人满意。

from opus import decoder as opus_decoder
from opus import encoder as opus_encoder

class OpusCodec():
    def __init__(self, *args, **kwargs):
        self.chunk = 960
        self.channels = 1
        self.rate = 48000
        self.encoder = opus_encoder.Encoder(
            self.rate, 
            self.channels,
            opus_encoder.APPLICATION_TYPES_MAP['voip']
        )
        self.decoder = opus_decoder.Decoder(
            self.rate, 
            self.channels, 
        )

    def encode(self, data, **kwargs):
        if not 'frame_size' in kwargs:
            kwargs['frame_size'] = self.chunk
        out = self.encoder.encode(data, frame_size=self.chunk)
        return out

    def decode(self, data, **kwargs):
        if not 'frame_size' in kwargs:
            kwargs['frame_size'] = self.chunk
        out = self.decoder.decode(data, **kwargs)
        return out[0:int(len(out)/2)] # hackety hack :D