Ejemplo n.º 1
0
 def analyze_tails(self):
     self.__currentRow = 0
     self.__thresholdLinear = AudioEditTools.decibel_to_linear(
         self.spbCutThreshold.value())
     self.__batchProcessor = BatchProcessor(self.__soundFiles,
                                            self.analyze_file)
     self.__batchProcessor.start()
Ejemplo n.º 2
0
 def copy_selection_to_active_objects(self):
     if WaapiTools.Client is None:
         return
     selection = WaapiTools.get_selected_objects()
     if len(selection) > 0:
         self.batchProcessor = BatchProcessor(
             self.activeObjects,
             lambda obj: WaapiTools.copy_object(selection[0], obj))
         self.batchProcessor.start()
Ejemplo n.º 3
0
 def move_active_objects_to_selection(self):
     if WaapiTools.Client is None:
         return
     selection = WaapiTools.get_selected_objects()
     if len(selection) > 0:
         self.batchProcessor = BatchProcessor(
             self.activeObjects,
             lambda obj: WaapiTools.move_object(obj, selection[0]))
         self.batchProcessor.start()
Ejemplo n.º 4
0
 def apply_naming_convention(self, naming_rule: int):
     if naming_rule == 0:
         self.batchProcessor = BatchProcessor(self.activeObjects,
                                              rename_to_lower_case)
     elif naming_rule == 1:
         self.batchProcessor = BatchProcessor(self.activeObjects,
                                              rename_to_title_case)
     else:
         self.batchProcessor = BatchProcessor(self.activeObjects,
                                              rename_to_upper_case)
     self.batchProcessor.start()
Ejemplo n.º 5
0
 def localize_languages(self):
     if WaapiTools.Client is None:
         return
     self.batchProcessor = BatchProcessor(self.activeObjects,
                                          localize_languages)
     self.batchProcessor.start()
Ejemplo n.º 6
0
 def rename_original_to_wwise(self):
     if WaapiTools.Client is None:
         return
     self.batchProcessor = BatchProcessor(self.activeObjects,
                                          rename_original_to_wwise)
     self.batchProcessor.start()
Ejemplo n.º 7
0
 def reset_source_editor(self):
     self.batchProcessor = BatchProcessor(self.activeObjects,
                                          reset_source_editor)
     self.batchProcessor.start()
Ejemplo n.º 8
0
 def clear_bank_inclusions(self):
     self.batchProcessor = BatchProcessor(self.activeObjects,
                                          clear_bank_inclusions)
     self.batchProcessor.start()
Ejemplo n.º 9
0
 def create_play_event(self):
     self.batchProcessor = BatchProcessor(self.activeObjects,
                                          create_play_event)
     self.batchProcessor.start()
Ejemplo n.º 10
0
 def apply_fader_edits_downstream(self):
     self.batchProcessor = BatchProcessor(self.activeObjects,
                                          apply_fader_edits_downstream)
     self.batchProcessor.start()
Ejemplo n.º 11
0
 def delete_all_objects(self):
     self.batchProcessor = BatchProcessor(self.activeObjects,
                                          WaapiTools.delete_object)
     self.batchProcessor.start()
     self.clear_object_list()
Ejemplo n.º 12
0
 def set_inclusion(self, included: bool):
     self.batchProcessor = BatchProcessor(
         self.activeObjects, lambda obj: WaapiTools.set_object_property(
             obj, 'Inclusion', True if included else False))
     self.batchProcessor.start()
Ejemplo n.º 13
0
 def convert_to_type(self, target_type: str):
     self.batchProcessor = BatchProcessor(
         self.activeObjects,
         lambda obj: WaapiTools.convert_to_type(obj, target_type))
     self.batchProcessor.start()
Ejemplo n.º 14
0
class MainWindow(QMainWindow, Ui_MainWindow):
    def __init__(self):
        super().__init__()
        LogTool.init_log()
        self.setupUi(self)
        self.setup_triggers()
        self.cacheObjects = []  # 通过Get Seletion、Find Children等操作获得的obj列表
        self.activeObjects = []  # cacheObjects经过filter过滤后的obj列表
        self.batchProcessor = None
        self.statusbar.showMessage('Wwise无法连接...')
        self.tblActiveObjects.horizontalHeader().setSectionResizeMode(
            QHeaderView.ResizeToContents)
        # 初始化默认尝试连接wwise
        self.connect_to_wwise()

    def setup_triggers(self):
        self.tblActiveObjects.itemDoubleClicked.connect(
            self.show_object_in_wwise)

        self.btnWaapiConnect.clicked.connect(self.connect_to_wwise)
        self.btnGetSelectedObjects.clicked.connect(self.get_selected_objects)
        self.btnUpdateObjects.clicked.connect(self.update_list_objects)
        self.btnRemoveSelection.clicked.connect(self.remove_table_selection)
        self.btnClearObjects.clicked.connect(self.clear_object_list)
        self.btnMultiEditor.clicked.connect(self.open_in_multi_editor)
        self.btnBatchRename.clicked.connect(self.open_in_batch_rename)

        self.actConvertToWorkUnit.triggered.connect(
            lambda: self.convert_to_type('WorkUnit'))
        self.actConvertToActorMixer.triggered.connect(
            lambda: self.convert_to_type('ActorMixer'))
        self.actConvertToVirtualFolder.triggered.connect(
            lambda: self.convert_to_type('Folder'))
        self.actConvertToBlendContainer.triggered.connect(
            lambda: self.convert_to_type('BlendContainer'))
        self.actConvertToRandomSequenceContainer.triggered.connect(
            lambda: self.convert_to_type('RandomSequenceContainer'))
        self.actConvertToSwitchContainer.triggered.connect(
            lambda: self.convert_to_type('SwitchContainer'))

        self.btnFindParent.clicked.connect(self.find_parent)
        self.btnFindChildren.clicked.connect(self.find_children)

        self.reset_filter()

        self.actUndo.triggered.connect(WaapiTools.undo)
        self.actRedo.triggered.connect(WaapiTools.redo)
        self.actSetIncluded.triggered.connect(lambda: self.set_inclusion(True))
        self.actSetExcluded.triggered.connect(
            lambda: self.set_inclusion(False))
        self.actFilterIncluded.triggered.connect(
            lambda: self.filter_by_inclusion(True))
        self.actFilterExcluded.triggered.connect(
            lambda: self.filter_by_inclusion(False))
        self.actDeleteObjects.triggered.connect(self.delete_all_objects)

        self.actMoveListToSelection.triggered.connect(
            self.move_active_objects_to_selection)
        self.actCopySelectionToList.triggered.connect(
            self.copy_selection_to_active_objects)
        self.actChangeToLowerCase.triggered.connect(
            lambda: self.apply_naming_convention(0))
        self.actChangeToTitleCase.triggered.connect(
            lambda: self.apply_naming_convention(1))
        self.actChangeToUpperCase.triggered.connect(
            lambda: self.apply_naming_convention(2))
        # self.actWwiseSilenceAdd.triggered.connect(self.create_wwise_silence)
        # self.actWwiseSilenceRemove.triggered.connect(self.remove_wwise_silence)

        self.actImportFromFile.triggered.connect(self.import_from_file)
        self.actExportToFile.triggered.connect(self.export_to_file)

        self.actBatchReplaceTool.triggered.connect(self.batch_replace_tool)

        self.actApplyEditsToOriginal.triggered.connect(self.apply_source_edits)
        self.actResetSourceEdits.triggered.connect(self.reset_source_editor)
        self.actTrimTailSilence.triggered.connect(self.trim_tail_silence)
        self.actRenameOriginalToWwise.triggered.connect(
            self.rename_original_to_wwise)
        self.actDeleteUnusedAKDFiles.triggered.connect(delete_unused_akd_files)
        self.actLocalizeLanguages.triggered.connect(self.localize_languages)

        self.actBreakContainer.triggered.connect(self.break_container)
        self.actReplaceParent.triggered.connect(self.replace_parent)
        self.actAssignSwitchMappings.triggered.connect(
            self.assign_switch_mappings)
        self.actRemoveAllSwitchAssignments.triggered.connect(
            self.remove_all_switch_mappings)
        self.actSplitByNetRole.triggered.connect(self.split_by_net_role)
        self.actApplyFaderEditsDownstream.triggered.connect(
            self.apply_fader_edits_downstream)
        self.actCreatePlayEvent.triggered.connect(self.create_play_event)

        self.actCalculateBankSize.triggered.connect(
            self.calculate_bank_total_size)
        self.actCreateOrAddToBank.triggered.connect(self.create_or_add_to_bank)
        self.actAddToSelectedBank.triggered.connect(self.add_to_selected_bank)
        self.actClearInclusions.triggered.connect(self.clear_bank_inclusions)
        self.actIncludeMediaOnly.triggered.connect(
            lambda: self.set_bank_inclusion_type(['media']))
        self.actIncludeEventsAndStructures.triggered.connect(
            lambda: self.set_bank_inclusion_type(['events', 'structures']))
        self.actIncludeAll.triggered.connect(
            lambda: self.set_bank_inclusion_type(
                ['events', 'structures', 'media']))
        self.actBankAssignmentMatrix.triggered.connect(
            self.bank_assignment_matrix)

        self.actTempTool.triggered.connect(
            lambda: temp_tool(self.activeObjects))
        self.actUnityResourceManager.triggered.connect(
            self.manage_unity_resources)

    # 重置筛选条件
    def reset_filter(self):
        try:
            self.rbnFilterByName.toggled.disconnect(self.filter_and_show_list)
            self.rbnFilterByPath.toggled.disconnect(self.filter_and_show_list)
            self.rbnFilterByInclude.toggled.disconnect(
                self.filter_and_show_list)
            self.rbnFilterByExclude.toggled.disconnect(
                self.filter_and_show_list)
            self.cbxUseRegularExpression.toggled.disconnect(
                self.filter_and_show_list)
            self.cbxMatchWholeWord.toggled.disconnect(
                self.filter_and_show_list)
            self.cbxCaseSensitive.toggled.disconnect(self.filter_and_show_list)
            self.cbbDescendantType.currentIndexChanged.disconnect(
                self.filter_and_show_list)
            self.iptSelectionFilter.textChanged.disconnect(
                self.filter_and_show_list)
        except:
            print("filter toggle first disconnect fail")

        self.rbnFilterByName.setChecked(True)
        self.rbnFilterByInclude.setChecked(True)
        self.cbxUseRegularExpression.setChecked(False)
        self.cbxMatchWholeWord.setChecked(False)
        self.cbxCaseSensitive.setChecked(False)
        self.cbbDescendantType.setCurrentIndex(0)
        self.iptSelectionFilter.setText("")

        self.rbnFilterByName.toggled.connect(self.filter_and_show_list)
        self.rbnFilterByPath.toggled.connect(self.filter_and_show_list)
        self.rbnFilterByInclude.toggled.connect(self.filter_and_show_list)
        self.rbnFilterByExclude.toggled.connect(self.filter_and_show_list)
        self.cbxUseRegularExpression.toggled.connect(self.filter_and_show_list)
        self.cbxMatchWholeWord.toggled.connect(self.filter_and_show_list)
        self.cbxCaseSensitive.toggled.connect(self.filter_and_show_list)
        self.cbbDescendantType.currentIndexChanged.connect(
            self.filter_and_show_list)
        self.iptSelectionFilter.textChanged.connect(self.filter_and_show_list)

    # 通过指定的端口连接到Wwise
    def connect_to_wwise(self):
        url = 'ws://127.0.0.1:' + str(self.spbWaapiPort.value()) + '/waapi'
        try:
            WaapiTools.Client = WaapiClient(url=url)
            WaapiTools.Client.subscribe('ak.wwise.core.project.postClosed',
                                        self.on_wwise_closed)
            self.statusbar.showMessage('Wwise已连接到' +
                                       WaapiTools.get_project_directory())
            self.btnWaapiConnect.setEnabled(False)
        except CannotConnectToWaapiException:
            self.statusbar.showMessage('无法连接到Wwise工程...')

    # Wwise关闭时的回调
    def on_wwise_closed(self):
        self.statusbar.showMessage('Wwise未连接...')
        self.btnWaapiConnect.setEnabled(True)
        WaapiTools.Client = None

    # 获取Wwise中选中的对象
    def get_selected_objects(self):
        if WaapiTools.Client is None:
            return
        self.cacheObjects = WaapiTools.get_selected_objects()
        self.reset_filter()
        self.filter_and_show_list()

    # 根据filter刷新当前列表内容
    def update_list_objects(self):
        self.cacheObjects = self.activeObjects
        self.reset_filter()
        self.filter_and_show_list()

    # 删去表格中选中的对象
    def remove_table_selection(self):
        for item in self.tblActiveObjects.selectedItems():
            self.activeObjects.pop(item.row())
            self.tblActiveObjects.removeRow(item.row())

    # 清空操作对象列表
    def clear_object_list(self):
        self.cacheObjects = []
        self.activeObjects = []
        self.tblActiveObjects.setRowCount(0)

    # 实时筛选列表内容
    def filter_obj_list(self):
        self.activeObjects = ScriptingTools.filter_objects(
            self.cacheObjects, self.iptSelectionFilter.text(),
            self.cbxCaseSensitive.isChecked(),
            self.cbxMatchWholeWord.isChecked(),
            self.cbxUseRegularExpression.isChecked(),
            self.cbbDescendantType.currentText(),
            self.rbnFilterByName.isChecked(),
            self.rbnFilterByInclude.isChecked())

    def show_obj_list(self):
        self.tblActiveObjects.setRowCount(0)
        for obj in self.activeObjects:
            row_count = self.tblActiveObjects.rowCount()
            self.tblActiveObjects.insertRow(row_count)
            self.tblActiveObjects.setItem(row_count, 0,
                                          QTableWidgetItem(obj['name']))
            self.tblActiveObjects.setItem(row_count, 1,
                                          QTableWidgetItem(obj['type']))
            self.tblActiveObjects.setItem(row_count, 2,
                                          QTableWidgetItem(obj['path']))

    # 在表中显示筛选过的对象
    def filter_and_show_list(self):
        self.filter_obj_list()
        self.show_obj_list()

    # 查找父级对象
    def find_parent(self):
        if WaapiTools.Client is None:
            return
        if self.cbxKeepSelf.isChecked():
            all_parents = self.activeObjects
        else:
            all_parents = []
        for obj in self.activeObjects:
            for parent in WaapiTools.get_parent_objects(
                    obj, self.cbxRecursiveFind.isChecked()):
                all_parents.append(parent)
        self.cacheObjects = all_parents
        self.reset_filter()
        self.filter_and_show_list()

    # 查找子级对象
    def find_children(self):
        if WaapiTools.Client is None:
            return
        if self.cbxKeepSelf.isChecked():
            all_children = self.activeObjects
        else:
            all_children = []
        for obj in self.activeObjects:
            for child in WaapiTools.get_children_objects(
                    obj, self.cbxRecursiveFind.isChecked()):
                all_children.append(child)
        self.cacheObjects = all_children
        self.reset_filter()
        self.filter_and_show_list()

    def on_ui_executed(*args, **kwargs):
        # 获取对象类型
        obj_type = kwargs.get("object", {}).get("type")
        # 获取之前的名字
        old_name = kwargs.get("oldName")
        # 获取新名字
        new_name = kwargs.get("newName")

        # 使用 format 格式化函数进行输出信息(其中的{}代表 format() 函数中的对应变量),告知用户XXX类型的对象从 A 改名到了 B
        print("Object '{}' (of type '{}') was renamed to '{}'\n".format(
            old_name, obj_type, new_name))

    def show_object_in_wwise(self, item: QTableWidgetItem):
        if WaapiTools.Client is None:
            return
        if item.column() != 2:
            item = self.tblActiveObjects.item(item.row(), 2)
            WaapiTools.open_item_in_wwise_by_path(item.text())

    # 通用操作
    def convert_to_type(self, target_type: str):
        self.batchProcessor = BatchProcessor(
            self.activeObjects,
            lambda obj: WaapiTools.convert_to_type(obj, target_type))
        self.batchProcessor.start()

    def set_inclusion(self, included: bool):
        self.batchProcessor = BatchProcessor(
            self.activeObjects, lambda obj: WaapiTools.set_object_property(
                obj, 'Inclusion', True if included else False))
        self.batchProcessor.start()

    def filter_by_inclusion(self, included: bool):
        self.cacheObjects = ScriptingTools.filter_objects_by_inclusion(
            self.activeObjects, True if included else False)
        self.reset_filter()
        self.filter_and_show_list()

    def delete_all_objects(self):
        self.batchProcessor = BatchProcessor(self.activeObjects,
                                             WaapiTools.delete_object)
        self.batchProcessor.start()
        self.clear_object_list()

    def apply_naming_convention(self, naming_rule: int):
        if naming_rule == 0:
            self.batchProcessor = BatchProcessor(self.activeObjects,
                                                 rename_to_lower_case)
        elif naming_rule == 1:
            self.batchProcessor = BatchProcessor(self.activeObjects,
                                                 rename_to_title_case)
        else:
            self.batchProcessor = BatchProcessor(self.activeObjects,
                                                 rename_to_upper_case)
        self.batchProcessor.start()

    def create_wwise_silence(self):
        WwiseSilenceTool.WwiseSilenceInstance.Add()

    def remove_wwise_silence(self):
        WwiseSilenceTool.WwiseSilenceInstance.Remove()

    # 导出为文件
    def export_to_file(self):
        if WaapiTools.Client is None:
            return
        saveDict = {
            "selectObject":
            WaapiTools.get_selected_objects(),
            "activeObjects":
            self.activeObjects,
            "cacheObjects":
            self.cacheObjects,
            "radioBtn_group_key_name":
            self.radioBtn_group_key_name.isChecked(),
            "radioBtn_group_ope_include":
            self.radioBtn_group_ope_include.isChecked(),
            "cbxUseRegularExpression":
            self.cbxUseRegularExpression.isChecked(),
            "cbxMatchWholeWord":
            self.cbxMatchWholeWord.isChecked(),
            "cbxCaseSensitive":
            self.cbxCaseSensitive.isChecked(),
            "cbbDescendantType":
            self.cbbDescendantType.currentIndex(),
            "iptSelectionFilter":
            self.iptSelectionFilter.text()
        }
        FileTools.export_to_file(saveDict)

    # 从文件导入,并刷新UI
    def import_from_file(self):
        loadDict = FileTools.import_from_file()
        if loadDict:
            try:
                self.radioBtn_group_key_name.setChecked(
                    loadDict["radioBtn_group_key_name"])
                self.radioBtn_group_ope_include.setChecked(
                    loadDict["radioBtn_group_ope_include"])
                self.cbxUseRegularExpression.setChecked(
                    loadDict["cbxUseRegularExpression"])
                self.cbxMatchWholeWord.setChecked(
                    loadDict["cbxMatchWholeWord"])
                self.cbxCaseSensitive.setChecked(loadDict["cbxCaseSensitive"])
                self.cbbDescendantType.setCurrentIndex(
                    loadDict["cbbDescendantType"])
                self.iptSelectionFilter.setText(loadDict["iptSelectionFilter"])
                self.activeObjects = loadDict["activeObjects"]
                self.cacheObjects = loadDict["cacheObjects"]
                self.show_obj_list()
            except:
                print("import from file failed")

    def move_active_objects_to_selection(self):
        if WaapiTools.Client is None:
            return
        selection = WaapiTools.get_selected_objects()
        if len(selection) > 0:
            self.batchProcessor = BatchProcessor(
                self.activeObjects,
                lambda obj: WaapiTools.move_object(obj, selection[0]))
            self.batchProcessor.start()

    def copy_selection_to_active_objects(self):
        if WaapiTools.Client is None:
            return
        selection = WaapiTools.get_selected_objects()
        if len(selection) > 0:
            self.batchProcessor = BatchProcessor(
                self.activeObjects,
                lambda obj: WaapiTools.copy_object(selection[0], obj))
            self.batchProcessor.start()

    def break_container(self):
        self.batchProcessor = BatchProcessor(self.activeObjects,
                                             break_container)
        self.batchProcessor.start()

    def replace_parent(self):
        self.batchProcessor = BatchProcessor(self.activeObjects,
                                             replace_parent)
        self.batchProcessor.start()

    def open_in_multi_editor(self):
        if WaapiTools.Client is None:
            return
        WaapiTools.execute_ui_command(self.activeObjects, 'ShowMultiEditor')

    def open_in_batch_rename(self):
        if WaapiTools.Client is None:
            return
        WaapiTools.execute_ui_command(self.activeObjects, 'ShowBatchRename')

    # 音频文件操作
    def apply_source_edits(self):
        self.batchProcessor = BatchProcessor(self.activeObjects,
                                             apply_source_edit)
        self.batchProcessor.start()

    def reset_source_editor(self):
        self.batchProcessor = BatchProcessor(self.activeObjects,
                                             reset_source_editor)
        self.batchProcessor.start()

    def trim_tail_silence(self):
        trim_window = AudioTailTrimmer(self.activeObjects)
        trim_window.show()
        trim_window.exec_()

    def batch_replace_tool(self):
        if WaapiTools.Client is None:
            return
        replace_window = BatchReplaceTool(self)
        replace_window.show()
        replace_window.exec_()

    def rename_original_to_wwise(self):
        if WaapiTools.Client is None:
            return
        self.batchProcessor = BatchProcessor(self.activeObjects,
                                             rename_original_to_wwise)
        self.batchProcessor.start()

    def localize_languages(self):
        if WaapiTools.Client is None:
            return
        self.batchProcessor = BatchProcessor(self.activeObjects,
                                             localize_languages)
        self.batchProcessor.start()

    # LogicContainer操作
    def assign_switch_mappings(self):
        self.batchProcessor = BatchProcessor(self.activeObjects,
                                             auto_assign_switch_mappings)
        self.batchProcessor.start()

    def remove_all_switch_mappings(self):
        self.batchProcessor = BatchProcessor(self.activeObjects,
                                             remove_all_switch_assignments)
        self.batchProcessor.start()

    def apply_fader_edits_downstream(self):
        self.batchProcessor = BatchProcessor(self.activeObjects,
                                             apply_fader_edits_downstream)
        self.batchProcessor.start()

    def split_by_net_role(self):
        self.batchProcessor = BatchProcessor(self.activeObjects,
                                             split_by_net_role)
        self.batchProcessor.start()

    # Event操作
    def create_play_event(self):
        self.batchProcessor = BatchProcessor(self.activeObjects,
                                             create_play_event)
        self.batchProcessor.start()

    # SoundBank操作
    def create_or_add_to_bank(self):
        self.batchProcessor = BatchProcessor(self.activeObjects,
                                             create_or_add_to_bank)
        self.batchProcessor.start()

    def calculate_bank_total_size(self):
        if WaapiTools.Client is None:
            return
        get_bank_size(self.activeObjects)

    def add_to_selected_bank(self):
        if WaapiTools.Client is None:
            return
        banks = WaapiTools.get_selected_objects()
        if len(banks) == 0 or banks[0]['type'] != 'SoundBank':
            return
        add_objects_to_bank(banks[0], self.activeObjects, ['media'])

    def clear_bank_inclusions(self):
        self.batchProcessor = BatchProcessor(self.activeObjects,
                                             clear_bank_inclusions)
        self.batchProcessor.start()

    def set_bank_inclusion_type(self, inclusion_type: list):
        self.batchProcessor = BatchProcessor(
            self.activeObjects,
            lambda obj: set_inclusion_type(obj, inclusion_type))
        self.batchProcessor.start()

    def bank_assignment_matrix(self):
        if WaapiTools.Client is None:
            return
        matrix_window = BankAssignmentMatrix(self)
        matrix_window.show()
        matrix_window.exec_()

    # 其他操作
    def manage_unity_resources(self):
        resource_manager = UnityResourceManager()
        resource_manager.show()
        resource_manager.exec_()

    def closeEvent(self, close_event):
        self.batchProcessor = None
Ejemplo n.º 15
0
 def start_trimming(self):
     self.__currentRow = 0
     self.__fadeDuration = self.spbFadeDuration.value()
     processor = BatchProcessor(self.__soundFiles, self.trim_file)
     processor.start()
Ejemplo n.º 16
0
 def assign_switch_mappings(self):
     self.batchProcessor = BatchProcessor(self.activeObjects,
                                          auto_assign_switch_mappings)
     self.batchProcessor.start()
Ejemplo n.º 17
0
 def remove_all_switch_mappings(self):
     self.batchProcessor = BatchProcessor(self.activeObjects,
                                          remove_all_switch_assignments)
     self.batchProcessor.start()
Ejemplo n.º 18
0
 def break_container(self):
     self.batchProcessor = BatchProcessor(self.activeObjects,
                                          break_container)
     self.batchProcessor.start()
Ejemplo n.º 19
0
 def split_by_net_role(self):
     self.batchProcessor = BatchProcessor(self.activeObjects,
                                          split_by_net_role)
     self.batchProcessor.start()
Ejemplo n.º 20
0
 def replace_parent(self):
     self.batchProcessor = BatchProcessor(self.activeObjects,
                                          replace_parent)
     self.batchProcessor.start()
Ejemplo n.º 21
0
 def create_or_add_to_bank(self):
     self.batchProcessor = BatchProcessor(self.activeObjects,
                                          create_or_add_to_bank)
     self.batchProcessor.start()
Ejemplo n.º 22
0
 def apply_source_edits(self):
     self.batchProcessor = BatchProcessor(self.activeObjects,
                                          apply_source_edit)
     self.batchProcessor.start()
Ejemplo n.º 23
0
 def set_bank_inclusion_type(self, inclusion_type: list):
     self.batchProcessor = BatchProcessor(
         self.activeObjects,
         lambda obj: set_inclusion_type(obj, inclusion_type))
     self.batchProcessor.start()
Ejemplo n.º 24
0
class AudioTailTrimmer(QDialog, Ui_AudioTailTrimmer):
    def __init__(self, active_objects: list):
        super().__init__()
        self.setupUi(self)
        self.setup_triggers()
        self.__thresholdLinear = 0
        self.__currentRow = 0
        self.__fadeDuration = 0
        self.__batchProcessor = None
        self.__soundFiles = []
        # 主窗口中有wwise对象,从中查找音频源文件
        if len(active_objects) > 0 and WaapiTools.Client:
            self.populate_from_wwise(active_objects)
        self.tblFileList.horizontalHeader().setSectionResizeMode(
            QHeaderView.ResizeToContents)

    def setup_triggers(self):
        self.btnImportFiles.clicked.connect(self.import_files)
        self.btnAnalyzeTails.clicked.connect(self.analyze_tails)
        self.btnStartTrim.clicked.connect(self.start_trimming)
        self.tblFileList.viewport().installEventFilter(self)

    # 从外界拖动文件到窗口中
    def eventFilter(self, source, event):
        if event.type() == QEvent.DragEnter:
            self.reset()
            for url in event.mimeData().urls():
                self.add_file(url.toLocalFile())
            return True
        return super().eventFilter(source, event)

    # 重置所有
    def reset(self):
        for sound_file in self.__soundFiles:
            sound_file.close()
        self.__soundFiles = []
        self.__batchProcessor = None
        self.tblFileList.setRowCount(0)

    # 根据主窗口的对象寻找音频文件
    def populate_from_wwise(self, objects: list):
        for obj in objects:
            if obj['type'] == 'Sound':
                self.add_file(WaapiTools.get_original_wave_path(obj))
            else:
                self.populate_from_wwise(
                    WaapiTools.get_children_objects(obj, False))

    # 导入音频文件
    def import_files(self):
        self.reset()
        files = QFileDialog.getOpenFileNames(filter='WAV(*.wav)')
        file_count = len(files[0])
        if file_count == 0:
            return
        for file_path in files[0]:
            self.add_file(file_path)
        self.tblFileList.repaint()

    # 添加单个文件回调
    def add_file(self, file_path: str):
        # only accept wav files
        if not file_path.endswith('.wav'):
            return
        # update row count
        row = self.tblFileList.rowCount()
        self.tblFileList.setRowCount(row + 1)
        # get audio info
        sound_file = soundfile.SoundFile(file=file_path)
        self.__soundFiles.append(sound_file)
        # print to table
        self.tblFileList.setItem(row, 0,
                                 QTableWidgetItem(os.path.basename(file_path)))

    # 分析音尾时长
    def analyze_tails(self):
        self.__currentRow = 0
        self.__thresholdLinear = AudioEditTools.decibel_to_linear(
            self.spbCutThreshold.value())
        self.__batchProcessor = BatchProcessor(self.__soundFiles,
                                               self.analyze_file)
        self.__batchProcessor.start()

    # 分析单个音频文件回调
    def analyze_file(self, sound_file):
        sound_file.seek(0)
        audio_data = sound_file.read()
        # calculate total duration
        duration = round(sound_file.frames / sound_file.samplerate * 1.0, 3)
        self.tblFileList.setItem(self.__currentRow, 1,
                                 QTableWidgetItem(str(duration)))
        # calculate silence duration
        silence_frames = AudioEditTools.get_tail_silence_duration_frames(
            audio_data, 256, self.__thresholdLinear)
        silence_duration = round(
            silence_frames * 256 / sound_file.samplerate * 1.0, 3)
        self.tblFileList.setItem(self.__currentRow, 2,
                                 QTableWidgetItem(str(silence_duration)))
        if silence_duration > self.spbFadeDuration.value():
            status_item = QTableWidgetItem('可裁剪')
            status_item.setForeground(QColor(200, 0, 0))
        else:
            status_item = QTableWidgetItem('无需裁剪')
        self.tblFileList.setItem(self.__currentRow, 3, status_item)
        self.__currentRow += 1

    # 开始裁剪
    def start_trimming(self):
        self.__currentRow = 0
        self.__fadeDuration = self.spbFadeDuration.value()
        processor = BatchProcessor(self.__soundFiles, self.trim_file)
        processor.start()

    # 裁剪单个音频文件回调
    def trim_file(self, sound_file):
        silence_duration = float(
            self.tblFileList.item(self.__currentRow, 2).text())
        if silence_duration > self.__fadeDuration:
            sound_file.seek(0)
            audio_data = sound_file.read()
            trim_samples = int((silence_duration - self.__fadeDuration) *
                               sound_file.samplerate)
            # 进行裁剪
            audio_data = audio_data[:len(audio_data) - trim_samples]
            # 把保留的静音时长做个淡出
            AudioEditTools.apply_fade(
                audio_data, int(self.__fadeDuration * sound_file.samplerate),
                -1)
            soundfile.write(file=sound_file.name,
                            data=audio_data,
                            samplerate=sound_file.samplerate)
            # 更新列表显示
            self.tblFileList.setItem(
                self.__currentRow, 2,
                QTableWidgetItem(str(self.__fadeDuration)))
            status_item = QTableWidgetItem('已裁剪')
            status_item.setForeground(QColor(0, 200, 0))
            self.tblFileList.setItem(self.__currentRow, 3, status_item)
        self.__currentRow += 1

    def closeEvent(self, close_event):
        self.reset()