Glue everything toghether

Add trainer
Add brain to dinos and use it to control movement
Connect players and ga individuals

src/ai/Trainer.js

@@ -0,0 +1,25 @@
+import { GeneticAlgorithm } from './ga';
+import { Network } from './ann/Network';
+const { weightsForArchitecture } = require('./ann/math');
+const { times } = require('lodash');
+export class Trainer {
+  constructor(populationSize) {
+    // this.population = times(populationSize, () => ({}));
+
+    const networkArchitecture = [[2], [4, 'sigmoid'], [4, 'sigmoid'], [2, 'sigmoid']];
+
+    this.ga = new GeneticAlgorithm(populationSize, {
+      dimentions: weightsForArchitecture(networkArchitecture.map(([n]) => n)),
+      min: -10,
+      max: 10,
+    });
+    this.population = this.ga.population.map((individual, i) => {
+      individual.brain = Network(networkArchitecture);
+      individual.brain.distributeWights(this.ga.population[i]._genome);
+      return individual;
+    });
+  }
+  async train() {
+    this.ga.evolve();
+  }
+}

src/ai/ga.js

@@ -30,10 +30,10 @@ class GeneticAlgorithm {
   }
   populate(populationSize, problem) {
     const { dimentions, min, max } = problem;
-    this.population = times(
-      populationSize,
-      () => new Individual(times(dimentions, () => random(min, max, true))),
-    );
+    this.population = times(populationSize, () => {
+      const genome = times(dimentions, () => random(min, max, true));
+      return new Individual(genome);
+    });
   }
   evolve() {
     //calculate fitness before this

src/prefabs/player/InputManager.js

@@ -7,10 +7,12 @@ class InputManager {
    * Creates an instance of InputManager
    * @param {Player} player - The Player to which this InputManager belongs
    */
-  constructor(player) {
+  constructor(player, brain) {
     this.player = player;
+    this.brain = brain;
     this.scene = player.scene;
     this.cursors = player.scene.input.keyboard.createCursorKeys();
+    [this._shouldGoUp, this.shouldGoDown] = [0, 0];
   }
 
   /**
@@ -19,6 +21,10 @@ class InputManager {
   update() {
     const { player } = this;
 
+    [this._shouldGoUp, this.shouldGoDown] = this.brain
+      .forward(this.player.perception)
+      .map(Math.round);
+
     if (player.isDead) {
       return;
     }
@@ -51,7 +57,7 @@ class InputManager {
    * @returns {boolean}
    */
   get isDuckKeyPressed() {
-    return this.cursors.down.isDown;
+    return this._shouldGoDown;
   }
 
   /**
@@ -60,13 +66,7 @@ class InputManager {
    * @returns {boolean}
    */
   get isJumpKeyPressed() {
-    return Math.round(Math.random());
-    // const { activePointer } = this.scene.input;
-    // return (
-    //   this.cursors.up.isDown ||
-    //   this.cursors.space.isDown ||
-    //   (activePointer.isDown && activePointer.wasTouch)
-    // );
+    return this._shouldGoUp;
   }
 }
 

src/prefabs/player/Player.js

@@ -19,14 +19,17 @@ class Player extends Phaser.Physics.Arcade.Sprite {
    * @param {number} [x=Player.CONFIG.POS.INITIAL_X] - The horizontal position of this Player in the world
    * @param {number} [y=Player.CONFIG.POS.Y] - The vertical position of this Player in the world
    */
-  constructor(scene, x = Player.CONFIG.POS.INITIAL_X, y = Player.CONFIG.POS.Y) {
+  constructor(scene, x = Player.CONFIG.POS.INITIAL_X, y = Player.CONFIG.POS.Y, trainingData) {
     super(scene, x, y, 'dino', Player.CONFIG.FRAMES.INITIAL);
 
+    this.fitness = trainingData._fitness;
+    this.brain = trainingData.brain;
+
     this.isInitialJump = true;
 
     // Init managers
     this.physicsManager = new PhysicsManager(this);
-    this.inputManager = new InputManager(this);
+    this.inputManager = new InputManager(this, this.brain);
     this.animationManager = new AnimationManager(this);
 
     // Init image
@@ -121,7 +124,7 @@ class Player extends Phaser.Physics.Arcade.Sprite {
    * Set player dead | Handle gameover
    */
   die(score) {
-    // if (!this.isDead) console.log(score);
+    this.fitness.value = score;
     this.setState(Player.CONFIG.STATES.DEAD);
     this.animationManager.update();
     this.physicsManager.reset();

src/scenes/game/GameScene.js

@@ -11,6 +11,8 @@ import Intro from './Intro';
 import Player from '../../prefabs/player/Player';
 import Horizon from '../../prefabs/horizon/Horizon';
 import isTelegramMode from '../../utils/telegram/isTelegramMode';
+import { Trainer } from './../../ai/Trainer';
+import { times } from 'lodash';
 
 /**
  * Main game scene
@@ -81,7 +83,11 @@ class GameScene extends Phaser.Scene {
     this.ui = new UI(this);
     this.intro = new Intro(this.events);
 
-    this.players = Array.from(Array(10)).map(() => new Player(this));
+    const populationSize = 10;
+    this.trainer = new Trainer(populationSize);
+    this.players = times(populationSize, i => {
+      return new Player(this, void 0, void 0, this.trainer.population[i]);
+    });
 
     this.horizon = new Horizon(this);
     this.ground = this.horizon.ground;
@@ -111,12 +117,12 @@ class GameScene extends Phaser.Scene {
 
   update() {
     const obstacle = this.obstacles.getLast(true) || { x: this.players[0].x, y: this.players[0].y };
-    this._score = Phaser.Math.Distance.Between(
-      this.players[0].x,
-      this.players[0].y,
-      obstacle.x,
-      obstacle.y,
-    );
+    this.players.forEach(player => {
+      const xDistanceToNextObstacle = obstacle.x - player.x;
+      const yDistanceToNextObstacle = obstacle.y - player.y;
+      const gapBetweenObstacles = this.obstacles.getGap;
+      player.perception = [yDistanceToNextObstacle, xDistanceToNextObstacle];
+    });
     const { gameSize } = this.scale;
     const isMobile = gameSize.width === CONFIG.GAME.WIDTH.PORTRAIT;
 
@@ -224,6 +230,8 @@ class GameScene extends Phaser.Scene {
    * Handle gameover
    */
   onGameOver() {
+    this.trainer.train();
+    console.log(this.trainer.population[0].genome);
     const { width: gameWidth, height: gameHeight } = this.scale.gameSize;
 
     this.isPlaying = false;