rikhoffbauer2 commited on
Commit
fbca0dd
Β·
verified Β·
1 Parent(s): 75a0e92

Add 3D reconstruction module, update __init__.py

Browse files
Files changed (1) hide show
  1. floorplan/__init__.py +16 -6
floorplan/__init__.py CHANGED
@@ -1,11 +1,13 @@
1
  """
2
- floorplan β€” Parse floor plan images into structured wall/door/window data.
 
3
 
4
  Architecture:
5
- schema.py β€” Pydantic data models (Wall, Opening, Room, FloorPlan, Corrections)
6
- geometry.py β€” Computational geometry (wall polygons, room detection, centerline ops)
7
- renderer.py β€” SVG + PIL rendering for visual verification and overlay
8
- parser.py β€” VLM-based parsing with iterative correction loop
 
9
  """
10
 
11
  from .schema import (
@@ -43,8 +45,14 @@ from .parser import (
43
  FloorPlanParser,
44
  parse_floorplan,
45
  )
 
 
 
 
 
 
46
 
47
- __version__ = "0.1.0"
48
 
49
  __all__ = [
50
  # Schema
@@ -60,4 +68,6 @@ __all__ = [
60
  "render_floorplan_svg", "render_to_image", "overlay_on_image",
61
  # Parser
62
  "FloorPlanParser", "parse_floorplan",
 
 
63
  ]
 
1
  """
2
+ floorplan β€” Parse floor plan images into structured wall/door/window data,
3
+ and generate 3D models from the structured representation.
4
 
5
  Architecture:
6
+ schema.py β€” Pydantic data models (Wall, Opening, Room, FloorPlan, Corrections)
7
+ geometry.py β€” Computational geometry (wall polygons, room detection, centerline ops)
8
+ renderer.py β€” SVG + PIL rendering for visual verification and overlay
9
+ parser.py β€” VLM-based parsing with iterative correction loop
10
+ reconstruction.py β€” 3D model generation (extrude walls, cut openings, export GLB/OBJ)
11
  """
12
 
13
  from .schema import (
 
45
  FloorPlanParser,
46
  parse_floorplan,
47
  )
48
+ from .reconstruction import (
49
+ generate_3d_model,
50
+ export_glb,
51
+ export_obj,
52
+ export_gltf,
53
+ )
54
 
55
+ __version__ = "0.2.0"
56
 
57
  __all__ = [
58
  # Schema
 
68
  "render_floorplan_svg", "render_to_image", "overlay_on_image",
69
  # Parser
70
  "FloorPlanParser", "parse_floorplan",
71
+ # 3D Reconstruction
72
+ "generate_3d_model", "export_glb", "export_obj", "export_gltf",
73
  ]