appsara.angular_model

import numpy as np
import matplotlib.pyplot as plt

from irfpy.cena import empirical

def main():

    fsmodel11 = empirical.AngularAS11()
    fsmodel13 = empirical.AngularAV13()

    fsmodel = fsmodel11
    #fsmodel = fsmodel13

    szalist = np.linspace(0, 90, 6, endpoint=False) + 90 / 6 / 2  # deg

    azlist = np.linspace(0, 360, 24, endpoint=False) + 360 / 24 / 2
    ellist = np.linspace(0, 90, 6, endpoint=False) + 90 / 6 / 2

    print(azlist)
    print(ellist)

    for sza in szalist:

        fs = []
        for phi in azlist:
            fs2 = []
            for theta in ellist:
                fs2.append(fsmodel.fs(sza, phi, theta))
            fs.append(fs2)

        print(sza, np.min(fs), np.max(fs))

    szalist2 = np.linspace(0, np.pi / 2, 100)

    f011 = fsmodel11._f0(szalist2)
    f013 = fsmodel13._f0(szalist2)

    plt.plot(szalist2, f011, 'b.')
    plt.plot(szalist2, f013, 'r.')

    plt.show()


if __name__ == "__main__":
    main()