From a816d919769d79e51fb2e5df3edc48a869929eb5 Mon Sep 17 00:00:00 2001
From: Matt Wala <wala1@illinois.edu>
Date: Thu, 31 Aug 2017 19:01:57 -0500
Subject: [PATCH] Add full control over expansion text labels. Closes #22.

---
 sumpy/toys.py | 22 ++++++++++++++++++----
 1 file changed, 18 insertions(+), 4 deletions(-)

diff --git a/sumpy/toys.py b/sumpy/toys.py
index 8caa657c..b32613d3 100644
--- a/sumpy/toys.py
+++ b/sumpy/toys.py
@@ -315,9 +315,14 @@ class ExpansionPotentialSource(PotentialSource):
     .. attribute:: radius
 
         Not used mathematically. Just for visualization, purely advisory.
+
+    .. attribute:: text_kwargs
+
+       Passed to matplotlib.text(). Used for customizing the expansion
+       label. Just for visualization, purely advisory.
     """
     def __init__(self, toy_ctx, center, order, coeffs, derived_from,
-            radius=None, expn_style=None):
+            radius=None, expn_style=None, text_kwargs=None):
         super(ExpansionPotentialSource, self).__init__(toy_ctx)
         self.center = np.asarray(center)
         self.order = order
@@ -326,6 +331,7 @@ class ExpansionPotentialSource(PotentialSource):
         self.derived_from = derived_from
         self.radius = radius
         self.expn_style = expn_style
+        self.text_kwargs = text_kwargs
 
 
 class MultipoleExpansion(ExpansionPotentialSource):
@@ -554,10 +560,18 @@ class SchematicVisitor(object):
             else:
                 raise ValueError("unknown expn_style: %s" % self.expn_style)
 
+        text_kwargs = dict(
+                x=psource.center[0],
+                y=psource.center[1],
+                text=type(psource).__name__[0],
+                verticalalignment="center",
+                horizontalalignment="center")
+
+        if psource.text_kwargs is not None:
+            text_kwargs.update(psource.text_kwargs)
+
         import matplotlib.pyplot as plt
-        plt.gca().text(psource.center[0], psource.center[1],
-                type(psource).__name__[0],
-                verticalalignment='center', horizontalalignment='center')
+        plt.gca().text(**text_kwargs)
 
         if psource.derived_from is not None:
             xmin, xmax = plt.xlim()
-- 
GitLab