File GPAgent.hpp

File List > Agents > GP > GPAgent.hpp

Go to the documentation of this file

#pragma once

#include <cassert>
#include <iostream>
#include <map>
#include <random>
#include <string>

#include "../../core/AgentBase.hpp"
#include "./GPAgentSensors.hpp"

namespace cowboys {

class GPAgent : public cse491::AgentBase {
 protected:
  const std::vector<std::string> predefinedMovement = {
      "down",  "down",  "down",  "down",  "right", "right", "up",    "up",
      "up",    "up",    "right", "right", "right", "right", "right", "right",
      "right", "right", "right", "right", "right", "left",  "left",  "left"};

  size_t movementIndex = 0;  

 public:
  GPAgent(size_t id, const std::string &name) : AgentBase(id, name) {
    std::random_device rd;
    std::mt19937 gen(rd());

    std::uniform_int_distribution<int> dist(
        0,
        3);  // size of the random set to 3 as we only have 4 actions initally
  }
  ~GPAgent() override = default;

  bool Initialize() override { return true; }

  size_t SelectAction(
      [[maybe_unused]] const cse491::WorldGrid &grid,
      [[maybe_unused]] const cse491::type_options_t &type_options,
      [[maybe_unused]] const cse491::item_map_t &item_map,
      [[maybe_unused]] const cse491::agent_map_t &agent_map) override {
    if (movementIndex >= predefinedMovement.size()) {
      return 0;  // do nothing if it is out of bound for defined movement
    }

    Sensors::wallDistance(grid, *this, SensorDirection::LEFT);
    Sensors::wallDistance(grid, *this, SensorDirection::RIGHT);
    Sensors::wallDistance(grid, *this, SensorDirection::ABOVE);
    Sensors::wallDistance(grid, *this, SensorDirection::BELOW);

    auto action = action_map[predefinedMovement[movementIndex++]];
    return action;
  }
};

}  // namespace cowboys