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]])