chiark / gitweb /
helixish: angles are unconventionally represented in "Railway problem"
[moebius3.git] / visual
1 #!/usr/bin/python
2
3 from __future__ import print_function
4
5 import signal
6 signal.signal(signal.SIGINT, signal.SIG_DFL)
7
8 from visual import *
9 from bezier import BezierSegment
10 import numpy as np
11 import sys
12
13 scene.width = 1200
14 scene.height = 1200
15 scene.fov = 1E-3
16
17 from moebius import *
18 from moedebug import *
19
20 nv = 40
21 nw = 40
22
23 class Visdebug(BaseVisdebug):
24   def curve(vd, fn, color=color.yellow):
25     # fn(t) => 3-tuple, 3-list, or similar, for t \elem [0,1]
26     ps = [ fn(t) for t in np.linspace(0,1,11) ]
27     #dbg('Visdebug', ps)
28     curve(pos = ps, color=color)
29   def arrow(vd, p, d, **kw):
30     print('arrow %s' % repr(kw), file=sys.stderr)
31     bodge_color = kw.get('color', (1,0,0))
32     try: del kw['color']
33     except KeyError: pass
34     vd.line(p, p+d, color=bodge_color, **kw)
35     ring(pos=p, axis=d, radius=0.25, thickness=0.005, color=bodge_color, **kw)
36   def crashing(vd, msg):
37     print("CRASHING - VISDEBUG", msg, file=sys.stderr)
38     rate(1E-8)
39
40 dbg_file(sys.stderr)
41 vdbg_enable(Visdebug())
42
43 m = Moebius(nv, nw)
44
45 vs = range(0, nv+1)
46 ws = range(0, nw+1)
47
48 #ts = np.linspace(0, 1, 40)
49
50 #thetas = np.linspace(0, tau, 40)
51 #c0 = curve( color=color.red, pos = [ m.edge.point(th) for th in thetas ] )
52 #c1 = curve( color=color.blue, pos = [ m.midline.point(th) for th in thetas ] )
53
54 for v in vs:
55   c2 = curve( color=color.blue, pos = [ m.point(v,w) for w in ws ] )
56
57 #for w in ws:
58 #  c3 = curve( color=color.blue, pos = [ m.point(v,w) for v in vs ] )
59
60
61 #for v in vs:
62 #  c4 = curve( color=color.blue,
63 #              pos = [ m.point_offset(v,w, 0.1) for w in ws ] )
64
65 #for w in ws:
66 #  c3 = curve( color=(1,0,1), pos = [ m.point_offset(v,w, -0.1) for v in vs ] )
67