feat: lazy boundconds aggregator
We also fixed the loose socket sorting bug w/some choice choices of `inputs|outputs.move()`.main
parent
624914f8cb
commit
830b316e01
|
@ -479,6 +479,16 @@ class MaxwellSimNode(bpy.types.Node, bl_instance.BLInstance):
|
||||||
self._prune_inactive_sockets()
|
self._prune_inactive_sockets()
|
||||||
self._add_new_active_sockets()
|
self._add_new_active_sockets()
|
||||||
|
|
||||||
|
for i, socket_name in enumerate(self.active_socket_defs('input')):
|
||||||
|
current_idx_of_socket = self.inputs.find(socket_name)
|
||||||
|
if i != current_idx_of_socket:
|
||||||
|
self.inputs.move(current_idx_of_socket, i)
|
||||||
|
|
||||||
|
for i, socket_name in enumerate(self.active_socket_defs('output')):
|
||||||
|
current_idx_of_socket = self.outputs.find(socket_name)
|
||||||
|
if i != current_idx_of_socket:
|
||||||
|
self.outputs.move(current_idx_of_socket, i)
|
||||||
|
|
||||||
####################
|
####################
|
||||||
# - Event Methods
|
# - Event Methods
|
||||||
####################
|
####################
|
||||||
|
|
|
@ -98,6 +98,7 @@ class BoundCondsNode(base.MaxwellSimNode):
|
||||||
@events.computes_output_socket(
|
@events.computes_output_socket(
|
||||||
'BCs',
|
'BCs',
|
||||||
kind=ct.FlowKind.Value,
|
kind=ct.FlowKind.Value,
|
||||||
|
# Loaded
|
||||||
input_sockets={'X', 'Y', 'Z', '+X', '-X', '+Y', '-Y', '+Z', '-Z'},
|
input_sockets={'X', 'Y', 'Z', '+X', '-X', '+Y', '-Y', '+Z', '-Z'},
|
||||||
input_sockets_optional={
|
input_sockets_optional={
|
||||||
'X': True,
|
'X': True,
|
||||||
|
@ -110,23 +111,26 @@ class BoundCondsNode(base.MaxwellSimNode):
|
||||||
'+Z': True,
|
'+Z': True,
|
||||||
'-Z': True,
|
'-Z': True,
|
||||||
},
|
},
|
||||||
|
output_sockets={'BCs'},
|
||||||
|
output_socket_kinds={'BCs': ct.FlowKind.Params},
|
||||||
)
|
)
|
||||||
def compute_boundary_conds(self, input_sockets) -> td.BoundarySpec:
|
def compute_bcs_value(self, input_sockets, output_sockets) -> td.BoundarySpec:
|
||||||
"""Compute the simulation boundary conditions, by combining the individual input by specified half axis."""
|
"""Compute the simulation boundary conditions, by combining the individual input by specified half axis."""
|
||||||
log.debug(
|
output_params = output_sockets['BCs']
|
||||||
'%s: Computing Boundary Conditions (Input Sockets = %s)',
|
has_output_params = not ct.FlowSignal.check(output_params)
|
||||||
self.sim_node_name,
|
|
||||||
str(input_sockets),
|
|
||||||
)
|
|
||||||
|
|
||||||
# Deduce "Doubledness"
|
# Deduce "Doubledness"
|
||||||
## -> A "doubled" axis defines the same bound cond both ways
|
## -> A "doubled" axis defines the same bound cond both ways
|
||||||
has_doubled_x = not ct.FlowSignal.check(input_sockets['X'])
|
x = input_sockets['X']
|
||||||
has_doubled_y = not ct.FlowSignal.check(input_sockets['Y'])
|
y = input_sockets['Y']
|
||||||
has_doubled_z = not ct.FlowSignal.check(input_sockets['Z'])
|
z = input_sockets['Z']
|
||||||
|
|
||||||
|
has_doubled_x = not ct.FlowSignal.check_single(x, ct.FlowSignal.NoFlow)
|
||||||
|
has_doubled_y = not ct.FlowSignal.check_single(y, ct.FlowSignal.NoFlow)
|
||||||
|
has_doubled_z = not ct.FlowSignal.check_single(z, ct.FlowSignal.NoFlow)
|
||||||
|
|
||||||
# Deduce +/- of Each Axis
|
# Deduce +/- of Each Axis
|
||||||
## +/- X
|
## -> +/- X
|
||||||
if has_doubled_x:
|
if has_doubled_x:
|
||||||
x_pos = input_sockets['X']
|
x_pos = input_sockets['X']
|
||||||
x_neg = input_sockets['X']
|
x_neg = input_sockets['X']
|
||||||
|
@ -134,7 +138,10 @@ class BoundCondsNode(base.MaxwellSimNode):
|
||||||
x_pos = input_sockets['+X']
|
x_pos = input_sockets['+X']
|
||||||
x_neg = input_sockets['-X']
|
x_neg = input_sockets['-X']
|
||||||
|
|
||||||
## +/- Y
|
has_x_pos = not ct.FlowSignal.check(x_pos)
|
||||||
|
has_x_neg = not ct.FlowSignal.check(x_neg)
|
||||||
|
|
||||||
|
## -> +/- Y
|
||||||
if has_doubled_y:
|
if has_doubled_y:
|
||||||
y_pos = input_sockets['Y']
|
y_pos = input_sockets['Y']
|
||||||
y_neg = input_sockets['Y']
|
y_neg = input_sockets['Y']
|
||||||
|
@ -142,7 +149,10 @@ class BoundCondsNode(base.MaxwellSimNode):
|
||||||
y_pos = input_sockets['+Y']
|
y_pos = input_sockets['+Y']
|
||||||
y_neg = input_sockets['-Y']
|
y_neg = input_sockets['-Y']
|
||||||
|
|
||||||
## +/- Z
|
has_y_pos = not ct.FlowSignal.check(y_pos)
|
||||||
|
has_y_neg = not ct.FlowSignal.check(y_neg)
|
||||||
|
|
||||||
|
## -> +/- Z
|
||||||
if has_doubled_z:
|
if has_doubled_z:
|
||||||
z_pos = input_sockets['Z']
|
z_pos = input_sockets['Z']
|
||||||
z_neg = input_sockets['Z']
|
z_neg = input_sockets['Z']
|
||||||
|
@ -150,6 +160,19 @@ class BoundCondsNode(base.MaxwellSimNode):
|
||||||
z_pos = input_sockets['+Z']
|
z_pos = input_sockets['+Z']
|
||||||
z_neg = input_sockets['-Z']
|
z_neg = input_sockets['-Z']
|
||||||
|
|
||||||
|
has_z_pos = not ct.FlowSignal.check(z_pos)
|
||||||
|
has_z_neg = not ct.FlowSignal.check(z_neg)
|
||||||
|
|
||||||
|
if (
|
||||||
|
has_x_pos
|
||||||
|
and has_x_neg
|
||||||
|
and has_y_pos
|
||||||
|
and has_y_neg
|
||||||
|
and has_z_pos
|
||||||
|
and has_z_neg
|
||||||
|
and has_output_params
|
||||||
|
and not output_params.symbols
|
||||||
|
):
|
||||||
return td.BoundarySpec(
|
return td.BoundarySpec(
|
||||||
x=td.Boundary(
|
x=td.Boundary(
|
||||||
plus=x_pos,
|
plus=x_pos,
|
||||||
|
@ -164,6 +187,197 @@ class BoundCondsNode(base.MaxwellSimNode):
|
||||||
minus=z_neg,
|
minus=z_neg,
|
||||||
),
|
),
|
||||||
)
|
)
|
||||||
|
return ct.FlowSignal.FlowPending
|
||||||
|
|
||||||
|
####################
|
||||||
|
# - FlowKind.Func
|
||||||
|
####################
|
||||||
|
@events.computes_output_socket(
|
||||||
|
'BCs',
|
||||||
|
kind=ct.FlowKind.Func,
|
||||||
|
input_sockets={'X', 'Y', 'Z', '+X', '-X', '+Y', '-Y', '+Z', '-Z'},
|
||||||
|
input_socket_kinds={
|
||||||
|
'X': ct.FlowKind.Func,
|
||||||
|
'Y': ct.FlowKind.Func,
|
||||||
|
'Z': ct.FlowKind.Func,
|
||||||
|
'+X': ct.FlowKind.Func,
|
||||||
|
'-X': ct.FlowKind.Func,
|
||||||
|
'+Y': ct.FlowKind.Func,
|
||||||
|
'-Y': ct.FlowKind.Func,
|
||||||
|
'+Z': ct.FlowKind.Func,
|
||||||
|
'-Z': ct.FlowKind.Func,
|
||||||
|
},
|
||||||
|
input_sockets_optional={
|
||||||
|
'X': True,
|
||||||
|
'Y': True,
|
||||||
|
'Z': True,
|
||||||
|
'+X': True,
|
||||||
|
'-X': True,
|
||||||
|
'+Y': True,
|
||||||
|
'-Y': True,
|
||||||
|
'+Z': True,
|
||||||
|
'-Z': True,
|
||||||
|
},
|
||||||
|
)
|
||||||
|
def compute_bcs_func(self, input_sockets) -> ct.ParamsFlow | ct.FlowSignal:
|
||||||
|
"""Compute the simulation boundary conditions, by combining the individual input by specified half axis."""
|
||||||
|
# Deduce "Doubledness"
|
||||||
|
## -> A "doubled" axis defines the same bound cond both ways
|
||||||
|
x = input_sockets['X']
|
||||||
|
y = input_sockets['Y']
|
||||||
|
z = input_sockets['Z']
|
||||||
|
|
||||||
|
has_doubled_x = not ct.FlowSignal.check_single(x, ct.FlowSignal.NoFlow)
|
||||||
|
has_doubled_y = not ct.FlowSignal.check_single(y, ct.FlowSignal.NoFlow)
|
||||||
|
has_doubled_z = not ct.FlowSignal.check_single(z, ct.FlowSignal.NoFlow)
|
||||||
|
|
||||||
|
# Deduce +/- of Each Axis
|
||||||
|
## -> +/- X
|
||||||
|
if has_doubled_x:
|
||||||
|
x_pos = input_sockets['X']
|
||||||
|
x_neg = input_sockets['X']
|
||||||
|
else:
|
||||||
|
x_pos = input_sockets['+X']
|
||||||
|
x_neg = input_sockets['-X']
|
||||||
|
|
||||||
|
has_x_pos = not ct.FlowSignal.check(x_pos)
|
||||||
|
has_x_neg = not ct.FlowSignal.check(x_neg)
|
||||||
|
|
||||||
|
## -> +/- Y
|
||||||
|
if has_doubled_y:
|
||||||
|
y_pos = input_sockets['Y']
|
||||||
|
y_neg = input_sockets['Y']
|
||||||
|
else:
|
||||||
|
y_pos = input_sockets['+Y']
|
||||||
|
y_neg = input_sockets['-Y']
|
||||||
|
|
||||||
|
has_y_pos = not ct.FlowSignal.check(y_pos)
|
||||||
|
has_y_neg = not ct.FlowSignal.check(y_neg)
|
||||||
|
|
||||||
|
## -> +/- Z
|
||||||
|
if has_doubled_z:
|
||||||
|
z_pos = input_sockets['Z']
|
||||||
|
z_neg = input_sockets['Z']
|
||||||
|
else:
|
||||||
|
z_pos = input_sockets['+Z']
|
||||||
|
z_neg = input_sockets['-Z']
|
||||||
|
|
||||||
|
has_z_pos = not ct.FlowSignal.check(z_pos)
|
||||||
|
has_z_neg = not ct.FlowSignal.check(z_neg)
|
||||||
|
|
||||||
|
if (
|
||||||
|
has_x_pos
|
||||||
|
and has_x_neg
|
||||||
|
and has_y_pos
|
||||||
|
and has_y_neg
|
||||||
|
and has_z_pos
|
||||||
|
and has_z_neg
|
||||||
|
):
|
||||||
|
return (x_pos | x_neg | y_pos | y_neg | z_pos | z_neg).compose_within(
|
||||||
|
enclosing_func=lambda els: td.BoundarySpec(
|
||||||
|
x=td.Boundary(
|
||||||
|
plus=els[0],
|
||||||
|
minus=els[1],
|
||||||
|
),
|
||||||
|
y=td.Boundary(
|
||||||
|
plus=els[2],
|
||||||
|
minus=els[3],
|
||||||
|
),
|
||||||
|
z=td.Boundary(
|
||||||
|
plus=els[4],
|
||||||
|
minus=els[5],
|
||||||
|
),
|
||||||
|
),
|
||||||
|
supports_jax=False,
|
||||||
|
)
|
||||||
|
return ct.FlowSignal.FlowPending
|
||||||
|
|
||||||
|
####################
|
||||||
|
# - FlowKind.Params
|
||||||
|
####################
|
||||||
|
@events.computes_output_socket(
|
||||||
|
'BCs',
|
||||||
|
kind=ct.FlowKind.Params,
|
||||||
|
input_sockets={'X', 'Y', 'Z', '+X', '-X', '+Y', '-Y', '+Z', '-Z'},
|
||||||
|
input_socket_kinds={
|
||||||
|
'X': ct.FlowKind.Params,
|
||||||
|
'Y': ct.FlowKind.Params,
|
||||||
|
'Z': ct.FlowKind.Params,
|
||||||
|
'+X': ct.FlowKind.Params,
|
||||||
|
'-X': ct.FlowKind.Params,
|
||||||
|
'+Y': ct.FlowKind.Params,
|
||||||
|
'-Y': ct.FlowKind.Params,
|
||||||
|
'+Z': ct.FlowKind.Params,
|
||||||
|
'-Z': ct.FlowKind.Params,
|
||||||
|
},
|
||||||
|
input_sockets_optional={
|
||||||
|
'X': True,
|
||||||
|
'Y': True,
|
||||||
|
'Z': True,
|
||||||
|
'+X': True,
|
||||||
|
'-X': True,
|
||||||
|
'+Y': True,
|
||||||
|
'-Y': True,
|
||||||
|
'+Z': True,
|
||||||
|
'-Z': True,
|
||||||
|
},
|
||||||
|
)
|
||||||
|
def compute_bcs_params(self, input_sockets) -> ct.ParamsFlow | ct.FlowSignal:
|
||||||
|
"""Compute the simulation boundary conditions, by combining the individual input by specified half axis."""
|
||||||
|
# Deduce "Doubledness"
|
||||||
|
## -> A "doubled" axis defines the same bound cond both ways
|
||||||
|
x = input_sockets['X']
|
||||||
|
y = input_sockets['Y']
|
||||||
|
z = input_sockets['Z']
|
||||||
|
|
||||||
|
has_doubled_x = not ct.FlowSignal.check_single(x, ct.FlowSignal.NoFlow)
|
||||||
|
has_doubled_y = not ct.FlowSignal.check_single(y, ct.FlowSignal.NoFlow)
|
||||||
|
has_doubled_z = not ct.FlowSignal.check_single(z, ct.FlowSignal.NoFlow)
|
||||||
|
|
||||||
|
# Deduce +/- of Each Axis
|
||||||
|
## -> +/- X
|
||||||
|
if has_doubled_x:
|
||||||
|
x_pos = input_sockets['X']
|
||||||
|
x_neg = input_sockets['X']
|
||||||
|
else:
|
||||||
|
x_pos = input_sockets['+X']
|
||||||
|
x_neg = input_sockets['-X']
|
||||||
|
|
||||||
|
has_x_pos = not ct.FlowSignal.check(x_pos)
|
||||||
|
has_x_neg = not ct.FlowSignal.check(x_neg)
|
||||||
|
|
||||||
|
## -> +/- Y
|
||||||
|
if has_doubled_y:
|
||||||
|
y_pos = input_sockets['Y']
|
||||||
|
y_neg = input_sockets['Y']
|
||||||
|
else:
|
||||||
|
y_pos = input_sockets['+Y']
|
||||||
|
y_neg = input_sockets['-Y']
|
||||||
|
|
||||||
|
has_y_pos = not ct.FlowSignal.check(y_pos)
|
||||||
|
has_y_neg = not ct.FlowSignal.check(y_neg)
|
||||||
|
|
||||||
|
## -> +/- Z
|
||||||
|
if has_doubled_z:
|
||||||
|
z_pos = input_sockets['Z']
|
||||||
|
z_neg = input_sockets['Z']
|
||||||
|
else:
|
||||||
|
z_pos = input_sockets['+Z']
|
||||||
|
z_neg = input_sockets['-Z']
|
||||||
|
|
||||||
|
has_z_pos = not ct.FlowSignal.check(z_pos)
|
||||||
|
has_z_neg = not ct.FlowSignal.check(z_neg)
|
||||||
|
|
||||||
|
if (
|
||||||
|
has_x_pos
|
||||||
|
and has_x_neg
|
||||||
|
and has_y_pos
|
||||||
|
and has_y_neg
|
||||||
|
and has_z_pos
|
||||||
|
and has_z_neg
|
||||||
|
):
|
||||||
|
return x_pos | x_neg | y_pos | y_neg | z_pos | z_neg
|
||||||
|
return ct.FlowSignal.FlowPending
|
||||||
|
|
||||||
|
|
||||||
####################
|
####################
|
||||||
|
|
|
@ -79,6 +79,16 @@ class MaxwellBoundCondBLSocket(base.MaxwellSimSocket):
|
||||||
def value(self, value: ct.BoundCondType) -> None:
|
def value(self, value: ct.BoundCondType) -> None:
|
||||||
self.default = value
|
self.default = value
|
||||||
|
|
||||||
|
@bl_cache.cached_bl_property(depends_on={'value'})
|
||||||
|
def lazy_func(self) -> ct.FuncFlow:
|
||||||
|
return ct.FuncFlow(
|
||||||
|
func=lambda: self.value,
|
||||||
|
)
|
||||||
|
|
||||||
|
@bl_cache.cached_bl_property()
|
||||||
|
def params(self) -> ct.FuncFlow:
|
||||||
|
return ct.ParamsFlow()
|
||||||
|
|
||||||
|
|
||||||
####################
|
####################
|
||||||
# - Socket Configuration
|
# - Socket Configuration
|
||||||
|
|
Loading…
Reference in New Issue