开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > Windows编程 > 通讯编程 > 查看源码
wnetmodel.cc
资源名称：ns-allinone-2.33.tar.gz　[点击查看]
上传用户：rrhhcc
上传日期：2015-12-11
资源大小：54129k
文件大小：10k
源码类别：

通讯编程

开发平台：

Visual C++

wnetmodel.cc：源码内容
  1. /*
  2.  * Network model with Wireless layout
  3.  *
  4.  */
  6. #include <stdlib.h>
  7. #include <math.h>
  8. #include <float.h>
  10. #include "random.h"
  11. #include "view.h"
  12. #include "netview.h"
  13. #include "animation.h"
  14. #include "queue.h"
  15. #include "edge.h"
  16. #include "node.h"
  17. #include "agent.h"
  18. #include "sincos.h"
  19. #include "state.h"
  20. #include "packet.h"
  21. #include "wnetmodel.h"
  23. class WirelessNetworkModelClass : public TclClass {
  24. public:
  25.   WirelessNetworkModelClass() : TclClass("NetworkModel/Wireless") {}
  26.   TclObject* create(int argc, const char*const* argv) {
  27.     if (argc < 5) 
  28.       return 0;
  29.     return (new WirelessNetModel(argv[4]));
  30.   }
  31. } wirelessnetworkmodel_class;
  33. WirelessNetModel::WirelessNetModel(const char *animator) :
  34.   NetModel(animator), edges_(0) {
  35.   bind("Wpxmax_", &pxmax_);
  36.   bind("Wpymax_", &pymax_);
  37. }
  39. WirelessNetModel::~WirelessNetModel() {
  40. }
  42. int WirelessNetModel::command(int argc, const char *const *argv) {
  43. if (argc == 2) {
  44. if (strcmp(argv[1], "layout") == 0) {
  45. /*
  46.  * <net> layout
  47.  */
  48. layout();
  49. NetModel::set_wireless();
  50. return (TCL_OK);
  51. }
  52. } 
  53. if (argc == 5) {
  54. if (strcmp(argv[1], "select-pkt") == 0) {
  55. NetModel::selectPkt(atoi(argv[2]), atoi(argv[3]), atoi(argv[4]));
  56. return (TCL_OK);
  57. }
  58. } 
  60. return (NetModel::command(argc, argv));
  61. }
  63. void WirelessNetModel::layout()
  64. {
  66. Node *n;
  67. for (n = nodes_; n != 0; n = n->next_)
  68.                 for (Edge* e = n->links(); e != 0; e = e->next_)  
  69.                         placeEdge(e, n); 
  71. nymax_ = pymax_ ;
  72. nymin_ = 0 ;
  74. }
  76. void WirelessNetModel::BoundingBox(BBox& bb)
  77. {                       
  78.         bb.xmin = bb.ymin = 0;
  79. bb.xmax = pxmax_;
  80. bb.ymax = pymax_;
  82.         for (Animation* a = drawables_; a != 0; a = a->next())
  83.                 a->merge(bb);
  84. }
  86. void WirelessNetModel::update(double now)
  87. {               
  88.         Animation *a, *n;
  89.         for (a = animations_; a != 0; a = n) {
  90.                 n = a->next();
  91.                 a->update(now);
  92.         }
  94. for (Node* x = nodes_; x != 0; x = x->next_) {
  95. x->update(now);
  96. moveNode(x);
  97. }
  98.         
  99.         /*
  100.          * Draw all animations and drawables on display to reflect
  101.          * current time.
  102.          */
  103.         now_ = now;
  104.         for (View* p = views_; p != 0; p = p->next_) {
  105.                 p->draw(); 
  106.         }               
  107. }       
  110. void WirelessNetModel::moveNode(Node *n)
  111. {
  113.         for (Edge *e = n->links(); e != 0; e = e->next_) {
  114.                 e->unmark();
  115.                 placeEdge(e, n);
  116.                 Node *dst = e->neighbor();
  117.                 // Should place reverse edges too
  118.                 Edge *p = dst->find_edge(n->num());
  119.                 if (p != NULL) {
  120.                         p->unmark();
  121.                         placeEdge(p, dst);
  122.                 }
  124.         }
  125.         for (Agent *a = n->agents(); a != NULL; a = a->next()) {
  126.                 a->mark(0), a->angle(NO_ANGLE);
  127.                 placeAgent(a, n);
  128.         }
  129. }
  131. // we need to place edge using the edge's real angle, instead of the
  132. // given one
  133. void WirelessNetModel::placeEdge(Edge* e, Node* src) const
  134. {
  135.         if (e->marked() == 0) {
  136.                 double hyp, dx, dy;
  137.                 Node *dst = e->neighbor();
  138.                 dx=dst->x()-src->x();
  139.                 dy=dst->y()-src->y();
  140.                 hyp=sqrt(dx*dx + dy*dy);
  141.            //   e->setAngle(atan2(dy,dx));
  142.                 double x0 = src->x() + src->size() * (dx/hyp) * .75;
  143.                 double y0 = src->y() + src->size() * (dy/hyp) * .75;
  144.                 double x1 = dst->x() - dst->size() * (dx/hyp) * .75;
  145.                 double y1 = dst->y() - dst->size() * (dy/hyp) * .75;
  146.                 e->place(x0, y0, x1, y1);
  148.                 /* Place the queue here too.  */
  149.                 EdgeHashNode *h = lookupEdgeHashNode(e->src(), e->dst());
  150. if (h != 0) {
  151.                     if (h->queue != 0)
  152.                         h->queue->place(e->size(), e->x0(), e->y0());
  153. }
  155.                 e->mark();
  156.         }
  157. }
  161. //----------------------------------------------------------------------
  162. // void 
  163. // WirelessNetModel::handle(const TraceEvent& e, double now, int direction)
  164. //   - packet handling stuff. use new packet
  165. //   - Given a trace event parses the event type and adds wireless
  166. //     extensions
  167. //----------------------------------------------------------------------
  168. void 
  169. WirelessNetModel::handle(const TraceEvent& e, double now, int direction) {
  170.   EdgeHashNode *ehn;
  171.   Edge* newEdge ;
  172.   Node *src, *dst, * n;
  173.   View * v;
  175.   switch (e.tt) {
  176.     case 'a': // Agent
  177.       NetModel::handle(e, now, direction);
  178.       // recalculate bounding box so that all agents will 
  179.       // be within view
  180.       for (v = views_; v != 0; v = v->next_) {
  181.         v->redrawModel();
  182.       }
  183.       break;
  185.     case 'n': // Node
  186.       NetModel::handle(e, now, direction);
  187.       // no need to reposition node if the node event
  188.       // is only for color change or label change
  189.       if ((strncmp(e.ne.node.state, "COLOR", 5) == 0) ||
  190.           (strncmp(e.ne.node.state, "DLABEL", 6) == 0)) {
  191.         return;
  192.       }
  194.       // placing the node position
  195.       n = lookupNode(e.ne.src);
  196.       if (n == 0)
  197.         return;
  199.       n->place(e.ne.x,e.ne.y);
  200.       n->placeorig(e.ne.x,e.ne.y);
  201.       n->setstart(now);
  202.       n->setend(e.ne.stoptime+now);
  203.       n->setvelocity(e.ne.x_vel_, e.ne.y_vel_);
  205.       moveNode(n);
  206.       break;
  207.   
  208.     case 'd':
  209.     case 'r':
  210.       src = lookupNode(e.pe.src);
  211.       if  (!src) {
  212.         fprintf(stderr,"node %d is not defined... ", e.pe.src);
  213.         break;
  214.       }
  216.       if (!(dst = lookupNode(e.pe.dst))) {
  217.         if (e.pe.dst != -1) { // broadcasting address
  218.           fprintf(stderr,"node %d is not defined... ", e.pe.dst);
  219.           break;
  220.         }
  221.       }
  223.       // Set node color based upon energy levels
  224.       /*
  225.       switch (e.pe.pkt.energy) {
  226.         case 3:
  227.           src->color("green");
  228.           break; 
  229.         case 2:
  230.           src->color("yellow");
  231.           break; 
  232.         case 1:
  233.           src->color("red");
  234.           break;
  235.         case 0:
  236.         src->color("black");
  237.         break;
  238.       } 
  239.       */
  241.       NetModel::handle(e, now, direction);
  242.       break;
  244.     case 'h':
  245.     case '+':
  246.     case '_':
  247.       if ( !(src= lookupNode(e.pe.src)) ) {
  248.         fprintf(stderr, "node %d is not defined... ", e.pe.src);
  249.         break;
  250.       } else { 
  251.         src->size(src->nsize());
  252.       }
  254.       if ( !(dst= lookupNode(e.pe.dst)) ) {
  255.         if (e.pe.dst != -1) { //broadcasting address
  256.           fprintf(stderr,"node %d is not defined... ", e.pe.dst);
  257.           break;
  258.         } else {
  259.           /*
  260.           if (e.pe.pkt.energy == 3) src->color("green");
  261.           if (e.pe.pkt.energy == 2) src->color("yellow");
  262.           if (e.pe.pkt.energy == 1) src->color("red");
  263.           if (e.pe.pkt.energy == 0) src->color("black");
  264.           */
  265.           NetModel::handle(e, now, direction);
  266.           break; 
  267.         }
  268.       } else {
  269.         dst->size(dst->nsize());
  270.       }
  272.       /*
  273.       if (e.pe.pkt.energy == 3) src->color("green");
  274.       if (e.pe.pkt.energy == 2) src->color("yellow");
  275.       if (e.pe.pkt.energy == 1) src->color("red");
  276.       if (e.pe.pkt.energy == 0) src->color("black");
  277.       */
  279.       // When a wireless packet is transferred then
  280.       // lookup to see if an edge has been created for that destination
  281.       ehn = lookupEdgeHashNode(e.pe.src, e.pe.dst);
  282.       if (ehn == 0 ) {
  283.         // if not then create one dynamically
  284.         //double bw = 1000000;
  285.         //double delay = 1.0000000000000008e-02;
  286.         double bw = 10.0;
  287.         double delay = 10.0;
  288.         double length = 0;
  289. // why it is 3? 
  290.         double dsize = 3;
  291.         double angle = 8.275783586691418e-313;
  292.         newEdge = new Edge(src, dst, dsize, bw, delay, length, angle, 1);
  294.         newEdge->init_color("black");
  295.         newEdge->visible(0);
  296.         enterEdge(newEdge);
  297.         newEdge->Animation::insert(&drawables_);
  298.         src->add_link(newEdge);
  299.       }
  301.       NetModel::handle(e, now, direction);
  302.       break;  
  304.     case 'l':
  305.       if ((strncmp(e.le.link.state, "in", 2) == 0) || 
  306.           (strncmp(e.le.link.state, "out", 3) == 0)) {
  307.         
  308.          // wireless process: dynamic link
  309.         if (strncmp(e.le.link.state, "in", 2)==0) {
  310.           if (direction==FORWARDS) {
  311.             Tcl& tcl = Tcl::instance();
  312.             tcl.evalf("%s add_link {%s}", name(), e.image);
  313.           } else {
  314.             // remove the link
  315.             removeLink(e);
  316.           } 
  317.         } 
  319.         if (strncmp(e.le.link.state, "out", 3)==0) {
  320.           if (direction==FORWARDS) {
  321.             removeLink(e);
  322.           } else {
  323.             Tcl& tcl = Tcl::instance();
  324.             tcl.evalf("%s add_link {%s}", name(), e.image);
  325.           }
  326.         }
  327.       } else {
  328.         NetModel::handle(e, now, direction);
  329.       }      
  330.       break;
  332.     default:
  333.       NetModel::handle(e, now, direction);
  334.       break;
  335.   }
  336. }
  338. void WirelessNetModel::addToEdgePool(Edge* e)
  339. {
  340.     WEdgeList* p = new WEdgeList;
  341.     p->wEdge = e;
  342.     p->next = edges_ ;
  343.     edges_ = p;
  344. }
  346. Edge* WirelessNetModel::removeFromEdgePool()
  347. {
  348.     if (edges_) {
  349.         WEdgeList* p = edges_ ;
  350.         edges_ = p->next;
  351. Edge* w = p->wEdge ;
  352.         delete p ;
  353. return w ;
  354.     }
  355.     else return (Edge*) 0 ;
  356. }
  359. //----------------------------------------------------------------------
  360. // void 
  361. // WirelessNetModel::removeLink(const TraceEvent& e)
  362. //----------------------------------------------------------------------
  363. void WirelessNetModel::removeLink(const TraceEvent& e) {
  364.   int src = e.le.src;
  365.   int dst = e.le.dst;
  367.   EdgeHashNode * h = lookupEdgeHashNode(src, dst);
  368.   EdgeHashNode * g = lookupEdgeHashNode(dst, src);
  369.   if (h == 0 || g == 0) {
  370.     // h,g = 0 or h,g !=0
  371.     return;
  372.   }
  373.                 
  374.   Edge * e1 = h->edge; 
  375.   Edge * e2 = g->edge;
  376.            
  377.   removeEdge(e1);
  378.   removeEdge(e2);
  380.   e1->detach();
  381.   e2->detach();
  382.  
  383.   Node* srcnode = e1->start();
  384.   Node* dstnode = e2->start();
  386.   // it is a duplex by default
  387.   srcnode->delete_link(e1);
  388.   dstnode->delete_link(e2);
  389.                         
  390.   delete e1;
  391.   delete e2;
  392. }
  394. void WirelessNetModel::removeLink(int src, int dst)
  395. {
  396.                     EdgeHashNode *h = lookupEdgeHashNode(src, dst);
  397.                     if (h == 0 ) return;
  398.                
  399.                     Edge* e1 = h->edge; 
  400.     if (e1 == 0) return ;
  401.            
  402.     e1->dec_usage();
  403.     if (e1->used() != 0) return;
  405.                     removeEdge(e1);
  407.                     e1->detach();
  408.  
  409.                     Node* srcnode = e1->start();
  411.                     srcnode->delete_link(e1);
  412.                         
  413.                     delete e1;
  415. }