Add 3D reconstruction module, update __init__.py
Browse files- 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
|
| 6 |
-
geometry.py
|
| 7 |
-
renderer.py
|
| 8 |
-
parser.py
|
|
|
|
| 9 |
"""
|
| 10 |
|
| 11 |
from .schema import (
|
|
@@ -43,8 +45,14 @@ from .parser import (
|
|
| 43 |
FloorPlanParser,
|
| 44 |
parse_floorplan,
|
| 45 |
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 46 |
|
| 47 |
-
__version__ = "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 |
]
|