| import json, os |
| from devcore.parser import parse_description |
|
|
| TEMPLATE_PATH = os.path.join(os.path.dirname(__file__), "templates", "viewer.html") |
|
|
|
|
| def description_to_blueprint(text): |
| data = parse_description(text) |
| return _build_threejs_html(data) |
|
|
|
|
| def _build_threejs_html(data): |
| scene = data["structure"] |
| rooms = data["rooms"] |
| w = scene["width"] |
| d = scene["depth"] |
| s = scene["stories"] |
| roof_style = scene.get("roof_style", "gable") |
|
|
| wall_height = 8 if s <= 1 else 9 |
| floor_height = 0.5 |
|
|
| room_meshes = [] |
| wall_meshes = [] |
| label_data = [] |
| measurements = [] |
|
|
| for room in rooms: |
| rx, rz = room["x"], room["z"] |
| rw, rd = room["w"], room["d"] |
| floor_n = room.get("floor", 0) |
| cy = floor_n * (wall_height + floor_height) + floor_height / 2 |
| exterior = room.get("exterior", False) |
| color = room["color"] |
|
|
| room_meshes.append({ |
| "x": rx, "z": rz, "w": rw, "d": rd, |
| "y": cy, "h": floor_height, |
| "color": color, "name": room["name"], "exterior": exterior, |
| }) |
|
|
| lx = rx + rw / 2 |
| lz = rz + rd / 2 |
| label_data.append({"x": lx, "z": lz, "y": cy + 0.3, "text": room["name"]}) |
|
|
| measurements.append({ |
| "x": rx + rw / 2, "z": rz - 1, "y": 0.1, |
| "text": f"{rw:.0f}'", |
| "dim": "width", |
| }) |
| measurements.append({ |
| "x": rx - 1, "z": rz + rd / 2, "y": 0.1, |
| "text": f"{rd:.0f}'", |
| "dim": "depth", |
| }) |
|
|
| if not exterior: |
| for side in ["north", "south", "east", "west"]: |
| wall_meshes.append({ |
| "side": side, "x": rx, "z": rz, |
| "w": rw, "d": rd, "h": wall_height, |
| "room_name": room["name"], |
| }) |
|
|
| roof_h = 4 if roof_style == "flat" else 6 |
| roof_elev = s * (wall_height + floor_height) |
| roof_info = { |
| "w": w * 1.15, "d": d * 1.15, "y": roof_elev, |
| "h": roof_h, "style": roof_style, "color": "#5a4a3a", |
| "pitch": 0.5 if roof_style == "steep-gable" else (0.3 if roof_style == "low-gable" else 0.4), |
| } |
|
|
| blueprint = { |
| "scene": { |
| "width": w, "depth": d, "stories": s, |
| "wall_height": wall_height, "floor_height": floor_height, |
| "roof_style": roof_style, "style": scene.get("style", "modern"), |
| }, |
| "floors": room_meshes, |
| "walls": wall_meshes, |
| "roof": roof_info, |
| "labels": label_data, |
| "measurements": measurements, |
| "summary": data["summary"], |
| "features": data["features"], |
| } |
|
|
| return _render_html(blueprint) |
|
|
|
|
| def _render_html(blueprint): |
| bp_json = json.dumps(blueprint) |
| with open(TEMPLATE_PATH, "r") as f: |
| template = f.read() |
| return template.replace("/* BLUEPRINT_DATA */", f"const BLUEPRINT = {bp_json};") |
|
|
|
|
| def blueprint_to_json(text): |
| data = parse_description(text) |
| return json.dumps(data, indent=2) |
|
|
