Undermind/proxy/cpp/ClientModule/ClientModule.h

Go to the documentation of this file.

#pragma once

#include "aimodule.pb.h"
#include <BWAPI.h>
#include <boost/format.hpp>
#include <set>
#include <BWTA.h>
#include <google/protobuf/repeated_field.h>
#include <hash_map>
#include <hash_set>
#include <string>
#include "netstream.h"

#include <fstream>

#include <google/protobuf/io/zero_copy_stream_impl.h>
#include "terrainanalysis.h"


class ClientModule : public BWAPI::AIModule
{
        messages::FrameMessage frameMessage;
        NetStream* proxyBotSocket;
        TerrainAnalysis terrain;
        stdext::hash_map<int,BWAPI::Unit*> unitMap;
        std::set<BWAPI::Unit*> registeredUnits;
        
        std::string stringbuf;
        std::ofstream log_;

        void log(const std::string& out) { log(boost::format(out)); }
        void log(const boost::format& format);

        bool writeMessageToStream(std::ostream& out, const google::protobuf::Message& message);
        bool readMessageFromStream(std::istream& out, google::protobuf::Message& message);
        void handleCommands(const messages::FrameCommands& commands);
        void registerUnit(messages::FrameMessage& frameMessage, BWAPI::Unit* u);
        void handleBotDefeated(const std::string& botName, const std::string& botCode, const std::string& playerName);
public:
        ClientModule();
        virtual void onStart();
        virtual void onFrame();
        virtual void onEnd(bool isWinner);
        virtual void onUnitCreate(BWAPI::Unit* unit);
        virtual void onUnitDestroy(BWAPI::Unit* unit);
        virtual void onSendText(std::string text);


        virtual void onReceiveText(BWAPI::Player* player, std::string text);

        virtual void onPlayerLeft(BWAPI::Player* player);

        virtual void onNukeDetect(BWAPI::Position target);

        virtual void onUnitMorph(BWAPI::Unit* unit);

        virtual void onUnitShow(BWAPI::Unit* unit);

        virtual void onUnitHide(BWAPI::Unit* unit);

        virtual void onUnitRenegade(BWAPI::Unit* unit);

        // TODO: Add Doxygen documentation
        virtual void onSaveGame(std::string gameName);

};


// needs to be a macro to get laziness
#define SET_ID(source, target) if( (source) != 0) target->set_id( (source)->getID())

template<typename Container, typename P2>
void set_ids(const Container& c, google::protobuf::RepeatedPtrField<P2>* ptr) {
        ptr->Reserve(c.size());
        for(typename Container::const_iterator iter = c.begin(); iter != c.end(); ++iter) {
                SET_ID(*iter, ptr->Add());
        }
}