src/engine/MessagePool.h

Go to the documentation of this file.

#ifndef MESSAGEPOOL_H
#define MESSAGEPOOL_H

#include <vector>
#include <string>
#include <map>
#include <boost/shared_ptr.hpp>
#include "MessageInstance.h"
#include "util/Error.h"


namespace green
{
        class MsgFormat {
                public:
                        Key mEntityType;
                        Key mMessageType;
                        
                MsgFormat() : 
                                mEntityType(0),
                                mMessageType(0)
                                {}
                                
                        MsgFormat(Key entType, Key msgType) : mEntityType(entType), mMessageType(msgType) {}
                        
                        bool operator<(const MsgFormat& rhs) const {
                                        return ((mEntityType < rhs.mEntityType) ||
                                                                        ((mEntityType == rhs.mEntityType ) && (mMessageType < rhs.mMessageType)));
                        }
        };
        
        class MessagePool
        {
        public:
                 
                virtual ~MessagePool() {}
                virtual void AddMessage(Key targetType, MessageInstancePtr msg)
                {
                        CHECK_SHARED(msg);
                        int msgKey = msg->GetTemplate();
                        if(msgKey != 1)
                                printf("Adding msg targetType-%d \n",msgKey);
                        
                        ERROR_ASSERT(msgKey < 2, "msgKey was too high");
                        

                        MsgFormat format(targetType, msg->GetTemplate());
                        MsgMap::iterator i = _getList(format);
                        
                        mMsgMap[format].push_back(msg);
                }
                 
                virtual size_t GetNumMessagesOfFormat(MsgFormat format)
                {
                        MsgFormat defaultFormat(0, format.mMessageType);
                        MsgMap::const_iterator i = _getConstList(format);
                        MsgMap::const_iterator j = _getConstList(defaultFormat);
                        
                        size_t exactSize = 0;
                        size_t defaultSize = 0;
                        
                        if(i != mMsgMap.end())
                                exactSize = i->second.size();
                        if(j != mMsgMap.end())
                                defaultSize = j->second.size();
                        
                        return exactSize + defaultSize;
                }
                        
                virtual MessageInstancePtr
                                GetMessageOfFormat(size_t n, MsgFormat format) const
                {
                        MsgFormat defaultFormat(0, format.mMessageType);
                        MsgMap::const_iterator i = _getConstList(format);
                        MsgMap::const_iterator j = _getConstList(defaultFormat);
                        
                        size_t exactSize = 0;
                        size_t defaultSize = 0;
                        
                        if(i != mMsgMap.end())
                                exactSize = i->second.size();
                        if(j != mMsgMap.end())
                                defaultSize = j->second.size();
                        
                        ERROR_ASSERT(n < exactSize + defaultSize, "invalid index");
                        
                        if(n < exactSize)
                                return i->second[n];
                        else
                                return j->second[n - exactSize];
                }
                 
                virtual void Clear()
                {
                        mMsgMap.clear();
                }
                 
        protected:
                typedef std::vector<MessageInstancePtr> MsgList;
                typedef std::map<MsgFormat, MsgList> MsgMap;

                MsgMap::iterator _getList(MsgFormat format)
                {
                        MsgMap::iterator i = mMsgMap.find(format);
                        return i;
                }
                
                MsgMap::const_iterator _getConstList(MsgFormat format) const
                {
                        MsgMap::const_iterator i = mMsgMap.find(format);
                        return i;
                }
                
                MsgMap mMsgMap;
        };
        typedef boost::shared_ptr<MessagePool> MessagePoolPtr;
}

#endif

---