add simple two room example

Author: Crowder44

src/facility_layout/__init__.py

@@ -0,0 +1 @@
+# Copyright 2025, Battelle Energy Alliance, LLC All Rights Reserved

src/facility_layout/models/__init__.py

@@ -0,0 +1 @@
+# Copyright 2025, Battelle Energy Alliance, LLC All Rights Reserved

src/facility_layout/models/connections.py

@@ -0,0 +1,17 @@
+# Copyright 2025, Battelle Energy Alliance, LLC All Rights Reserved
+
+from dataclasses import dataclass
+from typing import Optional, List, Tuple
+
+
+@dataclass
+class Connection:
+    """Represents a connection requirement between equipment."""
+
+    id: str
+    from_equipment_id: str
+    to_equipment_id: str
+    connection_type: str  # "pipe", "cable", "duct"
+    from_port_index: int  # Index into from_equipment.ports
+    to_port_index: int  # Index into to_equipment.ports
+    path: Optional[List[Tuple[float, float]]] = None  # Computed path points

src/facility_layout/models/equipment.py

@@ -0,0 +1,47 @@
+# Copyright 2025, Battelle Energy Alliance, LLC All Rights Reserved
+
+from dataclasses import dataclass
+from typing import List, Tuple, Optional
+from enum import Enum
+
+
+class PortType(Enum):
+    """Types of connections that can be made between equipment."""
+
+    PIPE_IN = "pipe_in"
+    PIPE_OUT = "pipe_out"
+    ELECTRICAL = "electrical"
+    CONTROL = "control"
+
+
+@dataclass
+class Port:
+    """Connection point on equipment."""
+
+    port_type: PortType
+    side: str  # "north", "south", "east", "west"
+    relative_position: float  # 0.0 to 1.0 along the side
+
+
+@dataclass
+class Equipment:
+    """Represents a piece of equipment to be placed."""
+
+    id: str
+    equipment_type: str
+    dimensions: Tuple[float, float]  # (width, height) in meters
+    clearance_required: float  # meters of clear space needed around equipment
+    ports: List[Port]
+    assigned_space: Optional[str] = None  # ID of space this must be placed in
+
+    @property
+    def width(self) -> float:
+        return self.dimensions[0]
+
+    @property
+    def height(self) -> float:
+        return self.dimensions[1]
+
+    @property
+    def area(self) -> float:
+        return self.width * self.height

src/facility_layout/models/layout.py

@@ -0,0 +1,62 @@
+# Copyright 2025, Battelle Energy Alliance, LLC All Rights Reserved
+
+from dataclasses import dataclass, field
+from typing import Dict, List, Tuple, Optional
+from .equipment import Equipment
+from .spatial import Space, Building, SpaceConstraint
+from .connections import Connection
+
+
+@dataclass
+class Placement:
+    """Represents the placement of equipment within a space."""
+
+    equipment_id: str
+    space_id: str
+    position: Tuple[float, float]  # (x, y) relative to space origin
+    rotation: float = 0.0  # degrees
+
+
+@dataclass
+class Layout:
+    """Complete facility layout solution."""
+
+    buildings: List[Building]
+    spaces: List[Space]
+    equipment: List[Equipment]
+    placements: Dict[str, Placement]  # equipment_id -> Placement
+    connections: List[Connection]
+    space_constraints: List[SpaceConstraint] = field(default_factory=list)
+    score: Optional[float] = None
+    metadata: Dict = field(default_factory=dict)
+
+    def get_equipment_absolute_position(
+        self, equipment_id: str
+    ) -> Optional[Tuple[float, float]]:
+        """Calculate absolute position of equipment considering all hierarchy levels."""
+        if equipment_id not in self.placements:
+            return None
+
+        placement = self.placements[equipment_id]
+        space = next((s for s in self.spaces if s.id == placement.space_id), None)
+        if not space:
+            return None
+
+        # Start with position within space
+        x, y = placement.position
+
+        # Add space position within building
+        if space.position:
+            x += space.position[0]
+            y += space.position[1]
+
+        # Add building position on site
+        if space.parent_building:
+            building = next(
+                (b for b in self.buildings if b.id == space.parent_building), None
+            )
+            if building and building.position:
+                x += building.position[0]
+                y += building.position[1]
+
+        return (x, y)

src/facility_layout/models/spatial.py

@@ -0,0 +1,61 @@
+# Copyright 2025, Battelle Energy Alliance, LLC All Rights Reserved
+
+from dataclasses import dataclass, field
+from typing import List, Tuple, Optional, Dict
+from enum import Enum
+
+
+class SpaceRelation(Enum):
+    """How two spaces relate to each other."""
+
+    ADJACENT_NORTH = "adjacent_north"
+    ADJACENT_SOUTH = "adjacent_south"
+    ADJACENT_EAST = "adjacent_east"
+    ADJACENT_WEST = "adjacent_west"
+    SEPARATED = "separated"
+
+
+@dataclass
+class Space:
+    """Represents a room or area within a building."""
+
+    id: str
+    dimensions: Tuple[float, float]  # (width, height) in meters
+    space_type: str  # "room", "corridor", "outdoor", etc.
+    parent_building: Optional[str] = None  # ID of containing building
+    position: Optional[Tuple[float, float]] = None  # (x, y) relative to building origin
+
+    @property
+    def width(self) -> float:
+        return self.dimensions[0]
+
+    @property
+    def height(self) -> float:
+        return self.dimensions[1]
+
+    @property
+    def area(self) -> float:
+        return self.width * self.height
+
+
+@dataclass
+class Building:
+    """Represents a building containing multiple spaces."""
+
+    id: str
+    position: Optional[Tuple[float, float]] = None  # (x, y) on site
+    spaces: List[Space] = field(default_factory=list)
+
+    def add_space(self, space: Space) -> None:
+        space.parent_building = self.id
+        self.spaces.append(space)
+
+
+@dataclass
+class SpaceConstraint:
+    """Defines a relationship constraint between two spaces."""
+
+    space1_id: str
+    space2_id: str
+    relation: SpaceRelation
+    gap: float = 0.0  # meters of gap between spaces (for corridors, etc.)

src/facility_layout/solvers/__init__.py

@@ -0,0 +1 @@
+# Copyright 2025, Battelle Energy Alliance, LLC All Rights Reserved