# I. Giannakis, A. Giannopoulos and N. Davidson,
# "Incorporating dispersive electrical properties in FDTD GPR models
# using a general Cole-Cole dispersion function," 
# 2012 14th International Conference on Ground Penetrating Radar (GPR), 2012, pp. 232-236
import os, sys
sys.path.append(os.path.join(os.path.dirname(__file__), '..'))
from Debye_Fit import HavriliakNegami

if __name__ == "__main__":
    # set Havrilak-Negami function with initial parameters
    setup = HavriliakNegami(f_min=1e4, f_max=1e11,
                            alpha=0.3, beta=1,
                            e_inf=3.4, de=2.7, tau_0=.8e-10,
                            sigma=0.45e-3, mu=1, mu_sigma=0,
                            material_name="dry_sand", f_n=100,
                            plot=True, save=False,
                            optimizer_options={'swarmsize':30,
                                               'maxiter':100,
                                               'omega':0.5,
                                               'phip':1.4,
                                               'phig':1.4,
                                               'minstep':1e-8,
                                               'minfun':1e-8,
                                               'seed': 111,
                                               'pflag': True})
    ### Dry Sand in case of 3, 5
    # and automatically set number of Debye poles (-1)
    for number_of_debye_poles in [3, 5, -1]:
        setup.number_of_debye_poles = number_of_debye_poles
        setup.run()

    ### Moist sand
    # set Havrilak-Negami function parameters
    setup.material_name="moist_sand"
    setup.alpha = 0.25
    setup.beta = 1
    setup.e_inf = 5.6
    setup.de = 3.3
    setup.tau_0 = 1.1e-10,
    setup.sigma = 2e-3
    # calculate for different number of Debye poles
    for number_of_debye_poles in [3, 5, -1]:
        setup.number_of_debye_poles = number_of_debye_poles
        setup.run()