D4rkst3randClaude Opus 4.8
f9e3fe4223
Kern aus EcoGame tools/blender_tree_gen.py regeneriert. Ast-Richtung jetzt ueber Tangent/Normal von 'Curve to Points' (Goldener Winkel) statt Weltkoordinaten -> Sub-Aeste wachsen entlang ihres Elternastes statt quer dazu. 'Tip Blunt' neu. Headless getestet: alle 4 Presets, Stufen aufsteigend, Modifier-Apply ok. Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
713 lines
30 KiB
Python
713 lines
30 KiB
Python
bl_info = {
|
|
"name": "Stylized Tree Generator",
|
|
"author": "D4rkst3r",
|
|
"version": (1, 1, 0),
|
|
"blender": (4, 2, 0),
|
|
"location": "View3D > Sidebar > Tree Gen",
|
|
"description": "Parametrischer Baum-/Palmen-/Busch-Generator (Geometry Nodes) mit Wachstums-Stufen",
|
|
"category": "Add Mesh",
|
|
}
|
|
|
|
# Parametrischer Stylized-Baum als Geometry-Nodes-Gruppe (Stamm + Aeste).
|
|
# Blaetter/Krone macht der Nutzer selbst - dieses Addon liefert das Geaest.
|
|
#
|
|
# HINWEIS: Der Node-Aufbau unten ist identisch mit EcoGame tools/blender_tree_gen.py
|
|
# (dort als CLI-Variante). Aenderungen bitte in beiden pflegen.
|
|
#
|
|
# LIVE BEARBEITEN: Nach dem Erzeugen liegen alle Regler am Modifier "GN_Tree" -
|
|
# im Viewport ziehen, der Baum aktualisiert sich sofort.
|
|
|
|
import bpy
|
|
import sys
|
|
import os
|
|
|
|
GROUP_NAME = "GN_StylizedTree"
|
|
OBJ_NAME = "StylizedTree"
|
|
|
|
# ============================ REGLER (Defaults) ============================
|
|
DEFAULTS = {
|
|
"Seed": 0,
|
|
"Height": 5.0,
|
|
"Trunk Radius": 0.13, # schlanker Stamm; zu dick wirkt sofort "stumpf"
|
|
"Bend": 0.5,
|
|
"Taper": 0.9,
|
|
"Branch Count": 9,
|
|
"Branch Length": 1.3,
|
|
"Branch Up": 1.3, # >1 = Aeste zeigen nach oben; negativ = haengend
|
|
"Branch Start": 0.35, # ab wo am Stamm Aeste sitzen (0..1)
|
|
"Branch End": 0.95,
|
|
"Branch Bend": 0.45, # Variation je Ast (0 = gerade Staebe)
|
|
"Branch Droop": 0.0, # >0 haengt nach unten (Palme), <0 kruemmt nach oben (Kaktus)
|
|
"Branch Thickness": 0.32, # Ast-Dicke relativ zum Stamm
|
|
"Branch Taper": 1.0, # 1 = laeuft spitz zu, 0.2 = bleibt dick (Kaktus)
|
|
"Sub Count": 3, # Sub-Aeste je Hauptast (0 = aus)
|
|
"Sub Length": 0.55,
|
|
"Sub Up": 1.4,
|
|
"Sides": 6,
|
|
"Tip Blunt": 0.0, # 0 = laeuft spitz aus (Baum), 0.45 = stumpfes Ende (Kaktus)
|
|
"Ribs": 0.0, # senkrechte Kanneluren (0 = glatt); Kaktus ~9
|
|
"Rib Depth": 0.0,
|
|
}
|
|
|
|
# Presets: nur die Abweichungen von DEFAULTS.
|
|
PRESETS = {
|
|
"baum": {},
|
|
"palme": {
|
|
"Height": 7.0, "Trunk Radius": 0.16, "Bend": 1.1, "Taper": 0.55,
|
|
"Branch Count": 11, "Branch Length": 2.6, "Branch Up": 0.55,
|
|
"Branch Start": 0.93, "Branch End": 1.0,
|
|
"Branch Bend": 0.25, "Branch Droop": 1.5,
|
|
"Sub Count": 0, # Wedel-Fiederung macht der User als Blattwerk
|
|
"Sides": 6,
|
|
},
|
|
"busch": {
|
|
"Height": 1.6, "Trunk Radius": 0.07, "Bend": 0.4, "Taper": 0.8,
|
|
"Branch Count": 14, "Branch Length": 0.9, "Branch Up": 1.0,
|
|
"Branch Start": 0.15, "Branch End": 0.95,
|
|
"Branch Bend": 0.6, "Branch Droop": 0.0,
|
|
"Sub Count": 2, "Sub Length": 0.35, "Sub Up": 1.6, "Sides": 5,
|
|
},
|
|
# Kaktus: dicker, kaum verjuengter Stamm, wenige Arme, die per NEGATIVEM
|
|
# Droop nach oben kruemmen (Saguaro-Silhouette). Keine Sub-Aeste.
|
|
"kaktus": {
|
|
"Height": 3.0, "Trunk Radius": 0.28, "Bend": 0.08, "Taper": 0.10,
|
|
"Branch Count": 2, "Branch Length": 1.6, "Branch Up": 0.05,
|
|
"Branch Start": 0.28, "Branch End": 0.5,
|
|
"Branch Bend": 0.0,
|
|
"Branch Droop": -2.2, # negativ = Arme kruemmen nach OBEN (Saguaro)
|
|
"Branch Thickness": 0.72, # Arme fast so dick wie der Stamm
|
|
"Branch Taper": 0.12, # Arme bleiben dick statt spitz zuzulaufen
|
|
"Sub Count": 0,
|
|
"Tip Blunt": 0.45,
|
|
"Sides": 18, "Ribs": 9.0, "Rib Depth": 0.09,
|
|
},
|
|
}
|
|
# ===========================================================================
|
|
|
|
|
|
def _sock(node, *names):
|
|
"""Falle 1: Socket nach Namen holen, mit Alternativen."""
|
|
for n in names:
|
|
if n in node.inputs:
|
|
return node.inputs[n]
|
|
raise KeyError("Socket %s nicht in %s" % (names, node.bl_idname))
|
|
|
|
|
|
def _out(node, *names):
|
|
for n in names:
|
|
if n in node.outputs:
|
|
return node.outputs[n]
|
|
return node.outputs[0]
|
|
|
|
|
|
def build_group():
|
|
old = bpy.data.node_groups.get(GROUP_NAME)
|
|
if old:
|
|
bpy.data.node_groups.remove(old)
|
|
ng = bpy.data.node_groups.new(GROUP_NAME, "GeometryNodeTree")
|
|
N, L = ng.nodes.new, ng.links.new
|
|
iface = ng.interface
|
|
|
|
def _new_any(*bl_idnames):
|
|
"""Falle 2: erster Node-Typ, den diese Blender-Version kennt."""
|
|
for bid in bl_idnames:
|
|
try:
|
|
return N(bid)
|
|
except Exception:
|
|
continue
|
|
return None
|
|
|
|
iface.new_socket("Geometry", in_out='OUTPUT', socket_type='NodeSocketGeometry')
|
|
|
|
def add_in(name, stype, default, mn=None, mx=None):
|
|
s = iface.new_socket(name, in_out='INPUT', socket_type=stype)
|
|
s.default_value = default
|
|
if mn is not None:
|
|
s.min_value = mn
|
|
if mx is not None:
|
|
s.max_value = mx
|
|
|
|
add_in("Seed", 'NodeSocketInt', DEFAULTS["Seed"], 0, 9999)
|
|
add_in("Height", 'NodeSocketFloat', DEFAULTS["Height"], 0.5, 30.0)
|
|
add_in("Trunk Radius", 'NodeSocketFloat', DEFAULTS["Trunk Radius"], 0.01, 3.0)
|
|
add_in("Bend", 'NodeSocketFloat', DEFAULTS["Bend"], 0.0, 3.0)
|
|
add_in("Taper", 'NodeSocketFloat', DEFAULTS["Taper"], 0.0, 1.0)
|
|
add_in("Branch Count", 'NodeSocketInt', DEFAULTS["Branch Count"], 0, 60)
|
|
add_in("Branch Length", 'NodeSocketFloat', DEFAULTS["Branch Length"], 0.1, 10.0)
|
|
add_in("Branch Up", 'NodeSocketFloat', DEFAULTS["Branch Up"], -3.0, 3.0)
|
|
add_in("Branch Start", 'NodeSocketFloat', DEFAULTS["Branch Start"], 0.0, 1.0)
|
|
add_in("Branch End", 'NodeSocketFloat', DEFAULTS["Branch End"], 0.0, 1.0)
|
|
add_in("Branch Bend", 'NodeSocketFloat', DEFAULTS["Branch Bend"], 0.0, 2.0)
|
|
# Droop darf NEGATIV sein: dann kruemmen sich die Aeste nach OBEN
|
|
# (= Kaktus-Arme statt Palmwedel).
|
|
add_in("Branch Droop", 'NodeSocketFloat', DEFAULTS["Branch Droop"], -3.0, 3.0)
|
|
add_in("Branch Thickness", 'NodeSocketFloat', DEFAULTS["Branch Thickness"], 0.05, 1.0)
|
|
add_in("Branch Taper", 'NodeSocketFloat', DEFAULTS["Branch Taper"], 0.0, 1.0)
|
|
add_in("Sub Count", 'NodeSocketInt', DEFAULTS["Sub Count"], 0, 12)
|
|
add_in("Sub Length", 'NodeSocketFloat', DEFAULTS["Sub Length"], 0.05, 5.0)
|
|
add_in("Sub Up", 'NodeSocketFloat', DEFAULTS["Sub Up"], -3.0, 3.0)
|
|
add_in("Sides", 'NodeSocketInt', DEFAULTS["Sides"], 3, 32)
|
|
add_in("Tip Blunt", 'NodeSocketFloat', DEFAULTS["Tip Blunt"], 0.0, 0.9)
|
|
add_in("Ribs", 'NodeSocketFloat', DEFAULTS["Ribs"], 0.0, 20.0)
|
|
add_in("Rib Depth", 'NodeSocketFloat', DEFAULTS["Rib Depth"], 0.0, 0.5)
|
|
|
|
gin = N("NodeGroupInput"); gin.location = (-1400, 0)
|
|
gout = N("NodeGroupOutput"); gout.location = (1400, 0)
|
|
V = gin.outputs
|
|
|
|
# ---------- Stamm ----------
|
|
line = N("GeometryNodeCurvePrimitiveLine"); line.location = (-1150, 200)
|
|
top = N("ShaderNodeCombineXYZ"); top.location = (-1300, 120)
|
|
L(V["Height"], top.inputs["Z"])
|
|
L(top.outputs[0], _sock(line, "End"))
|
|
|
|
res = N("GeometryNodeResampleCurve"); res.location = (-950, 200)
|
|
_sock(res, "Count").default_value = 24 # genug Punkte, damit die Kuppel rund wird
|
|
L(_out(line, "Curve"), _sock(res, "Curve"))
|
|
|
|
spar = N("GeometryNodeSplineParameter"); spar.location = (-950, -60)
|
|
pos = N("GeometryNodeInputPosition"); pos.location = (-1150, -220)
|
|
|
|
noise = N("ShaderNodeTexNoise"); noise.location = (-950, -260)
|
|
noise.noise_dimensions = '4D'
|
|
_sock(noise, "Scale").default_value = 0.55
|
|
L(pos.outputs[0], _sock(noise, "Vector"))
|
|
L(V["Seed"], _sock(noise, "W"))
|
|
|
|
nsub = N("ShaderNodeVectorMath"); nsub.location = (-750, -260)
|
|
nsub.operation = 'SUBTRACT'; nsub.inputs[1].default_value = (0.5, 0.5, 0.5)
|
|
L(noise.outputs["Color"], nsub.inputs[0])
|
|
|
|
# Falle 5: Z des Offsets platt machen
|
|
flat = N("ShaderNodeVectorMath"); flat.location = (-580, -260)
|
|
flat.operation = 'MULTIPLY'; flat.inputs[1].default_value = (1.0, 1.0, 0.0)
|
|
L(nsub.outputs[0], flat.inputs[0])
|
|
|
|
# Falle 4: Offset mit Spline-Faktor skalieren -> Fuss bleibt stehen
|
|
bendf = N("ShaderNodeMath"); bendf.location = (-750, -60)
|
|
bendf.operation = 'MULTIPLY'
|
|
L(V["Bend"], bendf.inputs[0])
|
|
L(spar.outputs["Factor"], bendf.inputs[1])
|
|
|
|
offs = N("ShaderNodeVectorMath"); offs.location = (-400, -200)
|
|
offs.operation = 'SCALE'
|
|
L(flat.outputs[0], offs.inputs[0])
|
|
L(bendf.outputs[0], _sock(offs, "Scale"))
|
|
|
|
setpos = N("GeometryNodeSetPosition"); setpos.location = (-250, 200)
|
|
L(_out(res, "Curve"), _sock(setpos, "Geometry"))
|
|
L(offs.outputs[0], _sock(setpos, "Offset"))
|
|
|
|
# Radius: TrunkRadius * (1 - Taper * factor)
|
|
tf = N("ShaderNodeMath"); tf.location = (-580, 60); tf.operation = 'MULTIPLY'
|
|
L(V["Taper"], tf.inputs[0]); L(spar.outputs["Factor"], tf.inputs[1])
|
|
inv = N("ShaderNodeMath"); inv.location = (-420, 60); inv.operation = 'SUBTRACT'
|
|
inv.inputs[0].default_value = 1.0
|
|
L(tf.outputs[0], inv.inputs[1])
|
|
trad0 = N("ShaderNodeMath"); trad0.location = (-260, 60); trad0.operation = 'MULTIPLY'
|
|
L(V["Trunk Radius"], trad0.inputs[0]); L(inv.outputs[0], trad0.inputs[1])
|
|
|
|
# Kuppel-Profil: (1 - f^3)^0.5 -> breite, gleichmaessige Kuppe. Mit einem
|
|
# zu spaeten/steilen Profil (f^8) trifft die Rundung nur 1-2 Resample-
|
|
# Punkte und wird zum KEGEL. Breit + genug Punkte = runde Kuppe.
|
|
# Bei starkem Taper (Baum) dominiert ohnehin die lineare Verjuengung.
|
|
def _dome(spar_node, x, y):
|
|
pw = N("ShaderNodeMath"); pw.location = (x, y); pw.operation = 'POWER'
|
|
pw.inputs[1].default_value = 3.0
|
|
L(spar_node.outputs["Factor"], pw.inputs[0])
|
|
s = N("ShaderNodeMath"); s.location = (x + 150, y); s.operation = 'SUBTRACT'
|
|
s.inputs[0].default_value = 1.0
|
|
L(pw.outputs[0], s.inputs[1])
|
|
rt = N("ShaderNodeMath"); rt.location = (x + 300, y); rt.operation = 'POWER'
|
|
rt.inputs[1].default_value = 0.5
|
|
L(s.outputs[0], rt.inputs[0])
|
|
# Radius NIE auf 0 laufen lassen: sonst entsteht immer eine Spitze,
|
|
# egal wie fein resampled wird. Mit Mindestradius + Fill Caps endet
|
|
# der Arm stumpf/gerundet statt als Dorn.
|
|
mx = N("ShaderNodeMath"); mx.location = (x + 450, y); mx.operation = 'MAXIMUM'
|
|
L(rt.outputs[0], mx.inputs[0])
|
|
L(V["Tip Blunt"], mx.inputs[1])
|
|
return mx
|
|
|
|
tdome = _dome(spar, -900, 220)
|
|
trad = N("ShaderNodeMath"); trad.location = (-100, 60); trad.operation = 'MULTIPLY'
|
|
L(trad0.outputs[0], trad.inputs[0]); L(tdome.outputs[0], trad.inputs[1])
|
|
|
|
setrad = N("GeometryNodeSetCurveRadius"); setrad.location = (-80, 200)
|
|
L(_out(setpos, "Geometry"), _sock(setrad, "Curve"))
|
|
L(trad.outputs[0], _sock(setrad, "Radius"))
|
|
|
|
# ---------- Ast-Ursprünge auf dem oberen Stamm ----------
|
|
trim = N("GeometryNodeTrimCurve"); trim.location = (100, 320)
|
|
L(_out(setrad, "Curve"), _sock(trim, "Curve"))
|
|
# Ansatzbereich der Aeste: Baum = breit gestreut, Palme = alles ganz oben.
|
|
try:
|
|
L(V["Branch Start"], _sock(trim, "Start"))
|
|
L(V["Branch End"], _sock(trim, "End"))
|
|
except KeyError:
|
|
pass
|
|
|
|
c2p = N("GeometryNodeCurveToPoints"); c2p.location = (280, 320)
|
|
try:
|
|
c2p.mode = 'COUNT' # Falle 3
|
|
except Exception:
|
|
pass
|
|
L(_out(trim, "Curve"), _sock(c2p, "Curve"))
|
|
L(V["Branch Count"], _sock(c2p, "Count"))
|
|
|
|
# ---------- Ast-Geometrie (verjüngte Linie) ----------
|
|
bline = N("GeometryNodeCurvePrimitiveLine"); bline.location = (100, 40)
|
|
btop = N("ShaderNodeCombineXYZ"); btop.location = (-60, -20)
|
|
L(V["Branch Length"], btop.inputs["Z"])
|
|
L(btop.outputs[0], _sock(bline, "End"))
|
|
bres = N("GeometryNodeResampleCurve"); bres.location = (280, 40)
|
|
_sock(bres, "Count").default_value = 14 # zu wenig -> Kuppel wird zum Kegel
|
|
L(_out(bline, "Curve"), _sock(bres, "Curve"))
|
|
|
|
# Ast-Verjuengung: 1 - BranchTaper*factor. Bei Kakteen klein halten,
|
|
# sonst laufen die Arme spitz zu wie Dornen.
|
|
bspar = N("GeometryNodeSplineParameter"); bspar.location = (280, -140)
|
|
btap = N("ShaderNodeMath"); btap.location = (360, -240); btap.operation = 'MULTIPLY'
|
|
L(V["Branch Taper"], btap.inputs[0])
|
|
L(bspar.outputs["Factor"], btap.inputs[1])
|
|
binv = N("ShaderNodeMath"); binv.location = (440, -140)
|
|
binv.operation = 'SUBTRACT'; binv.inputs[0].default_value = 1.0
|
|
L(btap.outputs[0], binv.inputs[1])
|
|
bmul = N("ShaderNodeMath"); bmul.location = (600, -140); bmul.operation = 'MULTIPLY'
|
|
L(V["Trunk Radius"], bmul.inputs[0]); L(binv.outputs[0], bmul.inputs[1])
|
|
# Ast-Dicke relativ zum Stamm. Beim Kaktus muessen die Arme fast so dick
|
|
# sein wie der Stamm (0.7+), beim Baum deutlich duenner (~0.3).
|
|
bscl0 = N("ShaderNodeMath"); bscl0.location = (760, -140); bscl0.operation = 'MULTIPLY'
|
|
L(bmul.outputs[0], bscl0.inputs[0])
|
|
L(V["Branch Thickness"], bscl0.inputs[1])
|
|
bdome = _dome(bspar, 600, -600)
|
|
bscl = N("ShaderNodeMath"); bscl.location = (1080, -140); bscl.operation = 'MULTIPLY'
|
|
L(bscl0.outputs[0], bscl.inputs[0]); L(bdome.outputs[0], bscl.inputs[1])
|
|
|
|
bsetr = N("GeometryNodeSetCurveRadius"); bsetr.location = (600, 40)
|
|
L(_out(bres, "Curve"), _sock(bsetr, "Curve"))
|
|
L(bscl.outputs[0], _sock(bsetr, "Radius"))
|
|
|
|
# ---------- Ausrichtung relativ zur ELTERNKURVE ----------
|
|
# Curve to Points liefert Tangent + Normal des Ansatzpunktes gratis mit.
|
|
# Die Normale um die Tangente zu drehen (Index * Goldener Winkel) verteilt
|
|
# die Aeste spiralig um den Elternast (Phyllotaxis) UND folgt automatisch
|
|
# dessen Biegung. Vorher wurde die Richtung aus Weltkoordinaten gerechnet ->
|
|
# Aeste standen wie angeklebte Nadeln quer zum Ast.
|
|
def _child_dir(points_node, up_socket, x, y):
|
|
i = N("GeometryNodeInputIndex"); i.location = (x, y + 220)
|
|
a = N("ShaderNodeMath"); a.location = (x + 150, y + 220); a.operation = 'MULTIPLY'
|
|
a.inputs[1].default_value = 2.399963 # Goldener Winkel
|
|
L(i.outputs[0], a.inputs[0])
|
|
vr = N("ShaderNodeVectorRotate"); vr.location = (x + 320, y + 120)
|
|
vr.rotation_type = 'AXIS_ANGLE'
|
|
L(_out(points_node, "Normal"), vr.inputs["Vector"])
|
|
L(_out(points_node, "Tangent"), vr.inputs["Axis"])
|
|
L(a.outputs[0], vr.inputs["Angle"])
|
|
lean = N("ShaderNodeVectorMath"); lean.location = (x + 320, y - 60)
|
|
lean.operation = 'SCALE'
|
|
L(_out(points_node, "Tangent"), lean.inputs[0])
|
|
L(up_socket, _sock(lean, "Scale"))
|
|
add = N("ShaderNodeVectorMath"); add.location = (x + 500, y + 40)
|
|
add.operation = 'ADD'
|
|
L(vr.outputs[0], add.inputs[0]); L(lean.outputs[0], add.inputs[1])
|
|
return add
|
|
|
|
# ---------- radiale Ausrichtung je Ast ----------
|
|
dirv = _child_dir(c2p, V["Branch Up"], 100, 560)
|
|
|
|
align = _new_any("FunctionNodeAlignRotationToVector", "FunctionNodeAlignEulerToVector")
|
|
if align is not None:
|
|
align.location = (900, 560)
|
|
try:
|
|
align.axis = 'Z'
|
|
except Exception:
|
|
pass
|
|
L(dirv.outputs[0], _sock(align, "Vector"))
|
|
|
|
inst = N("GeometryNodeInstanceOnPoints"); inst.location = (1050, 320)
|
|
L(_out(c2p, "Points"), _sock(inst, "Points"))
|
|
L(_out(bsetr, "Curve"), _sock(inst, "Instance"))
|
|
if align is not None:
|
|
L(align.outputs[0], _sock(inst, "Rotation"))
|
|
|
|
real = N("GeometryNodeRealizeInstances"); real.location = (1200, 320)
|
|
L(_out(inst, "Instances", "Geometry"), _sock(real, "Geometry"))
|
|
|
|
# ---------- Ast-Biegung NACH dem Realize ----------
|
|
# Trick fuer Variation je Ast: hier ist die Position bereits die WELT-Position
|
|
# des jeweiligen Astes. Dieselbe Noise liefert damit pro Ast einen anderen
|
|
# Wert — vor dem Realize haetten alle Instanzen identische lokale Coords
|
|
# (und wuerden exakt gleich gebogen).
|
|
rspar = N("GeometryNodeSplineParameter"); rspar.location = (1200, 60)
|
|
rpos = N("GeometryNodeInputPosition"); rpos.location = (1200, -100)
|
|
rnoise = N("ShaderNodeTexNoise"); rnoise.location = (1350, -140)
|
|
rnoise.noise_dimensions = '4D'
|
|
_sock(rnoise, "Scale").default_value = 0.9
|
|
L(rpos.outputs[0], _sock(rnoise, "Vector"))
|
|
L(V["Seed"], _sock(rnoise, "W"))
|
|
rsub = N("ShaderNodeVectorMath"); rsub.location = (1500, -140)
|
|
rsub.operation = 'SUBTRACT'; rsub.inputs[1].default_value = (0.5, 0.5, 0.5)
|
|
L(rnoise.outputs["Color"], rsub.inputs[0])
|
|
|
|
# Staerke waechst zur Astspitze -> Ansatz bleibt am Stamm
|
|
rfac = N("ShaderNodeMath"); rfac.location = (1350, 60); rfac.operation = 'MULTIPLY'
|
|
L(V["Branch Bend"], rfac.inputs[0])
|
|
L(rspar.outputs["Factor"], rfac.inputs[1])
|
|
rscale = N("ShaderNodeVectorMath"); rscale.location = (1650, -140)
|
|
rscale.operation = 'SCALE'
|
|
L(rsub.outputs[0], rscale.inputs[0])
|
|
L(rfac.outputs[0], _sock(rscale, "Scale"))
|
|
|
|
# Droop: Spitzen haengen nach unten (Palmwedel) -- quadratisch = schoene Kurve
|
|
dsq = N("ShaderNodeMath"); dsq.location = (1350, -320); dsq.operation = 'MULTIPLY'
|
|
L(rspar.outputs["Factor"], dsq.inputs[0]); L(rspar.outputs["Factor"], dsq.inputs[1])
|
|
dmul = N("ShaderNodeMath"); dmul.location = (1500, -320); dmul.operation = 'MULTIPLY'
|
|
L(V["Branch Droop"], dmul.inputs[0]); L(dsq.outputs[0], dmul.inputs[1])
|
|
dneg = N("ShaderNodeMath"); dneg.location = (1650, -320); dneg.operation = 'MULTIPLY'
|
|
dneg.inputs[1].default_value = -1.0
|
|
L(dmul.outputs[0], dneg.inputs[0])
|
|
dvec = N("ShaderNodeCombineXYZ"); dvec.location = (1800, -320)
|
|
L(dneg.outputs[0], dvec.inputs["Z"])
|
|
|
|
roffs = N("ShaderNodeVectorMath"); roffs.location = (1800, -140)
|
|
roffs.operation = 'ADD'
|
|
L(rscale.outputs[0], roffs.inputs[0])
|
|
L(dvec.outputs[0], roffs.inputs[1])
|
|
|
|
rsetp = N("GeometryNodeSetPosition"); rsetp.location = (1950, 320)
|
|
L(_out(real, "Geometry"), _sock(rsetp, "Geometry"))
|
|
L(roffs.outputs[0], _sock(rsetp, "Offset"))
|
|
|
|
# ---------- Sub-Aeste (zweite Verzweigungsebene) ----------
|
|
# Sitzen auf den bereits gebogenen Hauptaesten. Bei "Sub Count" = 0 liefert
|
|
# Curve to Points keine Punkte -> es entsteht schlicht nichts (kein Fehler).
|
|
strim = N("GeometryNodeTrimCurve"); strim.location = (2100, 520)
|
|
L(_out(rsetp, "Geometry"), _sock(strim, "Curve"))
|
|
try:
|
|
_sock(strim, "Start").default_value = 0.3
|
|
_sock(strim, "End").default_value = 0.9
|
|
except KeyError:
|
|
pass
|
|
sp = N("GeometryNodeCurveToPoints"); sp.location = (2250, 520)
|
|
try:
|
|
sp.mode = 'COUNT'
|
|
except Exception:
|
|
pass
|
|
L(_out(strim, "Curve"), _sock(sp, "Curve"))
|
|
L(V["Sub Count"], _sock(sp, "Count"))
|
|
|
|
sline = N("GeometryNodeCurvePrimitiveLine"); sline.location = (2100, 760)
|
|
stop = N("ShaderNodeCombineXYZ"); stop.location = (1950, 700)
|
|
L(V["Sub Length"], stop.inputs["Z"])
|
|
L(stop.outputs[0], _sock(sline, "End"))
|
|
sres = N("GeometryNodeResampleCurve"); sres.location = (2250, 760)
|
|
_sock(sres, "Count").default_value = 6
|
|
L(_out(sline, "Curve"), _sock(sres, "Curve"))
|
|
|
|
sspar = N("GeometryNodeSplineParameter"); sspar.location = (2250, 900)
|
|
sinv = N("ShaderNodeMath"); sinv.location = (2400, 900)
|
|
sinv.operation = 'SUBTRACT'; sinv.inputs[0].default_value = 1.0
|
|
L(sspar.outputs["Factor"], sinv.inputs[1])
|
|
srad = N("ShaderNodeMath"); srad.location = (2550, 900); srad.operation = 'MULTIPLY'
|
|
L(V["Trunk Radius"], srad.inputs[0]); L(sinv.outputs[0], srad.inputs[1])
|
|
srad2 = N("ShaderNodeMath"); srad2.location = (2700, 900); srad2.operation = 'MULTIPLY'
|
|
srad2.inputs[1].default_value = 0.14 # deutlich duenner als Hauptaeste
|
|
L(srad.outputs[0], srad2.inputs[0])
|
|
ssetr = N("GeometryNodeSetCurveRadius"); ssetr.location = (2550, 760)
|
|
L(_out(sres, "Curve"), _sock(ssetr, "Curve"))
|
|
L(srad2.outputs[0], _sock(ssetr, "Radius"))
|
|
|
|
# Richtung: radial gestreut (Index) + "Sub Up"
|
|
sdir = _child_dir(sp, V["Sub Up"], 2100, 1060)
|
|
|
|
salign = _new_any("FunctionNodeAlignRotationToVector", "FunctionNodeAlignEulerToVector")
|
|
if salign is not None:
|
|
salign.location = (2700, 1060)
|
|
try:
|
|
salign.axis = 'Z'
|
|
except Exception:
|
|
pass
|
|
L(sdir.outputs[0], _sock(salign, "Vector"))
|
|
|
|
sinst = N("GeometryNodeInstanceOnPoints"); sinst.location = (2850, 520)
|
|
L(_out(sp, "Points"), _sock(sinst, "Points"))
|
|
L(_out(ssetr, "Curve"), _sock(sinst, "Instance"))
|
|
if salign is not None:
|
|
L(salign.outputs[0], _sock(sinst, "Rotation"))
|
|
sreal = N("GeometryNodeRealizeInstances"); sreal.location = (3000, 520)
|
|
L(_out(sinst, "Instances", "Geometry"), _sock(sreal, "Geometry"))
|
|
|
|
join = N("GeometryNodeJoinGeometry"); join.location = (3150, 200)
|
|
L(_out(sreal, "Geometry"), join.inputs[0])
|
|
L(_out(rsetp, "Geometry"), join.inputs[0])
|
|
L(_out(setrad, "Curve"), join.inputs[0])
|
|
|
|
# ---------- Curve -> Mesh ----------
|
|
# Falle 6: Der Profil-Radius wird mit dem Curve-Radius MULTIPLIZIERT.
|
|
# Profil deshalb auf 1.0 lassen waere richtig -- aber nur, wenn der
|
|
# Curve-Radius bereits die echte Staerke ist. Hier ist er das, also 1.0.
|
|
circ = N("GeometryNodeCurvePrimitiveCircle"); circ.location = (1350, -100)
|
|
L(V["Sides"], _sock(circ, "Resolution"))
|
|
_sock(circ, "Radius").default_value = 1.0
|
|
|
|
# ---------- Rippen (senkrechte Kanneluren, Kaktus-Signatur) ----------
|
|
# Das Profil bekommt eine radiale Welle: Offset entlang der Punktrichtung,
|
|
# moduliert mit cos(Rippen * Winkel). Rib Depth = 0 -> glatter Kreis.
|
|
pspar = N("GeometryNodeSplineParameter"); pspar.location = (1350, -420)
|
|
pang = N("ShaderNodeMath"); pang.location = (1500, -420); pang.operation = 'MULTIPLY'
|
|
pang.inputs[1].default_value = 6.283185
|
|
L(pspar.outputs["Factor"], pang.inputs[0])
|
|
pribs = N("ShaderNodeMath"); pribs.location = (1650, -420); pribs.operation = 'MULTIPLY'
|
|
L(pang.outputs[0], pribs.inputs[0])
|
|
L(V["Ribs"], pribs.inputs[1])
|
|
pcos = N("ShaderNodeMath"); pcos.location = (1800, -420); pcos.operation = 'COSINE'
|
|
L(pribs.outputs[0], pcos.inputs[0])
|
|
pamp = N("ShaderNodeMath"); pamp.location = (1950, -420); pamp.operation = 'MULTIPLY'
|
|
L(pcos.outputs[0], pamp.inputs[0])
|
|
L(V["Rib Depth"], pamp.inputs[1])
|
|
ppos = N("GeometryNodeInputPosition"); ppos.location = (1650, -560)
|
|
pdir = N("ShaderNodeVectorMath"); pdir.location = (1800, -560)
|
|
pdir.operation = 'NORMALIZE'
|
|
L(ppos.outputs[0], pdir.inputs[0])
|
|
poff = N("ShaderNodeVectorMath"); poff.location = (2100, -520)
|
|
poff.operation = 'SCALE'
|
|
L(pdir.outputs[0], poff.inputs[0])
|
|
L(pamp.outputs[0], _sock(poff, "Scale"))
|
|
pset = N("GeometryNodeSetPosition"); pset.location = (2250, -100)
|
|
L(_out(circ, "Curve"), _sock(pset, "Geometry"))
|
|
L(poff.outputs[0], _sock(pset, "Offset"))
|
|
|
|
c2m = N("GeometryNodeCurveToMesh"); c2m.location = (1500, 200)
|
|
L(_out(join, "Geometry"), _sock(c2m, "Curve"))
|
|
L(_out(pset, "Geometry"), _sock(c2m, "Profile Curve"))
|
|
# Falle 7 (Blender 5.x!): "Curve to Mesh" wertet das Radius-Attribut NICHT
|
|
# mehr implizit aus, sondern hat einen eigenen "Scale"-Eingang. Ohne diese
|
|
# Verbindung bleibt der Stamm immer bei Profil-Radius 1.0 (= 2 m dick),
|
|
# egal was "Trunk Radius" sagt. Gemessen: Stammbreite konstant 1.995.
|
|
if "Scale" in c2m.inputs:
|
|
radattr = N("GeometryNodeInputRadius"); radattr.location = (1350, -260)
|
|
L(radattr.outputs[0], _sock(c2m, "Scale"))
|
|
try:
|
|
_sock(c2m, "Fill Caps").default_value = True
|
|
except KeyError:
|
|
pass
|
|
|
|
shade = N("GeometryNodeSetShadeSmooth"); shade.location = (1650, 200)
|
|
L(_out(c2m, "Mesh", "Geometry"), _sock(shade, "Geometry"))
|
|
L(_out(shade, "Geometry"), gout.inputs[0])
|
|
return ng
|
|
|
|
|
|
# Wachstums-Stufen: Faktoren/Werte fuer t = 0 (Setzling) .. 1 (ausgewachsen).
|
|
# Gleicher Seed -> dieselbe Baum-"Identitaet", nur juenger. Genau das braucht
|
|
# ein Wachstums-System, damit Stufe 1 und Stufe 4 wie DERSELBE Baum wirken.
|
|
GROWTH_YOUNG = {
|
|
"Height": 0.16, # Faktoren (werden mit dem Zielwert multipliziert)
|
|
"Trunk Radius": 0.30,
|
|
"Branch Length": 0.30,
|
|
"Branch Count": 0.35,
|
|
"Sub Count": 0.0, # Setzling hat noch keine Sub-Aeste
|
|
}
|
|
|
|
|
|
def growth_values(preset, t):
|
|
"""Parameter fuer Wachstums-Fortschritt t (0 = Setzling, 1 = ausgewachsen)."""
|
|
full = dict(DEFAULTS)
|
|
full.update(PRESETS.get(preset, {}))
|
|
out = dict(full)
|
|
t = max(0.0, min(1.0, t))
|
|
for key, young_factor in GROWTH_YOUNG.items():
|
|
if key not in full:
|
|
continue
|
|
target = full[key]
|
|
young = target * young_factor
|
|
v = young + (target - young) * t
|
|
out[key] = max(1, int(round(v))) if isinstance(target, int) and key != "Sub Count" \
|
|
else (int(round(v)) if isinstance(target, int) else v)
|
|
return out
|
|
|
|
|
|
def socket_ids(ng):
|
|
"""Name -> Modifier-Key ('Socket_3'). Fuer Presets/Skripting."""
|
|
out = {}
|
|
for s in ng.interface.items_tree:
|
|
if getattr(s, "in_out", "") == 'INPUT':
|
|
out[s.name] = s.identifier
|
|
return out
|
|
|
|
|
|
def make_object(ng, preset="baum", name=None, growth=None, seed=None):
|
|
name = name or OBJ_NAME
|
|
old = bpy.data.objects.get(name)
|
|
if old:
|
|
bpy.data.objects.remove(old, do_unlink=True)
|
|
me = bpy.data.meshes.new(name + "Mesh")
|
|
ob = bpy.data.objects.new(name, me)
|
|
bpy.context.scene.collection.objects.link(ob)
|
|
md = ob.modifiers.new("GN_Tree", 'NODES')
|
|
md.node_group = ng
|
|
|
|
ids = socket_ids(ng)
|
|
if growth is None:
|
|
values = dict(DEFAULTS)
|
|
values.update(PRESETS.get(preset, {}))
|
|
else:
|
|
values = growth_values(preset, growth)
|
|
if seed is not None:
|
|
values["Seed"] = seed
|
|
for k, v in values.items():
|
|
if k in ids:
|
|
md[ids[k]] = v
|
|
|
|
bpy.context.view_layer.objects.active = ob
|
|
return ob
|
|
|
|
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# Addon-UI
|
|
# ---------------------------------------------------------------------------
|
|
|
|
from bpy.props import (
|
|
IntProperty, FloatProperty, EnumProperty, BoolProperty, PointerProperty,
|
|
)
|
|
from bpy.types import Operator, Panel, PropertyGroup
|
|
|
|
|
|
def _preset_enum(self, context):
|
|
return [(k, k.capitalize(), "") for k in PRESETS]
|
|
|
|
|
|
class TreeGenSettings(PropertyGroup):
|
|
preset: EnumProperty(name="Preset", items=_preset_enum)
|
|
seed: IntProperty(name="Seed", default=0, min=0, max=9999)
|
|
use_growth: BoolProperty(
|
|
name="Wachstums-Stufen", default=False,
|
|
description="Erzeugt mehrere Stufen (Setzling .. ausgewachsen) mit gleichem Seed",
|
|
)
|
|
stages: IntProperty(name="Anzahl Stufen", default=4, min=2, max=8)
|
|
spacing: FloatProperty(name="Abstand", default=4.0, min=0.0, max=20.0)
|
|
apply_modifier: BoolProperty(
|
|
name="Modifier anwenden", default=False,
|
|
description="Ergebnis als normales Mesh einfrieren (fuer den Export)",
|
|
)
|
|
|
|
|
|
def _place(ob, x):
|
|
ob.location.x = x
|
|
|
|
|
|
class TREEGEN_OT_create(Operator):
|
|
bl_idname = "object.treegen_create"
|
|
bl_label = "Baum erzeugen"
|
|
bl_description = "Erzeugt einen parametrischen Baum (Regler danach am Modifier)"
|
|
bl_options = {'REGISTER', 'UNDO'}
|
|
|
|
def execute(self, context):
|
|
s = context.scene.tree_gen_settings
|
|
try:
|
|
ng = build_group()
|
|
except Exception as exc: # noqa: BLE001
|
|
self.report({'ERROR'}, "Node-Gruppe fehlgeschlagen: %s" % exc)
|
|
return {'CANCELLED'}
|
|
try:
|
|
ng.asset_mark()
|
|
ng.asset_data.description = "Parametrischer Stylized-Baum (Stamm + Aeste)"
|
|
except Exception:
|
|
pass
|
|
|
|
created = []
|
|
x = 0.0
|
|
try:
|
|
if s.use_growth:
|
|
for i in range(s.stages):
|
|
t = i / float(s.stages - 1) if s.stages > 1 else 1.0
|
|
ob = make_object(ng, preset=s.preset, growth=t, seed=s.seed,
|
|
name="%s_%s_stage%d" % (OBJ_NAME, s.preset, i))
|
|
_place(ob, x); x += s.spacing
|
|
created.append(ob)
|
|
else:
|
|
ob = make_object(ng, preset=s.preset, seed=s.seed,
|
|
name="%s_%s" % (OBJ_NAME, s.preset))
|
|
created.append(ob)
|
|
except Exception as exc: # noqa: BLE001
|
|
self.report({'ERROR'}, "Erzeugen fehlgeschlagen: %s" % exc)
|
|
return {'CANCELLED'}
|
|
|
|
if s.apply_modifier:
|
|
for ob in created:
|
|
context.view_layer.objects.active = ob
|
|
try:
|
|
bpy.ops.object.modifier_apply(modifier="GN_Tree")
|
|
except RuntimeError as exc:
|
|
self.report({'WARNING'}, "Modifier-Apply: %s" % exc)
|
|
|
|
context.view_layer.update()
|
|
total = 0
|
|
dg = context.evaluated_depsgraph_get()
|
|
for ob in created:
|
|
try:
|
|
total += len(ob.evaluated_get(dg).data.vertices)
|
|
except Exception:
|
|
pass
|
|
self.report({'INFO'}, "%d Objekt(e) erzeugt, %d Verts gesamt." % (len(created), total))
|
|
return {'FINISHED'}
|
|
|
|
|
|
class VIEW3D_PT_tree_generator(Panel):
|
|
bl_label = "Tree Generator"
|
|
bl_idname = "VIEW3D_PT_tree_generator"
|
|
bl_space_type = 'VIEW_3D'
|
|
bl_region_type = 'UI'
|
|
bl_category = "Tree Gen"
|
|
|
|
def draw(self, context):
|
|
layout = self.layout
|
|
s = context.scene.tree_gen_settings
|
|
|
|
box = layout.box()
|
|
box.label(text="Vorlage", icon='PRESET')
|
|
box.prop(s, "preset", text="")
|
|
box.prop(s, "seed")
|
|
|
|
box = layout.box()
|
|
box.label(text="Wachstum")
|
|
box.prop(s, "use_growth")
|
|
sub = box.column(align=True)
|
|
sub.enabled = s.use_growth
|
|
sub.prop(s, "stages")
|
|
sub.prop(s, "spacing")
|
|
|
|
box = layout.box()
|
|
box.label(text="Export")
|
|
box.prop(s, "apply_modifier")
|
|
|
|
layout.separator()
|
|
layout.operator("object.treegen_create", icon='OUTLINER_OB_MESH')
|
|
layout.label(text="Feinjustierung: am Modifier 'GN_Tree'", icon='INFO')
|
|
|
|
|
|
classes = (
|
|
TreeGenSettings,
|
|
TREEGEN_OT_create,
|
|
VIEW3D_PT_tree_generator,
|
|
)
|
|
|
|
|
|
def register():
|
|
for cls in classes:
|
|
bpy.utils.register_class(cls)
|
|
bpy.types.Scene.tree_gen_settings = PointerProperty(type=TreeGenSettings)
|
|
|
|
|
|
def unregister():
|
|
del bpy.types.Scene.tree_gen_settings
|
|
for cls in reversed(classes):
|
|
bpy.utils.unregister_class(cls)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
register()
|