Getfloorplan AI户型图：智能空间生成

import json  class FloorPlanAnalyzer:    """户型图分析器：解析房间数据、计算面积、分类房型、生成布局、推荐家具"""     ROOM_TYPES = {        "bedroom": {"label": "卧室", "min_area": 8, "furniture": ["双人床", "衣柜", "床头柜", "梳妆台"]},        "living_room": {"label": "客厅", "min_area": 15, "furniture": ["沙发", "茶几", "电视柜", "书架"]},        "kitchen": {"label": "厨房", "min_area": 4, "furniture": ["橱柜", "冰箱", "灶台", "操作台"]},        "bathroom": {"label": "卫生间", "min_area": 3, "furniture": ["马桶", "洗手台", "淋浴房", "浴柜"]},    }     def __init__(self, rooms_data):        self.rooms = []        for room in rooms_data:            parsed = self._parse_room(room)            self.rooms.append(parsed)     def _parse_room(self, room):        name = room.get("name", "未知房间")        length = float(room.get("length", 0))        width = float(room.get("width", 0))        room_type = room.get("room_type", "")        area = round(length * width, 2)        return {"name": name, "length": length, "width": width, "area": area, "room_type": room_type}     def calculate_areas(self):        total = sum(r["area"] for r in self.rooms)        return {"rooms": [{"name": r["name"], "area": r["area"]} for r in self.rooms], "total_floor_area": round(total, 2)}     def classify_rooms(self):        classified = []        for r in self.rooms:            rtype = r["room_type"] if r["room_type"] in self.ROOM_TYPES else "unknown"            info = self.ROOM_TYPES.get(rtype, {"label": "其他", "furniture": []})            classified.append({"name": r["name"], "classified_type": info["label"], "area": r["area"], "meets_min_area": r["area"] >= info.get("min_area", 0) if rtype in self.ROOM_TYPES else True})        return classified     def generate_text_layout(self):        lines = []        lines.append("=" * 50)        lines.append("           户 型 平 面 布 局 图")        lines.append("=" * 50)        for r in self.rooms:            label = self.ROOM_TYPES.get(r["room_type"], {}).get("label", r["name"])            block_w = max(int(r["length"] * 3), 6)            block_h = max(int(r["width"] * 2), 3)            lines.append(f"\n  [{label}] {r['name']}  ({r['length']}m x {r['width']}m = {r['area']}m2)")            lines.append("  +" + "-" * block_w + "+")            for i in range(block_h):                if i == block_h // 2:                    content = f"{r['area']}m2"                    padding = block_w - len(content)                    left = padding // 2                    right = padding - left                    lines.append("  |" + " " * left + content + " " * right + "|")                else:                    lines.append("  |" + " " * block_w + "|")            lines.append("  +" + "-" * block_w + "+")        lines.append("\n" + "=" * 50)        return "\n".join(lines)     def recommend_furniture(self):        recommendations = []        for r in self.rooms:            rtype = r["room_type"]            if rtype in self.ROOM_TYPES:                info = self.ROOM_TYPES[rtype]                furniture_list = info["furniture"]                placed = []                remaining_area = r["area"]                for item in furniture_list:                    item_area = 2.0 if rtype != "bathroom" else 1.0                    if remaining_area >= item_area:                        placed.append({"item": item, "estimated_area": item_area})                        remaining_area -= item_area                recommendations.append({"room": r["name"], "type": info["label"], "area": r["area"], "furniture_placed": placed, "remaining_area": round(remaining_area, 2)})        return recommendations  def main():    rooms_data = [        {"name": "主卧", "length": 4.2, "width": 3.6, "room_type": "bedroom"},        {"name": "次卧", "length": 3.3, "width": 3.0, "room_type": "bedroom"},        {"name": "客厅", "length": 5.5, "width": 4.0, "room_type": "living_room"},        {"name": "厨房", "length": 3.0, "width": 2.5, "room_type": "kitchen"},        {"name": "卫生间", "length": 2.5, "width": 2.0, "room_type": "bathroom"},    ]    analyzer = FloorPlanAnalyzer(rooms_data)    areas = analyzer.calculate_areas()    print("=== 面积计算 ===")    print(json.dumps(areas, ensure_ascii=False, indent=2))    classified = analyzer.classify_rooms()    print("\n=== 房间分类 ===")    print(json.dumps(classified, ensure_ascii=False, indent=2))    layout = analyzer.generate_text_layout()    print("\n" + layout)    furniture = analyzer.recommend_furniture()    print("\n=== 家具推荐 ===")    print(json.dumps(furniture, ensure_ascii=False, indent=2))  if __name__ == "__main__":    main()