mirror of
https://github.com/H-uru/korman.git
synced 2025-07-14 22:36:52 +00:00
Implement exclude region nodes
This commit is contained in:
@ -22,6 +22,7 @@ import nodeitems_utils
|
||||
from .node_avatar import *
|
||||
from .node_conditions import *
|
||||
from .node_core import *
|
||||
from .node_logic import *
|
||||
from .node_messages import *
|
||||
from .node_python import *
|
||||
from .node_responder import *
|
||||
|
||||
@ -195,7 +195,7 @@ class PlasmaNodeBase:
|
||||
|
||||
# If this is a multiple input node, make sure we have exactly one empty socket
|
||||
if (not socket.is_output and options.get("spawn_empty", False) and not socket.alias in done):
|
||||
empty_sockets = [i for i in sockets if i.bl_idname == socket.bl_idname and not i.links]
|
||||
empty_sockets = [i for i in sockets if i.bl_idname == socket.bl_idname and not i.is_used]
|
||||
if not empty_sockets:
|
||||
dbg = sockets.new(socket.bl_idname, socket.name, socket.alias)
|
||||
else:
|
||||
@ -230,6 +230,10 @@ class PlasmaNodeSocketBase:
|
||||
return tuple(self.bl_color[0], self.bl_color[1], self.bl_color[2], 1.0)
|
||||
return self.bl_color
|
||||
|
||||
@property
|
||||
def is_used(self):
|
||||
return bool(self.links)
|
||||
|
||||
|
||||
class PlasmaNodeTree(bpy.types.NodeTree):
|
||||
bl_idname = "PlasmaNodeTree"
|
||||
|
||||
130
korman/nodes/node_logic.py
Normal file
130
korman/nodes/node_logic.py
Normal file
@ -0,0 +1,130 @@
|
||||
# This file is part of Korman.
|
||||
#
|
||||
# Korman is free software: you can redistribute it and/or modify
|
||||
# it under the terms of the GNU General Public License as published by
|
||||
# the Free Software Foundation, either version 3 of the License, or
|
||||
# (at your option) any later version.
|
||||
#
|
||||
# Korman is distributed in the hope that it will be useful,
|
||||
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
# GNU General Public License for more details.
|
||||
#
|
||||
# You should have received a copy of the GNU General Public License
|
||||
# along with Korman. If not, see <http://www.gnu.org/licenses/>.
|
||||
|
||||
import bpy
|
||||
from bpy.props import *
|
||||
from collections import OrderedDict
|
||||
from PyHSPlasma import *
|
||||
|
||||
from .node_core import *
|
||||
from ..properties.modifiers.physics import bounds_types, bounds_type_index
|
||||
|
||||
class PlasmaExcludeRegionNode(PlasmaNodeBase, bpy.types.Node):
|
||||
bl_category = "LOGIC"
|
||||
bl_idname = "PlasmaExcludeRegionNode"
|
||||
bl_label = "Exclude Region"
|
||||
bl_width_default = 195
|
||||
|
||||
# ohey, this can be a Python attribute
|
||||
pl_attribs = {"ptAttribExcludeRegion"}
|
||||
|
||||
def _get_bounds(self):
|
||||
bo = bpy.data.objects.get(self.region, None)
|
||||
if bo is not None:
|
||||
return bounds_type_index(bo.plasma_modifiers.collision.bounds)
|
||||
return bounds_type_index("hull")
|
||||
def _set_bounds(self, value):
|
||||
bo = bpy.data.objects.get(self.region, None)
|
||||
if bo is not None:
|
||||
bo.plasma_modifiers.collision.bounds = value
|
||||
|
||||
region = StringProperty(name="Region",
|
||||
description="Region object's name")
|
||||
bounds = EnumProperty(name="Bounds",
|
||||
description="Region bounds",
|
||||
items=bounds_types,
|
||||
get=_get_bounds,
|
||||
set=_set_bounds)
|
||||
block_cameras = BoolProperty(name="Block Cameras",
|
||||
description="The region blocks cameras when it has been cleared")
|
||||
|
||||
input_sockets = OrderedDict([
|
||||
("safe_point", {
|
||||
"type": "PlasmaExcludeSafePointSocket",
|
||||
"text": "Safe Point",
|
||||
"spawn_empty": True,
|
||||
# This never links to anything...
|
||||
"valid_link_sockets": frozenset(),
|
||||
}),
|
||||
("msg", {
|
||||
"type": "PlasmaExcludeMessageSocket",
|
||||
"text": "Message",
|
||||
"spawn_empty": True,
|
||||
}),
|
||||
])
|
||||
|
||||
output_sockets = OrderedDict([
|
||||
("keyref", {
|
||||
"text": "References",
|
||||
"type": "PlasmaPythonReferenceNodeSocket",
|
||||
"valid_link_nodes": {"PlasmaPythonFileNode"},
|
||||
}),
|
||||
])
|
||||
|
||||
def draw_buttons(self, context, layout):
|
||||
layout.prop_search(self, "region", bpy.data, "objects", icon="MESH_DATA")
|
||||
layout.prop(self, "bounds")
|
||||
layout.prop(self, "block_cameras")
|
||||
|
||||
def get_key(self, exporter, parent_so):
|
||||
region_bo = bpy.data.objects.get(self.region, None)
|
||||
if region_bo is None:
|
||||
self.raise_error("invalid region object '{}'".format(self.region))
|
||||
return exporter.mgr.find_create_key(plExcludeRegionModifier, bl=region_bo, name=self.key_name)
|
||||
|
||||
def harvest_actors(self):
|
||||
return [i.safepoint_name for i in self.find_input_sockets("safe_points")]
|
||||
|
||||
def export(self, exporter, bo, parent_so):
|
||||
region_bo = bpy.data.objects.get(self.region, None)
|
||||
if region_bo is None:
|
||||
self.raise_error("invalid region object '{}'".format(self.region))
|
||||
region_so = exporter.mgr.find_create_object(plSceneObject, bl=region_bo)
|
||||
excludergn = exporter.mgr.find_create_object(plExcludeRegionModifier, so=region_so, name=self.key_name)
|
||||
excludergn.setFlag(plExcludeRegionModifier.kBlockCameras, self.block_cameras)
|
||||
|
||||
# Safe points
|
||||
for i in self.find_input_sockets("safe_point"):
|
||||
if not i.safepoint_name:
|
||||
continue
|
||||
safept = bpy.data.objects.get(i.safepoint_name, None)
|
||||
if safept is None:
|
||||
self.raise_error("invalid SafePoint '{}'".format(i.safepoint_name))
|
||||
excludergn.addSafePoint(exporter.mgr.find_create_key(plSceneObject, bl=safept))
|
||||
|
||||
# Ensure the region is exported
|
||||
phys_name = "{}_XRgn".format(self.region)
|
||||
simIface, physical = exporter.physics.generate_physical(region_bo, region_so, self.bounds, phys_name)
|
||||
simIface.setProperty(plSimulationInterface.kPinned, True)
|
||||
physical.setProperty(plSimulationInterface.kPinned, True)
|
||||
physical.LOSDBs |= plSimDefs.kLOSDBUIBlockers
|
||||
|
||||
|
||||
class PlasmaExcludeSafePointSocket(PlasmaNodeSocketBase, bpy.types.NodeSocket):
|
||||
bl_color = (0.0, 0.0, 0.0, 1.0)
|
||||
|
||||
safepoint_name = StringProperty(name="Safe Point",
|
||||
description="A point outside of this exclude region to move the avatar to")
|
||||
|
||||
def draw(self, context, layout, node, text):
|
||||
layout.prop_search(self, "safepoint_name", bpy.data, "objects", icon="EMPTY_DATA")
|
||||
|
||||
@property
|
||||
def is_used(self):
|
||||
return bpy.data.objects.get(self.safepoint_name, None) is not None
|
||||
|
||||
|
||||
class PlasmaExcludeMessageSocket(PlasmaNodeSocketBase, bpy.types.NodeSocket):
|
||||
bl_color = (0.467, 0.576, 0.424, 1.0)
|
||||
@ -268,6 +268,35 @@ class PlasmaEnableMsgNode(PlasmaMessageNode, bpy.types.Node):
|
||||
layout.prop(self, "settings")
|
||||
|
||||
|
||||
class PlasmaExcludeRegionMsg(PlasmaMessageNode, bpy.types.Node):
|
||||
bl_category = "MSG"
|
||||
bl_idname = "PlasmaExcludeRegionMsg"
|
||||
bl_label = "Exclude Region"
|
||||
|
||||
output_sockets = OrderedDict([
|
||||
("region", {
|
||||
"text": "Region",
|
||||
"type": "PlasmaExcludeMessageSocket"
|
||||
}),
|
||||
])
|
||||
|
||||
cmd = EnumProperty(name="Command",
|
||||
description="Exclude Region State",
|
||||
items=[("kClear", "Clear", "Clear all avatars from the region"),
|
||||
("kRelease", "Release", "Allow avatars to enter the region")],
|
||||
default="kClear")
|
||||
|
||||
def convert_message(self, exporter, so):
|
||||
msg = plExcludeRegionMsg()
|
||||
for i in self.find_outputs("region"):
|
||||
msg.addReceiver(i.get_key(exporter, so))
|
||||
msg.cmd = getattr(plExcludeRegionMsg, self.cmd)
|
||||
return msg
|
||||
|
||||
def draw_buttons(self, context, layout):
|
||||
layout.prop(self, "cmd", text="Cmd")
|
||||
|
||||
|
||||
class PlasmaOneShotMsgNode(PlasmaMessageNode, bpy.types.Node):
|
||||
bl_category = "MSG"
|
||||
bl_idname = "PlasmaOneShotMsgNode"
|
||||
|
||||
@ -28,6 +28,9 @@ bounds_types = (
|
||||
("trimesh", "Triangle Mesh", "Use the exact triangle mesh (SLOW!)")
|
||||
)
|
||||
|
||||
def bounds_type_index(key):
|
||||
return list(zip(*bounds_types))[0].index(key)
|
||||
|
||||
def _set_phys_prop(prop, sim, phys, value=True):
|
||||
"""Sets properties on plGenericPhysical and plSimulationInterface (seeing as how they are duped)"""
|
||||
sim.setProperty(prop, value)
|
||||
|
||||
Reference in New Issue
Block a user