예제 #1
0
simu.domain.read_regions_file(reg_filename)

inter = Interpreter()
eq = inter.build_static_EM_eq(simu)
g = eq['sol_vec']

my_solver = Solver()
fields = my_solver.solve_stationary(simu, eq)
quads = my_solver.substract_1(simu.domain.elements.quads.el_set)
quads = quads[:,1:]

field3 =  zeros((simu.domain.nodes.n,3))
field3[:,0:2] = fields
fields = field3

dir_sol = zeros(g.shape[0])
remove = eq['dir_positions']
g = my_solver.build_solution(dir_sol, g, remove, True)
print g
field2 =  zeros((simu.domain.nodes.n,3))
field2[:,0:2] = g
g_sol = field2



write_vtk(filename +'.vtk', 'MyTitle', 'UNSTRUCTURED_GRID' ,simu.domain.nodes.coords,\
               quads, ['VECTORS', ['sol'], [fields]])
write_vtk(filename+'_dirichlet' +'.vtk', 'MyTitle', 'UNSTRUCTURED_GRID' ,simu.domain.nodes.coords,\
               quads, ['VECTORS', ['sol'], [g_sol]])