25bd7db348
default in config file and OPTIONS. Remove ANNOUNCEMENT_LANGUAGE from [GLOBAL] Replace with ANNOUNCEMENT_LANGUAGES - Comma Separated list Current list: ANNOUNCEMENT_LANGUAGES: en_GB,en_GB_2,en_US,es_ES,es_ES_2,fr_FR,de_DE,dk_DK,it_IT,no_NO,pl_PL,se_SE Add: ANNOUNCEMENT_LANGUAGE to MASTER definition. If using GENERATOR, this becomes the template default. To change via OPTIONS add LANG=<language code> take codes from list above
159 lines
6.3 KiB
Python
159 lines
6.3 KiB
Python
from bitarray import bitarray
|
|
from itertools import islice
|
|
import os
|
|
import glob
|
|
|
|
class readAMBE:
|
|
|
|
def __init__(self, lang,path):
|
|
self.langcsv = lang
|
|
self.langs = lang.split(',')
|
|
self.path = path
|
|
|
|
def _make_bursts(self,data):
|
|
it = iter(data)
|
|
for i in range(0, len(data), 108):
|
|
yield bitarray([k for k in islice(it, 108)] )
|
|
|
|
#Read indexed files
|
|
def readfiles(self):
|
|
|
|
_AMBE_LENGTH = 9
|
|
|
|
_wordBADictofDicts = {}
|
|
|
|
for _lang in self.langs:
|
|
|
|
_prefix = self.path+_lang
|
|
_wordBADict = {}
|
|
|
|
indexDict = {}
|
|
|
|
if os.path.isdir(_prefix):
|
|
ambeBytearray = {}
|
|
_wordBitarray = bitarray(endian='big')
|
|
_wordBADict = {}
|
|
_glob = _prefix + "/*.ambe"
|
|
for ambe in glob.glob(_glob):
|
|
basename = os.path.basename(ambe)
|
|
_voice,ext = basename.split('.')
|
|
inambe = open(ambe,'rb')
|
|
_wordBitarray.frombytes(inambe.read())
|
|
inambe.close()
|
|
_wordBADict[_voice] = []
|
|
pairs = 1
|
|
_lastburst = ''
|
|
for _burst in self._make_bursts(_wordBitarray):
|
|
#Not sure if we need to pad or not? Seems to make little difference.
|
|
if len(_burst) < 108:
|
|
pad = (108 - len(_burst))
|
|
for i in range(0,pad,1):
|
|
_burst.append(False)
|
|
if pairs == 2:
|
|
_wordBADict[_voice].append([_lastburst,_burst])
|
|
_lastburst = ''
|
|
pairs = 1
|
|
next
|
|
else:
|
|
pairs = pairs + 1
|
|
_lastburst = _burst
|
|
|
|
_wordBitarray.clear()
|
|
_wordBADict['silence'] = ([
|
|
[bitarray('101011000000101010100000010000000000001000000000000000000000010001000000010000000000100000000000100000000000'),
|
|
bitarray('001010110000001010101000000100000000000010000000000000000000000100010000000100000000001000000000001000000000')]
|
|
])
|
|
_wordBADictofDicts[_lang] = _wordBADict
|
|
else:
|
|
try:
|
|
with open(_prefix+'.indx') as index:
|
|
for line in index:
|
|
(voice,start,length) = line.split()
|
|
indexDict[voice] = [int(start) * _AMBE_LENGTH ,int(length) * _AMBE_LENGTH]
|
|
index.close()
|
|
except IOError:
|
|
return False
|
|
|
|
ambeBytearray = {}
|
|
_wordBitarray = bitarray(endian='big')
|
|
_wordBADict = {}
|
|
try:
|
|
with open(_prefix+'.ambe','rb') as ambe:
|
|
for _voice in indexDict:
|
|
ambe.seek(indexDict[_voice][0])
|
|
_wordBitarray.frombytes(ambe.read(indexDict[_voice][1]))
|
|
#108
|
|
_wordBADict[_voice] = []
|
|
pairs = 1
|
|
_lastburst = ''
|
|
for _burst in self._make_bursts(_wordBitarray):
|
|
#Not sure if we need to pad or not? Seems to make little difference.
|
|
if len(_burst) < 108:
|
|
pad = (108 - len(_burst))
|
|
for i in range(0,pad,1):
|
|
_burst.append(False)
|
|
if pairs == 2:
|
|
_wordBADict[_voice].append([_lastburst,_burst])
|
|
_lastburst = ''
|
|
pairs = 1
|
|
next
|
|
else:
|
|
pairs = pairs + 1
|
|
_lastburst = _burst
|
|
|
|
_wordBitarray.clear()
|
|
ambe.close()
|
|
except IOError:
|
|
return False
|
|
_wordBADict['silence'] = ([
|
|
[bitarray('101011000000101010100000010000000000001000000000000000000000010001000000010000000000100000000000100000000000'),
|
|
bitarray('001010110000001010101000000100000000000010000000000000000000000100010000000100000000001000000000001000000000')]
|
|
])
|
|
_wordBADictofDicts[_lang] = _wordBADict
|
|
|
|
return _wordBADictofDicts
|
|
|
|
#Read a single ambe file from the audio directory
|
|
def readSingleFile(self,filename):
|
|
ambeBytearray = {}
|
|
_wordBitarray = bitarray(endian='big')
|
|
_wordBA= []
|
|
try:
|
|
with open(self.path+filename,'rb') as ambe:
|
|
_wordBitarray.frombytes(ambe.read())
|
|
#108
|
|
_wordBA = []
|
|
pairs = 1
|
|
_lastburst = ''
|
|
for _burst in self._make_bursts(_wordBitarray):
|
|
#Not sure if we need to pad or not? Seems to make little difference.
|
|
if len(_burst) < 108:
|
|
pad = (108 - len(_burst))
|
|
for i in range(0,pad,1):
|
|
_burst.append(False)
|
|
if pairs == 2:
|
|
_wordBA.append([_lastburst,_burst])
|
|
_lastburst = ''
|
|
pairs = 1
|
|
next
|
|
else:
|
|
pairs = pairs + 1
|
|
_lastburst = _burst
|
|
|
|
_wordBitarray.clear()
|
|
ambe.close()
|
|
except IOError:
|
|
return(False)
|
|
|
|
return(_wordBA)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
|
#test = readAMBE('en_GB','./Audio/')
|
|
|
|
#print(test.readfiles())
|
|
test = readAMBE('en_GB_2','./Audio/')
|
|
print(test.readfiles())
|
|
print(test.readSingleFile('44xx.ambe'))
|