# -*- coding: utf-8 -*-
import argparse
import os
import numpy as np
FRAMESIZE_729A = 10
def print_info(s):
print(s)
def print_error(s):
print(s)
def get_file_list():
global args
if os.path.isfile(args.input):
return [args.input]
elif os.path.isdir(args.input):
file_list = []
for file in os.listdir(args.input):
full_path = os.path.join(args.input, file)
if os.path.isfile(full_path):
file_list.append(full_path)
return file_list
else:
return None
def process_file(fileName):
def lpc_extract_frame(content):
if type(content) == str:
content_t = [int(item.encode('hex'), 16) for item in content]
else:
content_t = content
a = content_t[0] & 0x7f
b = (content_t[1] >> 3) & 0x1f
c = ((content_t[1] << 2) & 0x1c) | ((content_t[2] >> 6) & 0x03)
return [a, b, c]
global args
file = open(fileName, "rb")
content = file.read()
file.close()
LPC = []
num_frame = int(len(content) / FRAMESIZE_729A)
for i in range(num_frame):
LPC.append(lpc_extract_frame(content[i * FRAMESIZE_729A : i * FRAMESIZE_729A + 3]))
LPC = np.array(LPC, dtype = int)
basename = os.path.splitext(os.path.basename(fileName))[0]
np.savetxt(os.path.join(args.output, "%s_LPC.txt" % (basename)), LPC, fmt='%d')
if __name__ == "__main__":
global args
parser = argparse.ArgumentParser(description = 'G729A编码文件参数提取工具')
parser.add_argument("input", help = "输入文件/文件夹", type = str)
parser.add_argument("output", help = "输出文件夹", type = str)
args = parser.parse_args()
if not os.path.isdir(args.output):
print_error("output not exists!")
exit(1)
print_info("Reading files in foler...")
file_list = get_file_list()
if file_list == None:
print_error("Input error！")
exit(1)
else:
print_info("%d files to be processed" % len(file_list))
print_info("Start FEAT extraction...")
for i in range(len(file_list)):
file = file_list[i]
print_info("%d/%d: %s" % (i + 1, len(file_list), file))
process_file(file)
print_info("Done.")
