From: Ian Jackson <ijackson@chiark.greenend.org.uk>
Date: Sun, 17 Dec 2017 11:33:43 +0000 (+0000)
Subject: visdebug: support caller-specified colour tuples
X-Git-Url: https://www.chiark.greenend.org.uk/ucgi/~ian/git?a=commitdiff_plain;h=d9363ecd4d27dfaaabdea4e82a2a7f52ae67939c;p=moebius3.git

visdebug: support caller-specified colour tuples

Signed-off-by: Ian Jackson <ijackson@chiark.greenend.org.uk>
---

diff --git a/moedebug.py b/moedebug.py
index f9f8561..8c80c00 100644
--- a/moedebug.py
+++ b/moedebug.py
@@ -19,10 +19,10 @@ def dbg(*args):
     f.flush()
 
 class BaseVisdebug():
-  def line(vd, p0, p1):
-    vd.curve(lambda t: p0 + t * (p1-p0))
-  def arrow(vd, p, d):
-    vd.line(p, p+d)
+  def line(vd, p0, p1, **kw):
+    vd.curve(lambda t: p0 + t * (p1-p0), **kw)
+  def arrow(vd, p, d, **kw):
+    vd.line(p, p+d, **kw)
 
 class NullVisdebug():
   def curve(*a, **kw): pass
@@ -34,8 +34,8 @@ class MatrixVisdebug(BaseVisdebug):
   def __init__(m, base, matrix):
     m._b = base
     m._matrix = matrix
-  def curve(m, fn):
-    m._b.curve(lambda t: moenp.augmatmultiply(m._matrix, fn(t)))
+  def curve(m, fn, **kw):
+    m._b.curve(lambda t: moenp.augmatmultiply(m._matrix, fn(t)), **kw)
   def crashing(m, *a, **kw): m._b.crashing(*a, **kw)
 
 _nullvis = NullVisdebug()
diff --git a/visual b/visual
index dcb57e1..667ce6a 100755
--- a/visual
+++ b/visual
@@ -17,11 +17,11 @@ nv = 40
 nw = 40
 
 class Visdebug(BaseVisdebug):
-  def curve(vd, fn):
+  def curve(vd, fn, colour=color.yellow):
     # fn(t) => 3-tuple, 3-list, or similar, for t \elem [0,1]
     ps = [ fn(t) for t in np.linspace(0,1,11) ]
     #dbg('Visdebug', ps)
-    curve(pos = ps, color=color.yellow)
+    curve(pos = ps, color=colour)
   def crashing(vd, msg):
     print("CRASHING - VISDEBUG", msg, file=sys.stderr)
     rate(1E-8)