import numpy as npimport matplotlib.pyplot as pltfrom matplotlib.animation import FuncAnimation# ----------------------------# PARAMETERSN = 400dt = 0.001steps_per_frame = 1tol = 2e-7clip_u = 60.0enable_eps_continuation = Trueeps_schedule = [0.2]u0_level_1 = -2.5u0_level_2 = -2.7init_mode_amp = 0.35# Dirac settingsenable_dirac = Truesigma1 = 0.18sigma2 = 0.18n1 =5n2 = 1seed = 0rng1 = np.random.d..