Update class_diagram_generator.py

class_diagram_generator.py

@@ -199,10 +199,11 @@
 #         if 'classes' not in data:
 #             raise ValueError("Missing required field: classes")
 
+#         # Configuración del diagrama con espaciado reducido
 #         dot = graphviz.Digraph(comment='Class Diagram')
-#         dot.attr(rankdir='TB', bgcolor='white', pad='0.5')
-#         dot.attr('node', shape='box', style='filled', fillcolor='white', color='black', fontname='Arial', fontsize='10')
-#         dot.attr('edge', color='black', fontname='Arial', fontsize='9')
+#         dot.attr(rankdir='TB', bgcolor='white', pad='0.5', nodesep='0.8', ranksep='1.2')
+#         dot.attr('node', shape='plaintext', fontname='Arial', fontsize='11')
+#         dot.attr('edge', color='black', fontname='Arial', fontsize='9', minlen='1')
         
 #         classes = data.get('classes', [])
 #         relationships = data.get('relationships', [])
@@ -216,19 +217,24 @@
 #             if not class_name:
 #                 continue
             
-#             label_parts = []
+#             # Crear tabla HTML para estructura de clase
+#             html_label = '<TABLE BORDER="1" CELLBORDER="0" CELLSPACING="0" CELLPADDING="5" BGCOLOR="white">'
             
+#             # Header con nombre de clase y estereotipo
 #             if class_type == 'abstract':
-#                 label_parts.append(f"&lt;&lt;abstract&gt;&gt;\\n{class_name}")
+#                 html_label += f'<TR><TD ALIGN="CENTER"><B>&lt;&lt;abstract&gt;&gt;<BR/>{class_name}</B></TD></TR>'
 #             elif class_type == 'interface':
-#                 label_parts.append(f"&lt;&lt;interface&gt;&gt;\\n{class_name}")
+#                 html_label += f'<TR><TD ALIGN="CENTER"><B>&lt;&lt;interface&gt;&gt;<BR/>{class_name}</B></TD></TR>'
 #             elif class_type == 'enum':
-#                 label_parts.append(f"&lt;&lt;enumeration&gt;&gt;\\n{class_name}")
+#                 html_label += f'<TR><TD ALIGN="CENTER"><B>&lt;&lt;enumeration&gt;&gt;<BR/>{class_name}</B></TD></TR>'
 #             else:
-#                 label_parts.append(class_name)
+#                 html_label += f'<TR><TD ALIGN="CENTER"><B>{class_name}</B></TD></TR>'
             
+#             # Línea separadora después del nombre
+#             html_label += '<HR/>'
+            
+#             # Sección de atributos
 #             if attributes:
-#                 attr_lines = []
 #                 for attr in attributes:
 #                     visibility = attr.get('visibility', '+')
 #                     name = attr.get('name', '')
@@ -239,15 +245,18 @@
 #                     if attr_type:
 #                         line += f" : {attr_type}"
 #                     if is_static:
-#                         line += " [static]"
+#                         line = f"<U>{line}</U>"  # Subrayado para elementos estáticos
                     
-#                     attr_lines.append(line)
-                
-#                 if attr_lines:
-#                     label_parts.append("\\n".join(attr_lines))
+#                     html_label += f'<TR><TD ALIGN="LEFT">{line}</TD></TR>'
+#             else:
+#                 # Espacio vacío si no hay atributos
+#                 html_label += '<TR><TD ALIGN="LEFT"> </TD></TR>'
+            
+#             # Línea separadora entre atributos y métodos
+#             html_label += '<HR/>'
             
+#             # Sección de métodos
 #             if methods:
-#                 method_lines = []
 #                 for method in methods:
 #                     visibility = method.get('visibility', '+')
 #                     name = method.get('name', '')
@@ -267,29 +276,21 @@
 #                     line += f") : {return_type}"
                     
 #                     if is_static:
-#                         line += " [static]"
+#                         line = f"<U>{line}</U>"  # Subrayado para métodos estáticos
 #                     if is_abstract:
-#                         line += " [abstract]"
+#                         line = f"<I>{line}</I>"  # Cursiva para métodos abstractos
                     
-#                     method_lines.append(line)
-                
-#                 if method_lines:
-#                     label_parts.append("\\n".join(method_lines))
-            
-#             label = "\\n\\n".join(label_parts)
-            
-#             if class_type == 'interface':
-#                 fillcolor = '#f5f5f5'
-#                 style = 'filled,dashed'
-#             elif class_type == 'abstract':
-#                 fillcolor = '#eeeeee'
-#                 style = 'filled'
+#                     html_label += f'<TR><TD ALIGN="LEFT">{line}</TD></TR>'
 #             else:
-#                 fillcolor = 'white'
-#                 style = 'filled'
+#                 # Espacio vacío si no hay métodos
+#                 html_label += '<TR><TD ALIGN="LEFT"> </TD></TR>'
             
-#             dot.node(class_name, label, fillcolor=fillcolor, style=style)
+#             html_label += '</TABLE>'
+            
+#             # Agregar nodo con la tabla HTML
+#             dot.node(class_name, f'<{html_label}>')
         
+#         # Procesar relaciones con líneas rectas
 #         for relationship in relationships:
 #             from_class = relationship.get('from')
 #             to_class = relationship.get('to')
@@ -301,7 +302,11 @@
 #             if not from_class or not to_class:
 #                 continue
             
-#             edge_attrs = {}
+#             edge_attrs = {
+#                 'splines': 'ortho',      # Líneas estrictamente ortogonales (solo horizontales/verticales)
+#                 'concentrate': 'false',   # Evita que las líneas se combinen
+#                 'constraint': 'true'      # Mantiene la estructura jerárquica
+#             }
             
 #             if label:
 #                 edge_attrs['label'] = label
@@ -312,27 +317,35 @@
 #             if multiplicity_to:
 #                 edge_attrs['headlabel'] = multiplicity_to
             
+#             # Configurar estilos de flecha según el tipo de relación
 #             if rel_type == 'inheritance':
 #                 edge_attrs['arrowhead'] = 'empty'
+#                 edge_attrs['color'] = 'black'
 #             elif rel_type == 'composition':
 #                 edge_attrs['arrowhead'] = 'normal'
 #                 edge_attrs['arrowtail'] = 'diamond'
 #                 edge_attrs['dir'] = 'both'
+#                 edge_attrs['color'] = 'black'
 #             elif rel_type == 'aggregation':
 #                 edge_attrs['arrowhead'] = 'normal'
 #                 edge_attrs['arrowtail'] = 'odiamond'
 #                 edge_attrs['dir'] = 'both'
+#                 edge_attrs['color'] = 'black'
 #             elif rel_type == 'realization':
 #                 edge_attrs['arrowhead'] = 'empty'
 #                 edge_attrs['style'] = 'dashed'
+#                 edge_attrs['color'] = 'black'
 #             elif rel_type == 'dependency':
 #                 edge_attrs['arrowhead'] = 'normal'
 #                 edge_attrs['style'] = 'dashed'
-#             else:
+#                 edge_attrs['color'] = 'black'
+#             else:  # association
 #                 edge_attrs['arrowhead'] = 'normal'
+#                 edge_attrs['color'] = 'black'
             
 #             dot.edge(from_class, to_class, **edge_attrs)
 
+#         # Renderizar el diagrama
 #         with NamedTemporaryFile(delete=False, suffix=f'.{output_format}') as tmp:
 #             dot.render(tmp.name, format=output_format, cleanup=True)
 #             return f"{tmp.name}.{output_format}"
@@ -543,11 +556,12 @@ def generate_class_diagram(json_input: str, output_format: str) -> str:
         if 'classes' not in data:
             raise ValueError("Missing required field: classes")
 
-        # Configuración del diagrama con espaciado reducido
+        # Configuración del diagrama con líneas estrictamente rectas
         dot = graphviz.Digraph(comment='Class Diagram')
         dot.attr(rankdir='TB', bgcolor='white', pad='0.5', nodesep='0.8', ranksep='1.2')
         dot.attr('node', shape='plaintext', fontname='Arial', fontsize='11')
         dot.attr('edge', color='black', fontname='Arial', fontsize='9', minlen='1')
+        dot.attr('graph', splines='none')  # Sin curvas, solo líneas rectas
         
         classes = data.get('classes', [])
         relationships = data.get('relationships', [])
@@ -647,7 +661,7 @@ def generate_class_diagram(json_input: str, output_format: str) -> str:
                 continue
             
             edge_attrs = {
-                'splines': 'ortho',      # Líneas estrictamente ortogonales (solo horizontales/verticales)
+                'splines': 'none',        # Sin curvas, líneas completamente rectas
                 'concentrate': 'false',   # Evita que las líneas se combinen
                 'constraint': 'true'      # Mantiene la estructura jerárquica
             }