time dependence changed with commit 4137ad2 about rydiqule HOT 2 CLOSED

from numpy import *
from matplotlib.pyplot import *
import rydiqule as rq


## straight from Intro to Rydiqule section 3
def turn_on_field(t):
    if t < 1:
        return 0.0
    else:
        return 1.0

basis_size = 4
sensor_time = rq.Sensor(basis_size)

probe    = {"states":(0,1), "detuning": 2, "rabi_frequency": 3, "label":"probe"}
coupling = {"states":(1,2), "detuning": 2, "rabi_frequency": 5, "label": "coupling"}
rf       = {"states":(2,3), "detuning": 0, "rabi_frequency": 7, "label": "rf",
            "time_dependence":turn_on_field}

sensor_time.add_couplings(probe, coupling, rf)

gamma = np.zeros((basis_size, basis_size))
gamma[1:,0] = 0.1
sensor_time.set_gamma_matrix(gamma)

end_time = 10 #microseconds
num_pts = 100

solution_time = rq.solve_time(sensor_time, end_time, num_pts)
absorption_time = rq.get_rho_ij(solution_time.rho,0,1).imag


## variation 1: late turn on

def turn_on_field_late(t):
    if t < 11:
        return 0.0
    else:
        return 1.0

sensor_time1 = rq.Sensor(basis_size)

rf1 = {"states":(2,3), "detuning": 0, "rabi_frequency": 7, "label": "rf",
       "time_dependence":turn_on_field_late}

sensor_time1.add_couplings(probe, coupling, rf1)
sensor_time1.set_gamma_matrix(gamma)
solution_time1 = rq.solve_time(sensor_time1, end_time, num_pts)
absorption_time1 = rq.get_rho_ij(solution_time1.rho,0,1).imag

## variation 2: \Omega_{RF} = 0

sensor_time2 = rq.Sensor(basis_size)

rf2 = {"states":(2,3), "detuning": 0, "rabi_frequency": 0, "label": "rf",
       "time_dependence":turn_on_field}

sensor_time2.add_couplings(probe, coupling, rf2)
sensor_time2.set_gamma_matrix(gamma)
solution_time2 = rq.solve_time(sensor_time2, end_time, num_pts)
absorption_time2 = rq.get_rho_ij(solution_time2.rho,0,1).imag

## generate comparison figure

clf()
plot(solution_time.t, absorption_time , label='turn on at 1 µs')
plot(solution_time.t, absorption_time1, label='turn on at 11 µs')
plot(solution_time.t, absorption_time2, label='$\Omega_{RF} = 0$')
title ("Comparison of time dependent calcs")
xlabel("time [µs]")
ylabel("signal")
grid('on')
legend()
show()

sensor_ss   = rq.Sensor(basis_size)
sensor_ss.add_couplings(probe, coupling)
sensor_ss.set_gamma_matrix(gamma)
solution_ss = rq.solve_steady_state(sensor_ss)