#!/usr/bin/env python3
import mpmath as mp
import numpy as np
Du_test = [
# // x, [Re(m), Im(m)], Qext, Qsca, test_name
[0.099, [0.75,0], 7.417859e-06, 7.417859e-06, 'a'],
[0.101, [0.75,0], 8.033538e-06, 8.033538e-06, 'b'],
[10,    [0.75,0],     2.232265, 2.232265, 'c'],
[1000,  [0.75,0],     1.997908, 1.997908, 'd'],
[100,   [1.33,-1e-5], 2.101321, 2.096594, 'e'],
[10000, [1.33,-1e-5], 2.004089, 1.723857, 'f'],
[0.055, [1.5, -1],    0.101491, 1.131687e-05, 'g'],
[0.056, [1.5, -1],   0.1033467, 1.216311e-05, 'h'],
[100,   [1.5, -1],    2.097502, 1.283697, 'i'],
[10000, [1.5, -1],    2.004368, 1.236574, 'j'],
[1,     [10,  -10],   2.532993, 2.049405, 'k'],
[100,   [10,  -10,],  2.071124, 1.836785, 'l'],
[10000, [10,  -10],   2.005914, 1.795393, 'm'],
[80, [1.05,  1],   0, 0, 'Yang'],
]
# // Dtest refractive index is m={1.05,1}, the size parameter is x = 80
n_list = [0,1,30,50,60,70,75,80,85,90,99,116,130];

def get_z_values(du_list):
    zlist = []
    for record in du_list:
        x = mp.mpf(str(record[0]))
        m = mp.mpc(str(record[1][0]), str(record[1][1]))
        z = x*m
        zlist.append(z)
    return zlist


# APPLIED OPTICS / Vol. 53, No. 31 / 1 November 2014, eq(13)
def LeRu_cutoff(z):
    x = mp.fabs(z)
    return int(x + 11 * x**(1/3) + 1)


def get_n_list(z, max_number_of_elements = 10):
    nmax = LeRu_cutoff(z)
    factor = nmax**(1/(max_number_of_elements-2))
    n_list = [int(factor**i) for i in range(max_number_of_elements-1) ]
    n_list.append(0)
    n_set = set(n_list)
    return sorted(n_set)

# Riccati-Bessel z*j_n(z)
def psi(n,z):
    return mp.sqrt( (mp.pi * z)/2 ) * mp.autoprec(mp.besselj)(n+1/2,z)


# to compare r(n,z) with Wolfram Alpha
# n=49, z=1.3-2.1i,  SphericalBesselJ[n-1,z]/SphericalBesselJ[n,z]
def r(n,z):
    return psi(n-1,z)/psi(n,z)


def D1(n,z):
    return r(n,z) - n/z


def main ():
    mp.mp.dps = 30
    output_dps = 10
    elements_of_n_list = 15
    out_filename = 'test_spec_functions_data.h'
    zlist = get_z_values(Du_test)
    output_str = ('// complex(z), n, complex(D1_n(z)), abs_err_real, abs_err_imag\n'+
            'D1_test_'+str(output_dps)+'digits = {\n')
    for z in zlist:
        n_list = get_n_list(z, elements_of_n_list)
        print(z, n_list)
        for n in n_list:
            try:
                D1nz = D1(n,z)
            except:
                pass
            print(z, n)
            output_str+=('{{'+
                  mp.nstr(z.real, output_dps)+','+
                  mp.nstr(z.imag, output_dps)+'},'+
                  str(n)+',{'+
                  mp.nstr(D1nz.real, output_dps)+','+
                  mp.nstr(D1nz.imag, output_dps)+'},'+
                  mp.nstr(mp.fabs(D1nz.real* 10**-output_dps),2)+',' +
                  mp.nstr(mp.fabs(D1nz.imag* 10**-output_dps),2)+
                  '},\n')
    output_str += '};\n'
    with open(out_filename, 'w') as out_file:
        out_file.write(output_str)
main()