def main():
cl_ctx = cl.create_some_context()
queue = cl.CommandQueue(cl_ctx)
from meshmode.mesh.generation import ( # noqa
generate_icosphere, generate_icosahedron, generate_torus)
#mesh = generate_icosphere(1, order=order)
mesh = generate_icosahedron(1, order=order)
#mesh = generate_torus(3, 1, order=order)
from meshmode.discretization import Discretization
from meshmode.discretization.poly_element import \
PolynomialWarpAndBlendGroupFactory
discr = Discretization(cl_ctx, mesh,
PolynomialWarpAndBlendGroupFactory(order))
from meshmode.discretization.visualization import make_visualizer
vis = make_visualizer(queue, discr, order)
vis.write_vtk_file("geometry.vtu", [
("f", discr.nodes()[0]),
])
from meshmode.discretization.visualization import \
write_nodal_adjacency_vtk_file
write_nodal_adjacency_vtk_file("adjacency.vtu", mesh)
def test_vtk_overwrite(ctx_getter):
pytest.importorskip("pyvisfile")
def _try_write_vtk(writer, obj):
import os
from meshmode import FileExistsError
filename = "test_vtk_overwrite.vtu"
if os.path.exists(filename):
os.remove(filename)
writer(filename, [])
with pytest.raises(FileExistsError):
writer(filename, [])
writer(filename, [], overwrite=True)
if os.path.exists(filename):
os.remove(filename)
ctx = ctx_getter()
queue = cl.CommandQueue(ctx)
target_order = 7
from meshmode.mesh.generation import generate_torus
mesh = generate_torus(10.0, 2.0, order=target_order)
from meshmode.discretization import Discretization
from meshmode.discretization.poly_element import \
InterpolatoryQuadratureSimplexGroupFactory
discr = Discretization(
queue.context, mesh,
InterpolatoryQuadratureSimplexGroupFactory(target_order))
from meshmode.discretization.visualization import make_visualizer
from meshmode.discretization.visualization import \
write_nodal_adjacency_vtk_file
from meshmode.mesh.visualization import write_vertex_vtk_file
vis = make_visualizer(queue, discr, 1)
_try_write_vtk(vis.write_vtk_file, discr)
_try_write_vtk(
lambda x, y, **kwargs: write_vertex_vtk_file(discr.mesh, x, **kwargs),
discr.mesh)
_try_write_vtk(
lambda x, y, **kwargs: write_nodal_adjacency_vtk_file(
x, discr.mesh, **kwargs), discr.mesh)
def test_vtk_overwrite(ctx_getter):
pytest.importorskip("pyvisfile")
def _try_write_vtk(writer, obj):
import os
from meshmode import FileExistsError
filename = "test_vtk_overwrite.vtu"
if os.path.exists(filename):
os.remove(filename)
writer(filename, [])
with pytest.raises(FileExistsError):
writer(filename, [])
writer(filename, [], overwrite=True)
if os.path.exists(filename):
os.remove(filename)
ctx = ctx_getter()
queue = cl.CommandQueue(ctx)
target_order = 7
from meshmode.mesh.generation import generate_torus
mesh = generate_torus(10.0, 2.0, order=target_order)
from meshmode.discretization import Discretization
from meshmode.discretization.poly_element import \
InterpolatoryQuadratureSimplexGroupFactory
discr = Discretization(
queue.context, mesh,
InterpolatoryQuadratureSimplexGroupFactory(target_order))
from meshmode.discretization.visualization import make_visualizer
from meshmode.discretization.visualization import \
write_nodal_adjacency_vtk_file
from meshmode.mesh.visualization import write_vertex_vtk_file
vis = make_visualizer(queue, discr, 1)
_try_write_vtk(vis.write_vtk_file, discr)
_try_write_vtk(lambda x, y, **kwargs:
write_vertex_vtk_file(discr.mesh, x, **kwargs), discr.mesh)
_try_write_vtk(lambda x, y, **kwargs:
write_nodal_adjacency_vtk_file(x, discr.mesh, **kwargs), discr.mesh)
def test_vtk_overwrite(actx_factory):
pytest.importorskip("pyvisfile")
def _try_write_vtk(writer, obj):
import os
filename = "vtk_overwrite_temp.vtu"
if os.path.exists(filename):
os.remove(filename)
writer(filename, [])
with pytest.raises(FileExistsError):
writer(filename, [])
writer(filename, [], overwrite=True)
if os.path.exists(filename):
os.remove(filename)
actx = actx_factory()
target_order = 7
mesh = mgen.generate_torus(10.0, 2.0, order=target_order)
from meshmode.discretization import Discretization
discr = Discretization(
actx, mesh, InterpolatoryQuadratureSimplexGroupFactory(target_order))
from meshmode.discretization.visualization import make_visualizer
from meshmode.discretization.visualization import \
write_nodal_adjacency_vtk_file
from meshmode.mesh.visualization import write_vertex_vtk_file
vis = make_visualizer(actx, discr, 1)
_try_write_vtk(vis.write_vtk_file, discr)
_try_write_vtk(
lambda x, y, **kwargs: write_vertex_vtk_file(discr.mesh, x, **kwargs),
discr.mesh)
_try_write_vtk(
lambda x, y, **kwargs: write_nodal_adjacency_vtk_file(
x, discr.mesh, **kwargs), discr.mesh)
def refine_and_generate_chart_function(mesh, filename, function):
from time import clock
cl_ctx = cl.create_some_context()
queue = cl.CommandQueue(cl_ctx)
print("NELEMENTS: ", mesh.nelements)
#print mesh
for i in range(len(mesh.groups[0].vertex_indices[0])):
for k in range(len(mesh.vertices)):
print(mesh.vertices[k, i])
#check_nodal_adj_against_geometry(mesh);
r = Refiner(mesh)
#random.seed(0)
#times = 3
num_elements = []
time_t = []
#nelements = mesh.nelements
while True:
print("NELS:", mesh.nelements)
#flags = get_corner_flags(mesh)
flags = get_function_flags(mesh, function)
nels = 0
for i in flags:
if i:
nels += 1
if nels == 0:
break
print("LKJASLFKJALKASF:", nels)
num_elements.append(nels)
#flags = get_corner_flags(mesh)
beg = clock()
mesh = r.refine(flags)
end = clock()
time_taken = end - beg
time_t.append(time_taken)
#if nelements == mesh.nelements:
#break
#nelements = mesh.nelements
#from meshmode.mesh.visualization import draw_2d_mesh
#draw_2d_mesh(mesh, True, True, True, fill=None)
#import matplotlib.pyplot as pt
#pt.show()
#poss_flags = np.zeros(len(mesh.groups[0].vertex_indices))
#for i in range(0, len(flags)):
# poss_flags[i] = flags[i]
#for i in range(len(flags), len(poss_flags)):
# poss_flags[i] = 1
import matplotlib.pyplot as pt
pt.xlabel('Number of elements being refined')
pt.ylabel('Time taken')
pt.plot(num_elements, time_t, "o")
pt.savefig(filename, format='pdf')
pt.clf()
print('DONE REFINING')
'''
flags = np.zeros(len(mesh.groups[0].vertex_indices))
flags[0] = 1
flags[1] = 1
mesh = r.refine(flags)
flags = np.zeros(len(mesh.groups[0].vertex_indices))
flags[0] = 1
flags[1] = 1
flags[2] = 1
mesh = r.refine(flags)
'''
#check_nodal_adj_against_geometry(mesh)
#r.print_rays(70)
#r.print_rays(117)
#r.print_hanging_elements(10)
#r.print_hanging_elements(117)
#r.print_hanging_elements(757)
#from meshmode.mesh.visualization import draw_2d_mesh
#draw_2d_mesh(mesh, False, False, False, fill=None)
#import matplotlib.pyplot as pt
#pt.show()
from meshmode.discretization import Discretization
from meshmode.discretization.poly_element import \
PolynomialWarpAndBlendGroupFactory
discr = Discretization(
cl_ctx, mesh, PolynomialWarpAndBlendGroupFactory(order))
from meshmode.discretization.visualization import make_visualizer
vis = make_visualizer(queue, discr, order)
remove_if_exists("connectivity2.vtu")
remove_if_exists("geometry2.vtu")
vis.write_vtk_file("geometry2.vtu", [
("f", discr.nodes()[0]),
])
from meshmode.discretization.visualization import \
write_nodal_adjacency_vtk_file
write_nodal_adjacency_vtk_file("connectivity2.vtu",
mesh)
def refine_and_generate_chart_function(mesh, filename, function):
from time import clock
cl_ctx = cl.create_some_context()
queue = cl.CommandQueue(cl_ctx)
print("NELEMENTS: ", mesh.nelements)
#print mesh
for i in range(len(mesh.groups[0].vertex_indices[0])):
for k in range(len(mesh.vertices)):
print(mesh.vertices[k, i])
#check_nodal_adj_against_geometry(mesh);
r = Refiner(mesh)
#random.seed(0)
#times = 3
num_elements = []
time_t = []
#nelements = mesh.nelements
while True:
print("NELS:", mesh.nelements)
#flags = get_corner_flags(mesh)
flags = get_function_flags(mesh, function)
nels = 0
for i in flags:
if i:
nels += 1
if nels == 0:
break
print("LKJASLFKJALKASF:", nels)
num_elements.append(nels)
#flags = get_corner_flags(mesh)
beg = clock()
mesh = r.refine(flags)
end = clock()
time_taken = end - beg
time_t.append(time_taken)
#if nelements == mesh.nelements:
#break
#nelements = mesh.nelements
#from meshmode.mesh.visualization import draw_2d_mesh
#draw_2d_mesh(mesh, True, True, True, fill=None)
#import matplotlib.pyplot as pt
#pt.show()
#poss_flags = np.zeros(len(mesh.groups[0].vertex_indices))
#for i in range(0, len(flags)):
# poss_flags[i] = flags[i]
#for i in range(len(flags), len(poss_flags)):
# poss_flags[i] = 1
import matplotlib.pyplot as pt
pt.xlabel('Number of elements being refined')
pt.ylabel('Time taken')
pt.plot(num_elements, time_t, "o")
pt.savefig(filename, format='pdf')
pt.clf()
print('DONE REFINING')
'''
flags = np.zeros(len(mesh.groups[0].vertex_indices))
flags[0] = 1
flags[1] = 1
mesh = r.refine(flags)
flags = np.zeros(len(mesh.groups[0].vertex_indices))
flags[0] = 1
flags[1] = 1
flags[2] = 1
mesh = r.refine(flags)
'''
#check_nodal_adj_against_geometry(mesh)
#r.print_rays(70)
#r.print_rays(117)
#r.print_hanging_elements(10)
#r.print_hanging_elements(117)
#r.print_hanging_elements(757)
#from meshmode.mesh.visualization import draw_2d_mesh
#draw_2d_mesh(mesh, False, False, False, fill=None)
#import matplotlib.pyplot as pt
#pt.show()
from meshmode.discretization import Discretization
from meshmode.discretization.poly_element import \
PolynomialWarpAndBlendGroupFactory
discr = Discretization(cl_ctx, mesh,
PolynomialWarpAndBlendGroupFactory(order))
from meshmode.discretization.visualization import make_visualizer
vis = make_visualizer(queue, discr, order)
remove_if_exists("connectivity2.vtu")
remove_if_exists("geometry2.vtu")
vis.write_vtk_file("geometry2.vtu", [
("f", discr.nodes()[0]),
])
from meshmode.discretization.visualization import \
write_nodal_adjacency_vtk_file
write_nodal_adjacency_vtk_file("connectivity2.vtu", mesh)