"""This script runs a simulation of the dynamical system
reproducing the work of Van Assche."""

from __future__ import division
import mpmath as mp

from simulation import *
from plotting import *
#from plottingwithtitles import *
from mpl_toolkits.mplot3d import Axes3D

ion()
close('all')

data = {# System constraint: decimal precision
        'dec': 100,
        # Parameters in the mathematical formulation
        't': mp.mpf('0.01'),
        'zeta': mp.mpf('1'),
        'N': mp.mpf('1'),
        'n_end': mp.mpf('50'),
        # Optional: specify other initial conditions
        'S0': 10,
        'R0': 10,
        # Plotting pretties
        'col1': 'blue',
        'mar1': '.',
        'col2': 'mintcream',
        'mar2': '<',
        }

mp.mp.dps = data['dec']

simulation = [gen_fp_timedep, 
              #gen_ic_standard,
              ic_declare,
              gen_standard,
              plot_3d,
              plot_2dprojection,
              plot_gens,
              gen_energy_eval,
              gen_energy_error,
              gen_energy_diff
]

for func in simulation:
    data = func(data)