100 lines
3.6 KiB
Python
100 lines
3.6 KiB
Python
|
# Stage 4: review renders + numeric profile gates for the hires sculpt.
|
||
|
|
# blender --background --python 04_review.py -- <03_sculpted.blend> <review_dir>
|
||
|
|
import bpy, sys, os, math, time
|
||
|
|
import numpy as np
|
||
|
|
from mathutils import Vector
|
||
|
|
|
||
|
|
argv = sys.argv[sys.argv.index("--") + 1:]
|
||
|
|
BLEND, OUT = argv[0], argv[1]
|
||
|
|
os.makedirs(OUT, exist_ok=True)
|
||
|
|
t0 = time.time()
|
||
|
|
|
||
|
|
|
||
|
|
def log(m):
|
||
|
|
print(f"[review {time.time()-t0:6.1f}s] {m}", flush=True)
|
||
|
|
|
||
|
|
|
||
|
|
bpy.ops.wm.open_mainfile(filepath=BLEND)
|
||
|
|
ob = max([o for o in bpy.data.objects if o.type == 'MESH'],
|
||
|
|
key=lambda o: len(o.data.vertices))
|
||
|
|
me = ob.data
|
||
|
|
n_v = len(me.vertices)
|
||
|
|
|
||
|
|
co = np.empty(n_v * 3, dtype=np.float64)
|
||
|
|
me.vertices.foreach_get("co", co)
|
||
|
|
co = co.reshape(-1, 3)
|
||
|
|
|
||
|
|
# ---------- numeric gates ----------
|
||
|
|
front = co[:, 1] < 0
|
||
|
|
print("\n=== PROFILE y(z) at sternum x=0 (front) ===")
|
||
|
|
for z0 in np.arange(0.58, 0.78, 0.01):
|
||
|
|
m = front & (np.abs(co[:, 0]) < 0.004) & (np.abs(co[:, 2] - z0) < 0.005)
|
||
|
|
if m.sum():
|
||
|
|
print(f" z={z0:.2f} y={co[m,1].min():+.4f}")
|
||
|
|
print("=== PROFILE y(z) at apex x=0.034 ===")
|
||
|
|
for z0 in np.arange(0.58, 0.78, 0.01):
|
||
|
|
m = front & (np.abs(co[:, 0] - 0.034) < 0.005) & (np.abs(co[:, 2] - z0) < 0.005)
|
||
|
|
if m.sum():
|
||
|
|
print(f" z={z0:.2f} y={co[m,1].min():+.4f}")
|
||
|
|
print("=== PROFILE y(x) at z=0.688 ===")
|
||
|
|
for x0 in np.arange(-0.10, 0.101, 0.01):
|
||
|
|
m = front & (np.abs(co[:, 0] - x0) < 0.005) & (np.abs(co[:, 2] - 0.688) < 0.006)
|
||
|
|
if m.sum():
|
||
|
|
print(f" x={x0:+.2f} y={co[m,1].min():+.4f}")
|
||
|
|
|
||
|
|
# ---------- renders ----------
|
||
|
|
scn = bpy.context.scene
|
||
|
|
w = bpy.data.worlds.new("W")
|
||
|
|
w.color = (0.22, 0.22, 0.24)
|
||
|
|
scn.world = w
|
||
|
|
key = bpy.data.objects.new("Key", bpy.data.lights.new("Key", 'SUN'))
|
||
|
|
key.data.energy = 3.0
|
||
|
|
key.data.use_shadow = False
|
||
|
|
bpy.context.collection.objects.link(key)
|
||
|
|
fill = bpy.data.objects.new("Fill", bpy.data.lights.new("Fill", 'SUN'))
|
||
|
|
fill.data.energy = 1.0
|
||
|
|
fill.data.use_shadow = False
|
||
|
|
bpy.context.collection.objects.link(fill)
|
||
|
|
cam = bpy.data.objects.new("Cam", bpy.data.cameras.new("Cam"))
|
||
|
|
cam.data.lens = 85
|
||
|
|
bpy.context.collection.objects.link(cam)
|
||
|
|
scn.camera = cam
|
||
|
|
scn.render.engine = 'BLENDER_EEVEE' if bpy.app.version >= (4, 2) else 'BLENDER_EEVEE_NEXT'
|
||
|
|
scn.render.resolution_x = scn.render.resolution_y = 1000
|
||
|
|
|
||
|
|
clay_mat = bpy.data.materials.new("Clay")
|
||
|
|
clay_mat.use_nodes = True
|
||
|
|
clay_mat.node_tree.nodes["Principled BSDF"].inputs["Base Color"].default_value = (0.62, 0.60, 0.58, 1)
|
||
|
|
clay_mat.node_tree.nodes["Principled BSDF"].inputs["Roughness"].default_value = 0.45
|
||
|
|
orig_mats = [ms.material for ms in ob.material_slots]
|
||
|
|
|
||
|
|
|
||
|
|
def shoot(tag, ctr, span, yaw_deg, clay):
|
||
|
|
if clay:
|
||
|
|
for ms in ob.material_slots:
|
||
|
|
ms.material = clay_mat
|
||
|
|
else:
|
||
|
|
for ms, m in zip(ob.material_slots, orig_mats):
|
||
|
|
ms.material = m
|
||
|
|
yaw = math.radians(yaw_deg)
|
||
|
|
dist = span * 3.0
|
||
|
|
cam.location = Vector(ctr) + Vector((math.sin(yaw) * dist, -math.cos(yaw) * dist, 0.02))
|
||
|
|
cam.rotation_euler = (Vector(ctr) - cam.location).to_track_quat('-Z', 'Y').to_euler()
|
||
|
|
key.rotation_euler = (math.radians(62), 0, math.radians(35 + yaw_deg))
|
||
|
|
fill.rotation_euler = (math.radians(75), 0, math.radians(yaw_deg - 110))
|
||
|
|
scn.render.filepath = os.path.join(OUT, f"{tag}.png")
|
||
|
|
bpy.ops.render.render(write_still=True)
|
||
|
|
log(f"render {tag}")
|
||
|
|
|
||
|
|
|
||
|
|
CHEST = (0.0, 0.0, 0.675)
|
||
|
|
HIP = (0.0, 0.0, 0.53)
|
||
|
|
FULL = (0.0, 0.0, 0.50)
|
||
|
|
for yaw in (0, 40, 90):
|
||
|
|
shoot(f"chest_clay_{yaw}", CHEST, 0.22, yaw, True)
|
||
|
|
shoot(f"chest_tex_{yaw}", CHEST, 0.22, yaw, False)
|
||
|
|
shoot("hip_clay_0", HIP, 0.22, 0, True)
|
||
|
|
shoot("full_clay_0", FULL, 0.55, 0, True)
|
||
|
|
shoot("full_clay_40", FULL, 0.55, 40, True)
|
||
|
|
print("REVIEW_DONE")
|