-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathmain.js
94 lines (77 loc) · 2.58 KB
/
main.js
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
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
const carCanvas = document.getElementById("carCanvas");
carCanvas.width = 200;
const networkCanvas = document.getElementById("networkCanvas");
networkCanvas.width = 300;
carWidth = 30;
carHeight = 50
const carCtx = carCanvas.getContext("2d")
const networkCtx = networkCanvas.getContext("2d")
const road = new Road(carCanvas.width/2, carCanvas.width * 0.9)
const cars = generateCars(100)//new Car(road.getLaneCenter(1), 100, carWidth, carHeight, "AI")
let bestCar = cars[0]
if(localStorage.getItem("bestBrain")){
for(let i = 0; i < cars.length; i++){
cars[i].brain = JSON.parse(localStorage.getItem("bestBrain"))
if(i !== 0){
NeuralNetwork.mutate(cars[i].brain, 0.10)
}
}
}
const traffic = [
new Car(road.getLaneCenter(1), -100, carWidth, carHeight, "DUMMY", 2),
new Car(road.getLaneCenter(0), -300, carWidth, carHeight, "DUMMY", 2),
new Car(road.getLaneCenter(2), -300, carWidth, carHeight, "DUMMY", 2),
new Car(road.getLaneCenter(1), -500, carWidth, carHeight, "DUMMY", 2),
new Car(road.getLaneCenter(0), -500, carWidth, carHeight, "DUMMY", 2),
new Car(road.getLaneCenter(2), -700, carWidth, carHeight, "DUMMY", 2),
new Car(road.getLaneCenter(4), -700, carWidth, carHeight, "DUMMY", 2)
]
animate();
function save(){
localStorage.setItem("bestBrain",
JSON.stringify(bestCar.brain));
console.log("Saved Brain")
}
function discard() {
localStorage.removeItem("bestBrain",
JSON.stringify(bestCar.brain));
console.log("Discarded Brain")
}
function generateCars(N){
const cars=[];
for(let i=1;i<=N;i++){
cars.push(new Car(road.getLaneCenter(1),100,30,50,"AI"));
}
return cars;
}
function animate(time){
for(let i = 0; i < traffic.length; i++){
traffic[i].update(road.borders, [])
}
for(let i = 0; i < cars.length; i++){
cars[i].update(road.borders, traffic);
}
bestCar = cars.find(
c=>c.y===Math.min(
...cars.map(c=>c.y)
)
)
carCanvas.height = window.innerHeight
networkCanvas.height = window.innerHeight
carCtx.save()
carCtx.translate(0, 0-bestCar.y+carCanvas.height*0.67)
road.draw(carCtx)
for(let i = 0; i < traffic.length; i++){
traffic[i].draw(carCtx, "green")
}
bestCar.draw(carCtx, "blue", true)
carCtx.globalAlpha = 0.2
for(let i = 1; i < cars.length; i++){
cars[i].draw(carCtx, "blue")
}
carCtx.globalAlpha = 1
carCtx.restore()
networkCtx.lineDashOffset = -time/100
Visualizer.drawNetwork(networkCtx, bestCar.brain)
requestAnimationFrame(animate)
}