mobgen.c

无线通信的各种运动模型。适用于移动通信、无线传感器网络等领域。 包括：Random walk、random waypoint、random direction、boundless simulation

/******************************************************************************
*   File Name: mobgen.c
*   Purpose: Random Waypoint mobility model generator.
*   Author: Jeff Boleng
*   Date Created:
*
*   Edited 12 July 2006 by Jon Petty:  
*   Changed seed code to rand((int)time(NULL) + (int)getpid())
*
*   Copyright (C) 2004  Toilers Research Group -- Colorado School of Mines
*
*   Please see COPYRIGHT.TXT and LICENSE.TXT for copyright and license
*   details.
******************************************************************************/ 

#include <stdlib.h>
#include <stdio.h>
#include <time.h>
#include <math.h>
#include <unistd.h>

#undef rand48

#define true  1
#define false 0

int main(int argc, char *argv[])
{
	int i;
	int numNodes=0;
	int nextNode=0;
	double maxX=0.0, maxY=0.0;
	double endTime=0.0, lowest;
	double speedMean=0.0, speedDelta=0.0;
	double pauseMean=0.0, pauseDelta=0.0;
	int *moving;
	double *nextEvent;
	double *xLoc, *yLoc;
	double newX, newY, speed, dist;
	double speedLow, pauseLow;
	double speedRange, pauseRange;
	char output;

	if (argc == 10)
	{
		numNodes   = atoi(argv[1]);
		maxX       = atof(argv[2]);
		maxY       = atof(argv[3]);
		endTime    = atof(argv[4]);
		speedMean  = atof(argv[5]);
		speedDelta = atof(argv[6]);
		pauseMean  = atof(argv[7]);
		pauseDelta = atof(argv[8]);
		output     = argv[9][0];

		fprintf(stdout, "#~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~\n");
		fprintf(stdout, "#\tnumNodes   = %6d\n", numNodes);
		fprintf(stdout, "#\tmaxX       = %9.2f\n", maxX);
		fprintf(stdout, "#\tmaxY       = %9.2f\n", maxY);
		fprintf(stdout, "#\tendTime    = %9.2f\n", endTime);
		fprintf(stdout, "#\tspeedMean  = %9.2f\n", speedMean);
		fprintf(stdout, "#\tspeedDelta = %9.2f\n", speedDelta);
		fprintf(stdout, "#\tpauseMean  = %9.2f\n", pauseMean);
		fprintf(stdout, "#\tpauseDelta = %9.2f\n", pauseDelta);
		fprintf(stdout, "#\toutput     = %6c\n", output);
		fprintf(stdout, "#~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~\n\n");
	} else {
		fprintf(stdout, "Usage:  mobgen <number of nodes>\n");
		fprintf(stdout, "               <max-x> <max-y> <end time>\n");
		fprintf(stdout, "               <speed mean> <speed delta>\n");
		fprintf(stdout, "               <pause time> <pause time delta>\n");
		fprintf(stdout, "               <'N' or 'G'>\n");
		fprintf(stdout, "               'N' implies NS2 mobility file\n");
		fprintf(stdout, "               'G' implies gnuplot path file\n");
		return -1;
	}

	if (output == 'N')
	{
		fprintf(stdout, "# output format is NS2\n");
	} else if (output == 'G') {
		fprintf(stdout, "# output format is gnuplot\n");
		if (numNodes != 1)
		{
			fprintf(stderr, "Gnuplot output is only possible with one mobile node.\n");
			return -1;
		} else {
			fprintf(stdout, "plot \'-\' notitle with linespoints\n");
		}
	} else {
		fprintf(stderr, "Unknown output type requested\n");
		return -1;
	}

  #ifdef rand48
	srand48((int)time(NULL) + (int)getpid());
  #else
	srand((int)time(NULL) + (int)getpid());
  #endif
	speedLow = speedMean-speedDelta;
	pauseLow = pauseMean-pauseDelta;
	speedRange = 2*speedDelta;
	pauseRange = 2*pauseDelta;

	moving = (int*)malloc(sizeof(int)*numNodes);
	nextEvent = (double*)malloc(sizeof(double)*numNodes);
	xLoc = (double*)malloc(sizeof(double)*numNodes);
	yLoc = (double*)malloc(sizeof(double)*numNodes);

	fprintf(stdout, "#\tInitial positions:\n");
	for (i=0; i<numNodes; i++)
	{
    #ifdef rand48
		moving[i] = (drand48()>=0.5)?true:false;
		xLoc[i] = drand48()*maxX;
		yLoc[i] = drand48()*maxY;
    #else
		moving[i] = (rand()>=(RAND_MAX/2))?true:false;
		/*
		fprintf(stdout, "$node_(%d) is ", i);
		if (moving[i])
	{
		  fprintf(stderr, "moving\n"); 
	} else {
		  fprintf(stderr, "not moving\n"); 
	}
		*/
		xLoc[i] = ((double)rand()/(double)RAND_MAX)*maxX;
		yLoc[i] = ((double)rand()/(double)RAND_MAX)*maxY;
    #endif
		/*fprintf(stdout, "%d:  %s moving\n", i, ((moving[i])?"is not":"is"));*/
		if (output == 'N')
		{
			fprintf(stdout, "$node_(%d) set X_ %.12f\n", i, xLoc[i]);
			fprintf(stdout, "$node_(%d) set Y_ %.12f\n", i, yLoc[i]);
			fprintf(stdout, "$node_(%d) set Z_ %.12f\n", i, 0.0);
		} else if (output == 'G') {
			xLoc[i] = maxX/2.0;
			yLoc[i] = maxY/2.0;
			fprintf(stdout, "%.12f %.12f\n", xLoc[i], yLoc[i]);
		}

		nextEvent[i] = 0.0;
	}
	if (output == 'N')
	{
		fprintf(stdout, "\n\n#\tMovements:\n");
	}

	lowest = endTime;  /* initialize high so all starting movements are
	                      scheduled and output */
	while (lowest <= endTime)
	{
		/* schedule the next event */
		if (moving[nextNode])
		{
			moving[nextNode] = false;
			speed = 0.0;
			if (output == 'N')
			{
				fprintf(stdout, "$ns_ at %.12f \"$node_(%d) setdest %.12f %.12f %.12f\"\n",
				        nextEvent[nextNode], nextNode, xLoc[nextNode], yLoc[nextNode], speed);
			}
      #ifdef rand48
			nextEvent[nextNode] += drand48()*pauseRange + pauseLow;
      #else
			nextEvent[nextNode] += ((double)rand()/(double)RAND_MAX)*pauseRange + pauseLow;
      #endif
		} else /* not moving */ {
			moving[nextNode] = true;
      #ifdef rand48
			speed = drand48()*speedRange + speedLow;
			newX = drand48()*maxX;
			newY = drand48()*maxY;
      #else
			speed = ((double)rand()/(double)RAND_MAX)*speedRange + speedLow;
			newX = ((double)rand()/(double)RAND_MAX)*maxX;
			newY = ((double)rand()/(double)RAND_MAX)*maxY;
      #endif
			if (output == 'N')
			{
				fprintf(stdout, "$ns_ at %.12f \"$node_(%d) setdest %.12f %.12f %.12f\"\n",
				        nextEvent[nextNode], nextNode, newX, newY, speed);
			} else if (output == 'G') {
				fprintf(stdout, "%.12f %.12f # node %d at %.10f speed=%.10f\n", newX, newY, nextNode, nextEvent[nextNode], speed);
			}

			dist = sqrt((newX-xLoc[nextNode])*(newX-xLoc[nextNode]) + (newY-yLoc[nextNode])*(newY-yLoc[nextNode]));
			xLoc[nextNode] = newX;
			yLoc[nextNode] = newY;
			nextEvent[nextNode] += dist/speed;
		} /* if (moving[nextNode]) */

		/* find new lowest */
		lowest = endTime + 1.0;
		for (i=0; i<numNodes; i++)
		{
			if (nextEvent[i] <= lowest)
			{
				lowest = nextEvent[i];
				nextNode = i;
			} /* if (nextEvent[i] <= lowest) */
		} /* for (i=0; i<numNodes; i++) */
	} /* while (lowest <= endTime) */

	if (output == 'N')
	{
		fprintf(stdout, "\n\n\n");
	} else if (output == 'G') {
		fprintf(stdout, "e\n\n");
	}

	free(nextEvent);
	free(xLoc);
	free(yLoc);

	return 0;
}