#include "solid_angle.h"
#include <math.h>
#include <stdio.h>
#include "refractive_index.h"

/* Table of some of the tabulated values of the refractive index of water as a
 * function of wavelength and temperature. In all cases, I just used the values
 * for standard atmospheric pressure and assume this corresponds approximately
 * to a density of 1000 kg/m^3.
 *
 * See Table 7 in https://aip.scitation.org/doi/pdf/10.1063/1.555859. */

struct refractive_index_results {
    double p;
    double T;
    double wavelength;
    double n;
} refractive_index_results[] = {
    {1000.0, 0 ,  361.05, 1.39468},
    {1000.0, 10,  361.05, 1.39439},
    {1000.0, 20,  361.05, 1.39353},
    {1000.0, 30,  361.05, 1.39224},
    {1000.0, 0 ,  404.41, 1.34431},
    {1000.0, 10,  404.41, 1.34404},
    {1000.0, 20,  404.41, 1.34329},
    {1000.0, 30,  404.41, 1.34218},
    {1000.0, 0 ,  589.00, 1.33447},
    {1000.0, 10,  589.00, 1.33422},
    {1000.0, 20,  589.00, 1.33350},
    {1000.0, 30,  589.00, 1.33243},
    {1000.0, 0 ,  632.80, 1.33321},
    {1000.0, 10,  632.80, 1.33296},
    {1000.0, 20,  632.80, 1.33224},
    {1000.0, 30,  632.80, 1.33118},
    {1000.0, 0 , 1013.98, 1.32626},
    {1000.0, 10, 1013.98, 1.32604},
    {1000.0, 20, 1013.98, 1.32537},
    {1000.0, 30, 1013.98, 1.32437},
    {1000.0, 0 , 2325.42, 1.27663},
    {1000.0, 10, 2325.42, 1.27663},
    {1000.0, 20, 2325.42, 1.27627},
    {1000.0, 30, 2325.42, 1.27563},
};

/* Table of the values of solid angle for various values of r0/r and L/r.
 *
 * See Table 1 in http://www.umich.edu/~ners312/CourseLibrary/SolidAngleOfADiskOffAxis.pdf. */

struct solid_angle_results {
    double L;
    double r0;
    double omega;
} solid_angle_results[] = {
    {0.5,0.0,3.4732594},
    {0.5,0.2,3.4184435},
    {0.5,0.4,3.2435434},
    {0.5,0.6,2.9185178},
    {0.5,0.8,2.4122535},
    {0.5,1.0,1.7687239},
    {0.5,1.2,1.1661307},
    {0.5,1.4,0.7428889},
    {0.5,1.6,0.4841273},
    {0.5,1.8,0.3287007},
    {0.5,2.0,0.2324189},
    {1.0,0.0,1.8403024},
    {1.0,0.2,1.8070933},
    {1.0,0.4,1.7089486},
    {1.0,0.6,1.5517370},
    {1.0,0.8,1.3488367},
    {1.0,1.0,1.1226876},
    {1.0,1.2,0.9003572},
    {1.0,1.4,0.7039130},
    {1.0,1.6,0.5436956},
    {1.0,1.8,0.4195415},
    {1.0,2.0,0.3257993},
    {1.5,0.0,1.0552591},
    {1.5,0.2,1.0405177},
    {1.5,0.4,0.9975504},
    {1.5,0.6,0.9301028},
    {1.5,0.8,0.8441578},
    {1.5,1.0,0.7472299},
    {1.5,1.2,0.6472056},
    {1.5,1.4,0.5509617},
    {1.5,1.6,0.4632819},
    {1.5,1.8,0.3866757},
    {1.5,2.0,0.3217142},
    {2.0,0.0,0.6633335},
    {2.0,0.2,0.6566352},
    {2.0,0.4,0.6370508},
    {2.0,0.6,0.6060694},
    {2.0,0.8,0.5659755},
    {2.0,1.0,0.5195359},
    {2.0,1.2,0.4696858},
    {2.0,1.4,0.4191714},
    {2.0,1.6,0.3702014},
    {2.0,1.8,0.3243908},
    {2.0,2.0,0.282707}
};

int isclose(double a, double b, double rel_tol, double abs_tol)
{
    /* Returns 1 if a and b are "close". This algorithm is taken from Python's
     * math.isclose() function.
     *
     * See https://www.python.org/dev/peps/pep-0485/. */
    return fabs(a-b) <= fmax(rel_tol*fmax(fabs(a),fabs(b)),abs_tol);
}

int test_refractive_index(char *err)
{
    /* Tests the get_index() function. */
    int i;
    double n;
    struct refractive_index_results result;

    for (i = 0; i < sizeof(refractive_index_results)/sizeof(struct refractive_index_results); i++) {
        result = refractive_index_results[i];
        n = get_index(result.p, result.wavelength, result.T);

        if (!isclose(n, result.n, 1e-4, 0)) {
            sprintf(err, "n = %.5f, but expected %.5f", n, result.n);
            return 1;
        }
    }

    return 0;
}

int test_solid_angle(char *err)
{
    /* Tests the get_solid_angle() function. */
    int i;
    double pmt[3] = {0,0,0};
    double pos[3] = {0,0,1};
    double n[3] = {0,0,1};
    double r = 1.0;
    double solid_angle;

    solid_angle = get_solid_angle(pos,pmt,n,r);

    if (!isclose(solid_angle, 2*M_PI*(1-1/sqrt(2)), 1e-9, 0)) {
        sprintf(err, "solid angle = %.5f, but expected %.5f", solid_angle, 2*M_PI*(1-1/sqrt(2)));
        return 1;
    }

    for (i = 0; i < sizeof(solid_angle_results)/sizeof(struct solid_angle_results); i++) {
        pos[0] = solid_angle_results[i].r0*r;
        pos[2] = solid_angle_results[i].L*r;

        solid_angle = get_solid_angle(pos,pmt,n,r);

        if (!isclose(solid_angle, solid_angle_results[i].omega, 1e-4, 0)) {
            sprintf(err, "solid angle = %.5f, but expected %.5f", solid_angle, solid_angle_results[i].omega);
            return 1;
        }
    }

    return 0;
}

int main(int argc, char **argv)
{
    char err[256];
    int retval = 0;

    if (!test_solid_angle(err)) {
        printf("[\033[92mok\033[0m] test_solid_angle\n");
    } else {
        printf("[\033[91mfail\033[0m] test_solid_angle: %s\n", err);
        retval = 1;
    }

    if (!test_refractive_index(err)) {
        printf("[\033[92mok\033[0m] test_refractive_index\n");
    } else {
        printf("[\033[91mfail\033[0m] test_refractive_index: %s\n", err);
        retval = 1;
    }

    return retval;
}