import numpy as np
import math
import matplotlib.pyplot as plt

nx = 25 # количество элементов АФАР по оси x
ny = 25 # количество элементов АФАР по оси y
wavelength = (3e8)/(2.4e9) # длина волны для несущей частоты 2.4 ГГц

d = wavelength / 2 
k = 2*np.pi/wavelength
thetaMx = 10*(np.pi/180)
thetaMy = 0*(np.pi/180)

thetas_x = np.arange(-90,91)*(np.pi/180)
for idx, item in enumerate(thetas_x):
    if item == 0:
        thetas_x[idx] = 0.001

thetas_y = np.arange(-90,91)*(np.pi/180)
for idx, item in enumerate(thetas_y):
    if item == 0:
        thetas_y[idx] = 0.001

F_xy = np.zeros((len(thetas_x),len(thetas_y)))
for idx1, theta1 in enumerate(thetas_x):
    for idx2, theta2 in enumerate(thetas_y):
        num1 = np.sin(n*k*d*(np.sin(theta1) - np.sin(thetaMx))/2) 
        denum1 = np.sin(k*d*(np.sin(theta1) - np.sin(thetaMx))/2)
        num2 = np.sin(n*k*d*(np.sin(theta2) - np.sin(thetaMy))/2)
        denum2 = np.sin(k*d*(np.sin(theta2) - np.sin(thetaMy))/2)
        F_xy[idx1, idx2] = (1 /(nx*ny))*(num1/denum1)*(num2/denum2)
        if math.isnan(F_xy[idx1, idx2]) == True:
            F_xy[idx1, idx2] = 1


fig, ax = plt.subplots(nrows=1, ncols=2, constrained_layout=True, figsize=(15, 5), dpi=150)

ax[0].plot(thetas_x*(180/np.pi), F_xy[0,:])
ax[0].set_title("Азимут")
ax[0].set_xlabel('Градусы')
ax[0].set_ylabel('F(θ)')
ax[0].grid()

ax[1].plot(thetas_x*(180/np.pi), F_xy[:,0])
ax[1].set_title("Элевация")
ax[1].set_xlabel('Градусы')
ax[1].set_ylabel('F(θ)')
ax[1].grid()

fig.savefig('RA')