-
Notifications
You must be signed in to change notification settings - Fork 23
/
main.py
130 lines (125 loc) · 6.67 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
#!/usr/bin/env python
# coding: utf-8
# 版本
from core.Utils import EDITION
from core.Exceptions import ArgumentError
from core.Exceptions import MainPrint
# 包导入
import sys
import os
# 文件路径
from core.Medias import MediaObj
# 配置
from core.ProjConfig import Config
from core.ScriptParser import RplGenLog,MediaDef,CharTable,Script
# 输出
from core.OutputType import PreviewDisplay,ExportVideo,ExportXML
# 主程序
class ReplayGenerator:
# 初始化参数
def __init__(self,args):
try:
# 外部输入文件
self.log_path = args.LogFile
self.media_path = args.MediaObjDefine
self.char_path = args.CharacterTable
self.output_path = args.OutputPath
self.json_path = args.Json
# 配置
if self.json_path != None:
# 如果是json输入
input_json = Script(json_input=self.json_input)
self.config = Config(dict_input=input_json.struct['config'])
self.config.execute()
# 读取输入工程结构
self.medef = MediaDef(dict_input=input_json.struct['medef'])
self.chartab = CharTable(dict_input=input_json.struct['chartab'])
self.rplgenlog = RplGenLog(dict_input=input_json.struct['rplgenlog'])
else:
# 通过args输入
self.config = Config(argparse_input=args)
self.config.execute()
# 检查输入文件可用性
for path in [self.log_path,self.media_path,self.char_path]:
if path is None:
raise ArgumentError('MissInput')
if os.path.isfile(path) == False:
raise ArgumentError('FileNotFound',path)
self.medef = MediaDef(file_input=self.media_path)
self.chartab = CharTable(file_input=self.char_path)
self.rplgenlog = RplGenLog(file_input=self.log_path)
# debug
except ArgumentError as E:
self.medef = MediaDef(file_input='./toy/MediaObject.txt')
self.chartab = CharTable(file_input='./toy/CharactorTable.tsv')
self.rplgenlog = RplGenLog(file_input='./toy/LogFile.rgl')
self.output_path = './test_output'
except Exception as E:
print(E)
sys.exit(1)
self.main()
# 执行输入文件
def execute_input(self):
# TODO：把这个对类对象的操作，想个办法封装起来
MediaObj.export_xml = False
MediaObj.output_path = self.output_path
# log
try:
# 初始化媒体
self.medef.execute()
# 初始化角色表
self.chartab.execute()
# 初始化log文件
self.rplgenlog.execute(media_define=self.medef,char_table=self.chartab,config=self.config)
except Exception as E:
print(E)
sys.exit(1)
# 主流程
def main(self):
# welcome
print(MainPrint('Welcome',EDITION))
# 执行输入
self.execute_input()
# 开始预览播放
PreviewDisplay(rplgenlog=self.rplgenlog,config=self.config,output_path=self.output_path)
# ExportXML(rplgenlog=self.rplgenlog,config=self.config,output_path=self.output_path)
# ExportVideo(rplgenlog=self.rplgenlog,config=self.config,output_path=self.output_path)
if __name__ == '__main__':
import argparse
# 外部参数输入
# 参数处理
ap = argparse.ArgumentParser(description="Generating your TRPG replay video from logfile.")
ap.add_argument("-l", "--LogFile", help='The standerd input of this programme, which is mainly composed of TRPG log.',type=str)
ap.add_argument("-d", "--MediaObjDefine", help='Definition of the media elements, using real python code.',type=str)
ap.add_argument("-t", "--CharacterTable", help='The correspondence between character and media elements, using tab separated text file or Excel table.',type=str)
ap.add_argument("-j", "--Json", help='Use the config from json file, instead of arguments from command.',type=str)
ap.add_argument("-o", "--OutputPath", help='Choose the destination directory to save the project timeline and breakpoint file.',type=str,default=None)
# 选项
ap.add_argument("-F", "--FramePerSecond", help='Set the FPS of display, default is 30 fps, larger than this may cause lag.',type=int,default=30)
ap.add_argument("-W", "--Width", help='Set the resolution of display, default is 1920, larger than this may cause lag.',type=int,default=1920)
ap.add_argument("-H", "--Height", help='Set the resolution of display, default is 1080, larger than this may cause lag.',type=int,default=1080)
ap.add_argument("-Z", "--Zorder", help='Set the display order of layers, not recommended to change the values unless necessary!',type=str,
default='BG2,BG1,Am3,Am2,Am1,AmS,Bb,BbS')
# 用于语音合成的key
ap.add_argument("-K", "--AccessKey", help='Your AccessKey, to use with --SynthsisAnyway',type=str,default="Your_AccessKey")
ap.add_argument("-S", "--AccessKeySecret", help='Your AccessKeySecret, to use with --SynthsisAnyway',type=str,default="Your_AccessKey_Secret")
ap.add_argument("-A", "--Appkey", help='Your Appkey, to use with --SynthsisAnyway',type=str,default="Your_Appkey")
ap.add_argument("-U", "--Azurekey", help='Your Azure TTS key.',type=str,default="Your_Azurekey")
ap.add_argument("-R", "--ServRegion", help='Service region of Azure.', type=str, default="eastasia")
# 用于导出视频的质量值
ap.add_argument("-Q", "--Quality", help='Choose the quality (ffmpeg crf) of output video, to use with --ExportVideo.',type=int,default=24)
# Flags
ap.add_argument('--ExportXML',help='Export a xml file to load in Premiere Pro, some .png file will be created at same time.',action='store_true')
ap.add_argument('--ExportVideo',help='Export MP4 video file, this will disables interface display',action='store_true')
ap.add_argument('--SynthesisAnyway',help='Execute speech_synthezier first, and process all unprocessed asterisk time label.',action='store_true')
ap.add_argument('--FixScreenZoom',help='Windows system only, use this flag to fix incorrect windows zoom.',action='store_true')
ap.add_argument("--ForceSplitClip", help='Force to separate clips at breakpoints while exporting PR sequence.',action='store_true' )
# 语言
ap.add_argument("--Language",help='Choose the language of running log',default='en',type=str)
# 主
args = ap.parse_args()
try:
ReplayGenerator(args=args)
except:
from traceback import print_exc
print_exc()