Skip to content

Allow Python File nodes to activate each other. #426

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 1 commit into from
Feb 8, 2025
Merged

Allow Python File nodes to activate each other. #426

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
44 changes: 35 additions & 9 deletions korman/nodes/node_python.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -13,12 +13,16 @@
# You should have received a copy of the GNU General Public License
# along with Korman. If not, see <http://www.gnu.org/licenses/>.

from __future__ import annotations

import bpy
from bpy.props import *

from collections.abc import Iterable
from contextlib import contextmanager
from pathlib import Path
from typing import *

from PyHSPlasma import *

from .. import enum_props
Expand Down Expand Up @@ -68,9 +72,12 @@
_attrib_key_types = {
"ptAttribSceneobject": plFactory.ClassIndex("plSceneObject"),
"ptAttribSceneobjectList": plFactory.ClassIndex("plSceneObject"),
"ptAttribActivator": plFactory.ClassIndex("plLogicModifier"),
"ptAttribActivatorList": plFactory.ClassIndex("plLogicModifier"),
"ptAttribNamedActivator": plFactory.ClassIndex("plLogicModifier"),
"ptAttribActivator": (plFactory.ClassIndex("plLogicModifier"),
plFactory.ClassIndex("plPythonFileMod")),
"ptAttribActivatorList": (plFactory.ClassIndex("plLogicModifier"),
plFactory.ClassIndex("plPythonFileMod")),
"ptAttribNamedActivator": (plFactory.ClassIndex("plLogicModifier"),
plFactory.ClassIndex("plPythonFileMod")),
"ptAttribResponder": plFactory.ClassIndex("plResponderModifier"),
"ptAttribResponderList": plFactory.ClassIndex("plResponderModifier"),
"ptAttribNamedResponder": plFactory.ClassIndex("plResponderModifier"),
Expand Down Expand Up @@ -187,6 +194,17 @@ class PlasmaPythonFileNode(PlasmaVersionedNode, bpy.types.Node):
bl_label = "Python File"
bl_width_default = 290

# Yas, a PythonFileMod can activate another PythonFileMod
pl_attrib = {"ptAttribActivator", "ptAttribActivatorList", "ptAttribNamedActivator"}

output_sockets: dict[str, dict[str, Any]] = {
"satisfies": {
"text": "Satisfies",
"type": "PlasmaConditionSocket",
"valid_link_nodes": "PlasmaPythonFileNode",
},
}

def _poll_pytext(self, value):
return value.name.endswith(".py")

Expand Down Expand Up @@ -304,7 +322,7 @@ def export(self, exporter, bo, so):
param.value = i

if not socket.is_simple_value:
self._export_key_attrib(exporter, bo, so, i, socket)
self._export_key_attrib(exporter, bo, so, pfm, i, socket)
pfm.addParameter(param)

def _export_ancillary_sceneobject(self, exporter, bo, so: plSceneObject) -> None:
Expand All @@ -317,7 +335,7 @@ def _export_ancillary_sceneobject(self, exporter, bo, so: plSceneObject) -> None
exporter.report.msg(f"Marking RT light '{so.key.name}' as animated due to usage in a Python File node", so.key.name)
light.setProperty(plLightInfo.kLPMovable, True)

def _export_key_attrib(self, exporter, bo, so : plSceneObject, key : plKey, socket) -> None:
def _export_key_attrib(self, exporter, bo, so: plSceneObject, pfm: plPythonFileMod, key: plKey, socket) -> None:
if key is None:
exporter.report.warn("Attribute '{}' didn't return a key and therefore will be unavailable to Python",
self.id_data.name, socket.links[0].name)
Expand All @@ -333,17 +351,21 @@ def _export_key_attrib(self, exporter, bo, so : plSceneObject, key : plKey, sock
self.id_data.name, socket.links[0].from_node.name,
plFactory.ClassName(key.type))

if isinstance(key.object, plSceneObject):
self._export_ancillary_sceneobject(exporter, bo, key.object)
key_object = key.object
if isinstance(key_object, plSceneObject):
self._export_ancillary_sceneobject(exporter, bo, key_object)
elif isinstance(key_object, plPythonFileMod):
key_object.addReceiver(pfm.key)

def _get_attrib_sockets(self, idx):
for i in self.inputs:
if i.attribute_id == idx:
yield i

def generate_valid_links_for(self, context, socket, is_output):
# Python nodes have no outputs...
assert is_output is False
if is_output:
yield from PlasmaNodeBase.generate_valid_links_for(self, context, socket, True)
return

attrib_type = socket.attribute_type
for i in bpy.types.Node.__subclasses__():
Expand Down Expand Up @@ -493,6 +515,10 @@ def update(self):
while len(unconnected) > 1:
self.inputs.remove(unconnected.pop())

# Make sure the output sockets are present and accounted for.
self._update_extant_sockets(self.output_sockets, self.outputs)
self._update_init_sockets(self.output_sockets, self.outputs)

@property
def latest_version(self):
return 2
Expand Down