load_and_plot.py

import numpy as np
import matplotlib.pyplot as plt
from elastic_flow import ElasticFlow
import os

# Initialize grid
N = 300
grid = np.linspace(0, 2 * np.pi, N, False)

# Recreate all plots for the oval
folder = "triple_eight_300_0.5h^2"
curve_folder = f"{folder}/arrays/curve/"

# Get all iteration files
iteration_files = [f for f in os.listdir(curve_folder) if f.endswith('.npy')]
iterations = [int(f.split('.')[0]) for f in iteration_files]
iterations.sort()

for iteration in iterations:
    curve_path = f"{curve_folder}{iteration}.npy"
    curvature_path = f"{folder}/arrays/curvature/{iteration}.npy"
    normal_path = f"{folder}/arrays/normal/{iteration}.npy"
    
    # Initialize ElasticFlow with the loaded curve
    elastic_flow = ElasticFlow(grid, [curve_path, curvature_path, normal_path], folder)
    
    # Plot and save the curve without axis
    elastic_flow.save(iteration, show_axis=False)
    
    print(f"Iteration {iteration} loaded, plotted, and saved successfully.")

print("All curves loaded, plotted, and saved successfully.")