-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathmain2.go
76 lines (66 loc) · 1.21 KB
/
main2.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
package main
import (
"fmt"
"time"
)
func main() {
cmds := parse()
out1 := make(chan int, 100000)
out2 := make(chan int, 100000)
p0 := NewVM(0, cmds, out1, out2)
p1 := NewVM(1, cmds, out2, out1)
go p0.Run()
go p1.Run()
time.Sleep(200 * time.Millisecond)
fmt.Println(p1.NumSend)
}
type VM struct {
ID int
Registers map[string]int
Cmds []Cmd
Pos int
In <-chan int
Out chan<- int
NumSend int
}
func NewVM(id int, cmds []Cmd, out chan<- int, in <-chan int) *VM {
return &VM{
ID: id,
Registers: map[string]int{"p": id},
Cmds: cmds,
In: in,
Out: out,
}
}
func (vm *VM) Run() {
for {
vm.Exec()
if vm.Pos < 0 || vm.Pos >= len(vm.Cmds) {
return
}
}
}
func (vm *VM) Exec() {
cmd := vm.Cmds[vm.Pos]
switch cmd.Op {
case "snd":
vm.NumSend++
vm.Out <- vm.getVal(cmd.X)
case "set":
vm.Registers[cmd.X] = vm.getVal(cmd.Y)
case "add":
vm.Registers[cmd.X] += vm.getVal(cmd.Y)
case "mul":
vm.Registers[cmd.X] *= vm.getVal(cmd.Y)
case "mod":
vm.Registers[cmd.X] %= vm.getVal(cmd.Y)
case "rcv":
vm.Registers[cmd.X] = <-vm.In
case "jgz":
if vm.getVal(cmd.X) > 0 {
vm.Pos += vm.getVal(cmd.Y)
return
}
}
vm.Pos++
}