ForceAlignmentScripts/trans_phone_ID_2_name.py at master · ys10/ForceAlignmentScripts · GitHub

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
# coding: UTF-8

import sys
import getopt


def usage():
    print("Usage: python trans_phone_ID_2_name.py [-p phones_file_path] [-i original_file_path] [-o output_file_path] ")


opts, args = getopt.getopt(sys.argv[1:], "hp:i:o:")
phone_file_path = ""
original_file_path = ""
output_file_path = ""
for op, value in opts:
    if op == "-p":
        phone_file_path = value
    elif op == "-i":
        original_file_path = value
    elif op == "-o":
        output_file_path = value
    elif op == "-h":
        usage()
        sys.exit()

# construct phone dict.
phone_dict = dict()
with open(phone_file_path) as phone_file:
    while 1:
        phone_lines = phone_file.readlines(100000)
        if not phone_lines:
            break
        for phone_line in phone_lines:
            phone_list = phone_line.split(' ')
            phone_name = phone_list[0]
            phone_ID = phone_list[1]
            phone_dict[phone_ID] = phone_name
            pass
        pass
#
with open(output_file_path, 'w') as result_file:
    #
    with open(original_file_path) as f:
        while 1:
            result_lines = list()
            lines = f.readlines(100000)
            if not lines:
                break
            for line in lines:
                # print(line)
                original_list = line.split(' ')
                # replace duration with end time
                start_time = float(original_list[2])
                duration = float(original_list[3])
                end_time = round(start_time + duration, 3)
                original_list[3] = str(end_time)
                # replace phone ID with phone name
                phone_ID = original_list[4]
                phone_name = phone_dict[phone_ID]
                original_list[4] = phone_name
                # remove channel info
                original_list.pop(1)
                # join & append to result lines
                result_line = "\t\t".join(original_list) + "\n"
                result_lines.append(result_line)
                pass
            result_file.writelines(result_lines)
            pass
        pass
    pass