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