61 lines
1.3 KiB
Python
61 lines
1.3 KiB
Python
import tidy3d as td
|
|
import sympy as sp
|
|
import sympy.physics.units as spu
|
|
|
|
from .... import contracts as ct
|
|
from .... import sockets
|
|
from ... import base
|
|
|
|
class BoxStructureNode(base.MaxwellSimNode):
|
|
node_type = ct.NodeType.BoxStructure
|
|
bl_label = "Box Structure"
|
|
|
|
####################
|
|
# - Sockets
|
|
####################
|
|
input_sockets = {
|
|
"Medium": sockets.MaxwellMediumSocketDef(),
|
|
"Center": sockets.PhysicalPoint3DSocketDef(),
|
|
"Size": sockets.PhysicalSize3DSocketDef(),
|
|
}
|
|
output_sockets = {
|
|
"Structure": sockets.MaxwellStructureSocketDef(),
|
|
}
|
|
|
|
####################
|
|
# - Output Socket Computation
|
|
####################
|
|
@base.computes_output_socket(
|
|
"Structure",
|
|
input_sockets={"Medium", "Center", "Size"},
|
|
)
|
|
def compute_simulation(self, input_sockets: dict) -> td.Box:
|
|
medium = input_sockets["Medium"]
|
|
_center = input_sockets["Center"]
|
|
_size = input_sockets["Size"]
|
|
|
|
center = tuple(spu.convert_to(_center, spu.um) / spu.um)
|
|
size = tuple(spu.convert_to(_size, spu.um) / spu.um)
|
|
|
|
return td.Structure(
|
|
geometry=td.Box(
|
|
center=center,
|
|
size=size,
|
|
),
|
|
medium=medium,
|
|
)
|
|
|
|
|
|
|
|
####################
|
|
# - Blender Registration
|
|
####################
|
|
BL_REGISTER = [
|
|
BoxStructureNode,
|
|
]
|
|
BL_NODES = {
|
|
ct.NodeType.BoxStructure: (
|
|
ct.NodeCategory.MAXWELLSIM_STRUCTURES_PRIMITIVES
|
|
)
|
|
}
|