chiark / gitweb /
fixed up at start of party
[trains.git] / hostside / stopgap-controller
1 #!/usr/bin/tclsh8.2
2
3 set testonly 0
4 #set testonly 1
5 set port /dev/ttya0
6
7 set ch(funcsevery) 10
8 set ch(speeddirnevery) 12
9 set ch(scale) 1
10
11 set ch(minint) 5000
12 # unset always
13 # set always 0
14 set nmrawhich 0
15 set lastptchosen xx
16
17 set polmsg(l) 908000
18 set polmsg(x) 90f802
19 set polmsg(y) 90807c
20 set pname l
21 set m {}
22 set segs {xx yy}
23 set segsasgot {xx yy}
24 set pq {} ;# unset: cdu charged and waiting
25 set speeddirn ff7f
26 #set speeddirn_fixed {speed126 2 80 0}
27 set funcs ff7f
28 # unset pointpos($point)
29 # unset segdetect($seg) ;# unset: shown D0; {}: shown D1; or: after id, D1->0
30
31 proc gui {m} {
32     puts "GUI $m"
33 }
34
35 proc gui_init {} {
36     global watchdog polarity segdetect
37     gui "M A2 0"
38     gui "M A5 0 J"
39     gui "M A6 0 J"
40     if {[info exists watchdog]} { gui "P 1" }
41     gui_polarity
42     foreach seg [array names segdetect] {
43         gui "D1 $seg"
44     }
45 }
46
47 proc debug {m} {
48     puts $m
49 }
50
51 proc tellpic_q {m} {
52     global p testonly
53     if {$testonly} return
54     set b [binary format H* $m]
55     puts -nonewline $p $b
56 }
57
58 proc tellpic {m} {
59     puts ">> $m"
60     tellpic_q $m
61 }
62
63 proc bgerror {m} {
64     if {[catch {
65         global errorInfo errorCode
66         puts stderr "$m\n$errorCode\n$errorInfo"
67         fail "bgerror $m"
68     } emsg]} {
69         exit 127
70     }
71 }
72
73 proc fail_now {} {
74     global p
75     debug "failing now"
76     fconfigure $p -blocking yes
77     gui "P 0"
78     tellpic 20
79     exit 1
80 }
81
82 proc fail {m} {
83     global watchdog p
84     catch { after cancel $watchdog; unset watchdog }
85     puts "failing $m"
86     tellpic 9801 ;# 16ms
87     after 2000 fail_now
88     fileevent $p readable {}
89 }
90
91 proc gui_polarity {} {
92     global pname
93     set 1 {}
94     switch -exact $pname {
95         l { lappend 0 X1 X3 X5 X7 X9; lappend 0 X2 X4 X6 X8 X10 }
96         x { lappend 1 X1 X3 X5 X7 X9; lappend 0 X2 X4 X6 X8 X10 }
97         y { lappend 0 X1 X3 X5 X7 X9; lappend 1 X2 X4 X6 X8 X10 }
98     }
99     foreach v {0 1} {
100         foreach seg [set $v] {
101             gui "R $v $seg"
102         }
103     }
104 }
105
106 proc polarity {newpname} {
107     global pname polmsg
108     debug "polarising $newpname"
109     if {![string compare $pname $newpname]} return
110     tellpic $polmsg($newpname)
111     set pname $newpname
112     gui_polarity
113 }
114
115 proc pt_now {how point pos xtra} {
116     set msg a0[lindex $point $pos]
117     debug "$how point $point pos=$pos msg=$msg$xtra"
118     gui "M [lindex $point 2] [expr {!$pos}]"
119     tellpic $msg
120 }
121 proc pt_must {point newpos} {
122     upvar #0 pointpos($point) pos
123     global pq
124     if {[info exists pos] && $pos == $newpos} return
125     set pos $newpos
126     if {[info exists pq]} {
127         lappend pq [list $point $pos]
128         debug "queue point $point pos=$pos l=[llength $pq]"
129         return
130     }
131     pt_now immed $point $pos {}
132     set pq {}
133 }
134 proc pt_ifthenmust {ifpoint ifposwant thenpoint thenpos} {
135     upvar #0 pointpos($ifpoint) ifpos
136     if {![info exists ifpos] || $ifpos != $ifposwant} return
137     pt_must $thenpoint $thenpos
138 }
139
140 proc pm_charged {} {
141     global pq
142     if {[llength $pq]} {
143         set v [lindex $pq 0]
144         set pq [lrange $pq 1 end]
145         pt_now nowdo [lindex $v 0] [lindex $v 1] " l=[llength $pq]"
146     } else {
147         debug "cdu-charged"
148         unset pq
149     }
150 }
151
152 proc randbyte {} {
153     global rand
154     set c [read $rand 1]; if {![string length $c]} { error "eof on rand" }
155     binary scan $c H* x
156     return $x
157 }
158
159 proc pt_maybe {point oneisright} {
160     global always lastptchosen
161     if {[info exists always]} {
162         set pos $always
163     } else {
164         if {![string compare $point $lastptchosen]} return
165         set lastptchosen $point
166         set x [randbyte]
167         set pos [expr [regexp {^[89a-f]} $x] ? 1 : 0]
168         debug "chose point $point pos=$pos (x=$x)"
169     }
170     pt_must $point $pos
171 }
172
173 proc s0 {v seg} {
174     upvar #0 segdetect($seg) segd
175     if {![info exists segd]} {
176         debug "segment $seg = already"
177     } elseif {[string length $segd]} {
178         debug "segment $seg = pending already"
179     } else {
180         debug "segment $seg = soon"
181         set segd [after 100 s0t $seg]
182     }
183 }
184 proc s0t {seg} {
185     upvar #0 segdetect($seg) segd
186     debug "segment $seg = now"
187     unset segd
188     gui "D0 $seg"
189 }
190 proc s1 {v seg} {
191     upvar #0 segdetect($seg) segd
192     if {![info exists segd]} {
193         pm_detect $v
194         debug "segment $seg ! (overwrites =)"
195     } elseif {[string length $segd]} {
196         debug "segment $seg ! (cancels =)"
197         after cancel $segd
198     } else {
199         debug "segment $seg ! already"
200         return
201     }
202     gui "D1 $seg"
203     set segd {}
204 }       
205
206 proc pm_maydetect {d seg} {
207     switch -exact $seg {
208         06 { s$d $seg X10 }
209         09 { s$d $seg X8 }
210         0a { s$d $seg X6 }
211         04 { s$d $seg X5 }
212         02 { s$d $seg X7 }
213         07 { s$d $seg X9 }
214         14 { s$d $seg A5 }
215         20 { s$d $seg A6 }
216         1a { s$d $seg A4 }
217         10 { s$d $seg A2 }
218         03 { s$d $seg X1 }
219         05 { s$d $seg X3 }
220         16 { s$d $seg A3 }
221         1c { s$d $seg A1 }
222         08 { s$d $seg X2 }
223         0b { s$d $seg X4 }
224     }
225 }
226
227 #proc pm_nodetect {seg} {
228 #    global segsasgot
229 #    if {![string compare $seg [lindex $segsasgot 1]]} {
230 #       set segsasgot [list [lindex $segsasgot 1] [lindex $segsasgot 0]]
231 #    }
232 #}
233
234 proc pm_detect {seg} {
235     global segs pname segsasgot
236     if {[string compare $seg [lindex $segsasgot 1]]} {
237         set segsasgot [list [lindex $segsasgot 1] $seg]
238     }
239     if {[lsearch -exact $segs $seg] < 0} {
240         set segs $segsasgot
241     }
242     debug "pm_detect $seg ($segsasgot) ($segs) $pname$seg"
243 #    if {[lsearch -exact {
244 #       06 09 0a 04 02 07 14 20
245 #       0b 08 1c 16
246 #       1a 10 03 05
247 #    } $seg] < 0} return
248     switch -exact $pname$seg {
249         l16 - l1c - l08 - l0b { polarity y }
250         l10 - l1a - l03 - l05 { polarity x }
251         x07 - x04 - x0a { polarity l }
252         x16 - x1c - x14 - x0b { polarity y }
253         y06 - y04 - y0a { polarity l }
254         y20 - y10 - y1a - y05 { polarity x }
255     }
256     switch -exact $seg {
257         04 - 0a { pt_must "00 01 X7" 1; pt_must "40 41 X8" 1 }
258         05 { pt_must "00 01 X7" 0 }
259         0b { pt_must "40 41 X8" 0 }
260         16 - 1c { pt_must "02 03 A5" 0 }
261         1a - 10 { pt_must "42 43 A6" 0 }
262         14 { pt_ifthenmust "02 03 A5" 1 "42 43 A6" 1 }
263         20 { pt_ifthenmust "42 43 A6" 1 "02 03 A5" 1 }
264     }
265     switch -exact [join $segs -] {
266         02-07 { pt_maybe "02 03 A5" 1 }
267         07-02 { pt_maybe "00 01 X7" 0 }
268         09-06 { pt_maybe "42 43 A6" 0 }
269         06-09 { pt_maybe "40 41 X8" 1 }
270     }
271 }
272
273 proc tellnmra {m} {
274     global nmrawhich speeddirn funcs
275     set m 0x$m
276     for {set i 0} {$i < $m} {incr i} {
277         tellpic_q [lindex [list $speeddirn $funcs] $nmrawhich]
278         set nmrawhich [expr {!$nmrawhich}]
279     }
280 }
281
282 proc watchdog {} {
283     global watchdog testonly
284     catch { after cancel $watchdog }
285     set watchdog [after 50 watchdog]
286     tellpic_q 9808 ;# 128ms
287 }
288
289 proc pm_hello {} {
290     debug "got hello, starting up"
291     tellpic 21
292     gui "P 1"
293     watchdog
294     changewhat
295     tellnmra 01
296 }
297
298 proc fp {m} {
299     debug "<< $m"
300 }
301
302 proc frompic {m} {
303     set v [lindex $m 1]
304     switch -glob [lindex $m 0] {
305         01 - 02 { tellnmra $m }
306         09 { fp $m; pm_hello }
307         07 { puts "short circuit"; exit 1 }
308         28 { fp $m; pm_charged }
309         9[0-7] { fp $m; pm_maydetect 0 $v }
310         9? { fp $m; pm_maydetect 1 $v }
311         0a - [234567]? { puts "pic debug $m" }
312         * { fp $m; fail "pic unknown $m" }
313     }
314 }
315
316 proc onreadp_test {} {
317     if {![gets stdin m]} { return }
318     frompic $m
319 }
320
321 proc onreadp {} {
322     global p m
323     while 1 {
324         set c [read $p 1]
325         if {![string length $c]} {
326             if {[eof $p]} { error "eof on device" }
327             return
328         }
329         binary scan $c H* x
330         lappend m $x
331         if {[regexp {^[0-7]} $x]} {
332             if {![regexp {^x} $m]} {
333                 frompic $m
334             }
335             set m {}
336         }
337     }
338 }
339
340 proc newspeeddirn {} {
341     set b1 0x[randbyte]
342     set speed [expr {round(($b1 * $b1) / 65535.0 * 100.0 + 26.0)}]
343     set b2 0x[randbyte]
344     set dirn [expr {$b2 / 128}]
345     debug "speeddirn b1=$b1 speed=$speed b2=$b2 dirn=$dirn"
346     return "speed126 2 $speed $dirn"
347 }
348
349 proc newfuncs {} {
350     set b3 0x[randbyte]
351     set value [expr {($b3 & 127) * 16}]
352     debug "funcs b3=$b3 value=[format %x $value]"
353     return "funcs5to8 2 $value"
354 }
355
356 proc maybechange {thing} {
357     global $thing ch
358     upvar #0 ${thing}_fixed fixed
359     if {![info exists fixed]} {
360         set rb 0x[randbyte][randbyte]
361         if {
362             $rb / 65536.0 >
363             1.0 / (($ch(${thing}every) - $ch(minint)*0.001) * $ch(scale))
364         } {
365             debug "maybechange $thing rb=$rb no"
366             return 0
367         }
368         debug "maybechange $thing rb=$rb yes ..."
369         set l [new$thing]
370     } else {
371         debug "fixed $thing $fixed"
372         set l $fixed
373     }
374     set bin [eval exec ./hostside-old -s/dev/stdout $l]
375     binary scan $bin H* x
376     debug "changed $thing=$x"
377     set $thing ff$x
378     return 1
379 }
380
381 proc changewhat {} {
382     global ch chwa
383     catch { after cancel $chwa }
384     if {[maybechange speeddirn] || [maybechange funcs]} {
385         set interval $ch(minint)
386     } else {
387         set interval 1000
388     }
389     set chwa [after $interval changewhat]
390 }
391
392 proc setup {} {
393     global port p rand testonly
394     if {!$testonly} {
395         set p [open $port {RDWR NONBLOCK} 0]
396     
397         exec stty -F $port min 1 time 0 -istrip -ocrnl -onlcr -onocr -opost \
398                        -ctlecho -echo -echoe -echok -echonl -iexten -isig \
399                        -icanon -icrnl \
400             9600 clocal cread crtscts -hup -parenb cs8 -cstopb \
401             -ixoff bs0 cr0 ff0 nl0 -ofill -olcuc
402
403         fconfigure $p -encoding binary -translation binary \
404                 -blocking false -buffering none
405
406         fileevent $p readable onreadp
407     } else {
408         set p stdin
409         fconfigure stdin -blocking false
410         fileevent stdin readable onreadp_test
411     }
412
413     set rand [open /dev/urandom {RDONLY} 0]
414     fconfigure $rand -encoding binary -translation binary
415 }
416
417 setup
418 gui_init
419 vwait end