Create utils/diagram_generator.py

utils/diagram_generator.py

@@ -0,0 +1,145 @@
+"""
+Diagram Generation Utilities for Power Systems
+Generates SVG diagrams for power system concepts
+"""
+
+import json
+from typing import Dict, List, Tuple, Optional
+from datetime import datetime
+
+class DiagramGenerator:
+    """
+    Generate SVG diagrams for power systems concepts
+    """
+    
+    def __init__(self):
+        self.svg_width = 800
+        self.svg_height = 600
+        self.grid_size = 20
+        
+    def create_svg_header(self, width: int = None, height: int = None) -> str:
+        """Create SVG header with proper dimensions"""
+        w = width or self.svg_width
+        h = height or self.svg_height
+        
+        return f"""<svg width="{w}" height="{h}" viewBox="0 0 {w} {h}" xmlns="http://www.w3.org/2000/svg">
+    <defs>
+        <style>
+            .line {{ stroke: #2563eb; stroke-width: 2; fill: none; }}
+            .bus {{ stroke: #dc2626; stroke-width: 4; }}
+            .text {{ font-family: Arial, sans-serif; font-size: 12px; fill: #374151; }}
+            .title {{ font-family: Arial, sans-serif; font-size: 16px; font-weight: bold; fill: #1f2937; }}
+            .component {{ stroke: #059669; stroke-width: 2; fill: #d1fae5; }}
+            .fault {{ stroke: #dc2626; stroke-width: 3; fill: #fecaca; }}
+            .protection {{ stroke: #7c3aed; stroke-width: 2; fill: #e9d5ff; }}
+            .ground {{ stroke: #374151; stroke-width: 2; }}
+        </style>
+        
+        <!-- Marker definitions for arrows -->
+        <marker id="arrowhead" markerWidth="10" markerHeight="7" 
+         refX="9" refY="3.5" orient="auto">
+            <polygon points="0 0, 10 3.5, 0 7" fill="#2563eb" />
+        </marker>
+        
+        <!-- Component symbols -->
+        <g id="generator">
+            <circle cx="0" cy="0" r="15" class="component"/>
+            <text x="0" y="4" text-anchor="middle" class="text">G</text>
+        </g>
+        
+        <g id="transformer">
+            <circle cx="-10" cy="0" r="8" class="component"/>
+            <circle cx="10" cy="0" r="8" class="component"/>
+            <text x="0" y="25" text-anchor="middle" class="text">T</text>
+        </g>
+        
+        <g id="load">
+            <path d="M-10,-10 L10,-10 L5,10 L-5,10 Z" class="component"/>
+            <text x="0" y="25" text-anchor="middle" class="text">Load</text>
+        </g>
+        
+        <g id="fault-symbol">
+            <circle cx="0" cy="0" r="8" class="fault"/>
+            <line x1="-6" y1="-6" x2="6" y2="6" class="fault"/>
+            <line x1="-6" y1="6" x2="6" y2="-6" class="fault"/>
+        </g>
+    </defs>
+"""
+
+    def create_svg_footer(self) -> str:
+        """Create SVG footer"""
+        return "</svg>"
+
+    def generate_single_line_diagram(self, system_config: Dict) -> str:
+        """Generate a single line diagram"""
+        svg = self.create_svg_header()
+        
+        # Title
+        svg += f'<text x="400" y="30" text-anchor="middle" class="title">Single Line Diagram</text>\n'
+        
+        # Main bus (horizontal line)
+        svg += f'<line x1="100" y1="100" x2="700" y2="100" class="bus"/>\n'
+        svg += f'<text x="400" y="85" text-anchor="middle" class="text">Main Bus (33kV)</text>\n'
+        
+        # Generator
+        svg += f'<use href="#generator" transform="translate(150,100)"/>\n'
+        svg += f'<line x1="135" y1="100" x2="100" y2="100" class="line"/>\n'
+        svg += f'<text x="150" y="140" text-anchor="middle" class="text">Generator</text>\n'
+        svg += f'<text x="150" y="155" text-anchor="middle" class="text">100MVA</text>\n'
+        
+        # Transformer
+        svg += f'<use href="#transformer" transform="translate(300,100)"/>\n'
+        svg += f'<line x1="290" y1="100" x2="250" y2="100" class="line"/>\n'
+        svg += f'<line x1="310" y1="100" x2="350" y2="100" class="line"/>\n'
+        svg += f'<text x="300" y="140" text-anchor="middle" class="text">Power Transformer</text>\n'
+        svg += f'<text x="300" y="155" text-anchor="middle" class="text">33/11kV, 50MVA</text>\n'
+        
+        # Distribution lines
+        svg += f'<line x1="400" y1="100" x2="400" y2="200" class="line"/>\n'
+        svg += f'<line x1="350" y1="200" x2="450" y2="200" class="line"/>\n'
+        svg += f'<text x="400" y="215" text-anchor="middle" class="text">Distribution Bus (11kV)</text>\n'
+        
+        # Loads
+        positions = [375, 425]
+        load_names = ["Industrial Load", "Commercial Load"]
+        load_powers = ["15MW", "8MW"]
+        
+        for i, (pos, name, power) in enumerate(zip(positions, load_names, load_powers)):
+            svg += f'<use href="#load" transform="translate({pos},250)"/>\n'
+            svg += f'<line x1="{pos}" y1="200" x2="{pos}" y2="240" class="line"/>\n'
+            svg += f'<text x="{pos}" y="285" text-anchor="middle" class="text">{name}</text>\n'
+            svg += f'<text x="{pos}" y="300" text-anchor="middle" class="text">{power}</text>\n'
+        
+        # Protection devices
+        svg += f'<rect x="195" y="95" width="10" height="10" class="protection"/>\n'
+        svg += f'<text x="200" y="120" text-anchor="middle" class="text">CB1</text>\n'
+        
+        svg += f'<rect x="345" y="95" width="10" height="10" class="protection"/>\n'
+        svg += f'<text x="350" y="120" text-anchor="middle" class="text">CB2</text>\n'
+        
+        # Legend
+        svg += f'<text x="50" y="450" class="title">Legend:</text>\n'
+        svg += f'<line x1="50" y1="470" x2="80" y2="470" class="bus"/>\n'
+        svg += f'<text x="90" y="475" class="text">Bus</text>\n'
+        svg += f'<line x1="50" y1="490" x2="80" y2="490" class="line"/>\n'
+        svg += f'<text x="90" y="495" class="text">Transmission Line</text>\n'
+        svg += f'<rect x="50" y="505" width="10" height="10" class="protection"/>\n'
+        svg += f'<text x="70" y="515" class="text">Circuit Breaker</text>\n'
+        
+        svg += self.create_svg_footer()
+        return svg
+
+    def generate_fault_analysis_diagram(self, fault_type: str = "line_to_ground") -> str:
+        """Generate fault analysis diagram with sequence networks"""
+        svg = self.create_svg_header(900, 700)
+        
+        # Title
+        svg += f'<text x="450" y="30" text-anchor="middle" class="title">Fault Analysis - {fault_type.replace("_", " ").title()}</text>\n'
+        
+        if fault_type == "line_to_ground":
+            # Positive sequence network
+            svg += f'<text x="150" y="80" text-anchor="middle" class="title">Positive Sequence</text>\n'
+            svg += self._draw_sequence_network(150, 100, "Z1", "#059669")
+            
+            # Negative sequence network  
+            svg += f'<text x="450" y="80" text-anchor="middle" class="