From 69b83350f5f8e3eb718508b2bdd0a3d76a70d75c Mon Sep 17 00:00:00 2001 From: Andreas Kloeckner <inform@tiker.net> Date: Sat, 9 Jan 2016 15:46:53 -0600 Subject: [PATCH] Visualizer interface sanity --- meshmode/discretization/visualization.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/meshmode/discretization/visualization.py b/meshmode/discretization/visualization.py index 5e3772a..d68ec0c 100644 --- a/meshmode/discretization/visualization.py +++ b/meshmode/discretization/visualization.py @@ -77,10 +77,10 @@ class Visualizer(object): .. automethod:: write_vtk_file """ - def __init__(self, discr, vis_discr, connection): - self.discr = discr - self.vis_discr = vis_discr + def __init__(self, connection): self.connection = connection + self.discr = connection.from_discr + self.vis_discr = connection.to_discr def _resample_and_get(self, queue, vec): from pytools.obj_array import with_object_array_or_scalar @@ -240,9 +240,8 @@ def make_visualizer(queue, discr, vis_order): real_dtype=discr.real_dtype) from meshmode.discretization.connection import \ make_same_mesh_connection - cnx = make_same_mesh_connection(vis_discr, discr) - return Visualizer(discr, vis_discr, cnx) + return Visualizer(make_same_mesh_connection(vis_discr, discr)) # }}} -- GitLab