import sys
import os
sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))

import numpy as np
from array_design.base_array import *
from array_design.array_gen import *

d = 0.5

ula = create_ula(12, d)
ula_str_mat = ula.steering_matrix(1, [np.pi/4, np.pi/3])
ula_rot_2D = ula.rotate([-600, -1000])
ula_rot_3D = ula.rotate([-600, -1000, -1400])

ura = create_ura((3,4), d)
ura_str_mat = ura.steering_matrix(1, np.array([[np.pi/4],[np.pi/3]]))
ura_rot = ura.rotate([-600, -1000, -1400])

uca = create_uca(12, d)
uca_str_mat = uca.steering_matrix(1, np.array([[np.pi/4],[np.pi/3]]))
uca_rot = uca.rotate([-600, -1000, -1400])

cyl_array = create_cyl_array_with_uca(12, 1, 5, 0.5)
cyl_array_str_mat = cyl_array.steering_matrix(1, np.array([[np.pi/4],[np.pi/3]]))
cyl_array_rot = cyl_array.rotate([-600, -1000, -1400])

a = 2