CodeAkinori · December 8, 2023 21:04 · Dec 8, 2023
diff --git a/script.js b/script.js
@@ -0,0 +1,88 @@
+import utils from './utils.js';
+import RNA from './RNA.js';
+import controls from './controls.js';
+
+const SAMPLES = 10; // Número de amostras (agentes) no algoritmo genético
+const game = Runner.instance_; // Instância do jogo "Runner"
+let dinoList = []; // Lista de dinossauros
+let dinoIndex = 0; // Índice do dinossauro atual na lista
+
+let bestScore = 0; // Melhor pontuação encontrada durante o treinamento
+let bestRNA = null; // Melhor RNA (rede neural) encontrada durante o treinamento
+
+function fillDinoList () {
+  for (let i=0; i<SAMPLES; i++) {
+    dinoList[i] = new RNA(3, [10, 10, 2]); // Cria um novo dinossauro com uma RNA de 3 camadas
+    dinoList[i].load(bestRNA); // Carrega a melhor RNA encontrada anteriormente
+    if (i > 0) dinoList[i].mutate(0.2); // Mutação na RNA dos dinossauros, exceto o primeiro
+  }
+  console.log('Dino list created!');
+}
+
+setTimeout(() => {
+  fillDinoList();
+  controls.dispatch('jump'); // Faz o dinossauro executar um salto no jogo
+}, 1000);
+
+setInterval(() => {
+  if (!game.activated) return; // Verifica se o jogo está ativado
+
+  const dino = dinoList[dinoIndex]; // Seleciona o dinossauro atual
+
+  if (game.crashed) { // Verifica se o dinossauro colidiu no jogo
+    if (dino.score > bestScore) {
+      bestScore = dino.score;
+      bestRNA = dino.save(); // Salva a RNA do dinossauro com a melhor pontuação
+      console.log('bestScore:', bestScore);
+    }
+    dinoIndex++;
+
+    if (dinoIndex === SAMPLES) { // Se todos os dinossauros foram avaliados, preenche a lista novamente
+      fillDinoList();
+      dinoIndex = 0;
+      bestScore = 0;
+    }
+    game.restart(); // Reinicia o jogo
+  }
+
+  const { tRex, horizon, currentSpeed, distanceRan, dimensions } = game;
+  dino.score = distanceRan - 2000; // Calcula a pontuação do dinossauro
+
+  const player = {
+    x: tRex.xPos,
+    y: tRex.yPos,
+    speed: currentSpeed,
+  };
+
+  const [obstacle] = horizon.obstacles
+    .map((obstacle) => {
+      return {
+        x: obstacle.xPos,
+        y: obstacle.yPos,
+      }
+    })
+    .filter((obstacle) => obstacle.x > player.x)
+
+  if (obstacle) { // Verifica se há um obstáculo presente
+    const distance = 1 - (utils.getDistance(player, obstacle) / dimensions.WIDTH); // Calcula a distância relativa entre o jogador e o obstáculo
+    const speed = player.speed / 6; // Calcula a velocidade relativa do jogador
+    const height = Math.tanh(105 - obstacle.y); // Calcula a altura relativa do obstáculo
+
+    // Processa as informações no dinossauro atual
+    const [jump, crounch] = dino.compute([
+      distance,
+      speed,
+      height,
+    ]);
+
+    // Executa as ações com base nas probabilidades calculadas
+    if (jump === crounch) return; // Se a probabilidade de salto e agachamento forem iguais, nenhuma ação é tomada
+    if (jump) controls.dispatch('jump'); // Se a probabilidade de salto for verdadeira, o dinossauro executa um salto
+    if (crounch) controls.dispatch('crounch'); // Se a probabilidade de agachamento for verdadeira, o dinossauro se agacha
+  }
+}, 100);
+
+/* const s = document.createElement('script');
+s.type = 'module';
+s.src = 'http://localhost:5500/script.js';
+document.body.appendChild(s); */