n = shape(nodes)[0] #Number of nodes
bc_lines = elements[1] # Boundary condition is (bc) lines
triangles = elements[2]
potential = potential_2d('well', nodes, v0 = 2)
write_solver_input(glo_tag +'.msh', parameter = potential, dimension = 2, \
bc_type = 'Dir', \
sol_type = 'Stationary', eq = 'Schro', \
analysis_param = ['y', 'y', 4, 20, 15, 15, 1], \
bc_filename = glo_tag+'.bc')
k = read_solver_input(glo_tag +'.msh')
v, solution = schroedinger(glo_tag +'.msh')
triangles = substract_1(triangles)
triangles = triangles[:, 1:]
#k, solution = schroedinger(glo_tag +'.msh')
#z = zeros((k[0].shape[0], 3))
#print k[0].shape, z[:, 0:1].shape
#z[:, 0:2] = k[0]
#nodes = z
#triangles = k[1]
print v
write_vtk(glo_tag + '_1.vtk', 'this shit', '', nodes, triangles, \
['SCALARS', ['solution'], [solution]])
from math import pi
import numpy as np
from PostPro import plot_1d
from time import time
t0 = time()
nodes, elements = Mesh1D('simple',0,2*pi)
write_msh('lineSimple.msh', nodes, elements)
pot = Potential1D('well', nodes, V0=2)
write_solver_input('lineSimple.msh', parameter = pot, bc_type = 'Bloch',\
analysis_param = ['y', 'y', 4, 4, 50, 50, 1])
v, d = schroedinger('lineSimple.msh')
tf = time()
z = np.zeros((nodes.size, 3))
z[:, 0] = nodes
nodes = z
#write_vtk('test.vtk', 'thi shit', '', nodes, elements, \
# ['SCALARS', ['Pot', 'Psi'], [pot, d]])
plot_1d('', nodes = nodes,elements = elements, parameter = pot, bc_type = 'Dir', \
sol = d)
print 'time elapsed: \n', tf-t0, 'seconds'
##Nodes=Nodes[:,0]
##WriteMSH('lineSimple.msh',Nodes,Elems)
##Nodes,Elems=ReadVTK('test.vtk')
##Schroedinger('lineSimple.msh')
triangles = elements[2]
potential = potential_2d("well", nodes, v0=0)
write_solver_input(
glo_tag + ".msh",
parameter=potential,
dimension=2,
bc_type="Dir",
sol_type="Stationary",
eq="Schro",
analysis_param=["y", "y", 20, 20, 15, 15, 1],
bc_filename=glo_tag + ".bc",
)
k = read_solver_input(glo_tag + ".msh")
v, solution = schroedinger(glo_tag + ".msh")
triangles = substract_1(triangles)
triangles = triangles[:, 1:]
# k, solution = schroedinger(glo_tag +'.msh')
# z = zeros((k[0].shape[0], 3))
# print k[0].shape, z[:, 0:1].shape
# z[:, 0:2] = k[0]
# nodes = z
# triangles = k[1]
print v
write_vtk(glo_tag + "_dir.vtk", "this shit", "", nodes, triangles, ["SCALARS", ["solution"], [solution]])
print solution.shape
sol = test_sol(nodes, pi, 3 * pi)
write_vtk(glo_tag + "_sol.vtk", "this shit", "", nodes, triangles, ["SCALARS", ["sol"], [sol]])
