QuickVCD/QuickSoundScript.py at main · entropy-zero/QuickVCD · 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
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
# Very primitive python script to generate soundscript entries

input_file = open("input.csv")
data = dict()

lineNumber = 0
locationColumn = 0
vcdColumn = 3 # Need to separate this out for all line sheets!
wavPathColumn = 1
soundscriptColumn = 2
actorColumn = 4
lineTextColumn = 5

# This is mega lame, but let's just hard code caption colors for now
closeCaptionColors = dict()
closeCaptionColors["Clone Cop"] = "253,165,2"
closeCaptionColors["Bad Cop"] = "255,51,0"
closeCaptionColors["Advisor"] = "136,180,185"
closeCaptionColors["Will-E"] = "66,255,199"

for line in input_file:
    parts = line.split('^')

    # Skip header rows!
    if parts[locationColumn] == 'Location':
        continue

    lineNumber = lineNumber + 1

    line_vcd = parts[vcdColumn]
    line_actor = parts[actorColumn]
    line_text = parts[lineTextColumn].strip()
    line_text = line_text.rstrip('\n')
    wavPath = parts[wavPathColumn]
    soundscriptName = parts[soundscriptColumn]
    scene = data.get(line_vcd)
    if scene is None:
        lineNumber = 0
        scene = dict()
    actor = scene.get(line_actor)
    if actor is None:
        actor = []
    actor.append([lineNumber, line_text, wavPath, soundscriptName])
    scene[line_actor] = actor
    data[line_vcd] = scene

# Export soundscripts
soundscript_file = open("output_soundscripts.txt", "w")
captions_file = open("output_captions.txt", "w")

for scene_name in data.keys():
    scene = data.get(scene_name)

    for actor_name in scene:
        actor = scene.get(actor_name)
        for line in actor:
            line_number = line[0]
            line_text = line[1]
            soundscriptName = line[2]
            wavPath = line[3]
            captionColor = closeCaptionColors.get(actor_name)
            if captionColor is None:
                captionColor = "255,255,255"
            soundscript_file.write("\"" + soundscriptName + "\"\n")
            soundscript_file.write("{\n")
            soundscript_file.write("  \"channel\"\t\t\"CHAN_VOICE\"\n")
            soundscript_file.write("  \"volume\"\t\t\"0.650000\"\n")
            soundscript_file.write("  \"pitch\"\t\t\"PITCH_NORM\"\n")
            soundscript_file.write("  \"soundlevel\"\t\t\"SNDLVL_TALKING\"\n")
            soundscript_file.write("  \"wave\"\t\t\"" + wavPath + "\"\n")
            soundscript_file.write("}\n")
            captions_file.write("  \"" + soundscriptName + "\"  \"<clr:" + captionColor + ">" + line_text + "\"\n")

soundscript_file.close()
captions_file.close()

# Export placeholder VCDs
for scene_name in data.keys():
    scene = data.get(scene_name)
    output_file = open(scene_name + ".vcd", "w")

    for actor_name in scene:
        actor = scene.get(actor_name)
        output_file.write("actor \"" + actor_name + "\"\n")
        output_file.write("{\n")
        output_file.write("  channel \"Placeholder\"\n")
        output_file.write("  {\n")
        for line in actor:
            line_number = line[0]
            line_text = line[1]
            output_file.write("    event generic \"AI_GAMETEXT\"\n")
            output_file.write("    {\n")
            output_file.write("      time " + str(line_number * 3) + " " + str((line_number + 1) * 3) + "\n")
            output_file.write("      param \"AI_GAMETEXT\"\n")
            output_file.write("      param2 \"" + line_text + "\"\n")
            output_file.write("    }\n")
        output_file.write("  }\n")
        output_file.write("}\n")

    output_file.close()

print(data)