GamePageUIConnection.py

# -*- coding: utf-8 -*-

from Core import *
from GUI import *
from VNEngines import *


class GamePageUIConnection(object):

    def __init__(self):
        self.ui_game_page_connections()

    def ui_game_page_connections(self):
        self.ui.gamepage.kirikiri.check_resolution_btn.clicked.connect(self.kirikiri_check_resolution)
        self.ui.gamepage.artemis.check_resolution_btn.clicked.connect(self.artemis_check_resolution)
        self.ui.gamepage.run_btn.clicked.connect(self.game_page_run)

    def game_page_run(self):
        self.input_path = Path(self.ui.gamepage.select_input_folder_line_edit.text().strip())
        self.output_folder = Path(self.ui.gamepage.select_output_folder_line_edit.text().strip())
        # 子线程中运行
        self.game_page_runner = GamePageRunner(self)
        # pyside6信号需要通过实例绑定
        self.game_page_runner.start_sig.connect(self.start_game_page_runner_and_lock)
        self.game_page_runner.info_sig.connect(self.append_game_info_text_edit)
        self.game_page_runner.progress_sig.connect(self.update_game_progress_bar)
        self.game_page_runner.finish_sig.connect(self.finish_game_page_runner_and_unlock)
        self.game_page_runner.crash_sig.connect(self.crash_game_page_runner_and_unlock)
        self.game_page_runner.tip_sig.connect(self.show_game_page_tip)
        self.game_page_runner.start()

    def show_game_page_tip(self, _str):
        warn_msg = QMessageBox()
        reply = warn_msg.warning(self.ui, '提示', _str, QMessageBox.Yes)

    def start_game_page_runner_and_lock(self):
        # 开始时锁定，防止重复操作
        self.ui.gamepage.set_running_state(1)
        # 清空历史信息
        self.ui.gamepage.info_text_edit.clear()
        self.emit_info(format('开始处理', '=^76'))
        self.ui.gamepage.info_text_edit.append(format('开始处理', '=^76'))

    def append_game_info_text_edit(self, info_str):
        self.ui.gamepage.info_text_edit.append(info_str)

    def update_game_progress_bar(self, _percent, _left_time):
        self.ui.gamepage.set_running_state(2)
        self.ui.gamepage.status_progress_bar.setValue(_percent)
        left_time_str = seconds_format(_left_time)
        self.ui.gamepage.run_btn.setText(left_time_str)
        if _percent == 100:
            self.ui.gamepage.set_running_state(1)

    def finish_game_page_runner_and_unlock(self, info_str=''):
        self.ui.gamepage.set_running_state(3)
        self.ui.gamepage.info_text_edit.append(info_str)
        self.emit_info(format('结束处理', '=^76'))
        self.ui.gamepage.info_text_edit.append(format('结束处理', '=^76'))
        finish_info_msg = QMessageBox()
        reply = finish_info_msg.information(self.ui, '处理完成', f'{info_str}\n是否打开输出文件夹?', QMessageBox.Yes | QMessageBox.No, QMessageBox.Yes)
        show_folder(self.output_folder) if reply == QMessageBox.Yes else None

    def crash_game_page_runner_and_unlock(self, info_str):
        self.ui.gamepage.set_running_state(0)
        self.emit_info(format('中断处理', '=^76'))
        self.ui.gamepage.info_text_edit.append(format('中断处理', '=^76'))
        raise Exception(info_str)

    def kirikiri_check_resolution(self):
        input_folder = Path(self.ui.gamepage.select_input_folder_line_edit.text().strip())
        try:
            _kirikiri = Kirikiri()
            scwidth, scheight, encoding = _kirikiri.get_resolution_encoding(input_folder)
            self.ui.gamepage.kirikiri.set_game_resolution_encoding(scwidth, scheight, encoding)
        except:
            warn_msg = QMessageBox()
            reply = warn_msg.warning(self.ui, '提示', '未能找到游戏分辨率和主要编码格式!', QMessageBox.Yes)

    def artemis_check_resolution(self):
        input_folder = Path(self.ui.gamepage.select_input_folder_line_edit.text().strip())
        try:
            _artemis = Artemis()
            scwidth, scheight, encoding = _artemis.get_resolution_encoding(input_folder)
            self.ui.gamepage.artemis.set_game_resolution_encoding(scwidth, scheight, encoding)
        except:
            warn_msg = QMessageBox()
            reply = warn_msg.warning(self.ui, '提示', '未能找到游戏分辨率和主要编码格式!', QMessageBox.Yes)


class GamePageRunner(QThread):
    """用于处理高清重制等耗时任务"""

    # 开始信号
    start_sig = Signal()
    # 信息文本框
    info_sig = Signal(str)
    # 进度(0-100)
    progress_sig = Signal(int, int)
    # 结束弹窗信息
    finish_sig = Signal(str)
    # 崩溃错误信息
    crash_sig = Signal(str)
    # 检查路径提示等
    tip_sig = Signal(str)

    def __init__(self, vnu):
        QThread.__init__(self)
        self.vnu = vnu

    def path_pass(self, only_folder=True, check_output=True, warn_kwd=None) -> bool:
        """
        @brief      路径检查，发送提示

        @param      only_folder   输入路径只能是文件夹
        @param      check_output  检查输出文件夹
        @param      warn_kwd      禁止字样

        @return     Bool
        """
        warn_message = None
        if not self.vnu.input_path.exists():
            warn_message = '输入路径不存在!'
        if self.vnu.input_path == Path('./'):
            warn_message = '输入路径不能与工作目录相同!'
        if only_folder:
            if not self.vnu.input_path.is_dir():
                warn_message = '输入路径需要是文件夹!'
        if check_output:
            if self.vnu.output_folder == Path('./'):
                warn_message = '输出路径不能与工作目录相同!'
            if self.vnu.input_path == self.vnu.output_folder:
                warn_message = '输入路径和输出路径不能相同!'
        if warn_kwd is not None:
            parent_names = [self.vnu.input_path.name]
            if self.vnu.input_path.is_dir():
                parent_names += self.vnu.input_path.parent_names
            for parent_name in parent_names:
                if warn_kwd in parent_name:
                    warn_message = f'输入路径及其上级目录不能含有{warn_kwd}字样!'
                    break
        if warn_message is not None:
            self.tip_sig.emit(warn_message)
            return False
        else:
            # 通过时触发开始信号
            self.start_sig.emit()
            if check_output:
                if not self.vnu.output_folder.exists():
                    self.vnu.output_folder.mkdir(parents=True)
            return True

    def run(self):
        try:
            # Kirikiri
            if self.vnu.ui.gamepage.game_engine_area.currentWidget() is self.vnu.ui.gamepage.kirikiri:
                self.kirikiri_run()
            # Artemis
            elif self.vnu.ui.gamepage.game_engine_area.currentWidget() is self.vnu.ui.gamepage.artemis:
                self.artemis_run()
            # Majiro
            elif self.vnu.ui.gamepage.game_engine_area.currentWidget() is self.vnu.ui.gamepage.majiro:
                self.majiro_run()
        except Exception as e:
            self.crash_sig.emit(traceback.format_exc())

    def kirikiri_run(self):
        kirikiri = Kirikiri(self)
        # 给ui起个别名
        ugk = self.vnu.ui.gamepage.kirikiri
        if ugk.currentWidget() is ugk.hd_parts_frame:
            if self.path_pass(warn_kwd='patch'):
                # 设置放大部分
                run_dict = {}
                run_dict['script'] = ugk.text_part.isChecked()
                run_dict['image'] = ugk.image_part.isChecked()
                run_dict['animation'] = ugk.animation_part.isChecked()
                run_dict['video'] = ugk.video_part.isChecked()
                # 开始放大
                kirikiri.upscale(
                    game_data=self.vnu.input_path,
                    patch_folder=self.vnu.output_folder,
                    scale_ratio=ugk.custiom_ratio_spinbox.value(),
                    run_dict=run_dict,
                    encoding=ugk.main_encoding.text(),
                    resolution=[int(i) for i in ugk.before_resolution.text().split('x')],
                    # 高级选项
                    upscale_fg=True if ugk.upscale_fg_btn.isChecked() else False
                )
                self.finish_sig.emit('高清重制完成!')
        elif ugk.currentWidget() is ugk.work_up_frame:
            # 对话框头像坐标调整
            if ugk.stand_crt_btn.isChecked():
                if self.path_pass():
                    face_zoom = ugk.crt_ratio.value()
                    xpos_move = ugk.crt_movex.value()
                    kirikiri.stand_correction(self.vnu.input_path, self.vnu.output_folder, face_zoom, xpos_move)
                    self.finish_sig.emit('对话框头像坐标调整完成!')
            # scn格式转换
            elif ugk.scn_cvt_btn.isChecked():
                if self.path_pass(only_folder=False):
                    input_format = ugk.scn_in.currentText()
                    output_format = ugk.scn_out.currentText()
                    if input_format == 'scn':
                        kirikiri.scn_de_batch(self.vnu.input_path, self.vnu.output_folder)
                    elif input_format == 'json':
                        kirikiri.scn_en_batch(self.vnu.input_path, self.vnu.output_folder)
                    self.finish_sig.emit('scn转换完成!')
            # tlg图片格式转换
            elif ugk.tlg_convert_btn.isChecked():
                if self.path_pass(only_folder=False):
                    input_format = ugk.tlg_in.currentText()
                    output_format = ugk.tlg_out.currentText()
                    if input_format == 'tlg':
                        if output_format == 'png':
                            kirikiri.tlg2png_batch(self.vnu.input_path, self.vnu.output_folder)
                        else:
                            tlg5_mode = False if output_format == 'tlg6' else True
                            kirikiri.tlg2tlg_batch(self.vnu.input_path, self.vnu.output_folder, tlg5_mode)
                    elif input_format == 'png':
                        tlg5_mode = False if output_format == 'tlg6' else True
                        kirikiri.png2tlg_batch(self.vnu.input_path, self.vnu.output_folder, tlg5_mode)
                    self.finish_sig.emit('tlg图片转换完成!')
            # pimg格式转换
            elif ugk.pimg_cvt_btn.isChecked():
                if self.path_pass(only_folder=False):
                    input_format = ugk.pimg_in.currentText()
                    output_format = ugk.pimg_out.currentText()
                    if input_format == 'pimg':
                        kirikiri.pimg_de_batch(self.vnu.input_path, self.vnu.output_folder)
                    elif input_format == 'json&png':
                        kirikiri.pimg_en_batch(self.vnu.input_path, self.vnu.output_folder)
                    self.finish_sig.emit('pimg转换完成!')
            # amv动画格式转换
            elif ugk.amv_cvt_btn.isChecked():
                if self.path_pass(only_folder=False):
                    input_format = ugk.amv_in.currentText()
                    output_format = ugk.amv_out.currentText()
                    if input_format == 'amv' and output_format == 'json&png':
                        kirikiri.amv2png_batch(self.vnu.input_path, self.vnu.output_folder)
                    elif input_format == 'json&png' and output_format == 'amv':
                        kirikiri.png2amv_batch(self.vnu.input_path, self.vnu.output_folder)
                    self.finish_sig.emit('amv转换完成!')
            elif ugk.flat_patch_btn.isChecked():
                if self.path_pass(warn_kwd='patch'):
                    kirikiri.flat_kirikiri_patch_folder(self.vnu.input_path, self.vnu.output_folder)
                    self.finish_sig.emit('补丁文件平铺完成!')

    def artemis_run(self):
        artemis = Artemis(self)
        # 给ui起个别名
        uga = self.vnu.ui.gamepage.artemis
        if uga.currentWidget() is uga.hd_parts_frame:
            if self.path_pass():
                # 设置放大部分
                run_dict = {}
                run_dict['script'] = uga.text_part.isChecked()
                run_dict['image'] = uga.image_part.isChecked()
                run_dict['animation'] = uga.animation_part.isChecked()
                run_dict['video'] = uga.video_part.isChecked()
                artemis.upscale(
                    game_data=self.vnu.input_path,
                    patch_folder=self.vnu.output_folder,
                    scale_ratio=uga.custiom_ratio_spinbox.value(),
                    run_dict=run_dict,
                    encoding=uga.main_encoding.text(),
                    resolution=[int(i) for i in uga.before_resolution.text().split('x')]
                )
                self.finish_sig.emit('高清重制完成!')
        elif uga.currentWidget() is uga.work_up_frame:
            if uga.pfs_unpack_btn.isChecked():
                if self.path_pass(only_folder=False):
                    pfs_encoding = uga.pfs_encoding_line_edit.text().strip()
                    artemis.batch_extract_pfs(self.vnu.input_path, self.vnu.output_folder, pfs_encoding)
                    self.finish_sig.emit(f'拆包完成！请把游戏目录中类似script、movie等文件夹及*.ini文件也复制到：\n{self.vnu.output_folder}中')

    def majiro_run(self):
        majiro = Majiro(self)
        # 给ui起个别名
        ugm = self.vnu.ui.gamepage.majiro
        if ugm.currentWidget() is ugm.hd_parts_frame:
            if self.path_pass():
                # 设置放大部分
                run_dict = {}
                run_dict['script'] = ugm.text_part.isChecked()
                run_dict['image'] = ugm.image_part.isChecked()
                run_dict['video'] = ugm.video_part.isChecked()
                majiro.upscale(
                    game_data=self.vnu.input_path,
                    patch_folder=self.vnu.output_folder,
                    scale_ratio=ugm.custiom_ratio_spinbox.value(),
                    run_dict=run_dict,
                    encoding=ugm.main_encoding.text(),
                    resolution=[int(i) for i in ugm.before_resolution.text().split('x')]
                )
                self.finish_sig.emit('高清重制完成!')
        elif ugm.currentWidget() is ugm.work_up_frame:
            if ugm.mjo_convert_btn.isChecked():
                if self.path_pass(only_folder=False):
                    input_format = ugm.mjo_in.currentText()
                    output_format = ugm.mjo_out.currentText()
                    if input_format == 'mjo' and output_format == 'mjil&mjres':
                        majiro.mjo_de_batch(self.vnu.input_path, self.vnu.output_folder)
                    elif input_format == 'mjil&mjres' and output_format == 'mjo':
                        majiro.mjo_en_batch(self.vnu.input_path, self.vnu.output_folder)
                    self.finish_sig.emit('mjo转换完成!')