184 lines
6.6 KiB
Python
184 lines
6.6 KiB
Python
from __future__ import annotations
|
|
from typing import Any
|
|
import bpy
|
|
from bpy.props import CollectionProperty
|
|
from .bpy_ns_types import NfStringProperty, NfFloatProperty, NfIntProperty, NfEnumProperty, NfPointerProperty, NfFloatVectorProperty
|
|
|
|
from ...registry import register_class
|
|
from ...infra.blender.blender_slice_nodes_generic import SliceNodeSocket
|
|
from ..slice import SlicePaths
|
|
|
|
from .bpy_ns_types import notify_tree_changed
|
|
|
|
class LabelNodeSocket(SliceNodeSocket):
|
|
default = ""
|
|
def draw(self, context: Any, layout: Any, node: Any, text: str | None):
|
|
assert layout
|
|
layout.label(text=text or self.name)
|
|
|
|
class InputNodeSocket(SliceNodeSocket):
|
|
def draw(self, context: Any, layout: Any, node: Any, text: str | None):
|
|
assert layout
|
|
if self.is_output or self.is_linked:
|
|
layout.label(text=text or self.name)
|
|
else:
|
|
self._draw_prop(text, layout)
|
|
|
|
def _draw_prop(self, text, layout):
|
|
layout.prop(self, "value", text=text or self.name)
|
|
|
|
def _draw_as_ro(self, context: Any, layout: Any, node: Any, text: str | None):
|
|
layout.label(text=text or self.name)
|
|
|
|
def _draw_as_rw(self, context: Any, layout: Any, node: Any, text: str | None):
|
|
layout.prop(self, "value", text=text or self.name)
|
|
|
|
class ReadonlySocket():
|
|
def draw(self, context: Any, layout: Any, node: Any, text: str | None):
|
|
return self._draw_as_ro(context, layout, node, text) # pyright: ignore[reportAttributeAccessIssue]
|
|
|
|
|
|
|
|
@register_class
|
|
class SLICESOCK_Geometry(LabelNodeSocket):
|
|
bl_label = "Geometry"
|
|
default = None
|
|
bl_color = (0.0, 0.75, 0.75, 1.0)
|
|
|
|
def draw_color(self, context, node) -> tuple[float, float, float, float]:
|
|
def _get_upstream_color(sock: Any) -> tuple[float, float, float, float]:
|
|
if not sock or not getattr(sock, "is_linked", False) or not getattr(sock, "links", None):
|
|
return self.bl_color
|
|
from_sock = sock.links[0].from_socket
|
|
if hasattr(from_sock, "draw_color"):
|
|
try:
|
|
from_node = getattr(from_sock, "node", None)
|
|
return from_sock.draw_color(context, from_node)
|
|
except Exception:
|
|
pass
|
|
return getattr(from_sock, "bl_color", self.bl_color)
|
|
|
|
if getattr(self, "is_output", False):
|
|
if node and hasattr(node, "inputs"):
|
|
sock_name = getattr(self, "name", "")
|
|
matching_in = (node.inputs.get(sock_name) if sock_name else None) or node.inputs.get("Geometry")
|
|
if matching_in:
|
|
return _get_upstream_color(matching_in)
|
|
return self.bl_color
|
|
else:
|
|
return _get_upstream_color(self)
|
|
|
|
@register_class
|
|
class SLICESOCK_Mesh(LabelNodeSocket):
|
|
bl_label = "Mesh"
|
|
default = None
|
|
bl_color = (0, 214/255, 163/255, 1.0)
|
|
|
|
@register_class
|
|
class SLICESOCK_Paths(LabelNodeSocket):
|
|
bl_label = "Paths"
|
|
default = SlicePaths()
|
|
bl_color = (214/255, 0, 120/255, 1.0)
|
|
|
|
@register_class
|
|
class SLICESOCK_Intrinsics(LabelNodeSocket):
|
|
bl_label = "Intrinsics"
|
|
default = None
|
|
bl_color = (0.2, 0.2, 0.2, 1.0)
|
|
|
|
@register_class
|
|
class SLICESOCK_PrusaslicerConf(LabelNodeSocket):
|
|
bl_label = "PrusaSlicer Config"
|
|
default = None
|
|
bl_color = (0.95, 0., 0., 1.0) # #F26C21
|
|
|
|
@register_class
|
|
class SettingItem(bpy.types.PropertyGroup):
|
|
field: NfStringProperty(name="Field", default="") # pyright: ignore[reportInvalidTypeForm]
|
|
value_str: NfStringProperty(name="Value", default="") # pyright: ignore[reportInvalidTypeForm]
|
|
|
|
@property
|
|
def is_active(self):
|
|
return bool(self.field) or bool(self.value_str)
|
|
|
|
def draw_item(self, layout):
|
|
row = layout.row(align=True)
|
|
row.prop(self, "field", text="")
|
|
row.prop(self, "value_str", text="")
|
|
|
|
from ...infra.blender.mixins import CollectionMixin
|
|
class SLICESOCK_Settings_Common(CollectionMixin, InputNodeSocket):
|
|
bl_label = "Settings"
|
|
|
|
items: CollectionProperty(type=SettingItem) # pyright: ignore[reportInvalidTypeForm]
|
|
|
|
bl_color = (1.0, 1.0, 0.0, 1.0) # Yellow
|
|
|
|
@property
|
|
def value(self) -> dict[str, str]:
|
|
overrides = {}
|
|
for item in self.items:
|
|
if item.field:
|
|
overrides[item.field] = item.value_str
|
|
return overrides
|
|
|
|
@register_class
|
|
class SLICESOCK_Settings(SLICESOCK_Settings_Common): ...
|
|
@register_class
|
|
class SLICESOCK_SettingsRo(ReadonlySocket, SLICESOCK_Settings_Common): ...
|
|
|
|
@register_class
|
|
class SLICESOCK_ExecutionLink(LabelNodeSocket):
|
|
bl_label = "Execution Link"
|
|
default = None
|
|
bl_color = (0.5, 0.0, 0.5, 1.0) # Purple
|
|
|
|
@register_class
|
|
class SLICESOCK_Printer(InputNodeSocket):
|
|
bl_label = "Printer"
|
|
|
|
host: NfStringProperty(name="Host", default="localhost") # pyright: ignore[reportInvalidTypeForm]
|
|
port: NfIntProperty(name="Port", default=80, min=1, max=65535) # pyright: ignore[reportInvalidTypeForm]
|
|
prefix: NfStringProperty(name="Prefix", default="") # pyright: ignore[reportInvalidTypeForm]
|
|
label: NfStringProperty(name="Label", default="Printer") # pyright: ignore[reportInvalidTypeForm]
|
|
username: NfStringProperty(name="Username", default="") # pyright: ignore[reportInvalidTypeForm]
|
|
password: NfStringProperty(name="Password", default="", subtype='PASSWORD') # pyright: ignore[reportInvalidTypeForm]
|
|
|
|
host_type: NfEnumProperty(
|
|
name="Type",
|
|
items=[
|
|
('PRUSALINK', "PrusaLink", "PrusaLink API"),
|
|
('CREALITY', "Creality", "Creality Cloud API"),
|
|
('MOONRAKER', "Moonraker", "Moonraker (Klipper) API"),
|
|
],
|
|
default='PRUSALINK',
|
|
update=notify_tree_changed
|
|
) # pyright: ignore[reportInvalidTypeForm]
|
|
|
|
bl_color = (1.0, 0.45, 0.1, 1.0) # Orange
|
|
|
|
def draw(self, context, layout, node, text):
|
|
if self.is_output or self.is_linked:
|
|
layout.label(text=text or self.name)
|
|
else:
|
|
col = layout.column(align=True)
|
|
col.prop(self, "label", text="")
|
|
col.prop(self, "host_type", text="")
|
|
col.prop(self, "host")
|
|
col.prop(self, "port")
|
|
col.prop(self, "prefix")
|
|
col.prop(self, "username")
|
|
col.prop(self, "password")
|
|
|
|
@property
|
|
def value(self) -> Any:
|
|
from ...infra.networking.printer_backends import PrinterConfig
|
|
return PrinterConfig(
|
|
host=self.host,
|
|
port=self.port,
|
|
prefix=self.prefix,
|
|
label=self.label,
|
|
username=self.username,
|
|
password=self.password,
|
|
host_type=self.host_type
|
|
)
|