update speed of trains so users could see them

kobellem · Aug 7, 2016 · 36b2007 · 36b2007
1 parent dfc7104
commit 36b2007
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/infra/controller.rkt b/src/infra/controller.rkt
@@ -15,8 +15,12 @@
     ;abstraction
     (define (id o)(send o get-id))
     (define (position train)(send train get-position))
-    (define (stop train)(send requester request "set-speed" (list (send train get-id) 0 #t)))
-    (define (start train dir)(send requester request "set-speed" (list (send train get-id) 60 dir)))
+    (define (stop train)
+      (send requester request "set-speed" (list (send train get-id) 0 #t))
+      (send train set-speed! 0))
+    (define (start train dir)
+      (send requester request "set-speed" (list (send train get-id) 60 dir))
+      (send train set-speed! 60))
     ;public methods
     (define (add-train id_ pos)
       (set-add! trains (make-train id_ (send track get-segment pos)))