improved display of tunnels in network simulator

git-svn-id: http://svn.code.sf.net/p/retroshare/code/trunk@5643 b45a01b8-16f6-495d-af2f-9b41ad6348cc

libretroshare/src/tests/network_simulator/Network.cpp

@@ -40,7 +40,7 @@ bool Network::initRandom(uint32_t nb_nodes,float connexion_probability)
 	//
 	for(uint32_t i=0;i<nb_nodes;++i)
 	{
-		int nb_friends = (int)ceil(-log(1-drand48())) ;
+		int nb_friends = std::min((int)nb_nodes-1,(int)ceil(-log(1-drand48())/(1.00001-connexion_probability))) ;
 
 		for(uint32_t j=0;j<nb_friends;++j)
 		{

libretroshare/src/tests/network_simulator/NetworkViewer.cpp

@@ -11,15 +11,17 @@ NetworkViewer::NetworkViewer(QWidget *parent,Network&net)
 	:  QGLViewer(parent),_network(net) , timerId(0)
 {
 	_current_selected_node = -1 ;
+	_current_displayed_node = -1 ;
 	_dragging = false ;
+	_nodes_need_recomputing = true ;
 
 	_node_coords.resize(net.n_nodes()) ;
 	_node_speeds.resize(net.n_nodes()) ;
 
 	for(int i=0;i<_node_coords.size();++i)
 	{
-		_node_coords[i].x = drand48()*width() ;
+		_node_coords[i].x = drand48()*width()*5;
-		_node_coords[i].y = drand48()*height() ;
+		_node_coords[i].y = drand48()*height()*5 ;
 		_node_speeds[i].x = 0 ;
 		_node_speeds[i].y = 0 ;
 	}
@@ -30,6 +32,8 @@ NetworkViewer::NetworkViewer(QWidget *parent,Network&net)
 
 	action_ManageHash = new QAction(QString("Manage new random hash"),this) ;
 	QObject::connect(action_ManageHash,SIGNAL(triggered()),this,SLOT(actionManageHash())) ;
+
+	setMouseTracking(true) ;
 }
 
 void NetworkViewer::draw() 
@@ -52,6 +56,8 @@ void NetworkViewer::draw()
 
 	// Now, draw all edges
 
+	std::set<int> tunnel_nodes ;
+
 	glEnable(GL_LINE_SMOOTH) ;
 	glBegin(GL_LINES) ;
 
@@ -65,7 +71,11 @@ void NetworkViewer::draw()
 		for(std::set<uint32_t>::const_iterator it(neighs.begin());it!=neighs.end();++it)
 		{
 			if(traffic_info.local_src.find(_network.node(*it).id())!=traffic_info.local_src.end() || traffic_info.local_dst.find(_network.node(*it).id())!=traffic_info.local_dst.end())
+			{
 				glColor3f(0.9f,0.4f,0.2f) ;
+				tunnel_nodes.insert(i) ;
+				tunnel_nodes.insert(*it) ;
+			}
 			else
 				glColor3f(0.4f,0.4f,0.4f) ;
 
@@ -82,21 +92,64 @@ void NetworkViewer::draw()
 	// Draw all nodes.
 	//
 	glEnable(GL_POINT_SMOOTH) ;
+	glPointSize(20.0f) ;
+	glBegin(GL_POINTS) ;
+
+	if(_current_selected_node > -1)
+	{
+		glColor4f(1.0f,0.2f,0.1f,0.7f) ;
+		glVertex2f(_node_coords[_current_selected_node].x, _node_coords[_current_selected_node].y) ;
+	}
+	glEnd() ;
+
 	glPointSize(10.0f) ;
 	glBegin(GL_POINTS) ;
 
 	for(uint32_t i=0;i<_network.n_nodes();++i)
 	{
-		if((int)i == _current_selected_node)
+		float r = 0.8 ;
-			glColor3f(1.0f,0.2f,0.1f) ;
+		float g = 0.8 ;
-		else
+		float b = 0.8 ;
-			glColor3f(0.8f,0.8f,0.8f) ;
+
+		if(!_network.node(i).providedHashes().empty())
+			r *= 2.0, g /= 2.0, b /= 2.0f ;
+
+		if(!_network.node(i).managedHashes().empty())
+			g *= 2.0, b /= 2.0, r /= 2.0f ;
+
+		if(tunnel_nodes.find(i) != tunnel_nodes.end() && r==0.8f && g==0.8f && b==0.8f)
+			r = 0.9f, g=0.4f,b=0.2f ;
+
+		glColor3f(r,g,b) ;
 
 		glVertex2f(_node_coords[i].x, _node_coords[i].y) ;
 	}
 
 	glEnd() ;
 
+	// Draw info about current node under mouse.
+	//
+	if(_current_displayed_node > -1)
+	{
+		const PeerNode& node(_network.node(_current_displayed_node)) ;
+		int offset = 0 ;
+		int text_height = 15 ;
+
+		drawText(10+_node_coords[_current_displayed_node].x,offset + height()-_node_coords[_current_displayed_node].y, "Node id = " + QString::fromStdString(node.id())) ;
+		offset += text_height ;
+
+		for(std::set<TurtleFileHash>::const_iterator it(node.providedHashes().begin());it!=node.providedHashes().end();++it)
+		{
+			drawText(10+_node_coords[_current_displayed_node].x,offset + height()-_node_coords[_current_displayed_node].y, "Server for hash " + QString::fromStdString(*it) );
+			offset += text_height ;
+		}
+
+		for(std::set<TurtleFileHash>::const_iterator it(node.managedHashes().begin());it!=node.managedHashes().end();++it)
+		{
+			drawText(10+_node_coords[_current_displayed_node].x,offset + height()-_node_coords[_current_displayed_node].y, "Client for hash " + QString::fromStdString(*it) ) ;
+			offset += text_height ;
+		}
+	}
 
 	glMatrixMode(GL_MODELVIEW) ;
 	glPopMatrix() ;
@@ -222,6 +275,9 @@ void NetworkViewer::timerEvent(QTimerEvent *event)
 	 if(!isVisible())
 		 return ;
 
+	 if(_nodes_need_recomputing)
+	 {
+		 std::cerr << "Updating forces..."<< std::endl;
 		 static const int S = 256 ;
 		 static double *forceMap = new double[2*S*S] ;
 
@@ -267,7 +323,7 @@ void NetworkViewer::timerEvent(QTimerEvent *event)
 		 bool itemsMoved = false;
 		 for(uint32_t i=0;i<_node_coords.size();++i)
 		 {
-		 if( fabsf(_node_coords[i].x - new_coords[i].x) > 0.2 || fabsf(_node_coords[i].y - new_coords[i].y) > 0.2)
+			 if( fabsf(_node_coords[i].x - new_coords[i].x) > 1.0 || fabsf(_node_coords[i].y - new_coords[i].y) > 1.0)
 				 itemsMoved = true;
 
 			 //std::cerr << "Old i = " << _node_coords[i].x << ", new = " << new_coords[i].x << std::endl;
@@ -275,9 +331,10 @@ void NetworkViewer::timerEvent(QTimerEvent *event)
 		 }
 
 		 if (!itemsMoved) {
-        killTimer(timerId);
+	//		 killTimer(timerId);
 			 //#ifdef DEBUG_ELASTIC
 			 std::cerr << "Killing timr" << std::endl ;
+			 _nodes_need_recomputing = false ;
 			 //#endif
 			 timerId = 0;
 		 }
@@ -286,6 +343,9 @@ void NetworkViewer::timerEvent(QTimerEvent *event)
 			 updateGL() ;
 			 usleep(2000) ;
 		 }
+	 }
+	 else
+		 updateGL() ;
 
 }
 
@@ -296,11 +356,22 @@ void NetworkViewer::mouseMoveEvent(QMouseEvent *e)
 		_node_coords[_current_selected_node].x = e->x() ;
 		_node_coords[_current_selected_node].y = height() - e->y() ;
 
-		if(timerId == 0)
+		_nodes_need_recomputing = true ;
-			timerId = startTimer(1000/25) ;
 
 		updateGL() ;
 	}
+
+	float x = e->x() ;
+	float y = height()-e->y() ;
+
+	_current_displayed_node = -1 ;
+
+	for(uint32_t i=0;i<_node_coords.size();++i)
+		if( pow(_node_coords[i].x-x,2)+pow(_node_coords[i].y-y,2) < 5*5)
+		{
+			_current_displayed_node = i;
+			break ;
+		}
 }
 
 void NetworkViewer::mouseReleaseEvent(QMouseEvent *e)

libretroshare/src/tests/network_simulator/NetworkViewer.h

@@ -60,16 +60,19 @@ class NetworkViewer: public QGLViewer
 		std::vector<NodeCoord> _node_coords ;
 		std::vector<NodeCoord> _node_speeds ;
 
-		static const float MASS_FACTOR      = 7 ;
+		static const float MASS_FACTOR      = 10 ;
-		static const float FRICTION_FACTOR  = 9.8 ;
+		static const float FRICTION_FACTOR  = 15.8 ;
 		static const float REPULSION_FACTOR = 8  ;
 		static const float NODE_DISTANCE    = 30.0 ;
 
 		int timerId ;
 
 		int _current_selected_node ;
+		int _current_displayed_node ;
 		bool _dragging ;
+		bool _nodes_need_recomputing ;
 
 		QAction *action_ManageHash ;
 		QAction *action_ProvideHash ;
 };
+

libretroshare/src/tests/network_simulator/main.cpp

@@ -17,8 +17,12 @@ int main(int argc, char *argv[])
 #endif
 		argstream as(argc,argv) ;
 		bool show_gui = false;
+		int nb_nodes = 20 ;
+		float connexion_probability = 0.2 ;
 
 		as >> option('i',"gui",show_gui,"show gui (vs. do the pipeline automatically)")
+			>> parameter('n',"nodes",nb_nodes,"number of nodes in the network")
+			>> parameter('p',"connexion probability",connexion_probability,"probability that two nodes are connected (exponential law)")
 			>> help() ;
 
 		as.defaultErrorHandling() ;
@@ -27,7 +31,7 @@ int main(int argc, char *argv[])
 
 		Network network ;
 
-		network.initRandom(20,0.2) ;
+		network.initRandom(nb_nodes,connexion_probability) ;
 
 		rsPeers = new MonitoredRsPeers(network) ;