17
17
import * as tf from '@tensorflow/tfjs' ;
18
18
19
19
const CONTROLS = [ 'up' , 'down' , 'left' , 'right' ] ;
20
- const CONTROL_LABELS = [ '↑' , '↓' , '←' , '→' ] ;
21
20
const CONTROL_CODES = [ 38 , 40 , 37 , 39 ] ;
22
21
23
22
export function init ( ) {
24
23
document . getElementById ( 'controller' ) . style . display = '' ;
25
- statusElement . style . visibility = 'hidden ' ;
24
+ statusElement . style . display = 'none ' ;
26
25
}
27
26
28
27
// Set hyper params from UI values.
@@ -45,7 +44,7 @@ export function startPacman() {
45
44
46
45
export function predictClass ( classId ) {
47
46
google . pacman . keyPressed ( CONTROL_CODES [ classId ] ) ;
48
- status ( CONTROL_LABELS [ classId ] ) ;
47
+ document . body . setAttribute ( 'data-active' , CONTROLS [ classId ] ) ;
49
48
}
50
49
51
50
export function isPredicting ( ) {
@@ -55,10 +54,6 @@ export function donePredicting() {
55
54
statusElement . style . visibility = 'hidden' ;
56
55
}
57
56
58
- export function status ( msg ) {
59
- statusElement . innerText = msg ;
60
- }
61
-
62
57
export let addExampleHandler ;
63
58
export function setExampleHandler ( handler ) {
64
59
addExampleHandler = handler ;
@@ -80,9 +75,11 @@ async function handler(label) {
80
75
const total = document . getElementById ( className + '-total' ) ;
81
76
while ( mouseDown ) {
82
77
addExampleHandler ( label ) ;
78
+ document . body . setAttribute ( 'data-active' , CONTROLS [ label ] ) ;
83
79
total . innerText = totals [ label ] ++ ;
84
80
await tf . nextFrame ( ) ;
85
81
}
82
+ document . body . removeAttribute ( 'data-active' ) ;
86
83
}
87
84
88
85
upButton . addEventListener ( 'mousedown' , ( ) => handler ( 0 ) ) ;
0 commit comments