# quantum mechanical description
qm = mb.qm_description(6e24, H, u, relax_sop)
mat_ar = mb.material("AR_Song", qm)
mb.material.add_to_library(mat_ar)
# Song setup
dev = mb.device("Song")
dev.add_region(mb.region("Active region (single point)", mat_ar, 0, 0))
# scenario
ic_d = mb.ic_density_const(rho_init)
ic_e = mb.ic_field_const(0.0)
ic_m = mb.ic_field_const(0.0)
sce = mb.scenario("Basic", 1, 80e-15, ic_d, ic_e, ic_m, 10000)
sce.add_record(mb.record("e", 0.0, 0.0))
sce.add_record(mb.record("d11", mb.record.density, 1, 1, 0.0, 0.0))
sce.add_record(mb.record("d22", mb.record.density, 2, 2, 0.0, 0.0))
sce.add_record(mb.record("d33", mb.record.density, 3, 3, 0.0, 0.0))
# add source
sce.add_source(
mb.sech_pulse("sech", 0.0, mb.source.hard_source, 3.5471e9, 3.8118e14,
17.248, 1.76 / 5e-15, -math.pi / 2))
# run solver
sol = mb.solver.create_instance("cpu-fdtd-3lvl-cvr-rodr", dev, sce)
print('Solver ' + sol.get_name() + ' started')
tic = time.time()
sol.run()
toc = time.time()
qm = mb.qm_desc_2lvl(1e24, 2 * math.pi * 2e14, 6.24e-11, 1.0e10, 1.0e10, -1.0)
mat_ar = mb.material("AR_Ziolkowski", qm)
mb.material.add_to_library(mat_ar)
# Ziolkowski setup
dev = mb.device("Ziolkowski")
dev.add_region(mb.region("Vacuum left", mat_vac, 0, 7.5e-6))
dev.add_region(mb.region("Active region", mat_ar, 7.5e-6, 142.5e-6))
dev.add_region(mb.region("Vacuum right", mat_vac, 142.5e-6, 150e-6))
# initial density matrix
rho_init = mb.qm_operator([1, 0])
# scenario
sce = mb.scenario("Basic", 32768, 200e-15, rho_init)
sce.add_record(mb.record("inv12", 2.5e-15))
sce.add_record(mb.record("e", 2.5e-15))
# add source
sce.add_source(
mb.sech_pulse("sech", 0.0, mb.source.hard_source, 4.2186e9, 2e14, 10,
2e14))
# run solver
sol = mb.solver.create_instance("cpu-fdtd-red-2lvl-cvr-rodr", dev, sce)
print('Solver ' + sol.get_name() + ' started')
tic = time.time()
sol.run()
toc = time.time()
print('Solver ' + sol.get_name() + ' finished in ' + str(toc - tic) + ' sec')