#!/usr/bin/python

import numpy as np
import matplotlib.pyplot as plt 

fig = plt.figure()


# initial values
omega0 = 3.0
omegaP = 1.0
G = 1.0



omegamax=10.0
n = 100

t = np.linspace(0,omegamax,num=n)
omegaRe = np.zeros(n)
omegaIm = np.zeros(n)
z = np.zeros(n)

def REoscilator(omega): 
      return omegaP**2*(omega0**2-omega**2)/((omega0**2-omega**2)*(omega0**2-omega**2)+omega**2*G**2)
      
def IMoscilator(omega):
      return omegaP**2*G*omega/((omega0**2-omega**2)*(omega0**2-omega**2)+omega**2*G**2)

omegaRe = [REoscilator(t[l]) for l in range(n)]
omegaIm = [IMoscilator(t[l]) for l in range(n)]

plt.plot(t,omegaRe,'orange')
plt.plot(t,omegaIm,'blue')


plt.show()