esp-idf/examples/peripherals/i2s_adc_dac/tools/generate_audio_file.py

45 lines
1.5 KiB
Python
Raw Normal View History

2018-09-10 09:13:47 -04:00
from __future__ import print_function
import os
import struct
import wave
from builtins import range
2018-12-04 02:32:48 -05:00
def get_wave_array_str(filename, target_bits):
wave_read = wave.open(filename, 'r')
array_str = ''
nchannels, sampwidth, framerate, nframes, comptype, compname = wave_read.getparams()
sampwidth *= 8
for i in range(wave_read.getnframes()):
val, = struct.unpack('<H', wave_read.readframes(1))
scale_val = (1 << target_bits) - 1
cur_lim = (1 << sampwidth) - 1
2018-12-04 02:32:48 -05:00
# scale current data to 8-bit data
val = val * scale_val / cur_lim
2018-09-10 09:13:47 -04:00
val = int(val + ((scale_val + 1) // 2)) & scale_val
array_str += '0x%x, ' % (val)
if (i + 1) % 16 == 0:
array_str += '\n'
return array_str
2018-12-04 02:32:48 -05:00
def gen_wave_table(wav_file_list, target_file_name, scale_bits=8):
with open(target_file_name, 'w') as audio_table:
2018-09-10 09:13:47 -04:00
print('#include <stdio.h>', file=audio_table)
print('const unsigned char audio_table[] = {', file=audio_table)
for wav in wav_file_list:
print('processing: {}'.format(wav))
2018-12-04 02:32:48 -05:00
print(get_wave_array_str(filename=wav, target_bits=scale_bits), file=audio_table)
2018-09-10 09:13:47 -04:00
print('};\n', file=audio_table)
print('Done...')
2018-12-04 02:32:48 -05:00
if __name__ == '__main__':
print('Generating audio array...')
wav_list = []
for filename in os.listdir('./'):
if filename.endswith('.wav'):
wav_list.append(filename)
gen_wave_table(wav_file_list=wav_list, target_file_name='audio_example_file.h')