Update utils/diagram_generator.py

utils/diagram_generator.py

@@ -142,4 +142,371 @@ class DiagramGenerator:
             svg += self._draw_sequence_network(150, 100, "Z1", "#059669")
             
             # Negative sequence network  
-            svg += f'<text x="450" y="80" text-anchor="middle" class="
+            svg += f'<text x="450" y="80" text-anchor="middle" class="title">Negative Sequence</text>\n'
+            svg += self._draw_sequence_network(450, 100, "Z2", "#dc2626")
+            
+            # Zero sequence network
+            svg += f'<text x="750" y="80" text-anchor="middle" class="title">Zero Sequence</text>\n'
+            svg += self._draw_sequence_network(750, 100, "Z0", "#7c3aed")
+            
+            # Connection diagram
+            svg += f'<text x="450" y="320" text-anchor="middle" class="title">Network Connection</text>\n'
+            
+            # Series connection for L-G fault
+            svg += f'<line x1="400" y1="350" x2="400" y2="450" class="line"/>\n'
+            svg += f'<line x1="400" y1="370" x2="500" y2="370" class="line"/>\n'
+            svg += f'<line x1="400" y1="410" x2="500" y2="410" class="line"/>\n'
+            svg += f'<line x1="400" y1="450" x2="500" y2="450" class="line"/>\n'
+            
+            # Impedance boxes
+            svg += f'<rect x="480" y="360" width="40" height="20" class="component"/>\n'
+            svg += f'<text x="500" y="375" text-anchor="middle" class="text">Z1</text>\n'
+            
+            svg += f'<rect x="480" y="400" width="40" height="20" class="component"/>\n'
+            svg += f'<text x="500" y="415" text-anchor="middle" class="text">Z2</text>\n'
+            
+            svg += f'<rect x="480" y="440" width="40" height="20" class="component"/>\n'
+            svg += f'<text x="500" y="455" text-anchor="middle" class="text">Z0</text>\n'
+            
+            # Voltage source
+            svg += f'<circle cx="370" cy="350" r="15" class="component"/>\n'
+            svg += f'<text x="370" y="355" text-anchor="middle" class="text">Ea</text>\n'
+            
+            # Fault point
+            svg += f'<use href="#fault-symbol" transform="translate(530,410)"/>\n'
+            svg += f'<text x="530" y="440" text-anchor="middle" class="text">Fault</text>\n'
+            
+            # Current equation
+            svg += f'<text x="450" y="520" text-anchor="middle" class="title">Fault Current Calculation</text>\n'
+            svg += f'<text x="450" y="550" text-anchor="middle" class="text">I_fault = 3 × Ea / (Z1 + Z2 + Z0)</text>\n'
+            
+        elif fault_type == "line_to_line":
+            # L-L fault diagram
+            svg += self._draw_ll_fault_diagram()
+            
+        elif fault_type == "three_phase":
+            # 3-phase fault diagram
+            svg += self._draw_three_phase_fault_diagram()
+        
+        svg += self.create_svg_footer()
+        return svg
+
+    def _draw_sequence_network(self, x: int, y: int, impedance: str, color: str) -> str:
+        """Draw a sequence network"""
+        svg = ""
+        
+        # Voltage source
+        svg += f'<circle cx="{x-50}" cy="{y+50}" r="15" stroke="{color}" stroke-width="2" fill="white"/>\n'
+        svg += f'<text x="{x-50}" y="{y+55}" text-anchor="middle" class="text">E</text>\n'
+        
+        # Impedance
+        svg += f'<rect x="{x-10}" y="{y+40}" width="40" height="20" stroke="{color}" stroke-width="2" fill="white"/>\n'
+        svg += f'<text x="{x+10}" y="{y+55}" text-anchor="middle" class="text">{impedance}</text>\n'
+        
+        # Connecting lines
+        svg += f'<line x1="{x-35}" y1="{y+50}" x2="{x-10}" y2="{y+50}" stroke="{color}" stroke-width="2"/>\n'
+        svg += f'<line x1="{x+30}" y1="{y+50}" x2="{x+60}" y2="{y+50}" stroke="{color}" stroke-width="2"/>\n'
+        
+        # Ground/neutral
+        svg += f'<line x1="{x+60}" y1="{y+50}" x2="{x+60}" y2="{y+80}" stroke="{color}" stroke-width="2"/>\n'
+        svg += f'<line x1="{x+50}" y1="{y+80}" x2="{x+70}" y2="{y+80}" stroke="{color}" stroke-width="2"/>\n'
+        svg += f'<line x1="{x+52}" y1="{y+85}" x2="{x+68}" y2="{y+85}" stroke="{color}" stroke-width="2"/>\n'
+        svg += f'<line x1="{x+54}" y1="{y+90}" x2="{x+66}" y2="{y+90}" stroke="{color}" stroke-width="2"/>\n'
+        
+        return svg
+
+    def _draw_ll_fault_diagram(self) -> str:
+        """Draw line-to-line fault diagram"""
+        svg = ""
+        
+        # Positive and negative sequence in parallel
+        svg += f'<text x="300" y="80" text-anchor="middle" class="title">L-L Fault: Z1 and Z2 in Parallel</text>\n'
+        
+        # Parallel connection
+        svg += f'<circle cx="200" cy="150" r="15" class="component"/>\n'
+        svg += f'<text x="200" y="155" text-anchor="middle" class="text">Ea</text>\n'
+        
+        # Upper branch (Z1)
+        svg += f'<line x1="215" y1="140" x2="300" y2="140" class="line"/>\n'
+        svg += f'<rect x="300" y="130" width="40" height="20" class="component"/>\n'
+        svg += f'<text x="320" y="145" text-anchor="middle" class="text">Z1</text>\n'
+        svg += f'<line x1="340" y1="140" x2="400" y2="140" class="line"/>\n'
+        
+        # Lower branch (Z2)
+        svg += f'<line x1="215" y1="160" x2="300" y2="160" class="line"/>\n'
+        svg += f'<rect x="300" y="150" width="40" height="20" class="component"/>\n'
+        svg += f'<text x="320" y="165" text-anchor="middle" class="text">Z2</text>\n'
+        svg += f'<line x1="340" y1="160" x2="400" y2="160" class="line"/>\n'
+        
+        # Connection
+        svg += f'<line x1="400" y1="140" x2="400" y2="160" class="line"/>\n'
+        
+        # Fault
+        svg += f'<use href="#fault-symbol" transform="translate(420,150)"/>\n'
+        
+        # Current equation
+        svg += f'<text x="300" y="250" text-anchor="middle" class="title">I_fault = √3 × Ea / (Z1 + Z2)</text>\n'
+        
+        return svg
+
+    def _draw_three_phase_fault_diagram(self) -> str:
+        """Draw three-phase fault diagram"""
+        svg = ""
+        
+        svg += f'<text x="400" y="80" text-anchor="middle" class="title">Three-Phase Fault: Positive Sequence Only</text>\n'
+        
+        # Simple circuit
+        svg += f'<circle cx="250" cy="150" r="15" class="component"/>\n'
+        svg += f'<text x="250" y="155" text-anchor="middle" class="text">Ea</text>\n'
+        
+        svg += f'<line x1="265" y1="150" x2="350" y2="150" class="line"/>\n'
+        svg += f'<rect x="350" y="140" width="40" height="20" class="component"/>\n'
+        svg += f'<text x="370" y="155" text-anchor="middle" class="text">Z1</text>\n'
+        
+        svg += f'<line x1="390" y1="150" x2="450" y2="150" class="line"/>\n'
+        svg += f'<use href="#fault-symbol" transform="translate(470,150)"/>\n'
+        
+        # Current equation
+        svg += f'<text x="400" y="220" text-anchor="middle" class="title">I_fault = Ea / Z1</text>\n'
+        
+        return svg
+
+    def generate_protection_coordination_diagram(self) -> str:
+        """Generate time-current coordination curves"""
+        svg = self.create_svg_header(800, 600)
+        
+        # Title
+        svg += f'<text x="400" y="30" text-anchor="middle" class="title">Protection Coordination Curves</text>\n'
+        
+        # Axes
+        svg += f'<line x1="100" y1="500" x2="700" y2="500" class="line marker-end="url(#arrowhead)"/>\n'
+        svg += f'<line x1="100" y1="500" x2="100" y2="100" class="line marker-end="url(#arrowhead)"/>\n'
+        
+        # Axis labels
+        svg += f'<text x="400" y="530" text-anchor="middle" class="text">Current (A)</text>\n'
+        svg += f'<text x="50" y="300" text-anchor="middle" class="text" transform="rotate(-90 50 300)">Time (s)</text>\n'
+        
+        # Grid lines
+        for i in range(2, 7):
+            x = 100 + i * 100
+            svg += f'<line x1="{x}" y1="100" x2="{x}" y2="500" stroke="#e5e7eb" stroke-width="1"/>\n'
+            svg += f'<text x="{x}" y="520" text-anchor="middle" class="text">{10**(i-1)}</text>\n'
+        
+        for i in range(1, 5):
+            y = 500 - i * 80
+            svg += f'<line x1="100" y1="{y}" x2="700" y2="{y}" stroke="#e5e7eb" stroke-width="1"/>\n'
+            svg += f'<text x="85" y="{y+5}" text-anchor="end" class="text">{10**(i-1)}</text>\n'
+        
+        # Relay curves
+        # Primary relay (closer to load)
+        svg += self._draw_relay_curve(200, "Primary Relay", "#059669", "inverse")
+        
+        # Backup relay 
+        svg += self._draw_relay_curve(300, "Backup Relay", "#dc2626", "very_inverse")
+        
+        # Fuse curve
+        svg += self._draw_relay_curve(150, "Fuse", "#7c3aed", "fuse")
+        
+        # Legend
+        svg += f'<rect x="550" y="120" width="180" height="120" stroke="#374151" stroke-width="1" fill="white"/>\n'
+        svg += f'<text x="640" y="140" text-anchor="middle" class="text">Legend</text>\n'
+        
+        svg += f'<line x1="560" y1="160" x2="590" y2="160" stroke="#7c3aed" stroke-width="3"/>\n'
+        svg += f'<text x="600" y="165" class="text">Fuse</text>\n'
+        
+        svg += f'<line x1="560" y1="180" x2="590" y2="180" stroke="#059669" stroke-width="3"/>\n'
+        svg += f'<text x="600" y="185" class="text">Primary Relay</text>\n'
+        
+        svg += f'<line x1="560" y1="200" x2="590" y2="200" stroke="#dc2626" stroke-width="3"/>\n'
+        svg += f'<text x="600" y="205" class="text">Backup Relay</text>\n'
+        
+        svg += f'<text x="640" y="225" text-anchor="middle" class="text">Coordination Interval: 0.3s</text>\n'
+        
+        svg += self.create_svg_footer()
+        return svg
+
+    def _draw_relay_curve(self, x_offset: int, label: str, color: str, curve_type: str) -> str:
+        """Draw a time-current curve for a relay"""
+        svg = ""
+        
+        points = []
+        
+        if curve_type == "inverse":
+            # Standard inverse curve
+            for i in range(50):
+                current = 10 ** (i / 10.0)
+                time = 0.14 / ((current/100) ** 0.02 - 1)
+                if time > 0 and time < 1000:
+                    x = 100 + (i * 12)
+                    y = 500 - (time * 40)
+                    if 100 <= x <= 700 and 100 <= y <= 500:
+                        points.append(f"{x},{y}")
+        
+        elif curve_type == "very_inverse":
+            # Very inverse curve (higher up)
+            for i in range(50):
+                current = 10 ** (i / 10.0)
+                time = 13.5 / ((current/100) ** 1.0 - 1)
+                if time > 0 and time < 1000:
+                    x = 100 + (i * 12)
+                    y = 500 - (time * 20)
+                    if 100 <= x <= 700 and 100 <= y <= 500:
+                        points.append(f"{x},{y}")
+        
+        elif curve_type == "fuse":
+            # Fuse curve (faster, lower)
+            for i in range(40):
+                current = 10 ** (i / 8.0)
+                time = 0.01 / ((current/50) ** 2.0 - 1) if current > 50 else 1000
+                if time > 0 and time < 1000:
+                    x = 100 + (i * 15)
+                    y = 500 - (time * 60)
+                    if 100 <= x <= 700 and 100 <= y <= 500:
+                        points.append(f"{x},{y}")
+        
+        if points:
+            path = f'<polyline points="{" ".join(points)}" stroke="{color}" stroke-width="3" fill="none"/>\n'
+            svg += path
+            
+            # Label
+            if len(points) > 10:
+                mid_point = points[len(points)//2].split(',')
+                x, y = int(mid_point[0]), int(mid_point[1])
+                svg += f'<text x="{x+10}" y="{y-5}" class="text" fill="{color}">{label}</text>\n'
+        
+        return svg
+
+    def generate_phasor_diagram(self, fault_type: str = "balanced") -> str:
+        """Generate phasor diagrams for different fault conditions"""
+        svg = self.create_svg_header(600, 400)
+        
+        # Title
+        svg += f'<text x="300" y="30" text-anchor="middle" class="title">Phasor Diagram - {fault_type.title()} Conditions</text>\n'
+        
+        center_x, center_y = 300, 200
+        radius = 80
+        
+        if fault_type == "balanced":
+            # Three balanced phasors 120° apart
+            angles = [0, 120, 240]
+            colors = ["#dc2626", "#059669", "#2563eb"]
+            labels = ["Va", "Vb", "Vc"]
+            
+            for i, (angle, color, label) in enumerate(zip(angles, colors, labels)):
+                x_end = center_x + radius * cos(radians(angle))
+                y_end = center_y - radius * sin(radians(angle))
+                
+                svg += f'<line x1="{center_x}" y1="{center_y}" x2="{x_end}" y2="{y_end}" '
+                svg += f'stroke="{color}" stroke-width="3" marker-end="url(#arrowhead)"/>\n'
+                
+                # Label
+                label_x = center_x + (radius + 20) * cos(radians(angle))
+                label_y = center_y - (radius + 20) * sin(radians(angle))
+                svg += f'<text x="{label_x}" y="{label_y}" text-anchor="middle" class="text" fill="{color}">{label}</text>\n'
+        
+        elif fault_type == "unbalanced":
+            # Unbalanced phasors showing fault condition
+            # Phase A (affected by fault) - reduced magnitude
+            svg += f'<line x1="{center_x}" y1="{center_y}" x2="{center_x + 40}" y2="{center_y}" '
+            svg += f'stroke="#dc2626" stroke-width="3" marker-end="url(#arrowhead)"/>\n'
+            svg += f'<text x="{center_x + 60}" y="{center_y}" class="text" fill="#dc2626">Va (faulted)</text>\n'
+            
+            # Phase B - normal
+            x_b = center_x + radius * cos(radians(120))
+            y_b = center_y - radius * sin(radians(120))
+            svg += f'<line x1="{center_x}" y1="{center_y}" x2="{x_b}" y2="{y_b}" '
+            svg += f'stroke="#059669" stroke-width="3" marker-end="url(#arrowhead)"/>\n'
+            svg += f'<text x="{x_b-20}" y="{y_b-10}" class="text" fill="#059669">Vb</text>\n'
+            
+            # Phase C - normal  
+            x_c = center_x + radius * cos(radians(240))
+            y_c = center_y - radius * sin(radians(240))
+            svg += f'<line x1="{center_x}" y1="{center_y}" x2="{x_c}" y2="{y_c}" '
+            svg += f'stroke="#2563eb" stroke-width="3" marker-end="url(#arrowhead)"/>\n'
+            svg += f'<text x="{x_c-20}" y="{y_c+20}" class="text" fill="#2563eb">Vc</text>\n'
+        
+        # Center point
+        svg += f'<circle cx="{center_x}" cy="{center_y}" r="3" fill="#374151"/>\n'
+        
+        # Reference circle
+        svg += f'<circle cx="{center_x}" cy="{center_y}" r="{radius}" stroke="#e5e7eb" stroke-width="1" fill="none" stroke-dasharray="5,5"/>\n'
+        
+        svg += self.create_svg_footer()
+        return svg
+
+    def generate_impedance_diagram(self) -> str:
+        """Generate impedance diagram for distance protection"""
+        svg = self.create_svg_header(600, 500)
+        
+        # Title
+        svg += f'<text x="300" y="30" text-anchor="middle" class="title">Distance Protection - R-X Diagram</text>\n'
+        
+        center_x, center_y = 300, 250
+        
+        # Axes
+        svg += f'<line x1="100" y1="{center_y}" x2="500" y2="{center_y}" class="line" marker-end="url(#arrowhead)"/>\n'
+        svg += f'<line x1="{center_x}" y1="400" x2="{center_x}" y2="100" class="line" marker-end="url(#arrowhead)"/>\n'
+        
+        # Axis labels
+        svg += f'<text x="520" y="{center_y+5}" class="text">R (Ω)</text>\n'
+        svg += f'<text x="{center_x-10}" y="90" class="text">X (Ω)</text>\n'
+        
+        # Mho circle (Zone 1)
+        svg += f'<circle cx="{center_x+50}" cy="{center_y}" r="80" stroke="#059669" stroke-width="2" fill="none"/>\n'
+        svg += f'<text x="{center_x+90}" y="{center_y-90}" class="text" fill="#059669">Zone 1 (Mho)</text>\n'
+        
+        # Zone 2 (larger circle)
+        svg += f'<circle cx="{center_x+70}" cy="{center_y}" r="120" stroke="#dc2626" stroke-width="2" fill="none" stroke-dasharray="5,5"/>\n'
+        svg += f'<text x="{center_x+140}" y="{center_y-130}" class="text" fill="#dc2626">Zone 2</text>\n'
+        
+        # Load impedance area
+        svg += f'<path d="M{center_x+20},{center_y-20} Q{center_x+80},{center_y-40} {center_x+120},{center_y-10}" '
+        svg += f'stroke="#7c3aed" stroke-width="2" fill="none"/>\n'
+        svg += f'<text x="{center_x+70}" y="{center_y-50}" class="text" fill="#7c3aed">Load Region</text>\n'
+        
+        # Grid marks
+        for i in range(1, 5):
+            x = center_x + i * 40
+            svg += f'<line x1="{x}" y1="{center_y-5}" x2="{x}" y2="{center_y+5}" stroke="#374151" stroke-width="1"/>\n'
+            svg += f'<text x="{x}" y="{center_y+20}" text-anchor="middle" class="text">{i*5}</text>\n'
+            
+            y = center_y - i * 40
+            svg += f'<line x1="{center_x-5}" y1="{y}" x2="{center_x+5}" y2="{y}" stroke="#374151" stroke-width="1"/>\n'
+            svg += f'<text x="{center_x-20}" y="{y+5}" text-anchor="middle" class="text">{i*5}</text>\n'
+        
+        svg += self.create_svg_footer()
+        return svg
+
+# Helper functions for calculations
+def cos(angle_deg):
+    import math
+    return math.cos(math.radians(angle_deg))
+
+def sin(angle_deg):
+    import math
+    return math.sin(math.radians(angle_deg))
+
+def radians(angle_deg):
+    import math
+    return math.radians(angle_deg)
+
+# Example usage
+if __name__ == "__main__":
+    generator = DiagramGenerator()
+    
+    # Test diagram generation
+    diagrams = {
+        "single_line": generator.generate_single_line_diagram({}),
+        "fault_analysis": generator.generate_fault_analysis_diagram("line_to_ground"),
+        "protection_coordination": generator.generate_protection_coordination_diagram(),
+        "phasor": generator.generate_phasor_diagram("balanced"),
+        "impedance": generator.generate_impedance_diagram()
+    }
+    
+    # Save diagrams
+    for name, svg_content in diagrams.items():
+        with open(f"{name}_diagram.svg", "w") as f:
+            f.write(svg_content)
+        print(f"Generated {name} diagram")
+    
+    print("All diagrams generated successfully!")