72 lines
2.4 KiB
Python
72 lines
2.4 KiB
Python
|
# Copyright Epic Games, Inc. All Rights Reserved.
|
||
|
|
from epic_pose_wrangler.v2.model import base_action
|
||
|
|
|
||
|
|
|
||
|
|
class ExportSelectedAction(base_action.BaseAction):
|
||
|
|
__display_name__ = "Export Selected Solvers"
|
||
|
|
__tooltip__ = "Exports the currently selected solver nodes in the scene to a JSON file"
|
||
|
|
__category__ = "IO"
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def validate(cls, ui_context):
|
||
|
|
return bool(ui_context.current_solvers)
|
||
|
|
|
||
|
|
def execute(self, ui_context=None, **kwargs):
|
||
|
|
from PySide2 import QtWidgets
|
||
|
|
|
||
|
|
if not ui_context:
|
||
|
|
ui_context = self.api.get_ui_context()
|
||
|
|
if not ui_context:
|
||
|
|
return
|
||
|
|
file_path = QtWidgets.QFileDialog.getSaveFileName(None, "Pose Wrangler File", "", "*.json")[0]
|
||
|
|
# If no path is specified, exit early
|
||
|
|
if file_path == "":
|
||
|
|
return
|
||
|
|
self.api.serialize_to_file(file_path, ui_context.current_solvers)
|
||
|
|
|
||
|
|
|
||
|
|
class ExportAllAction(base_action.BaseAction):
|
||
|
|
__display_name__ = "Export All Solvers"
|
||
|
|
__tooltip__ = "Exports all solver nodes in the scene to a JSON file"
|
||
|
|
__category__ = "IO"
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def validate(cls, ui_context):
|
||
|
|
return bool(ui_context.current_solvers)
|
||
|
|
|
||
|
|
def execute(self, ui_context=None, **kwargs):
|
||
|
|
from PySide2 import QtWidgets
|
||
|
|
|
||
|
|
if not ui_context:
|
||
|
|
ui_context = self.api.get_ui_context()
|
||
|
|
if not ui_context:
|
||
|
|
return
|
||
|
|
file_path = QtWidgets.QFileDialog.getSaveFileName(None, "Pose Wrangler File", "", "*.json")[0]
|
||
|
|
# If no path is specified, exit early
|
||
|
|
if file_path == "":
|
||
|
|
return
|
||
|
|
self.api.serialize_to_file(file_path, None)
|
||
|
|
|
||
|
|
|
||
|
|
class ImportFromFileAction(base_action.BaseAction):
|
||
|
|
__display_name__ = "Import Solvers"
|
||
|
|
__tooltip__ = "Imports solver nodes into the scene from a JSON file"
|
||
|
|
__category__ = "IO"
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def validate(cls, ui_context):
|
||
|
|
return bool(ui_context.current_solvers)
|
||
|
|
|
||
|
|
def execute(self, ui_context=None, **kwargs):
|
||
|
|
from PySide2 import QtWidgets
|
||
|
|
|
||
|
|
if not ui_context:
|
||
|
|
ui_context = self.api.get_ui_context()
|
||
|
|
if not ui_context:
|
||
|
|
return
|
||
|
|
file_path = QtWidgets.QFileDialog.getOpenFileName(None, "Pose Wrangler File", "", "*.json")[0]
|
||
|
|
# If no path is specified, exit early
|
||
|
|
if file_path == "":
|
||
|
|
return
|
||
|
|
self.api.deserialize_from_file(file_path)
|