vikashmakeit commited on
Commit
dd071f2
·
verified ·
1 Parent(s): e25566e

3D view now built FROM actual 2D pattern pieces (triangulate → cylindrical wrap → Mesh3d)

Browse files
Files changed (1) hide show
  1. garment_3d.py +364 -529
garment_3d.py CHANGED
@@ -1,574 +1,409 @@
1
  """
2
- 3D Garment Visualizer — Anatomically Correct Mannequin + Garment Surfaces
3
 
4
- Creates realistic 3D garment previews on a shaped parametric body using Plotly.
5
- Uses two core primitives:
6
- 1. Revolution surfaces (body, torso, skirt, collar, waistband)
7
- 2. Tilted tubes (sleeves)
8
 
9
- All Z values are in cm, Z=0 at floor, Z=170 at head top.
10
- Garment radii are derived from circumference measurements:
11
- radius = circumference / (2*pi) (for circular cross-section)
12
- For elliptical: rx = circ / (2*pi) * 1.1, ry = circ / (2*pi) * 0.9
13
  """
14
  import numpy as np
15
  import plotly.graph_objects as go
16
- from typing import Dict, List, Tuple
17
 
18
- TWO_PI = 2 * np.pi
19
-
20
- # ── Anatomical body profile (female mannequin, 170 cm) ─────────────────────
21
- # 17 landmarks from feet to head (all radii in cm)
22
- _BODY_Z = [0, 7, 30, 45, 65, 88, 100, 104, 112, 120, 132, 140, 145, 150, 158, 162, 170]
23
- _BODY_RX = [4.0, 3.0, 4.5, 4.0, 6.5, 9.5, 8.0, 7.0, 8.5, 10.0, 9.0, 8.5, 11.5, 3.5, 3.0, 5.5, 4.5]
24
- _BODY_RY = [3.0, 2.5, 4.0, 4.0, 6.0, 8.5, 7.5, 6.5, 8.0, 9.0, 8.0, 7.5, 8.0, 3.5, 3.0, 5.0, 4.5]
25
-
26
- # Key anatomical Z heights (cm from floor)
27
- Z_FLOOR = 0
28
  Z_ANKLE = 7
29
  Z_KNEE = 45
 
30
  Z_HIP = 88
31
  Z_WAIST = 104
32
  Z_BUST = 120
33
  Z_SHOULDER = 145
34
- Z_NECK_BASE = 150
35
- Z_HEAD_TOP = 170
36
-
37
-
38
- # ── Helpers: circumference elliptical radii ──────────────────────────────
39
- def _circ_to_rx(circ):
40
- """Convert circumference to X-radius (wider front-back)."""
41
- return circ / TWO_PI * 1.10
42
-
43
- def _circ_to_ry(circ):
44
- """Convert circumference to Y-radius (narrower side-side)."""
45
- return circ / TWO_PI * 0.90
46
-
47
-
48
- # ── Primitive 1: Revolution surface ─────────────────────────────────────────
49
- def _revolution_surface(z_heights, rx_vals, ry_vals,
50
- n_theta=36, cx=0.0, cy=0.0):
51
- """Surface of revolution with elliptical cross-sections."""
52
- theta = np.linspace(0, 2 * np.pi, n_theta)
53
- z_arr = np.asarray(z_heights, float)
54
- rx = np.asarray(rx_vals, float)
55
- ry = np.asarray(ry_vals, float)
56
- Z = np.outer(z_arr, np.ones(n_theta))
57
- X = cx + np.outer(rx, np.cos(theta))
58
- Y = cy + np.outer(ry, np.sin(theta))
59
- return X, Y, Z
60
-
61
-
62
- # ── Primitive 2: Tilted tube (for sleeves) ──────────────────────────────────
63
- def _tilted_tube(start, end, r_start, r_end, n_theta=24, n_z=12):
64
- """Tapered tube between two 3D points."""
65
- s = np.array(start, float)
66
- e = np.array(end, float)
67
- axis = e - s
68
- length = np.linalg.norm(axis)
69
- if length < 0.01:
70
- return np.zeros((n_z, n_theta)), np.zeros((n_z, n_theta)), np.zeros((n_z, n_theta))
71
-
72
- axis_n = axis / length
73
- ref = np.array([0., 0., 1.]) if abs(axis_n[2]) < 0.9 else np.array([1., 0., 0.])
74
- right = np.cross(axis_n, ref)
75
- right /= np.linalg.norm(right)
76
- up = np.cross(right, axis_n)
77
-
78
- theta = np.linspace(0, 2 * np.pi, n_theta)
79
- t_vals = np.linspace(0, 1, n_z)
80
-
81
- X = np.zeros((n_z, n_theta))
82
- Y = np.zeros((n_z, n_theta))
83
- Z = np.zeros((n_z, n_theta))
84
-
85
- for i, t in enumerate(t_vals):
86
- center = s + t * axis
87
- r = r_start + t * (r_end - r_start)
88
- for j, th in enumerate(theta):
89
- offset = r * (np.cos(th) * right + np.sin(th) * up)
90
- X[i, j] = center[0] + offset[0]
91
- Y[i, j] = center[1] + offset[1]
92
- Z[i, j] = center[2] + offset[2]
93
-
94
- return X, Y, Z
95
-
96
-
97
- # ── Surface helper ──────────────────────────────────────────────────────────
98
- def _make_surface(X, Y, Z, color, name, opacity=0.88, lighting=None):
99
- """Create a Plotly Surface trace with fabric-like shading."""
100
- if lighting is None:
101
- lighting = dict(ambient=0.65, diffuse=0.85, specular=0.15, roughness=0.8)
102
- return go.Surface(
103
- x=X, y=Y, z=Z,
104
- surfacecolor=np.ones_like(X),
105
- colorscale=[[0, color], [1, color]],
106
- opacity=opacity,
107
- showscale=False,
108
- name=name,
109
- hoverinfo="name",
110
- lighting=lighting,
111
- lightposition=dict(x=100, y=200, z=300),
112
- )
113
 
114
 
115
- # ── Body mannequin ──────────────────────────────────────────────────────────
116
- def _make_body():
117
- """Create the translucent mannequin body."""
118
- X, Y, Z = _revolution_surface(_BODY_Z, _BODY_RX, _BODY_RY, n_theta=48)
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
119
  return go.Surface(
120
  x=X, y=Y, z=Z,
121
  surfacecolor=np.ones_like(X) * 0.8,
122
  colorscale=[[0, "#E8D0B0"], [1, "#E8D0B0"]],
123
- opacity=0.20,
124
- showscale=False,
125
- name="Body",
126
  hoverinfo="skip",
127
  lighting=dict(ambient=0.9, diffuse=0.1, specular=0, roughness=1.0),
128
  )
129
 
130
 
131
- # ── Helper: interpolate body radius at arbitrary Z ──────────────────────────
132
- def _body_rx(z_val):
133
- return float(np.interp(z_val, _BODY_Z, _BODY_RX))
134
-
135
- def _body_ry(z_val):
136
- return float(np.interp(z_val, _BODY_Z, _BODY_RY))
137
-
138
-
139
- # ── Garment builders ────────────────────────────────────────────────────────
140
-
141
- def _build_shirt(params, color="#4A90D9"):
142
- """Build shirt/blouse/top/t-shirt surfaces."""
143
- bust = params.get("bust", 92)
144
- waist = params.get("waist", 74)
145
- shoulder = params.get("shoulder_width", 42)
146
- bodice_len = params.get("bodice_length", 72)
147
- sleeve_len = params.get("sleeve_length", 60)
148
- bicep = params.get("bicep", 30)
149
- wrist_circ = params.get("wrist", 18)
150
- fit = params.get("fit", "regular")
151
- flare = params.get("flare", 0)
152
-
153
- ease = {"fitted": 0.5, "regular": 1.2, "oversized": 3.0, "loose": 2.2}.get(fit, 1.2)
154
-
155
- bust_rx = _circ_to_rx(bust) + ease
156
- bust_ry = _circ_to_ry(bust) + ease
157
- waist_rx = _circ_to_rx(waist) + ease
158
- waist_ry = _circ_to_ry(waist) + ease
159
- shoulder_rx = shoulder / 2
160
-
161
- surfaces = []
162
-
163
- extra_length = max(0, bodice_len - 42) * 0.5
164
- hem_z = max(Z_WAIST - extra_length, Z_HIP - 5)
165
-
166
- n_torso = 16
167
- torso_z = np.linspace(hem_z, Z_SHOULDER, n_torso)
168
- torso_rx_arr = []
169
- torso_ry_arr = []
170
-
171
- for z in torso_z:
172
- t = (z - hem_z) / (Z_SHOULDER - hem_z)
173
-
174
- if t < 0.25:
175
- rx = waist_rx + flare * (0.25 - t) / 0.25
176
- ry = waist_ry + flare * 0.8 * (0.25 - t) / 0.25
177
- elif t < 0.55:
178
- f = (t - 0.25) / 0.30
179
- rx = waist_rx + (bust_rx - waist_rx) * f
180
- ry = waist_ry + (bust_ry - waist_ry) * f
181
- elif t < 0.8:
182
- f = (t - 0.55) / 0.25
183
- rx = bust_rx + (bust_rx * 0.92 - bust_rx) * f
184
- ry = bust_ry + (bust_ry * 0.92 - bust_ry) * f
185
- else:
186
- f = (t - 0.8) / 0.2
187
- rx = bust_rx * 0.92 + (shoulder_rx - bust_rx * 0.92) * f
188
- ry = bust_ry * 0.92 * (1 - 0.15 * f)
189
-
190
- rx = max(rx, _body_rx(z) + 0.8)
191
- ry = max(ry, _body_ry(z) + 0.8)
192
- torso_rx_arr.append(rx)
193
- torso_ry_arr.append(ry)
194
-
195
- X, Y, Zt = _revolution_surface(torso_z, torso_rx_arr, torso_ry_arr, n_theta=36)
196
- surfaces.append(_make_surface(X, Y, Zt, color, "Torso", opacity=0.85))
197
-
198
- shoulder_attach_rx = torso_rx_arr[-1]
199
- shoulder_attach_z = Z_SHOULDER - 3
200
-
201
- sleeve_drop = sleeve_len * 0.45
202
- sleeve_spread = sleeve_len * 0.30
203
-
204
- r_bicep = bicep / TWO_PI + ease * 0.4
205
- r_wrist = wrist_circ / TWO_PI + ease * 0.3
206
-
207
- for side in [1, -1]:
208
- start = (side * shoulder_attach_rx, 0, shoulder_attach_z)
209
- end = (side * (shoulder_attach_rx + sleeve_spread), 0,
210
- shoulder_attach_z - sleeve_drop)
211
- Xs, Ys, Zs = _tilted_tube(start, end, r_bicep, r_wrist, n_theta=20, n_z=14)
212
- sleeve_color = _lighten_color(color, 0.08) if side > 0 else _darken_color(color, 0.08)
213
- surfaces.append(_make_surface(Xs, Ys, Zs, sleeve_color, "Sleeve", opacity=0.80))
214
-
215
- if params.get("has_collar", False):
216
- collar_h = params.get("collar_height", 4)
217
- n_c = 6
218
- collar_z = np.linspace(Z_NECK_BASE - 2, Z_NECK_BASE + collar_h, n_c)
219
- collar_rx = []
220
- collar_ry = []
221
- for z in collar_z:
222
- collar_rx.append(_body_rx(z) + 1.5)
223
- collar_ry.append(_body_ry(z) + 1.5)
224
- Xc, Yc, Zc = _revolution_surface(collar_z, collar_rx, collar_ry, n_theta=28)
225
- surfaces.append(_make_surface(Xc, Yc, Zc, "#FFFFFF", "Collar", opacity=0.92))
226
-
227
- return surfaces
228
-
229
-
230
- def _build_pants(params, color="#2C5F8A"):
231
- """Build pants/trousers/jeans surfaces."""
232
- waist = params.get("waist", 74)
233
- hip = params.get("hip", 96)
234
- thigh = params.get("thigh", 56)
235
- ankle = params.get("ankle", 24)
236
- pant_len = params.get("pant_length", 100)
237
- fit = params.get("fit", "regular")
238
- flare = params.get("flare", 0)
239
-
240
- ease = {"fitted": 0.5, "regular": 1.0, "oversized": 2.5, "loose": 2.0}.get(fit, 1.0)
241
-
242
- surfaces = []
243
-
244
- hip_rx = _circ_to_rx(hip) + ease
245
- hip_ry = _circ_to_ry(hip) + ease
246
- waist_rx = _circ_to_rx(waist) + ease
247
- waist_ry = _circ_to_ry(waist) + ease
248
-
249
- n_wb = 8
250
- wb_z = np.linspace(Z_HIP, Z_WAIST, n_wb)
251
- wb_rx_arr = []
252
- wb_ry_arr = []
253
- for z in wb_z:
254
- t = (z - Z_HIP) / (Z_WAIST - Z_HIP)
255
- rx = hip_rx + (waist_rx - hip_rx) * t
256
- ry = hip_ry + (waist_ry - hip_ry) * t
257
- rx = max(rx, _body_rx(z) + 0.8)
258
- ry = max(ry, _body_ry(z) + 0.8)
259
- wb_rx_arr.append(rx)
260
- wb_ry_arr.append(ry)
261
-
262
- Xw, Yw, Zw = _revolution_surface(wb_z, wb_rx_arr, wb_ry_arr, n_theta=36)
263
- surfaces.append(_make_surface(Xw, Yw, Zw, _darken_color(color, 0.12), "Waistband", opacity=0.90))
264
-
265
- hem_z = max(Z_HIP - pant_len * 0.85, Z_ANKLE - 5)
266
- hem_z = max(hem_z, 2)
267
-
268
- leg_cx = hip_rx * 0.45
269
-
270
- r_thigh = thigh / TWO_PI + ease
271
- r_ankle = ankle / TWO_PI + ease + flare * 0.15
272
-
273
- n_leg = 18
274
- leg_z = np.linspace(Z_HIP, hem_z, n_leg)
275
-
276
- for side_idx, side in enumerate([1, -1]):
277
- leg_rx = []
278
- leg_ry = []
279
- for i, z in enumerate(leg_z):
280
- t = i / (n_leg - 1)
281
- rx = r_thigh + (r_ankle - r_thigh) * t
282
- ry = rx * 0.85
283
- leg_rx.append(rx)
284
- leg_ry.append(ry)
285
-
286
- cx = side * leg_cx
287
- Xl, Yl, Zl = _revolution_surface(leg_z, leg_rx, leg_ry, n_theta=28, cx=cx)
288
- leg_color = color if side_idx == 0 else _lighten_color(color, 0.06)
289
- surfaces.append(_make_surface(Xl, Yl, Zl, leg_color, "Leg", opacity=0.85))
290
-
291
- return surfaces
292
-
293
-
294
- def _build_skirt(params, color="#C48BB8"):
295
- """Build skirt surfaces."""
296
- waist = params.get("waist", 72)
297
- hip = params.get("hip", 96)
298
- skirt_len = params.get("skirt_length", 55)
299
- flare = params.get("flare", 5)
300
- fit = params.get("fit", "regular")
301
-
302
- ease = {"fitted": 0.5, "regular": 1.0, "oversized": 2.5, "loose": 2.0}.get(fit, 1.0)
303
-
304
- surfaces = []
305
-
306
- hem_z = max(Z_WAIST - skirt_len, Z_ANKLE - 5)
307
- hem_z = max(hem_z, 2)
308
-
309
- waist_rx = _circ_to_rx(waist) + ease
310
- waist_ry = _circ_to_ry(waist) + ease
311
- hip_rx = _circ_to_rx(hip) + ease
312
- hip_ry = _circ_to_ry(hip) + ease
313
-
314
- n_sk = 16
315
- skirt_z = np.linspace(Z_WAIST, hem_z, n_sk)
316
- skirt_rx = []
317
- skirt_ry = []
318
-
319
- for i, z in enumerate(skirt_z):
320
- t = i / (n_sk - 1)
321
- if t < 0.3:
322
- f = t / 0.3
323
- rx = waist_rx + (hip_rx - waist_rx) * f
324
- ry = waist_ry + (hip_ry - waist_ry) * f
325
- else:
326
- f = (t - 0.3) / 0.7
327
- rx = hip_rx + flare * f
328
- ry = hip_ry + flare * 0.85 * f
329
-
330
- rx = max(rx, _body_rx(z) + 1.0)
331
- ry = max(ry, _body_ry(z) + 1.0)
332
- skirt_rx.append(rx)
333
- skirt_ry.append(ry)
334
-
335
- Xs, Ys, Zs = _revolution_surface(skirt_z, skirt_rx, skirt_ry, n_theta=36)
336
- surfaces.append(_make_surface(Xs, Ys, Zs, color, "Skirt", opacity=0.82))
337
-
338
- n_wb = 4
339
- wb_height = params.get("waistband_height", 4)
340
- wb_z = np.linspace(Z_WAIST, Z_WAIST + wb_height, n_wb)
341
- wb_rx = [waist_rx + 0.3] * n_wb
342
- wb_ry = [waist_ry + 0.3] * n_wb
343
- Xwb, Ywb, Zwb = _revolution_surface(wb_z, wb_rx, wb_ry, n_theta=36)
344
- surfaces.append(_make_surface(Xwb, Ywb, Zwb, _darken_color(color, 0.15), "Waistband", opacity=0.92))
345
-
346
- return surfaces
347
-
348
-
349
- def _build_dress(params, color="#8E44AD"):
350
- """Build dress = bodice top + skirt bottom."""
351
- surfaces = []
352
- bodice_params = dict(params)
353
- bodice_params["bodice_length"] = params.get("bodice_length", 42)
354
- surfaces.extend(_build_shirt(bodice_params, color=color))
355
-
356
- skirt_params = dict(params)
357
- skirt_params["skirt_length"] = params.get("skirt_length", 55)
358
- surfaces.extend(_build_skirt(skirt_params, color=_lighten_color(color, 0.05)))
359
-
360
- return surfaces
361
-
362
-
363
- def _build_hoodie(params, color="#27AE60"):
364
- """Build hoodie = shirt + hood."""
365
- surfaces = _build_shirt(params, color=color)
366
-
367
- n_hood = 10
368
- hood_z = np.linspace(Z_SHOULDER, Z_HEAD_TOP + 2, n_hood)
369
- hood_rx = []
370
- hood_ry = []
371
-
372
- for i, z in enumerate(hood_z):
373
- t = i / (n_hood - 1)
374
- if t < 0.2:
375
- rx = _body_rx(Z_SHOULDER) + 2.5
376
- elif t < 0.6:
377
- f = (t - 0.2) / 0.4
378
- rx = _body_rx(min(z, Z_HEAD_TOP)) + 3.0 + 2.0 * np.sin(f * np.pi)
379
- else:
380
- f = (t - 0.6) / 0.4
381
- rx = _body_rx(min(z, Z_HEAD_TOP)) + 3.0 * (1 - f * 0.6)
382
- hood_rx.append(rx)
383
- hood_ry.append(rx * 0.85)
384
-
385
- Xh, Yh, Zh = _revolution_surface(hood_z, hood_rx, hood_ry, n_theta=28)
386
- surfaces.append(_make_surface(Xh, Yh, Zh, _darken_color(color, 0.12), "Hood", opacity=0.78))
387
-
388
- return surfaces
389
-
390
-
391
- def _build_jacket(params, color="#7F8C8D"):
392
- """Build jacket/coat/blazer — like a shirt but longer, wider."""
393
- jacket_params = dict(params)
394
- jacket_len = params.get("jacket_length", params.get("bodice_length", 70))
395
- jacket_params["bodice_length"] = jacket_len
396
-
397
- original_fit = params.get("fit", "regular")
398
- if original_fit == "fitted":
399
- jacket_params["fit"] = "regular"
400
- elif original_fit == "regular":
401
- jacket_params["fit"] = "loose"
402
-
403
- surfaces = _build_shirt(jacket_params, color=color)
404
-
405
- if params.get("has_hood", False):
406
- surfaces.extend(_build_hood_only(params, color))
407
-
408
- return surfaces
409
-
410
-
411
- def _build_hood_only(params, color):
412
- """Just the hood part."""
413
- n_hood = 10
414
- hood_z = np.linspace(Z_SHOULDER, Z_HEAD_TOP + 2, n_hood)
415
- hood_rx = []
416
- hood_ry = []
417
-
418
- for i, z in enumerate(hood_z):
419
- t = i / (n_hood - 1)
420
- if t < 0.2:
421
- rx = _body_rx(Z_SHOULDER) + 2.5
422
- elif t < 0.6:
423
- f = (t - 0.2) / 0.4
424
- rx = _body_rx(min(z, Z_HEAD_TOP)) + 3.0 + 2.0 * np.sin(f * np.pi)
425
- else:
426
- f = (t - 0.6) / 0.4
427
- rx = _body_rx(min(z, Z_HEAD_TOP)) + 3.0 * (1 - f * 0.6)
428
- hood_rx.append(rx)
429
- hood_ry.append(rx * 0.85)
430
-
431
- Xh, Yh, Zh = _revolution_surface(hood_z, hood_rx, hood_ry, n_theta=28)
432
- return [_make_surface(Xh, Yh, Zh, _darken_color(color, 0.12), "Hood", opacity=0.78)]
433
-
434
-
435
- def _build_vest(params, color="#D4A574"):
436
- """Build vest — sleeveless bodice."""
437
- bust = params.get("bust", 92)
438
- waist = params.get("waist", 74)
439
- shoulder = params.get("shoulder_width", 42)
440
- fit = params.get("fit", "regular")
441
- flare = params.get("flare", 0)
442
-
443
- ease = {"fitted": 0.5, "regular": 1.2, "oversized": 3.0, "loose": 2.2}.get(fit, 1.2)
444
-
445
- bust_rx = _circ_to_rx(bust) + ease
446
- bust_ry = _circ_to_ry(bust) + ease
447
- waist_rx = _circ_to_rx(waist) + ease
448
- waist_ry = _circ_to_ry(waist) + ease
449
- shoulder_rx = shoulder / 2
450
-
451
- surfaces = []
452
- vest_len = params.get("vest_length", params.get("bodice_length", 55))
453
- extra = max(0, vest_len - 42) * 0.5
454
- hem_z = max(Z_WAIST - extra, Z_HIP)
455
-
456
- n = 14
457
- torso_z = np.linspace(hem_z, Z_SHOULDER, n)
458
- torso_rx = []
459
- torso_ry = []
460
-
461
- for z in torso_z:
462
- t = (z - hem_z) / (Z_SHOULDER - hem_z)
463
- if t < 0.25:
464
- rx = waist_rx + flare * (0.25 - t) / 0.25
465
- ry = waist_ry + flare * 0.8 * (0.25 - t) / 0.25
466
- elif t < 0.55:
467
- f = (t - 0.25) / 0.30
468
- rx = waist_rx + (bust_rx - waist_rx) * f
469
- ry = waist_ry + (bust_ry - waist_ry) * f
470
- elif t < 0.8:
471
- f = (t - 0.55) / 0.25
472
- rx = bust_rx * (1 - 0.08 * f)
473
- ry = bust_ry * (1 - 0.08 * f)
474
- else:
475
- f = (t - 0.8) / 0.2
476
- rx = bust_rx * 0.92 + (shoulder_rx - bust_rx * 0.92) * f
477
- ry = bust_ry * 0.92 * (1 - 0.15 * f)
478
-
479
- rx = max(rx, _body_rx(z) + 0.8)
480
- ry = max(ry, _body_ry(z) + 0.8)
481
- torso_rx.append(rx)
482
- torso_ry.append(ry)
483
-
484
- X, Y, Zt = _revolution_surface(torso_z, torso_rx, torso_ry, n_theta=36)
485
- surfaces.append(_make_surface(X, Y, Zt, color, "Vest", opacity=0.85))
486
-
487
- return surfaces
488
-
489
-
490
- # ── Color helpers ───────────────────────────────────────────────────────────
491
- def _hex_to_rgb(hex_color):
492
- hex_color = hex_color.lstrip('#')
493
- return tuple(int(hex_color[i:i + 2], 16) for i in (0, 2, 4))
494
-
495
- def _rgb_to_hex(r, g, b):
496
- return f"#{int(r):02x}{int(g):02x}{int(b):02x}"
497
-
498
- def _lighten_color(hex_color, amount=0.1):
499
- r, g, b = _hex_to_rgb(hex_color)
500
- return _rgb_to_hex(min(255, r + int(255 * amount)),
501
- min(255, g + int(255 * amount)),
502
- min(255, b + int(255 * amount)))
503
-
504
- def _darken_color(hex_color, amount=0.1):
505
- r, g, b = _hex_to_rgb(hex_color)
506
- return _rgb_to_hex(max(0, r - int(255 * amount)),
507
- max(0, g - int(255 * amount)),
508
- max(0, b - int(255 * amount)))
509
-
510
-
511
- # ── Default garment colors ─────────────────────────────────────────────────
512
- GARMENT_COLORS = {
513
- "shirt": "#4A90D9", "blouse": "#D98CB3", "top": "#5B9BD5",
514
- "t-shirt": "#4A90D9", "tee": "#4A90D9",
515
- "dress": "#8E44AD", "skirt": "#C48BB8",
516
- "pants": "#2C5F8A", "trousers": "#2C5F8A", "jeans": "#1A3D6C",
517
- "jacket": "#7F8C8D", "coat": "#5D6D7E", "blazer": "#6C757D",
518
- "hoodie": "#27AE60", "sweatshirt": "#27AE60",
519
- "vest": "#D4A574",
520
- }
521
-
522
- GARMENT_BUILDERS = {
523
- "shirt": _build_shirt, "blouse": _build_shirt,
524
- "top": _build_shirt, "t-shirt": _build_shirt, "tee": _build_shirt,
525
- "dress": _build_dress, "skirt": _build_skirt,
526
- "pants": _build_pants, "trousers": _build_pants, "jeans": _build_pants,
527
- "jacket": _build_jacket, "coat": _build_jacket, "blazer": _build_jacket,
528
- "hoodie": _build_hoodie, "sweatshirt": _build_hoodie,
529
- "vest": _build_vest,
530
- }
531
-
532
-
533
- # ── Main entry point ───────────────────────────────────────────────────────
534
- def create_3d_figure(analysis: Dict) -> go.Figure:
535
- """Create interactive 3D garment visualization on a mannequin body."""
536
  garment_type = analysis.get("garment_type", "shirt").lower()
537
  measurements = analysis.get("measurements", {})
538
  features = analysis.get("features", {})
539
- params = {**measurements, **features}
540
 
541
  fig = go.Figure()
 
542
 
543
- # Mannequin body
544
- fig.add_trace(_make_body())
 
545
 
546
- # Garment surfaces
547
- color = GARMENT_COLORS.get(garment_type, "#4A90D9")
548
- builder = GARMENT_BUILDERS.get(garment_type, _build_shirt)
549
- for surf in builder(params, color=color):
550
- fig.add_trace(surf)
551
 
552
- # Scene layout
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
553
  fig.update_layout(
554
  scene=dict(
555
  xaxis=dict(showgrid=False, showticklabels=False, title="",
556
- zeroline=False, range=[-35, 35]),
557
  yaxis=dict(showgrid=False, showticklabels=False, title="",
558
- zeroline=False, range=[-35, 35]),
559
  zaxis=dict(showgrid=False, showticklabels=False, title="",
560
  zeroline=False, range=[0, 180]),
561
  aspectmode="data",
562
- camera=dict(
563
- eye=dict(x=1.8, y=1.2, z=0.5),
564
- center=dict(x=0, y=0, z=-0.1),
565
- ),
566
  bgcolor="rgba(245,245,248,1)",
567
  ),
568
  margin=dict(l=0, r=0, t=35, b=0),
569
  height=550,
570
- title=dict(text=f"3D Preview: {garment_type.title()}", font=dict(size=14)),
 
571
  paper_bgcolor="#fafafa",
 
 
572
  )
573
-
574
- return fig
 
1
  """
2
+ 3D Garment Visualizer — Built FROM 2D Pattern Pieces
3
 
4
+ Takes the actual 2D sewing pattern pieces from pattern_generator.py,
5
+ triangulates them, and wraps each piece onto the correct body region
6
+ using cylindrical/cone projections.
 
7
 
8
+ Each 3D surface directly corresponds to a 2D pattern piece.
 
 
 
9
  """
10
  import numpy as np
11
  import plotly.graph_objects as go
12
+ from typing import Dict, List, Tuple, Optional
13
 
14
+ # ── Body constants (cm, Z=0 at floor, Z=170 at head top) ───────────────────
15
+ BODY_HEIGHT = 170.0
 
 
 
 
 
 
 
 
16
  Z_ANKLE = 7
17
  Z_KNEE = 45
18
+ Z_CROTCH = 82
19
  Z_HIP = 88
20
  Z_WAIST = 104
21
  Z_BUST = 120
22
  Z_SHOULDER = 145
23
+ Z_NECK = 150
24
+ Z_HEAD = 170
25
+
26
+ _BODY_Z = [0, 7, 30, 45, 65, 82, 88, 100, 104, 112, 120, 132, 140, 145, 150, 158, 162, 170]
27
+ _BODY_RX = [4, 3, 4.5, 4, 6.5, 8, 9.5, 8, 7, 8.5, 10, 9, 8.5, 11.5, 3.5, 3, 5.5, 4.5]
28
+ _BODY_RY = [3, 2.5, 4, 4, 6, 7.5, 8.5, 7.5, 6.5, 8, 9, 8, 7.5, 8, 3.5, 3, 5, 4.5]
29
+
30
+ R_NECK = 5.5
31
+ R_WRIST = 3.5
32
+
33
+ PIECE_COLORS = {
34
+ 'Front Bodice': '#5B9BD5', 'Back Bodice': '#4472C4',
35
+ 'Sleeve': '#6BB3E0', 'Collar': '#FFFFFF', 'Cuff': '#E8E8E8',
36
+ 'Front Skirt': '#C48BB8', 'Back Skirt': '#A86CA8',
37
+ 'Front Pant': '#2C5F8A', 'Back Pant': '#1A4570',
38
+ 'Waistband': '#333333', 'Hood': '#3A7BD5', 'Pocket': '#7A7A7A',
39
+ }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
40
 
41
 
42
+ def _point_in_polygon(x, y, polygon):
43
+ n = len(polygon)
44
+ inside = False
45
+ j = n - 1
46
+ for i in range(n):
47
+ xi, yi = polygon[i]
48
+ xj, yj = polygon[j]
49
+ if ((yi > y) != (yj > y)) and (x < (xj - xi) * (y - yi) / (yj - yi + 1e-12) + xi):
50
+ inside = not inside
51
+ j = i
52
+ return inside
53
+
54
+
55
+ def _triangulate_piece(pts_2d, nu=16, nv=16):
56
+ pts = np.array(pts_2d, dtype=float)
57
+ if len(pts) < 3:
58
+ return pts, np.zeros((0, 3), dtype=int)
59
+ if np.allclose(pts[0], pts[-1]):
60
+ pts = pts[:-1]
61
+
62
+ x_min, x_max = pts[:, 0].min(), pts[:, 0].max()
63
+ y_min, y_max = pts[:, 1].min(), pts[:, 1].max()
64
+ dx = max(x_max - x_min, 0.1)
65
+ dy = max(y_max - y_min, 0.1)
66
+
67
+ us = np.linspace(x_min - dx * 0.01, x_max + dx * 0.01, nu)
68
+ vs = np.linspace(y_min - dy * 0.01, y_max + dy * 0.01, nv)
69
+
70
+ grid_idx = -np.ones((nv, nu), dtype=int)
71
+ valid_pts = []
72
+ for i in range(nv):
73
+ for j in range(nu):
74
+ if _point_in_polygon(us[j], vs[i], pts):
75
+ grid_idx[i, j] = len(valid_pts)
76
+ valid_pts.append([us[j], vs[i]])
77
+
78
+ if len(valid_pts) < 3:
79
+ return _triangulate_fan(pts)
80
+
81
+ tris = []
82
+ for i in range(nv - 1):
83
+ for j in range(nu - 1):
84
+ a, b = grid_idx[i, j], grid_idx[i, j + 1]
85
+ c, d = grid_idx[i + 1, j], grid_idx[i + 1, j + 1]
86
+ if a >= 0 and b >= 0 and c >= 0:
87
+ tris.append([a, b, c])
88
+ if b >= 0 and c >= 0 and d >= 0:
89
+ tris.append([b, d, c])
90
+
91
+ if len(tris) == 0:
92
+ return _triangulate_fan(pts)
93
+ return np.array(valid_pts), np.array(tris)
94
+
95
+
96
+ def _triangulate_fan(pts):
97
+ n = len(pts)
98
+ if n < 3:
99
+ return pts, np.zeros((0, 3), dtype=int)
100
+ centroid = pts.mean(axis=0, keepdims=True)
101
+ all_pts = np.vstack([pts, centroid])
102
+ tris = np.array([[i, (i + 1) % n, n] for i in range(n)])
103
+ return all_pts, tris
104
+
105
+
106
+ def _wrap_cylinder(pts_2d, radius, z_bottom, z_top, angle_start, angle_range,
107
+ x_offset=0.0, y_offset=0.0):
108
+ pts = np.array(pts_2d, dtype=float)
109
+ x_min, x_max = pts[:, 0].min(), pts[:, 0].max()
110
+ y_min, y_max = pts[:, 1].min(), pts[:, 1].max()
111
+ x_norm = (pts[:, 0] - x_min) / max(x_max - x_min, 0.01)
112
+ y_norm = (pts[:, 1] - y_min) / max(y_max - y_min, 0.01)
113
+ theta = angle_start + x_norm * angle_range
114
+ z = z_bottom + y_norm * (z_top - z_bottom)
115
+ x_3d = radius * np.cos(theta) + x_offset
116
+ y_3d = radius * np.sin(theta) + y_offset
117
+ return np.stack([x_3d, y_3d, z], axis=1)
118
+
119
+
120
+ def _wrap_cone(pts_2d, r_top, r_bottom, z_top, z_bottom, angle_start, angle_range):
121
+ pts = np.array(pts_2d, dtype=float)
122
+ x_min, x_max = pts[:, 0].min(), pts[:, 0].max()
123
+ y_min, y_max = pts[:, 1].min(), pts[:, 1].max()
124
+ x_norm = (pts[:, 0] - x_min) / max(x_max - x_min, 0.01)
125
+ y_norm = (pts[:, 1] - y_min) / max(y_max - y_min, 0.01)
126
+ theta = angle_start + x_norm * angle_range
127
+ z = z_top - y_norm * (z_top - z_bottom)
128
+ radius = r_top + y_norm * (r_bottom - r_top)
129
+ x_3d = radius * np.cos(theta)
130
+ y_3d = radius * np.sin(theta)
131
+ return np.stack([x_3d, y_3d, z], axis=1)
132
+
133
+
134
+ def _wrap_sleeve(pts_2d, side='right', shoulder_x=14.0, shoulder_z=142.0,
135
+ arm_length=35.0, arm_angle_deg=25.0, r_top=5.5, r_bottom=3.5):
136
+ pts = np.array(pts_2d, dtype=float)
137
+ x_min, x_max = pts[:, 0].min(), pts[:, 0].max()
138
+ y_min, y_max = pts[:, 1].min(), pts[:, 1].max()
139
+ x_norm = (pts[:, 0] - x_min) / max(x_max - x_min, 0.01)
140
+ y_norm = (pts[:, 1] - y_min) / max(y_max - y_min, 0.01)
141
+ theta = x_norm * 2 * np.pi
142
+ t_arm = 1.0 - y_norm
143
+ radius = r_top + t_arm * (r_bottom - r_top)
144
+ lx = radius * np.cos(theta)
145
+ ly = -t_arm * arm_length
146
+ lz = radius * np.sin(theta)
147
+ ang = np.radians(arm_angle_deg)
148
+ rx = lx * np.cos(ang) - ly * np.sin(ang)
149
+ rz_local = lx * np.sin(ang) + ly * np.cos(ang)
150
+ sign = 1.0 if side == 'right' else -1.0
151
+ return np.stack([sign * (rx + shoulder_x), lz, rz_local + shoulder_z], axis=1)
152
+
153
+
154
+ def _wrap_pant_leg(pts_2d, side='right', hip_x=5.0, z_top=88.0, z_bottom=7.0,
155
+ r_top=8.5, r_bottom=4.5, is_front=True):
156
+ pts = np.array(pts_2d, dtype=float)
157
+ x_min, x_max = pts[:, 0].min(), pts[:, 0].max()
158
+ y_min, y_max = pts[:, 1].min(), pts[:, 1].max()
159
+ x_norm = (pts[:, 0] - x_min) / max(x_max - x_min, 0.01)
160
+ y_norm = (pts[:, 1] - y_min) / max(y_max - y_min, 0.01)
161
+ if is_front:
162
+ theta = -np.pi / 2 + x_norm * np.pi
163
+ else:
164
+ theta = np.pi / 2 + x_norm * np.pi
165
+ z = z_bottom + y_norm * (z_top - z_bottom)
166
+ radius = r_bottom + y_norm * (r_top - r_bottom)
167
+ sign = 1.0 if side == 'right' else -1.0
168
+ x_3d = sign * hip_x + radius * np.cos(theta)
169
+ y_3d = radius * np.sin(theta)
170
+ return np.stack([x_3d, y_3d, z], axis=1)
171
+
172
+
173
+ def _wrap_hood(pts_2d, z_base=145.0, z_top=172.0, r_base=12.0, r_top=7.0):
174
+ pts = np.array(pts_2d, dtype=float)
175
+ x_min, x_max = pts[:, 0].min(), pts[:, 0].max()
176
+ y_min, y_max = pts[:, 1].min(), pts[:, 1].max()
177
+ x_norm = (pts[:, 0] - x_min) / max(x_max - x_min, 0.01)
178
+ y_norm = (pts[:, 1] - y_min) / max(y_max - y_min, 0.01)
179
+ theta = x_norm * np.pi
180
+ z = z_base + y_norm * (z_top - z_base)
181
+ radius = r_base + y_norm * (r_top - r_base)
182
+ radius *= (1.0 - 0.3 * y_norm ** 2)
183
+ x_3d = radius * np.cos(theta)
184
+ y_3d = radius * np.sin(theta) - 3
185
+ return np.stack([x_3d, y_3d, z], axis=1)
186
+
187
+
188
+ def _match_piece_type(name):
189
+ name_lower = name.lower()
190
+ if 'front bodice' in name_lower: return 'front_bodice'
191
+ if 'back bodice' in name_lower: return 'back_bodice'
192
+ if 'sleeve' in name_lower: return 'sleeve'
193
+ if 'collar' in name_lower: return 'collar'
194
+ if 'cuff' in name_lower: return 'cuff'
195
+ if 'front skirt' in name_lower: return 'front_skirt'
196
+ if 'back skirt' in name_lower: return 'back_skirt'
197
+ if 'front pant' in name_lower: return 'front_pant'
198
+ if 'back pant' in name_lower: return 'back_pant'
199
+ if 'waistband' in name_lower: return 'waistband'
200
+ if 'hood' in name_lower: return 'hood'
201
+ if 'pocket' in name_lower: return 'pocket'
202
+ return 'unknown'
203
+
204
+
205
+ def _get_piece_color(name):
206
+ for key, color in PIECE_COLORS.items():
207
+ if key.lower() in name.lower():
208
+ return color
209
+ return '#888888'
210
+
211
+
212
+ def _make_body_trace():
213
+ theta = np.linspace(0, 2 * np.pi, 48)
214
+ z_arr = np.array(_BODY_Z, float)
215
+ rx = np.array(_BODY_RX, float)
216
+ ry = np.array(_BODY_RY, float)
217
+ Z = np.outer(z_arr, np.ones(48))
218
+ X = np.outer(rx, np.cos(theta))
219
+ Y = np.outer(ry, np.sin(theta))
220
  return go.Surface(
221
  x=X, y=Y, z=Z,
222
  surfacecolor=np.ones_like(X) * 0.8,
223
  colorscale=[[0, "#E8D0B0"], [1, "#E8D0B0"]],
224
+ opacity=0.20, showscale=False, name="Body",
 
 
225
  hoverinfo="skip",
226
  lighting=dict(ambient=0.9, diffuse=0.1, specular=0, roughness=1.0),
227
  )
228
 
229
 
230
+ def create_3d_figure(analysis: Dict, pattern_pieces: Optional[List[Dict]] = None) -> go.Figure:
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
231
  garment_type = analysis.get("garment_type", "shirt").lower()
232
  measurements = analysis.get("measurements", {})
233
  features = analysis.get("features", {})
 
234
 
235
  fig = go.Figure()
236
+ fig.add_trace(_make_body_trace())
237
 
238
+ if pattern_pieces is None or len(pattern_pieces) == 0:
239
+ _setup_layout(fig, garment_type)
240
+ return fig
241
 
242
+ for piece in pattern_pieces:
243
+ name = piece['name']
244
+ pts_2d = piece['points']
245
+ piece_type = _match_piece_type(name)
246
+ color = _get_piece_color(name)
247
 
248
+ mesh_pts, tris = _triangulate_piece(pts_2d)
249
+ if len(tris) == 0:
250
+ continue
251
+
252
+ traces = _wrap_piece(piece_type, mesh_pts, tris, name, color,
253
+ measurements, features, garment_type)
254
+ for trace in traces:
255
+ fig.add_trace(trace)
256
+
257
+ _setup_layout(fig, garment_type)
258
+ return fig
259
+
260
+
261
+ def _wrap_piece(piece_type, mesh_pts, tris, name, color,
262
+ measurements, features, garment_type):
263
+ traces = []
264
+ bust = measurements.get('bust', 92)
265
+ waist = measurements.get('waist', 74)
266
+ hip = measurements.get('hip', 96)
267
+ sleeve_len = measurements.get('sleeve_length', 60)
268
+ skirt_len = measurements.get('skirt_length', 55)
269
+ pant_len = measurements.get('pant_length', 100)
270
+
271
+ r_torso = bust / (2 * np.pi) + 1.5
272
+ r_waist = waist / (2 * np.pi) + 1.5
273
+ r_hip = hip / (2 * np.pi) + 1.5
274
+
275
+ if piece_type == 'front_bodice':
276
+ pts_3d = _wrap_cylinder(mesh_pts, r_torso, Z_WAIST, Z_SHOULDER,
277
+ -np.pi / 2, np.pi)
278
+ traces.append(_make_mesh3d(pts_3d, tris, color, name))
279
+
280
+ elif piece_type == 'back_bodice':
281
+ pts_3d = _wrap_cylinder(mesh_pts, r_torso, Z_WAIST, Z_SHOULDER,
282
+ np.pi / 2, np.pi)
283
+ traces.append(_make_mesh3d(pts_3d, tris, color, name))
284
+
285
+ elif piece_type == 'sleeve':
286
+ shoulder_x = r_torso + 1.0
287
+ for side in ['right', 'left']:
288
+ pts_3d = _wrap_sleeve(mesh_pts, side=side, shoulder_x=shoulder_x,
289
+ shoulder_z=Z_SHOULDER - 2,
290
+ arm_length=sleeve_len * 0.55,
291
+ arm_angle_deg=25, r_top=5.5, r_bottom=3.5)
292
+ traces.append(_make_mesh3d(pts_3d, tris, color,
293
+ f"{name} ({'R' if side == 'right' else 'L'})"))
294
+
295
+ elif piece_type == 'collar':
296
+ pts_3d = _wrap_cylinder(mesh_pts, R_NECK + 1.0, Z_NECK - 2, Z_NECK + 3,
297
+ -np.pi, 2 * np.pi)
298
+ traces.append(_make_mesh3d(pts_3d, tris, color, name))
299
+
300
+ elif piece_type == 'cuff':
301
+ for side in ['right', 'left']:
302
+ sign = 1.0 if side == 'right' else -1.0
303
+ wrist_x = sign * (r_torso + sleeve_len * 0.3 + 2)
304
+ wrist_z = Z_SHOULDER - sleeve_len * 0.5
305
+ pts_3d = _wrap_cylinder(mesh_pts, R_WRIST + 0.5,
306
+ wrist_z - 3, wrist_z,
307
+ 0, 2 * np.pi, x_offset=wrist_x)
308
+ traces.append(_make_mesh3d(pts_3d, tris, color,
309
+ f"{name} ({'R' if side == 'right' else 'L'})"))
310
+
311
+ elif piece_type == 'front_skirt':
312
+ hem_z = max(Z_WAIST - skirt_len, Z_ANKLE)
313
+ flare = measurements.get('flare', 5)
314
+ r_hem = r_hip + flare * 0.5
315
+ pts_3d = _wrap_cone(mesh_pts, r_waist, r_hem, Z_WAIST, hem_z,
316
+ -np.pi / 2, np.pi)
317
+ traces.append(_make_mesh3d(pts_3d, tris, color, name))
318
+
319
+ elif piece_type == 'back_skirt':
320
+ hem_z = max(Z_WAIST - skirt_len, Z_ANKLE)
321
+ flare = measurements.get('flare', 5)
322
+ r_hem = r_hip + flare * 0.5
323
+ pts_3d = _wrap_cone(mesh_pts, r_waist, r_hem, Z_WAIST, hem_z,
324
+ np.pi / 2, np.pi)
325
+ traces.append(_make_mesh3d(pts_3d, tris, color, name))
326
+
327
+ elif piece_type == 'front_pant':
328
+ hem_z = max(Z_HIP - pant_len * 0.85, Z_ANKLE)
329
+ thigh_circ = measurements.get('thigh', 56)
330
+ ankle_circ = measurements.get('ankle', 24)
331
+ r_thigh = thigh_circ / (2 * np.pi) + 1
332
+ r_ankle_r = ankle_circ / (2 * np.pi) + 1
333
+ for side in ['right', 'left']:
334
+ pts_3d = _wrap_pant_leg(mesh_pts, side=side, hip_x=r_hip * 0.4,
335
+ z_top=Z_HIP, z_bottom=hem_z,
336
+ r_top=r_thigh, r_bottom=r_ankle_r, is_front=True)
337
+ traces.append(_make_mesh3d(pts_3d, tris, color,
338
+ f"{name} ({'R' if side == 'right' else 'L'})"))
339
+
340
+ elif piece_type == 'back_pant':
341
+ hem_z = max(Z_HIP - pant_len * 0.85, Z_ANKLE)
342
+ thigh_circ = measurements.get('thigh', 56)
343
+ ankle_circ = measurements.get('ankle', 24)
344
+ r_thigh = thigh_circ / (2 * np.pi) + 1
345
+ r_ankle_r = ankle_circ / (2 * np.pi) + 1
346
+ for side in ['right', 'left']:
347
+ pts_3d = _wrap_pant_leg(mesh_pts, side=side, hip_x=r_hip * 0.4,
348
+ z_top=Z_HIP, z_bottom=hem_z,
349
+ r_top=r_thigh, r_bottom=r_ankle_r, is_front=False)
350
+ traces.append(_make_mesh3d(pts_3d, tris, color,
351
+ f"{name} ({'R' if side == 'right' else 'L'})"))
352
+
353
+ elif piece_type == 'waistband':
354
+ wb_h = measurements.get('waistband_height', 4)
355
+ pts_3d = _wrap_cylinder(mesh_pts, r_waist + 0.5, Z_WAIST, Z_WAIST + wb_h,
356
+ -np.pi, 2 * np.pi)
357
+ traces.append(_make_mesh3d(pts_3d, tris, color, name))
358
+
359
+ elif piece_type == 'hood':
360
+ pts_3d_r = _wrap_hood(mesh_pts, z_base=Z_SHOULDER, z_top=Z_HEAD + 2,
361
+ r_base=r_torso, r_top=7)
362
+ traces.append(_make_mesh3d(pts_3d_r, tris, color, f"{name} (R)"))
363
+ pts_3d_l = pts_3d_r.copy()
364
+ pts_3d_l[:, 0] = -pts_3d_l[:, 0]
365
+ traces.append(_make_mesh3d(pts_3d_l, tris, color, f"{name} (L)"))
366
+
367
+ elif piece_type == 'pocket':
368
+ pocket_z = (Z_WAIST + Z_BUST) / 2
369
+ pts_3d = _wrap_cylinder(mesh_pts, r_torso + 0.3,
370
+ pocket_z - 8, pocket_z, -0.3, 0.6)
371
+ traces.append(_make_mesh3d(pts_3d, tris, color, name))
372
+
373
+ return traces
374
+
375
+
376
+ def _make_mesh3d(pts_3d, tris, color, name):
377
+ return go.Mesh3d(
378
+ x=pts_3d[:, 0], y=pts_3d[:, 1], z=pts_3d[:, 2],
379
+ i=tris[:, 0], j=tris[:, 1], k=tris[:, 2],
380
+ color=color, opacity=0.85, name=name,
381
+ flatshading=False,
382
+ lighting=dict(ambient=0.55, diffuse=0.8, specular=0.2, roughness=0.6),
383
+ lightposition=dict(x=100, y=200, z=300),
384
+ hoverinfo="name",
385
+ )
386
+
387
+
388
+ def _setup_layout(fig, garment_type):
389
  fig.update_layout(
390
  scene=dict(
391
  xaxis=dict(showgrid=False, showticklabels=False, title="",
392
+ zeroline=False, range=[-40, 40]),
393
  yaxis=dict(showgrid=False, showticklabels=False, title="",
394
+ zeroline=False, range=[-40, 40]),
395
  zaxis=dict(showgrid=False, showticklabels=False, title="",
396
  zeroline=False, range=[0, 180]),
397
  aspectmode="data",
398
+ camera=dict(eye=dict(x=1.8, y=1.2, z=0.5),
399
+ center=dict(x=0, y=0, z=-0.1)),
 
 
400
  bgcolor="rgba(245,245,248,1)",
401
  ),
402
  margin=dict(l=0, r=0, t=35, b=0),
403
  height=550,
404
+ title=dict(text=f"3D Preview: {garment_type.title()} (from pattern pieces)",
405
+ font=dict(size=14)),
406
  paper_bgcolor="#fafafa",
407
+ showlegend=True,
408
+ legend=dict(font=dict(size=10), bgcolor="rgba(255,255,255,0.8)"),
409
  )