Update visualization.py

visualization.py

@@ -1,84 +1,98 @@
 import matplotlib.pyplot as plt
+from mpl_toolkits.mplot3d import Axes3D
+import numpy as np
 
-def visualize_results(simulator, length, width, thickness, stress, deformation):
+def visualize_results(simulation_type, length, width, thickness, stress, deformation):
     """
-    Generates 2D visualizations for simulation results.
+    Generate 2D visualization of stress and deformation.
+    """
+    x = np.linspace(0, length, 100)
+    stress_distribution = stress * (1 - (x / length))
+    deformation_distribution = deformation * (x / length)
 
-    Parameters:
-        simulator (str): Name of the simulator (e.g., 'Python-Based Solver').
-        length (float): Length of the object.
-        width (float): Width of the object.
-        thickness (float): Thickness of the object.
-        stress (float): Stress value.
-        deformation (float): Deformation value.
+    fig, ax1 = plt.subplots()
 
-    Returns:
-        str: Path to the 2D visualization image.
-    """
-    # Generate 2D bar chart
-    fig, ax = plt.subplots(figsize=(6, 4))
-    ax.bar(["Stress", "Deformation"], [stress, deformation], color=["red", "blue"])
-    ax.set_title(f"Simulation Results ({simulator})")
-    ax.set_ylabel("Magnitude")
-    ax.grid(True, linestyle="--", alpha=0.6)
-    
-    # Save the chart
-    output_path = "results_2d.png"
+    ax1.set_xlabel("Length (mm)")
+    ax1.set_ylabel("Stress (MPa)", color="tab:red")
+    ax1.plot(x, stress_distribution, color="tab:red", label="Stress")
+    ax1.tick_params(axis="y", labelcolor="tab:red")
+
+    ax2 = ax1.twinx()
+    ax2.set_ylabel("Deformation (mm)", color="tab:blue")
+    ax2.plot(x, deformation_distribution, color="tab:blue", label="Deformation")
+    ax2.tick_params(axis="y", labelcolor="tab:blue")
+
+    plt.title(f"{simulation_type} Results")
+    plt.legend(loc="upper right")
+    output_path = "2d_visualization.png"
     plt.savefig(output_path)
-    plt.close(fig)
+    plt.close()
 
-    return output_path, None
+    return output_path, {"stress": stress_distribution, "deformation": deformation_distribution}
 
-def visualize_end_product(simulation_type, length, width, thickness, deformation):
+def visualize_end_product(use_case, length, width, thickness, deformation):
+    """
+    Generate an image representation of the end product.
     """
-    Visualize the end product based on the parameters.
+    fig, ax = plt.subplots()
+    rect = plt.Rectangle((0, 0), length, width, edgecolor="black", facecolor="lightgrey")
+    ax.add_patch(rect)
+    ax.set_xlim(-10, length + 10)
+    ax.set_ylim(-10, width + 10)
+    ax.set_title(f"Final Product ({use_case})")
+    ax.set_xlabel("Length (mm)")
+    ax.set_ylabel("Width (mm)")
+    output_path = "end_product_visualization.png"
+    plt.savefig(output_path)
+    plt.close()
 
-    Parameters:
-        simulation_type (str): 'plate' or 'beam'.
-        length (float): Length of the product (mm).
-        width (float): Width of the product (mm).
-        thickness (float): Thickness of the product (mm).
-        deformation (float): Deformation (mm).
+    return output_path
 
+def create_3d_visualization(length, width, deformation):
+    """
+    Generate a 3D visualization of deformation based on length and width.
+    Parameters:
+        length (float): The length of the simulated object.
+        width (float): The width of the simulated object.
+        deformation (float): The deformation value to visualize.
     Returns:
-        str: Path to the saved image of the visualization.
+        str: Path to the saved 3D visualization image.
     """
-    fig, ax = plt.subplots(figsize=(6, 4))
-    
-    if simulation_type == "plate":
-        # Draw the plate
-        rect = plt.Rectangle((0, 0), length, width, color="lightblue", alpha=0.8)
-        ax.add_patch(rect)
-        # Add deformation as text
-        ax.text(
-            length / 2, width / 2,
-            f"Deformation: {deformation:.2f} mm",
-            color="red", fontsize=10, ha="center"
-        )
-    elif simulation_type == "beam":
-        # Draw the beam
-        rect = plt.Rectangle((0, 0), length, thickness, color="lightgreen", alpha=0.8)
-        ax.add_patch(rect)
-        # Add deformation as text
-        ax.text(
-            length / 2, thickness / 2,
-            f"Deflection: {deformation:.2f} mm",
-            color="red", fontsize=10, ha="center"
-        )
-    else:
-        raise ValueError("Invalid simulation type.")
-
-    # Add dimensions
-    ax.set_xlim(-10, length + 10)
-    ax.set_ylim(-10, max(width, thickness) + 10)
-    ax.set_title(f"Visualization of {simulation_type.capitalize()} End Product")
+    # Create grid data
+    x = np.linspace(0, length, 100)
+    y = np.linspace(0, width, 100)
+    X, Y = np.meshgrid(x, y)
+    Z = np.sin((X / length) * np.pi) * np.cos((Y / width) * np.pi) * deformation
+
+    # Create the 3D plot
+    fig = plt.figure()
+    ax = fig.add_subplot(111, projection='3d')
+    ax.plot_surface(X, Y, Z, cmap='viridis')
+    ax.set_title("3D Deformation Visualization")
     ax.set_xlabel("Length (mm)")
-    ax.set_ylabel("Width/Thickness (mm)")
-    ax.grid(True, linestyle="--", alpha=0.6)
+    ax.set_ylabel("Width (mm)")
+    ax.set_zlabel("Deformation (mm)")
 
-    # Save the visualization
-    output_path = f"{simulation_type}_end_product.png"
+    # Save the plot to a file
+    output_path = "3d_visualization.png"
     plt.savefig(output_path)
-    plt.close(fig)
+    plt.close()
 
     return output_path
+
+if __name__ == "__main__":
+    # Example usage
+    length = 200
+    width = 100
+    thickness = 10
+    stress = 150
+    deformation = 2.5
+
+    # Generate 2D visualization
+    visualize_results("Example Simulation", length, width, thickness, stress, deformation)
+
+    # Generate end-product visualization
+    visualize_end_product("Example Use Case", length, width, thickness, deformation)
+
+    # Generate 3D visualization
+    create_3d_visualization(length, width, deformation)