98 lines
2.0 KiB
Python
98 lines
2.0 KiB
Python
|
import functools
|
||
|
|
import typing as typ
|
||
|
|
import json
|
||
|
|
from pathlib import Path
|
||
|
|
|
||
|
|
import bpy
|
||
|
|
import sympy as sp
|
||
|
|
import pydantic as pyd
|
||
|
|
import tidy3d as td
|
||
|
|
|
||
|
|
from ... import contracts as ct
|
||
|
|
from ... import sockets
|
||
|
|
from .. import base
|
||
|
|
|
||
|
|
|
||
|
|
class ConsoleViewOperator(bpy.types.Operator):
|
||
|
|
bl_idname = "blender_maxwell.console_view_operator"
|
||
|
|
bl_label = "View Plots"
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def poll(cls, context):
|
||
|
|
return True
|
||
|
|
|
||
|
|
def execute(self, context):
|
||
|
|
node = context.node
|
||
|
|
node.print_data_to_console()
|
||
|
|
return {'FINISHED'}
|
||
|
|
|
||
|
|
class RefreshPlotViewOperator(bpy.types.Operator):
|
||
|
|
bl_idname = "blender_maxwell.refresh_plot_view_operator"
|
||
|
|
bl_label = "Refresh Plots"
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def poll(cls, context):
|
||
|
|
return True
|
||
|
|
|
||
|
|
def execute(self, context):
|
||
|
|
node = context.node
|
||
|
|
node.trigger_action("value_changed", "Data")
|
||
|
|
return {'FINISHED'}
|
||
|
|
|
||
|
|
####################
|
||
|
|
# - Node
|
||
|
|
####################
|
||
|
|
class ViewerNode(base.MaxwellSimNode):
|
||
|
|
node_type = ct.NodeType.Viewer
|
||
|
|
bl_label = "Viewer"
|
||
|
|
|
||
|
|
input_sockets = {
|
||
|
|
"Data": sockets.AnySocketDef(),
|
||
|
|
}
|
||
|
|
|
||
|
|
####################
|
||
|
|
# - UI
|
||
|
|
####################
|
||
|
|
def draw_operators(self, context, layout):
|
||
|
|
row = layout.row(align=True)
|
||
|
|
row.label(text="Console")
|
||
|
|
row.operator(ConsoleViewOperator.bl_idname, text="Print")
|
||
|
|
|
||
|
|
row = layout.row(align=True)
|
||
|
|
row.label(text="Plot")
|
||
|
|
row.operator(RefreshPlotViewOperator.bl_idname, text="", icon="FILE_REFRESH")
|
||
|
|
|
||
|
|
####################
|
||
|
|
# - Methods
|
||
|
|
####################
|
||
|
|
def print_data_to_console(self):
|
||
|
|
if not (data := self._compute_input("Data")):
|
||
|
|
return
|
||
|
|
|
||
|
|
if isinstance(data, sp.Basic):
|
||
|
|
sp.pprint(data, use_unicode=True)
|
||
|
|
|
||
|
|
print(str(data))
|
||
|
|
|
||
|
|
####################
|
||
|
|
# - Update
|
||
|
|
####################
|
||
|
|
@base.on_value_changed(socket_name="Data")
|
||
|
|
def on_value_changed__data(self):
|
||
|
|
self.trigger_action("show_plot")
|
||
|
|
|
||
|
|
|
||
|
|
####################
|
||
|
|
# - Blender Registration
|
||
|
|
####################
|
||
|
|
BL_REGISTER = [
|
||
|
|
ConsoleViewOperator,
|
||
|
|
RefreshPlotViewOperator,
|
||
|
|
ViewerNode,
|
||
|
|
]
|
||
|
|
BL_NODES = {
|
||
|
|
ct.NodeType.Viewer: (
|
||
|
|
ct.NodeCategory.MAXWELLSIM_OUTPUTS
|
||
|
|
)
|
||
|
|
}
|