72 lines
1.5 KiB
Python
72 lines
1.5 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 BoundCondsNode(base.MaxwellSimNode):
|
|
node_type = ct.NodeType.BoundConds
|
|
bl_label = "Bound Box"
|
|
#bl_icon = ...
|
|
|
|
####################
|
|
# - Sockets
|
|
####################
|
|
input_sockets = {
|
|
"+X": sockets.MaxwellBoundCondSocketDef(),
|
|
"-X": sockets.MaxwellBoundCondSocketDef(),
|
|
"+Y": sockets.MaxwellBoundCondSocketDef(),
|
|
"-Y": sockets.MaxwellBoundCondSocketDef(),
|
|
"+Z": sockets.MaxwellBoundCondSocketDef(),
|
|
"-Z": sockets.MaxwellBoundCondSocketDef(),
|
|
}
|
|
output_sockets = {
|
|
"BCs": sockets.MaxwellBoundCondsSocketDef(),
|
|
}
|
|
|
|
####################
|
|
# - Output Socket Computation
|
|
####################
|
|
@base.computes_output_socket(
|
|
"BCs",
|
|
input_sockets={"+X", "-X", "+Y", "-Y", "+Z", "-Z"}
|
|
)
|
|
def compute_simulation(self, input_sockets) -> td.BoundarySpec:
|
|
x_pos = input_sockets["+X"]
|
|
x_neg = input_sockets["-X"]
|
|
y_pos = input_sockets["+Y"]
|
|
y_neg = input_sockets["-Y"]
|
|
z_pos = input_sockets["+Z"]
|
|
z_neg = input_sockets["-Z"]
|
|
|
|
return td.BoundarySpec(
|
|
x=td.Boundary(
|
|
plus=x_pos,
|
|
minus=x_neg,
|
|
),
|
|
y=td.Boundary(
|
|
plus=y_pos,
|
|
minus=y_neg,
|
|
),
|
|
z=td.Boundary(
|
|
plus=z_pos,
|
|
minus=z_neg,
|
|
),
|
|
)
|
|
|
|
|
|
|
|
####################
|
|
# - Blender Registration
|
|
####################
|
|
BL_REGISTER = [
|
|
BoundCondsNode,
|
|
]
|
|
BL_NODES = {
|
|
ct.NodeType.BoundConds: (
|
|
ct.NodeCategory.MAXWELLSIM_BOUNDS
|
|
)
|
|
}
|