G729a_LPC_extract.py.zip_G.729a_LPC_py lpc_steganography

# -*- 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.")