forked from janrueth/SiriServerCore
-
Notifications
You must be signed in to change notification settings - Fork 0
/
speex.py
125 lines (110 loc) · 6.63 KB
/
speex.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
from ctypes import *
import sys
import platform
import ctypes.util
system = platform.system()
libspeex_name = ctypes.util.find_library('speex')
if libspeex_name == None:
print "Could not find libspeex"
exit()
libspeex = CDLL(libspeex_name)
#defines copied from: http://speex.org/docs/api/speex-api-reference/group__Codec.html
SPEEX_SET_ENH = 0 #Set enhancement on/off (decoder only)
SPEEX_GET_ENH = 1 #Get enhancement state (decoder only)
SPEEX_GET_FRAME_SIZE = 3 #Obtain frame size used by encoder/decoder
SPEEX_SET_QUALITY = 4 #Set quality value
SPEEX_SET_MODE = 6 #Set sub-mode to use
SPEEX_GET_MODE = 7 #Get current sub-mode in use
SPEEX_SET_LOW_MODE = 8 #Set low-band sub-mode to use (wideband only)
SPEEX_GET_LOW_MODE = 9 #Get current low-band mode in use (wideband only)
SPEEX_SET_HIGH_MODE = 10 #Set high-band sub-mode to use (wideband only)
SPEEX_GET_HIGH_MODE = 11 #Get current high-band mode in use (wideband only)
SPEEX_SET_VBR = 12 #Set VBR on (1) or off (0)
SPEEX_GET_VBR = 13 #Get VBR status (1 for on, 0 for off)
SPEEX_SET_VBR_QUALITY = 14 #Set quality value for VBR encoding (0-10)
SPEEX_GET_VBR_QUALITY = 15 #Get current quality value for VBR encoding (0-10)
SPEEX_SET_COMPLEXITY = 16 #Set complexity of the encoder (0-10)
SPEEX_GET_COMPLEXITY = 17 #Get current complexity of the encoder (0-10)
SPEEX_SET_BITRATE = 18 #Set bit-rate used by the encoder (or lower)
SPEEX_GET_BITRATE = 19 #Get current bit-rate used by the encoder or decoder
SPEEX_SET_HANDLER = 20 #Define a handler function for in-band Speex request
SPEEX_SET_USER_HANDLER = 22 #Define a handler function for in-band user-defined request
SPEEX_SET_SAMPLING_RATE = 24 #Set sampling rate used in bit-rate computation
SPEEX_GET_SAMPLING_RATE = 25 #Get sampling rate used in bit-rate computation
SPEEX_RESET_STATE = 26 #Reset the encoder/decoder memories to zero
SPEEX_GET_RELATIVE_QUALITY = 29 #Get VBR info (mostly used internally)
SPEEX_SET_VAD = 30 #Set VAD status (1 for on, 0 for off)
SPEEX_GET_VAD = 31 #Get VAD status (1 for on, 0 for off)
SPEEX_SET_ABR = 32 #Set Average Bit-Rate (ABR) to n bits per seconds
SPEEX_GET_ABR = 33 #Get Average Bit-Rate (ABR) setting (in bps)
SPEEX_SET_DTX = 34 #Set DTX status (1 for on, 0 for off)
SPEEX_GET_DTX = 35 #Get DTX status (1 for on, 0 for off)
SPEEX_SET_SUBMODE_ENCODING = 36 #Set submode encoding in each frame (1 for yes, 0 for no, setting to no breaks the standard)
SPEEX_GET_SUBMODE_ENCODING = 37 #Get submode encoding in each frame
SPEEX_GET_LOOKAHEAD = 39 #Returns the lookahead used by Speex
SPEEX_SET_PLC_TUNING = 40 #Sets tuning for packet-loss concealment (expected loss rate)
SPEEX_GET_PLC_TUNING = 41 #Gets tuning for PLC
SPEEX_SET_VBR_MAX_BITRATE = 42 #Sets the max bit-rate allowed in VBR mode
SPEEX_GET_VBR_MAX_BITRATE = 43 #Gets the max bit-rate allowed in VBR mode
SPEEX_SET_HIGHPASS = 44 #Turn on/off input/output high-pass filtering
SPEEX_GET_HIGHPASS = 45 #Get status of input/output high-pass filtering
SPEEX_GET_ACTIVITY = 47 #Get "activity level" of the last decoded frame, i.e. how much damage we cause if we remove the frame
SPEEX_SET_PF = 0 #Equivalent to SPEEX_SET_ENH
SPEEX_GET_PF = 1 #Equivalent to SPEEX_GET_ENH
SPEEX_MODE_FRAME_SIZE = 0 #Query the frame size of a mode
SPEEX_SUBMODE_BITS_PER_FRAME = 1 #Query the size of an encoded frame for a particular sub-mode
SPEEX_LIB_GET_MAJOR_VERSION = 1 #Get major Speex version
SPEEX_LIB_GET_MINOR_VERSION = 3 #Get minor Speex version
SPEEX_LIB_GET_MICRO_VERSION = 5 #Get micro Speex version
SPEEX_LIB_GET_EXTRA_VERSION = 7 #Get extra Speex version
SPEEX_LIB_GET_VERSION_STRING = 9 #Get Speex version string
SPEEX_NB_MODES = 3 #Number of defined modes in Speex
#Encoding/Decoding Modes:
SPEEX_MODEID_NB = 0 #modeID for the defined narrowband mode
SPEEX_MODEID_WB = 1 #modeID for the defined wideband mode
SPEEX_MODEID_UWB = 2 #modeID for the defined ultra-wideband mode
class SpeexBits(Structure):
_fields_ = [('chars', c_char_p)
, ('nbBits', c_int)
, ('charPtr', c_int)
, ('bitPtr', c_int)
, ('owner', c_int)
, ('overflow', c_int)
, ('buf_size', c_int)
, ('reserved1', c_int)
, ('reserved2', c_void_p)
]
class Decoder:
def __init__(self):
self.state = None
self.frame_size = None
self.bits = None
def initialize(self, mode=SPEEX_MODEID_UWB):
libspeex.speex_decoder_init.restype = c_void_p
libspeex.speex_decoder_init.argtypes = [c_void_p]
libspeex.speex_decode_int.restype = c_int
libspeex.speex_decode_int.argtypes = [c_void_p, POINTER(SpeexBits), POINTER(c_int16)]
libspeex.speex_bits_read_from.argtypes = [POINTER(SpeexBits), c_char_p, c_int]
libspeex.speex_bits_destroy.argtypes = [POINTER(SpeexBits)]
libspeex.speex_decoder_destroy.argtypes = [c_void_p]
libspeex.speex_decoder_ctl.argtypes = [c_void_p, c_int, c_void_p]
self.state = libspeex.speex_decoder_init(libspeex.speex_wb_mode)
self.frame_size = c_int()
result = libspeex.speex_decoder_ctl(self.state, SPEEX_GET_FRAME_SIZE, byref(self.frame_size));
self.bits = SpeexBits()
libspeex.speex_bits_init(byref(self.bits))
def decode(self, data):
self.buffer = create_string_buffer(1024)
decoded_frame = (c_int16*self.frame_size.value)()
out = ""
for i in range(0,len(data)):
self.buffer = data[i]
libspeex.speex_bits_read_from(byref(self.bits), self.buffer, len(data[i]))
while libspeex.speex_decode_int(self.state, byref(self.bits), decoded_frame) == 0:
out += string_at(decoded_frame, self.frame_size.value*2)
return out
def destroy(self):
if self.state:
libspeex.speex_bits_destroy(byref(self.bits))
libspeex.speex_decoder_destroy(self.state)
self.state = None