Fix resonance example

Author: rpep

examples/atomistic/dynamic_spectrum/deal.py renamed to examples/atomistic/dynamic_spectrum/plot_results.py

@@ -14,7 +14,7 @@ def deal_plot(Hy=0):
     #N = 8000
 
     dt = ts[1] - ts[0]
-    print 'dt=',dt
+    print('dt=',dt)
 
     freq = np.fft.fftshift(np.fft.fftfreq(N, dt))
 
@@ -33,7 +33,7 @@ def deal_plot(Hy=0):
     ix = (b*c-a*d)/(a*a+b*b)
 
     ind = np.argmax(ix)
-    print ind, freq[ind]*2*np.pi
+    print(ind, freq[ind]*2*np.pi)
     w_ix = np.array([freq*2*np.pi, ix])
     #np.savetxt('w_ix.txt',np.transpose(w_ix))
 

examples/atomistic/dynamic_spectrum/dyn.py renamed to examples/atomistic/dynamic_spectrum/run_simulation.py

@@ -9,7 +9,7 @@
 
 global_nx = 174
 global_ny = 150
-global_mesh = CuboidMesh(nx=global_nx,ny=global_ny, pbc='2d')
+global_mesh = CuboidMesh(nx=global_nx,ny=global_ny, periodicity=[True, True, False])
 
 def init_m(pos):
     x,y,z = pos
@@ -36,15 +36,13 @@ def init_m(pos):
         return m2
 
 def sinc_fun(t):
-
     w = 0.1
-
     return np.sinc(w*t)
 
 def relax_system(mesh, Hy=0):
 
     sim=Sim(mesh,name='relax')
-    sim.set_options(rtol=1e-10,atol=1e-12)
+    sim.driver.set_tols(rtol=1e-10,atol=1e-12)
     sim.driver.alpha = 0.5
     sim.driver.gamma = 1.0
     sim.mu_s = 1.0
@@ -66,15 +64,15 @@ def relax_system(mesh, Hy=0):
     zeeman = Zeeman([0,Hy,2e-2],name='H')
     sim.add(zeeman)
 
-    sim.relax(dt=2.0, stopping_dmdt=1e-8, max_steps=10000, save_m_steps=100, save_vtk_steps=50)
+    sim.relax(dt=2.0, stopping_dmdt=1e-7, max_steps=10000, save_m_steps=100, save_vtk_steps=50)
 
     np.save('m0.npy',sim.spin)
 
 def excite_system(mesh, Hy=0):
 
     sim=Sim(mesh,name='dyn')
 
-    sim.set_options(rtol=1e-10,atol=1e-12)
+    sim.driver.set_tols(rtol=1e-10,atol=1e-12)
     sim.driver.alpha = 0.04
     sim.driver.gamma = 1.0
     sim.mu_s = 1.0
@@ -99,13 +97,10 @@ def excite_system(mesh, Hy=0):
     dt = 5
     steps = 2001
     for i in range(steps):
-
         sim.run_until(i*dt)
-        #sim.save_m()
-
-        #print 'sim t=%g'%(i*dt)
+        sim.save_m()
+        print("step {}/{}".format(i, steps))
 
 if __name__=='__main__':
-
 	relax_system(global_mesh)
 	excite_system(global_mesh)