-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.py
222 lines (209 loc) · 12.8 KB
/
main.py
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
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
import re
debug = 0 # 0关闭调试，1仅显示转换为代码块的部分，2显示所有转换过程
class Conversion:
def __init__(self):
self.original = open('original.txt', 'r', encoding='utf-8')
self.result = open('result.txt', 'w+', encoding='utf-8')
self.line_num = 0 # 统计行数
# 记录已出现的东西
self.backgrounds: list[str] = []
self.BGMs: list[str] = []
self.standings: list[str] = []
self.CGs: list[str] = []
self.BGs: list[str] = []
def whereis_error(self):
self.result.seek(0, 0)
return '| 目标位置：第%s行' % sum(1 for _ in self.result)
@staticmethod
def eager_execution_block(line: str, span: tuple[int, int]) -> str:
"""将标题行转换为提前代码块"""
cn_num = {'一': '1', '二': '2', '三': '3', '四': '4', '五': '5', '六': '6', '七': '7', '八': '8', '九': '9'}
result = "@<|\n" \
"label('ch%s', '%s')\n" \
"|>\n" \
% (cn_num[line[span[0] + 1: span[1] - 1]], line)
return result
def lazy_execution_block(self, line: str) -> str | None:
# todo: 延迟代码块的加载，包括立绘、场景、bgm
# 首先进行分割
result = "<|\n" # 代码块开头
blocks = re.findall('【(.*?)】', line) # blocks为待处理的场景，类型为list
# 立绘部分
last_standing = []
character_change = {'贝拉': 'Bella', '向晚': 'Ava', '珈乐': 'Carol', '嘉然': 'Diana', '乃琳': 'Queen', '阿草': 'Acao',
'成年男人的剪影': 'Man', '男人的剪影': 'Man', '成年女人的剪影': 'Women', '女人的剪影': 'Women',
'小男孩的剪影': 'Boy', '小女孩的剪影': 'Girl', '医护人员的剪影': 'Medical',
'长发珈乐': 'Carol'} # todo: 此行需改进
cloth_change = {'常服': 'causal', '舞蹈服': 'dance', '团服': 'team', '画家': 'draw', '病服': 'sick', '礼服': 'lifu', '律师服': 'lawyer'}
appearance_change = {'通常': 'causal', '生气': 'angry', '微笑': 'smile', '惊讶': 'surprised', '失望': 'disappointed'}
trans = {'黑屏': "'bg_000_black'", '黑场': "'bg_000_black'", '白场': "'bg_000_white'", '溶解': "'bg_000_melt'"}
CG_trans = {'嘉然抱着贝拉哭泣': 'cg_002_贝贝嘉哭', '穿着常服坐在地上绝望的贝拉': 'cg_007_贝常服哭', '穿着团服坐在地上绝望的贝拉': 'cg_009_贝团服哭',
'墙上的海报': 'cg_006_嘉然海报', '嘉然画的嘉心糖': 'cg_004_嘉然壁画', '贝拉-CG5': 'cg_010_贝靠墙'}
bg_trans = {'画展': 'bg_002_画展', '舞蹈室-黄昏（色调更暗）': 'bg_005_舞蹈室_夜晚', '展览馆外景': 'bg_002_画展', '舞蹈室-黄昏': 'bg_001_舞蹈室',
'地铁站': 'bg_003_地铁站', '夜晚的病房': 'bg_009_医院病床', '夜晚的大街': 'bg_007_夜晚的大街'}
# self.diana_cloth = {'常服': 'causal','画家': 'draw','团服': 'team'}
for block in blocks:
if '立绘' in block:
for standing in self.standings: # 先将不用的立绘隐藏
result += "hide(%s)\n" % standing
self.standings = []
text = block[block.index('：') + 1:] + '/' # 末尾添加符号方便匹配
# text_t = block[block.index('：') + 1:]
if text == '无立绘/': # 无立绘不需要显示
print('\033[32mSUCCESS\033[0m | 延迟代码块：%s -- %s' % (line, block)) if debug > 0 else None
elif re.match('(.*?)-(.*?)-(.*?)/', text): # 主要角色的匹配
# if(character_change[text_t])
character = re.findall('(.*?)-(.*?)-(.*?)/', text)[0]
if character[2] == '50%灰度': # 如果第三个值是50%灰度，添加color参数
if len(last_standing) != 0:
# hide(last_standing)
result += "hide(%s)\n" % last_standing[0]
last_standing = []
result += "show(%s, '%s', pos_c, {0.299, 0.587, 0.114})\n" \
% (character_change[character[0]], cloth_change[character[1]])
last_standing.append(character_change[character[0]])
else:
print(character) if debug > 1 else None
print("show(" + character_change[character[0]] + ", '" + cloth_change[character[1]] + '_' +
appearance_change[character[2]] + "', pos_c)") if debug > 1 else None
if len(last_standing) != 0:
# hide(last_standing)
result += "hide(%s)\n" % last_standing[0]
last_standing = []
result += "show(%s, '%s_%s', pos_c)\n" \
% (character_change[character[0]], cloth_change[character[1]],
appearance_change[character[2]])
last_standing.append(character_change[character[0]])
self.standings.append(character_change[character[0]])
print('\033[32mSUCCESS\033[0m | 延迟代码块：%s -- %s' % (line, block)) if debug > 0 else None
# 下面是次要人物的匹配
elif re.match('(.*?)-50%灰度/', text): # 如果有50%灰度，添加color参数
character = re.findall('(.*?)-', text)
if len(last_standing) != 0:
# hide(last_standing)
result += "hide(%s)\n" % last_standing[0]
last_standing = []
result += "show(%s, 'default', pos_c, {0.299, 0.587, 0.114})\n" % (character_change[character[0]])
last_standing.append(character_change[character[0]])
elif re.match('(.*?)/', text):
character = re.findall('(.*?)/', text)
print(character) if debug > 1 else None
print("show(%s, 'default', pos_c)\n" % (character_change[character[0]])) if debug > 1 else None
if len(last_standing) != 0:
# hide(last_standing)
result += "hide(%s)\n" % last_standing[0]
last_standing = []
result += "show(%s, 'default', pos_c)\n" % (character_change[character[0]])
last_standing.append(character_change[character[0]])
self.standings.append(character_change[character[0]])
print('\033[32mSUCCESS\033[0m | 延迟代码块：%s -- %s' % (line, block)) if debug > 0 else None
else:
print('\033[33mWARNING\033[0m | 未完成自动转换的剧本行：%s | 原位置：第%s行' % (line, self.line_num), self.whereis_error())
print(text)
elif '转场' in block:
if 1 == 2:
# if re.findall('(.*?)转场', block):
tr = re.findall('(.*?)转场', block)[0]
print("anim:trans_fade(bg, " + trans[tr] + ")") if debug > 1 else None
result += "anim:trans_fade(bg, " + trans[tr] + ")\n"
print('\033[32mSUCCESS\033[0m | 延迟代码块：%s -- %s' % (line, block)) if debug > 0 else None
else:
print('\033[33mWARNING\033[0m | 未完成自动转换的转场剧本行：%s | 原位置：第%s行' % (line, self.line_num), self.whereis_error())
elif '过渡' in block:
if 1 == 2:
# if re.findall('(.*?)过渡', block):
tr = re.findall('(.*?)过渡', block)[0]
# print(block)
print("anim:trans_fade(bg, " + trans[tr] + ")") if debug > 1 else None
result += "anim:trans_fade(bg, %s)\n" % trans[tr]
print('\033[32mSUCCESS\033[0m | 延迟代码块：%s -- %s' % (line, block)) if debug > 0 else None
else:
print('\033[33mWARNING\033[0m | 未完成自动转换的过渡剧本行：%s | 原位置：第%s行' % (line, self.line_num), self.whereis_error())
elif '场景' in block:
# 转换场景的代码
text = block[block.index('：') + 1:]
self.BGs.append(text)
text = "'" + bg_trans[text] + "'"
result += "show(bg, %s, {0, 0, 1.4})\n" % text
print('\033[32mSUCCESS\033[0m | 延迟代码块：%s -- %s' % (line, block)) if debug > 0 else None
elif 'BGM' in block:
# 转换bgm的代码
text = block[block.find('：') + 1:]
if text == '无BGM': # 应停止所有bgm
for BGM in self.BGMs:
result += "stop(%s)\n" % BGM
self.BGMs = []
else:
result += "play(bgm, '%s')\n" % text
self.BGMs.append(text)
print('\033[32mSUCCESS\033[0m | 延迟代码块：%s -- %s' % (line, block)) if debug > 0 else None
elif 'CG' in block:
# 转换CG的代码
text = block[block.find('：') + 1:]
if re.match('(.*?)-(.*?)灰度', text):
list = re.findall('(.*?)-(.*?)灰度', text)
# print(list[0])
self.CGs.append(list[0][0])
print('show(bg ,' + "'" + CG_trans[
list[0][0]] + "'" + ', ' + 'pos_default, {0.299, 0.587, 0.114})') if debug > 1 else None
result += ('show(bg ,' + "'" + CG_trans[
list[0][0]] + "'" + ', ' + 'pos_default, {0.299, 0.587, 0.114})' + "\n")
else:
result += "show(bg, '%s')\n" % CG_trans[text]
self.CGs.append(text)
print('\033[32mSUCCESS\033[0m | 延迟代码块：%s -- %s' % (line, block)) if debug > 0 else None
elif '音效' in block:
text = block[block.find('：') + 1:]
if text == '无音效':
pass
else:
result += "sound('%s')\n" % text
print('\033[32mSUCCESS\033[0m | 延迟代码块：%s -- %s' % (line, block)) if debug > 0 else None
else:
self.result.seek(0, 0) # 显示错误时会显示输出到的行号
print('\033[31mERROR \033[0m | 自动转换出错的剧本行：%s | 原位置：第%s行' % (line, self.line_num), self.whereis_error())
return None # 不将错误的代码块写入结果文件
result += "|>\n" # 代码块结尾
if result != "<|\n|>\n": # 如果没有代码无需添加
self.result.write(result)
return result
def main(self):
"""主程序"""
line = self.original.readline()
while line:
self.line_num += 1
if line == '\n': # 删除空行
line = self.original.readline()
continue
line = line.strip('\n') # 删除末尾的换行符
if re.match('第(.*?)章', line): # 匹配文件开头的章节名
span = re.match('第(.*?)章', line).span()
self.result.write(self.eager_execution_block(line, span))
print('\033[32mSUCCESS\033[0m | 提前代码块：%s' % line) if debug > 0 else None
line = self.original.readline()
continue
if re.match('【(.*?)】', line): # 匹配场景配置
self.lazy_execution_block(line)
# print('\033[32mSUCCESS\033[0m | 延迟代码块：%s' % line) if debug > 0 and a is not None else None
line = self.original.readline()
continue
# 以下为对话和旁白
if re.match('旁白', line):
span = re.match('旁白', line).span()
aside = line[:span[0]] + line[span[1] + 1:] + "\n\n" # 处理旁白
self.result.write(aside)
print('\033[32mSUCCESS\033[0m | 旁白：%s' % line) if debug > 1 else None
line = self.original.readline()
continue
else:
line = line.replace('：', '：：', 1) + "\n\n"
self.result.write(line)
print('\033[32mSUCCESS\033[0m | 台词：%s' % line) if debug > 1 else None
line = self.original.readline()
continue
self.original.close()
self.result.close()
return 0
if __name__ == '__main__':
x = Conversion()
x.main()