event_dispatcher.cpp

#include <windows.h>

#include "event_dispatcher.h"


PyObject* _getSwigUnit(BWAPI::Unit* unit);
PyObject* _getSwigPlayer(BWAPI::Player* player);
PyObject* _getSwigPosition(BWAPI::Position* position);


EventDispatcher::EventDispatcher()
	:event_handler(NULL)
{
	LoadEventHandler();
}

EventDispatcher::~EventDispatcher()
{
	if (event_handler != NULL)
	{
		Py_DECREF( event_handler );
		event_handler = NULL;
	}
}


static void python_func_cleanup(PyObject* res)
{
	if (res != NULL)
	{
	  Py_DECREF( res );
	}
	PyErr_Clear();
}

void EventDispatcher::LoadEventHandler(bool call_reload)
{
	PyObject* pybw_module = PyImport_ImportModule("pybw");
	PyErr_Clear();
	if (pybw_module == NULL)
	{
		printf("Error importing pybw module\n");
		return;
	}

	if (event_handler != NULL)
	{
		Py_DECREF(event_handler);
		event_handler = NULL;
	}

	if (call_reload)
	{
		PyObject* reload_event_handler = PyObject_GetAttrString(pybw_module, "reload_event_handler");
		PyErr_Clear();
		if (reload_event_handler == NULL)
		{
			printf("Error getting reload_event_handler from pybw module\n");
			return;
		}

		python_func_cleanup( PyObject_CallFunction(reload_event_handler, NULL) );
		Py_DECREF(reload_event_handler);
	}

	event_handler = PyObject_GetAttrString(pybw_module, "event_handler");
	PyErr_Clear();
	Py_DECREF( pybw_module );

	if (event_handler == NULL)
	{
		printf("Error getting event_handler from pybw module\n");
		return;
	}

}
void EventDispatcher::reload()
{
	LoadEventHandler(true);
}

void EventDispatcher::dispatchEvent(BWAPI::Event* e)
{
    switch(e->getType())
	{
	case BWAPI::EventType::MatchStart:
		onMatchStart();
		break;
	case BWAPI::EventType::MatchEnd:
		onMatchEnd(e->isWinner());
		break;
    case BWAPI::EventType::MatchFrame:
        onMatchFrame();
        break;
    case BWAPI::EventType::MenuFrame:
        onMenuFrame();
        break;
    case BWAPI::EventType::SendText:
		onSendText(e->getText());
        break;
    case BWAPI::EventType::ReceiveText:
		onReceiveText(e->getPlayer(), e->getText());
        break;
    case BWAPI::EventType::PlayerLeft:
        onPlayerLeft(e->getPlayer());
        break;
    case BWAPI::EventType::NukeDetect:
		onNukeDetect(e->getPosition());
        break;
    case BWAPI::EventType::UnitDiscover:
        onUnitDiscover(e->getUnit());
        break;
    case BWAPI::EventType::UnitEvade:
        onUnitEvade(e->getUnit());
        break;
    case BWAPI::EventType::UnitShow:
        onUnitShow(e->getUnit());
        break;
    case BWAPI::EventType::UnitHide:
        onUnitHide(e->getUnit());
        break;
    case BWAPI::EventType::UnitCreate:
        onUnitCreate(e->getUnit());
        break;
    case BWAPI::EventType::UnitDestroy:
        onUnitDestroy(e->getUnit());
        break;
    case BWAPI::EventType::UnitMorph:
        onUnitMorph(e->getUnit());
        break;
    case BWAPI::EventType::UnitRenegade:
        onUnitRenegade(e->getUnit());
        break;
    case BWAPI::EventType::SaveGame:
        onSaveGame(e->getText());
        break;
    case BWAPI::EventType::UnitComplete:
        onUnitComplete(e->getUnit());
        break;
    case BWAPI::EventType::None:
        break;
    default:
        printf("error: Unknown event: %d\n", e->getType());
        break;
	}
}

void EventDispatcher::onMatchStart()
{
  if (event_handler != NULL)
  {
	  python_func_cleanup( PyObject_CallMethod(event_handler, "onMatchStart", NULL) );
  }
}

void EventDispatcher::onMatchEnd(bool isWinner)
{
	if (event_handler != NULL)
	{
		python_func_cleanup( PyObject_CallMethod(event_handler, "onMatchEnd", "i", isWinner) );
	}
}

void EventDispatcher::onMatchFrame()
{
  if (event_handler != NULL)
  {
	  python_func_cleanup( PyObject_CallMethod(event_handler, "onMatchFrame", NULL) );
  }
}

void EventDispatcher::onMenuFrame()
{
  if (event_handler != NULL)
  {
	  python_func_cleanup( PyObject_CallMethod(event_handler, "onMenuFrame", NULL) );
  }
}

void EventDispatcher::onSendText(std::string text)
{
  if (event_handler != NULL)
  {
	  python_func_cleanup( PyObject_CallMethod(event_handler, "onSendText", "s", text.c_str()) );
  }
}

void EventDispatcher::onReceiveText(BWAPI::Player* player, std::string text)
{
  if (event_handler != NULL)
  {
	  python_func_cleanup( PyObject_CallMethod(event_handler, "onReceiveText", "Os", _getSwigPlayer(player), text.c_str()) );
  }
}

void EventDispatcher::onSaveGame(std::string gameName)
{
  if (event_handler != NULL)
  {
	  python_func_cleanup( PyObject_CallMethod(event_handler, "onSaveGame", "s", gameName.c_str()) );
  }
}

void EventDispatcher::onPlayerLeft(BWAPI::Player* player)
{
	if (event_handler != NULL)
	{
		python_func_cleanup( PyObject_CallMethod(event_handler, "onPlayerLeft", "O", _getSwigPlayer(player)) );
	}
}

void EventDispatcher::onNukeDetect(BWAPI::Position target)
{
	if (event_handler != NULL)
	{
		python_func_cleanup( PyObject_CallMethod(event_handler, "onNukeDetect", NULL) );
	}
}

void EventDispatcher::onUnitCreate(BWAPI::Unit* unit)
{
	if (event_handler != NULL)
	{
		python_func_cleanup( PyObject_CallMethod(event_handler, "onUnitCreate", "O", _getSwigUnit(unit)) );
	}
}

void EventDispatcher::onUnitDestroy(BWAPI::Unit* unit)
{
	if (event_handler != NULL)
	{
		python_func_cleanup( PyObject_CallMethod(event_handler, "onUnitDestroy", "O", _getSwigUnit(unit)) );
	}
}

void EventDispatcher::onUnitEvade(BWAPI::Unit* unit)
{
	if (event_handler != NULL)
	{
		python_func_cleanup( PyObject_CallMethod(event_handler, "onUnitEvade", "O", _getSwigUnit(unit)) );
	}
}

void EventDispatcher::onUnitDiscover(BWAPI::Unit* unit)
{
	if (event_handler != NULL)
	{
		python_func_cleanup( PyObject_CallMethod(event_handler, "onUnitDiscover", "O", _getSwigUnit(unit)) );
	}
}


void EventDispatcher::onUnitMorph(BWAPI::Unit* unit)
{
	if (event_handler != NULL)
	{
		python_func_cleanup( PyObject_CallMethod(event_handler, "onUnitMorph", "O", _getSwigUnit(unit)) );
	}
}

void EventDispatcher::onUnitShow(BWAPI::Unit* unit)
{
	if (event_handler != NULL)
	{
		python_func_cleanup( PyObject_CallMethod(event_handler, "onUnitShow", "O", _getSwigUnit(unit)) );
	}
}

void EventDispatcher::onUnitHide(BWAPI::Unit* unit)
{
	if (event_handler != NULL)
	{
		python_func_cleanup( PyObject_CallMethod(event_handler, "onUnitHide", "O", _getSwigUnit(unit)) );
	}
}

void EventDispatcher::onUnitRenegade(BWAPI::Unit* unit)
{
	if (event_handler != NULL)
	{
		python_func_cleanup( PyObject_CallMethod(event_handler, "onUnitRenegade", "O", _getSwigUnit(unit)) );
	}
}

void EventDispatcher::onUnitComplete(BWAPI::Unit* unit)
{
	if (event_handler != NULL)
	{
		python_func_cleanup( PyObject_CallMethod(event_handler, "onUnitComplete", "O", _getSwigUnit(unit)) );
	}
}

void EventDispatcher::onConnect(BWAPI::Game* game)
{
	if (event_handler != NULL)
	{
		python_func_cleanup( PyObject_CallMethod(event_handler, "onConnect", NULL) );	// TODO !
	}
}

void EventDispatcher::onDisconnect()
{
	if (event_handler != NULL)
	{
		python_func_cleanup( PyObject_CallMethod(event_handler, "onDisconnect", NULL) );
	}
}