smConMgr.h

Go to the documentation of this file.

/*********************************************************************
** Copyright (c) 2005 MaK Technologies, Inc.
** All rights reserved.
*********************************************************************/
/*********************************************************************
** $RCSfile: smConMgr.h,v $ $Revision: 1.13 $ $State: Exp $
*********************************************************************/


#ifndef DtSmConnectionMgr_H_
#define DtSmConnectionMgr_H_

#include "connectMgr.h"
#include "smConnection.h"
#include <vlutil/vlFilename.h>

#define DtSmDEFAULT_SHARED_MEM_NAME "DtRtiSharedMessages"

class DtRtiShmQNetMgr;


class DT_DLL_LRC DtSmConnectionMgr : public DtConnectionMgr
{
protected:

   DtSmConnectionMgr(DtRIDParameters* params, bool isFederate = false, 
      DtBoost::shared_ptr<DtRtiAssistantLrcClient> assistantClient = 
      DtBoost::shared_ptr<DtRtiAssistantLrcClient>());

public:

   virtual ~DtSmConnectionMgr();

   virtual void initQMgr(DtU32 numBuckets, DtU32 payloadSize, bool reliable,
                         bool forceMgrSubscribe);


   virtual int send(const DtRtiMsg& msg, 
                    MAKRti::DtTransportType transport,
                    const MAKRti::DtInetAddr& addr = MAKRti::DtInetAddr::inaddrAny());

   virtual bool flush();


   virtual bool processConnection( DtRtiConnection* conn,
                                   MAKRti::DtTransportType trans );

   virtual bool processConnections();

   virtual DtRtiMsg* getMessage();

   virtual DtRtiMsg* getMessageWithNetInfo(MAKRti::DtInetAddr *addr, MAKRti::DtTransportType *trans,
                                           DtU32 *priority = NULL);
   virtual MAKRti::DtInetAddr getLastDestAddr() const;
   virtual MAKRti::DtTransportType getLastTransportType() const;

#ifdef DtIFSPEC1516
   virtual int readFileDescriptor(rti1516::TransportationType transport);
#elif defined(DtIFSPEC1516E)
   virtual int readFileDescriptor(rti1516e::TransportationType transport);
#else
   virtual int readFileDescriptor(RTI::TransportType transport);
#endif

#if defined(_WIN32)
#ifdef DtIFSPEC1516
   virtual WSAEVENT readFileEvent(rti1516::TransportationType transport);
#elif defined(DtIFSPEC1516E)
   virtual WSAEVENT readFileEvent(rti1516e::TransportationType transport);
#else
   virtual WSAEVENT readFileEvent(RTI::TransportType transport);
#endif
#endif

   virtual DtSmRtiConnection* sharedMemConn() const;

   virtual void setNetConnMgr( DtRtiShmQNetMgr *smNetConnMgr );

   virtual bool sm_shutdown(bool ruthless = false);

   virtual bool setupSelectParams(DtSelectParam* params);

   virtual void setMsgFactory(DtRtiMsgFactory* fact);

   virtual void listenToMulticastGroup(const MAKRti::DtInetAddr& addr,
      const MAKRti::DtInetAddr& netInterface = MAKRti::DtInetAddr::inaddrAny());
   virtual void dropMulticastGroup(const MAKRti::DtInetAddr& addr,
      const MAKRti::DtInetAddr& netInterface = MAKRti::DtInetAddr::inaddrAny());
   virtual void forceDropMulticastGroup(const MAKRti::DtInetAddr& addr,
      const MAKRti::DtInetAddr& netInterface = MAKRti::DtInetAddr::inaddrAny());

   virtual void connectFederation(
#ifdef DtIFSPEC1516
      rti1516::FederateAmbassador* fedAmb,
#elif defined(DtIFSPEC1516E)
       rti1516e::FederateAmbassador* fedAmb,
#else
      RTI::FederateAmbassador* fedAmb,
#endif
      DtLogicalTimeFactory* logicalTimeFactoryPtr,
      MAKRti::DtString fedExName, 
      DtFederateHandle fedHandle,
      int fedExHandle);

   virtual bool repairFederationConnection();

   virtual void disconnectFederation();

   virtual bool wait(MAKRti::DtTime period = 0.0, bool unlockWhileWaiting = false);

   virtual int subscriberCount() const;


protected:

   virtual void startSharedMemMgr();

   virtual int initConnections();

   virtual bool createSharedMemConn();

public:
 
   static DtConnectionMgr* create(DtRIDParameters* p,  bool isFederate = false, 
      DtBoost::shared_ptr<DtRtiAssistantLrcClient> assistantClient = 
      DtBoost::shared_ptr<DtRtiAssistantLrcClient>());

   static DtConnectionMgr* createQMgr(DtRIDParameters* params, DtU32 numBuckets,
                                      DtU32 payloadSize, bool queueReliable,
                                      bool forceMgrSubscribe, 
                                      bool isFederate = false, 
                                      DtBoost::shared_ptr<DtRtiAssistantLrcClient> assistantClient = 
                                         DtBoost::shared_ptr<DtRtiAssistantLrcClient>(),
                                      DtRtiShmQNetMgr *smNetConnMgr = 0);

protected:
   DtSmRtiConnection * mySharedMemConn;
   MAKRti::DtString            mySharedMemName;
   DtRtiShmQNetMgr   * mySharedMemNetMgr;

   bool  myIsQueueManager;   
   DtU32 myNumBuckets;     
   DtU32 myPayloadSize;    
   bool  myQueueReliable;  
   bool  myForceSubscribe; 
   DtU32 myReadMask;       

   MAKRti::DtInetAddr        myLastDestAddr;
   MAKRti::DtTransportType   myLastTransportType; 
};

#endif