ExampleAIModule.cpp

#include "ExampleAIModule.h"
#include <iostream>
#include <iomanip>
#include <numeric>

using namespace BWAPI;
using namespace Filter;

extern "C" void gameInit(BWAPI::Game *game) { BWAPI::BroodwarPtr = game; }
extern "C" BWAPI::AIModule *newAIModule()
{
  return new ExampleAIModule();
}

std::string env(std::string name, std::string def)
{
  const char *s = std::getenv(name.c_str());
  if (!s)
    return def;
  return s;
}


void ExampleAIModule::onStart()
{
  screen_width = 0;
}

void ExampleAIModule::onEnd(bool isWinner)
{
}

inline int getCharWidth(uint8_t c, uint8_t bSize)
{
  if (c == '\t')
    return 12 * bSize;
  if (c < ' ')
    return 0;
  if (c == ' ')
    return 3 * bSize;
  return 6 * bSize;
}

inline int getTextWidth(const char *pszStr, uint8_t bSize)
{
  // Retrieve size
  int size = 0;
  for (; *pszStr; ++pszStr)
    size += getCharWidth(*pszStr, bSize);

  return size;
}

void ExampleAIModule::onFrame()
{
  if (!screen_width)
  {
#ifdef OPENBW
    screen_width = Broodwar->getScreenSize().x;
    screen_height = Broodwar->getScreenSize().y;
#else
    screen_width = std::stoi(env("SCREEN_WIDTH", "1280"));
    screen_height = std::stoi(env("SCREEN_HEIGHT", "720"));
#endif
    if (screen_width)
    {
      auto startPosition = Position(0, 0);
      auto players = Broodwar->getPlayers();
      for (auto iter = players.cbegin(); iter != players.end(); ++iter)
      {
        if (!(*iter)->isObserver() && !(*iter)->isNeutral())
        {
          startPosition = Position((*iter)->getStartLocation());
        }
      }
      cameraModule.onStart(startPosition, screen_width, screen_height);
    }
    else
    {
      return;
    }
  }

  renderPlayerInfos();
  renderBuildQueues();

  cameraModule.onFrame();
}

void ExampleAIModule::renderPlayerInfos() 
{
  auto players = Broodwar->getPlayers();
  std::vector<std::string> lines;
  auto maxLength = 0;
  for (auto iter = players.cbegin(); iter != players.end(); ++iter)
  {
    auto player = *iter;
    if (!player->isObserver() && !player->isNeutral())
    {
      std::stringstream line;
      line << std::setfill('\x80') << Text::Align_Right;
      line << player->getTextColor();
      if (cameraModule.hasVision(player))
        line << "*";
      else
        line << "\x80";
      line << player->getName();
      line << std::left;

      auto playerUnits = player->getUnits();
      auto workers = std::count_if(playerUnits.cbegin(), playerUnits.cend(), [](BWAPI::Unit unit) { return unit->isCompleted() && unit->getType().isWorker(); });
      auto armySupply = std::accumulate(playerUnits.cbegin(), playerUnits.cend(), 0, [this](int sum, BWAPI::Unit unit) {
                          return sum + (unit->isCompleted() && cameraModule.isArmyUnit(unit) ? unit->getType().supplyRequired() : 0);
                        }) /
                        2.0;

      line << "   m " << std::setw(4) << player->minerals();
      line << "   g " << std::setw(4) << player->gas();
      line << "   s " << std::setw(3) << (player->supplyUsed() + 1) / 2 << "/" << std::setw(3) << player->supplyTotal() / 2;
      line << "   workers " << std::setw(3) << workers;
      line << "   army " << std::setw(5) << armySupply;
      auto linestr = line.str();
      auto strlen = getTextWidth(linestr.c_str(), Text::Size::Large);
      maxLength = maxLength > strlen ? maxLength : strlen;
      lines.push_back(linestr);
    }
  }
  Broodwar->drawBoxScreen(screen_width - maxLength - 16, 24, screen_width, 24 + lines.size() * 18, Colors::Black, true);
  Broodwar->setTextSize(Text::Size::Large);

  for (size_t y = 0; y < lines.size(); y++)
  {
    Broodwar->drawTextScreen(Position(0, y * 16 + 26), lines.at(y).c_str());
  }

  Broodwar->setTextSize();
}

int f2s(int frames) {
  return (frames * 42 + 999) / 1000;
}

void ExampleAIModule::renderBuildQueues() 
{
  using row = std::tuple<int, std::string>;
  auto players = Broodwar->getPlayers();
  std::vector<row> lines;

  for (auto iter = players.cbegin(); iter != players.end(); ++iter)
  {
    auto player = *iter;
    if (!player->isObserver() && !player->isNeutral())
    {

      for (auto u: player->getUnits()) {
        int bt = u->getRemainingBuildTime();
        if (bt) {
          std::stringstream line;
          line << std::setfill('\x80') << player->getTextColor();
          line << std::setw(3) << f2s(bt) << "s ";
          std::string name;
          if (u->getType() == UnitTypes::Zerg_Egg) {
            name = u->getBuildType().getName();
          } else {
            name = u->getType().getName();
          }
          line << name.substr(name.find('_') + 1);
          lines.push_back(std::make_tuple(bt, line.str()));
        }
        int ut = u->getRemainingUpgradeTime();
        if (ut) {
          std::stringstream line;
          line << std::setfill('\x80') << player->getTextColor();
          line << std::setw(3) << f2s(ut) << "s " << u->getUpgrade().getName();
          lines.push_back(std::make_tuple(ut, line.str()));
        }
        int tt = u->getRemainingResearchTime();
        if (tt) {
          std::stringstream line;
          line << std::setfill('\x80') << player->getTextColor();
          line << std::setw(3) << f2s(tt) << "s " << u->getTech().getName();
          lines.push_back(std::make_tuple(tt, line.str()));
        }
      }

    }
  }

  sort(lines.begin(), lines.end(), 
    [](const row& a , const row& b) {
  	  return std::get<0>(a) < std::get<0>(b);
    });

  for (size_t y = 0; y < lines.size() && y < 25; y++)
  {
    auto str = std::get<1>(lines.at(y)).c_str();
    auto text_width = getTextWidth(str, Text::Size::Default);
    Broodwar->drawBoxScreen(0, y * 10 + 39, text_width, y * 10 + 49, Colors::Black, true);
    Broodwar->drawTextScreen(Position(0, y * 10 + 40), str);
  }
}

void ExampleAIModule::onSendText(std::string text)
{

  // Send the text to the game if it is not being processed.
  Broodwar->sendText("%s", text.c_str());

  // Make sure to use %s and pass the text as a parameter,
  // otherwise you may run into problems when you use the %(percent) character!
}

void ExampleAIModule::onReceiveText(BWAPI::Player player, std::string text)
{
  // Parse the received text
  Broodwar << player->getName() << " said \"" << text << "\"" << std::endl;
}

void ExampleAIModule::onPlayerLeft(BWAPI::Player player)
{
}

void ExampleAIModule::onNukeDetect(BWAPI::Position target)
{

  // Check if the target is a valid position
  if (target)
  {
    // if so, print the location of the nuclear strike target
    /* Broodwar << "Nuclear Launch Detected at " << target << std::endl; */
  }
  else
  {
    // Otherwise, ask other players where the nuke is!
    /* Broodwar->sendText("Where's the nuke?"); */
  }

  // You can also retrieve all the nuclear missile targets using Broodwar->getNukeDots()!
}

void ExampleAIModule::onUnitDiscover(BWAPI::Unit unit)
{
}

void ExampleAIModule::onUnitEvade(BWAPI::Unit unit)
{
}

void ExampleAIModule::onUnitShow(BWAPI::Unit unit)
{
}

void ExampleAIModule::onUnitHide(BWAPI::Unit unit)
{
}

void ExampleAIModule::onUnitCreate(BWAPI::Unit unit)
{
  if (Broodwar->isReplay())
  {
    // if we are in a replay, then we will print out the build order of the structures
    if (unit->getType().isBuilding() && !unit->getPlayer()->isNeutral())
    {
      int seconds = Broodwar->getFrameCount() / 24;
      int minutes = seconds / 60;
      seconds %= 60;
      Broodwar->sendText("%.2d:%.2d: %s creates a %s", minutes, seconds, unit->getPlayer()->getName().c_str(), unit->getType().c_str());
    }
  }
}

void ExampleAIModule::onUnitDestroy(BWAPI::Unit unit)
{
  cameraModule.onUnitDestroy(unit);
}

void ExampleAIModule::onUnitMorph(BWAPI::Unit unit)
{
  if (Broodwar->isReplay())
  {
    // if we are in a replay, then we will print out the build order of the structures
    if (unit->getType().isBuilding() && !unit->getPlayer()->isNeutral())
    {
      int seconds = Broodwar->getFrameCount() / 24;
      int minutes = seconds / 60;
      seconds %= 60;
      Broodwar->sendText("%.2d:%.2d: %s morphs a %s", minutes, seconds, unit->getPlayer()->getName().c_str(), unit->getType().c_str());
    }
  }
}

void ExampleAIModule::onUnitRenegade(BWAPI::Unit unit)
{
}

void ExampleAIModule::onSaveGame(std::string gameName)
{
  Broodwar << "The game was saved to \"" << gameName << "\"" << std::endl;
}

void ExampleAIModule::onUnitComplete(BWAPI::Unit unit)
{
  cameraModule.moveCameraUnitCreated(unit);
}