h=input('Vstavi razdaljo med viroma (/lambda): ')
p=input('Vstavi fazno razliko v stopinjah: ')

print 'Pocakaj, gre zelo pocasi...'
import numpy as np
import matplotlib.pyplot as plt

def f(t):	#smerni diagram dveh izotropnih virov |F(theta)|
	return np.fabs(np.cos(h*np.pi*np.cos(t)+np.radians(p)/2))

t=np.arange(0.0, 2*np.pi, 0.001)

plt.plot(np.sin(t), np.cos(t), 'g--')
plt.plot(0.75*np.sin(t), 0.75*np.cos(t), 'g--')
plt.plot(0.5*np.sin(t), 0.5*np.cos(t), 'g--')
plt.plot(0.25*np.sin(t), 0.25*np.cos(t), 'g--')

plt.plot(f(t)*np.sin(t), f(t)*np.cos(t),'b-')

plt.axis([-1.1, 1.1, -1.1, 1.1])
plt.axes().set_aspect('equal')

plt.show()