helixish: remember to transform back from findcurve basis (!)

[moebius3.git] / moedebug.py

diff --git a/moedebug.py b/moedebug.py
index 97e8635f16d90e9f02d780b5bbec52fc1d998a6e..f5e039eb758699b73229ae6cb15f97624519cee6 100644 (file)
--- a/moedebug.py
+++ b/moedebug.py
@@ -1,6 +1,8 @@
 
 from __future__ import print_function
 
+import moenp
+
 _files = []
 
 def dbg_file(f):
@@ -16,12 +18,40 @@ def dbg(*args):
       print('// D ', l, file=f)
     f.flush()
 
+class BaseVisdebug():
+  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)
+  def basis(vd, basis, hue=(1,1,1)):
+    for ax in range(0,3):
+      vd.arrow(basis[0:3,3], basis[0:3,ax],
+               color= tuple([ c * (1.0 - ax*0.25) for c in hue ]))
+
 class NullVisdebug():
   def curve(*a, **kw): pass
   def line(*a, **kw): pass
+  def circle(*a, **kw): pass
   def arrow(*a, **kw): pass
+  def basis(*a, **kw): pass
   def crashing(*a, **kw): pass
 
+class MatrixVisdebug(BaseVisdebug):
+  def __init__(m, base, matrix):
+    m._b = base
+    m._matrix = matrix
+  def curve(m, fn, **kw):
+    m._b.curve(lambda t: moenp.augmatmultiply(m._matrix, fn(t)), **kw)
+  def circle(m, c, axis, **kw):
+    m._b.circle(moenp.augmatmultiply(m._matrix, c),
+                moenp.augmatmultiply(m._matrix, axis, 0),
+                **kw)
+  def arrow(m, p, d, **kw):
+    m._b.arrow(moenp.augmatmultiply(m._matrix, p),
+               moenp.augmatmultiply(m._matrix, d, 0),
+               **kw)
+  def crashing(m, *a, **kw): m._b.crashing(*a, **kw)
+
 _nullvis = NullVisdebug()
 _vis = [ _nullvis ]