k_coords, energy = my_solver.solve_bloch_Brillouin(simu, eq)
fig = plt.figure()
ax = fig.add_subplot(111)
plt.title("Dispersion relations for Square lattice")
k_div = k_coords.shape[0]/3.0
f = open('result.gz','w')
a = 1
print max(energy[:,-1])
for i in range(energy.shape[1]):
energy[:,i] = sqrt(around(energy[:,i],10))*a/(2*pi)
savetxt(f, around(energy[:,i],10),comments='energy profile '+str(i))
ax.plot(energy[:,i])
l = plt.axvline(x=k_div)
l = plt.axvline(x=k_div*2+1)
l = plt.axvline(x= k_coords.shape[0])
plt.ylim(0, max(energy[:,-1]))
plt.xlim(0, k_coords.shape[0])
ax.set_xticks((0,k_div,k_div*2+1, k_coords.shape[0]))
ax.set_xticklabels(('$\Gamma$','$\chi$','$M$','$\Gamma$'),fontsize = 20)
ax.set_ylabel(r'Frequency $ \frac{\omega a}{2\pi c} $', fontsize = 20)
#plt.savefig("img/r_01a_inHomogeneus_material_true.svg")
plt.show()
f.close
pl = open(filename + '.pkl','wb')
pickle.dump([energy, k_div], pl)
pl.close()
inform('Iḿ done runing')
print 'Finished.'
filename = 'unit_cell_1-02_two_reg'
write_solver_input(filename +'.msh',dimension = 2, bc_type = 'Bloch', \
parameter = [], eq = 'EM', sol_type = 'Stationary',analysis_param \
= ['y', 'y', 8, 8, 15, 15, 1], bc_filename = filename +'.bc')
simu = Simulation()
simu.read_solver_input(filename +'.msh')
simu.domain.read_mesh_file(filename +'.msh',simu)
inter = Interpreter(vectorial = True)
eq = inter.build_EM_bloch_eq(simu)
print simu.domain.regions[1]
my_solver = Solver()
k_mesh, energy = my_solver.solve_bloch(simu, eq)
from numpy import zeros, sqrt, around
nodes = zeros((k_mesh[0].shape[0],3))
nodes[:,0:2] = k_mesh[0]
triangles = k_mesh[1]
for i in range(energy.shape[1]):
energy[:,i] = sqrt(around(energy[:,i],10))
write_vtk('Bloch_periodic_r-a_1-02_two_reg_true'+str(i)+'.vtk','this shit','POLYDATA', nodes,\
triangles, ['SCALARS', ['sol'], [energy[:,i]]])
from inform import inform
inform('done runing')
print 'Finished.'