rtiSmSocket.h

Go to the documentation of this file.

/*********************************************************************
** Copyright (c) 2005 MaK Technologies, Inc.
** All rights reserved.
*********************************************************************/
/*********************************************************************
** $RCSfile: rtiSmSocket.h,v $ $Revision: 1.8 $ $State: Exp $
*********************************************************************/
#ifndef rtiSmSocket_H_
#define rtiSmSocket_H_


#include <smSocket__rti__.h>
#include "smSubscribableMessageQueue.h"


class DT_DLL_LRC DtRtiSmSocket : public DtSmSocket__rti__
{
public:
   DtRtiSmSocket(const char * sharename, DtBoost::shared_ptr<DtMsgHeaderReader> hdrReader);

   virtual ~DtRtiSmSocket();

    virtual bool initializeSmQueue(DtU32 id = DtSM_INVALID_SUBSCRIBER_ID,
                                   bool forceSubscribe = false);
    virtual bool initializeSmQueue(DtU32 numBuckets, DtU32 payloadSize,
                                   bool reliableQueue,
                                   DtU32 id = DtSM_INVALID_SUBSCRIBER_ID,
                                   bool forceSubscribe = false);

    virtual int sendMessage(caddr_t msg, size_t msgSize, bool deleteAfterSend,
       unsigned int flags, const DtInetAddr& destAddr = DtInetAddr::inaddrAny());

   virtual bool flush();

   virtual int getMsgTransportInfo(DtInetAddr *addr, DtU32 *transAndPriority);

   virtual DtSmSubscribableMessageQueue__rti__ * msgQueue() const;

   virtual bool queueIsReliable() const;


   virtual int waitForMessages(MAKRti::DtTime waitTime = 0.0);

   virtual int waitUntilFull(MAKRti::DtTime waitTime = 0.0);

   virtual void setReceiveMask( DtU32 rcvMask );

   virtual void setAnonFilter( bool onOrOff );

   virtual DtInetSockState state() const;

protected:
   
   MAKRti::DtString                mySharedMemName;
   DtRtiSubscribableMessageQueue * myRtiSmMsgQueue;
   DtU32                           myRecvMask;
   bool                            myAnonFilterEnabled;

   class DtRtiSmSocketBundleBuff;
   friend class DtRtiSmSocketBundleBuff;
   typedef std::vector<char> Buffer;
   
   class DtRtiSmSocketBundleBuff
   {
   public:
      DtRtiSmSocketBundleBuff() {};
      ~DtRtiSmSocketBundleBuff() {};
      unsigned int     bbFlags;
      Buffer bbBuff;
   };

   
   typedef std::map< MAKRti::DtInetAddr, DtRtiSmSocketBundleBuff > RtiBufferMap;
   RtiBufferMap myRtiBufferMap;

   DtInetSockState myState;

protected:
   DtRtiSmSocket(const DtRtiSmSocket &);

   DtRtiSmSocket &operator=(const DtRtiSmSocket &);

   virtual int recv(caddr_t *buffptr, DtInetEndpoint* ep);

   virtual bool filterAnonMsgs( caddr_t buffptr, DtU32 msgSenderId );

};

#endif