networkMapWidget.h

Go to the documentation of this file.

/*******************************************************************************
** Copyright (c) 2012 MAK Technologies, Inc.
** All rights reserved.
*******************************************************************************/

#pragma once

#include "forwarderInfo.h"
#include "assistantEnums.h"
#include "assistantConnectionInfo.h"

#include <QtOpenGL/QtOpenGL>
//#include <qgl.h>
#include <QtCore/QTimer>
#include <string>


class DtNetworkMapCentralNode;
class DtNetworkMapCamera;
class DtNetworkMapNode;


class DT_DLL_ASSISTANTAPP DtNetworkMapWidget : public QGLWidget
{
   Q_OBJECT
public:
   DtNetworkMapWidget( QWidget * parent );

   virtual ~DtNetworkMapWidget();

   virtual QSize sizeHint() const;

   void addPeerForwarderNode( const MAKRti::DtString& forwarderHostName, int port );
   void removeForwarderNode( const MAKRti::DtString& forwarderHostName, int port );
   void addExecNode();
   void removeExecNode();
   void addFederateNode( int fedHandle, const MAKRti::DtString& federationName,
      const MAKRti::DtString& processName, const MAKRti::DtString& fedType,
      const MAKRti::DtString& fedName, bool notJoined = false, bool isDropped = false);
   void removeFederateNode(int fedHandle, const MAKRti::DtString& federationName );
   void checkAndAddMainForwarder( const MAKRti::DtString& forwarderHostName, int port );
   void checkAndAddLightweight( const MAKRti::DtString& connName );

   // Update the federate information
   void updateFederateInfo( int fedHandle,
      const MAKRti::DtString& federationName, int queueStatus,
      bool disconnected, bool reconnecting, bool inErrorState = false );

   // Information about a node
   bool isPeerForwarder( const MAKRti::DtString& forwarderHostName, int port );

   // update / refresh all values for federates 
   void refreshPositions();

   bool doesFederateExist( int fedHandle, const MAKRti::DtString& federationName );

   void clearFederateFlags( int fedHandle, const MAKRti::DtString& federationName );

   void forceFederateSelection( int fedHandle, const MAKRti::DtString& federationName);

   void forceRtiexecSelection();

   void forceForwarderSelection(const MAKRti::DtString& forwarderName, int port);

   void forceClearSelection();

   virtual void tick( float delta );

   virtual void selection( int x, int y, bool onHover = false );

   virtual void clearList();

   virtual bool hasSelection();

   virtual inline int getSelectedHandle() { return mySelectedId; };

   virtual void getSelectedNodeInfo( MAKRti::DtString& federationName,
      int& federateHandle, bool& isForwarder, bool& isExec, bool& isLightweightCenter );

   virtual QColor getFederationColor( const MAKRti::DtString & federationName );

   virtual void changeFederationColor( const MAKRti::DtString& federationName,
      const QColor& newColor );

   virtual void enableFederationColors(bool enable);

   virtual void setFederateIdentifierType(DtFederateIdentifierType fedIdType);

   virtual void setContextMenu(QMenu* contextMenu);

   virtual bool hasForwarder() const;
   
   virtual bool hasExec() const;
   
   virtual bool isLightweight() const;

public slots:

   void selectAndHoverText();

   void showContextMenu(const QPoint & pos);

signals:

   void federateSelected(const MAKRti::DtString& federationName, int federateHandle);

   void rtiexecSelected();

   void forwarderSelected(const MAKRti::DtString& forwarderName, int port);

   void peerForwarderSelected(const MAKRti::DtString& forwarderName, int port);

   void switchToPeerForwarder(const MAKRti::DtString& forwarderName, int port);

   void lightweightSelected();

   void nothingSelected();

protected:
   virtual void resizeEvent( QResizeEvent* evt );

   virtual void closeEvent( QCloseEvent* evt );

   virtual void wheelEvent(QWheelEvent* evt);

   virtual void mouseMoveEvent(QMouseEvent* evt);

   virtual void keyPressEvent(QKeyEvent* evt);

   virtual void mousePressEvent(QMouseEvent* evt);

   virtual void mouseDoubleClickEvent(QMouseEvent* evt);

   virtual void initializeGL();

   virtual void paintGL();

   virtual void drawLines();

   virtual void drawNodes();

   virtual void drawForwarderNodes();

   virtual void drawForwarderNode(DtNetworkMapNode* node, unsigned int index);

   virtual void drawForwarderNodeLines();

   virtual void drawText();

   virtual QColor addFederationColor( const MAKRti::DtString& federationName );

   virtual void drawWarning( int textIndex );

protected:
   int myWidth;                                // widget width
   int myHeight;                               // widget height
   unsigned mySelectedId;                      // currently selected node
   unsigned myHoverId;                         // id of node hovering over
   float myCameraDelta;                        // Camera Zoom Delta
   QPoint myOldMouse;                          // Old Mouse Position
   QPoint myNewMouse;                          // New Mouse Position
   QPoint myDeltaMouse;                        // Delta Mouse Position
   DtNetworkMapCentralNode* myCentralNode;     // Our central node and its children
   DtNetworkMapCamera* myCamera;               // Our Camera Class

   DtFederateIdentifierType myFedIdType;

   std::vector<QColor> myDefaultColors;

   bool myDisplayFederationColors;

   bool myNodesAreMoving;

   unsigned myNextDefaultColor;

   std::set<QRgb> myInUseColors;

   QTimer* myHoverTimer;                      // Our hover timer

   std::map< MAKRti::DtString, QColor > myFederationColors;

   QMenu* myCurrentContextMenu;


   static const int theRtiexecId;
   static const int theForwarderId;
   static const int theLigthweightCenterId;
   static const int theEmptyNodeId;


   static const MAKRti::DtString theRtiexecName;
   static const MAKRti::DtString theForwarderName;
   static const MAKRti::DtString theLigthweightCenterName;
   static const MAKRti::DtString theEmptyNodeName;

   static const int theStandardNodeRadius;

   static const unsigned theInvalidMapNodeId;

   static const unsigned theInitialFwId;
};