Fix mgechev#13

mgechev · mgechev · commit feb63e322606 · 2015-02-22T20:25:21.000-08:00
diff --git a/src/data-structures/edge.js b/src/data-structures/edge.js
@@ -0,0 +1,20 @@
+(function (exports) {
+  'use strict';
+
+  /**
+   * Graph edge.
+   *
+   * @constructor
+   * @public
+   * @param {Vertex} e Vertex which this edge connects.
+   * @param {Vertex} v Vertex which this edge connects.
+   * @param {Number} distance Weight of the edge.
+   * @module data-structures/edge
+   */
+  exports.Edge = function (e, v, distance) {
+    this.e = e;
+    this.v = v;
+    this.distance = distance;
+  };
+
+})(typeof window === 'undefined' ? module.exports : window);
diff --git a/src/data-structures/vertex.js b/src/data-structures/vertex.js
@@ -0,0 +1,16 @@
+(function (exports) {
+  'use strict';
+
+  /**
+   * Graph vertex.
+   *
+   * @constructor
+   * @public
+   * @param {Number} id Id of the vertex.
+   * @module data-structures/vertex
+   */
+  exports.Vertex = function (id) {
+    this.id = id;
+  };
+
+})(typeof window === 'undefined' ? module.exports : window);
diff --git a/src/graphs/shortest-path/bellman-ford.js b/src/graphs/shortest-path/bellman-ford.js
@@ -35,20 +35,8 @@
 
   'use strict';
 
-  /**
-   * Graph edge.
-   *
-   * @constructor
-   * @public
-   * @param {Vertex} u Start vertex.
-   * @param {Vertex} v End vertex.
-   * @param {Number} weight Weight of the edge.
-   */
-  exports.Edge = function (u, v, weight) {
-    this.from = u;
-    this.to = v;
-    this.weight = weight;
-  };
+  exports.Vertex = require('../../data-structures/vertex').Vertex;
+  exports.Edge = require('../../data-structures/edge').Edge;
 
   /**
    * Computes shortest paths from a single source
diff --git a/src/graphs/spanning-trees/prim.js b/src/graphs/spanning-trees/prim.js
@@ -45,32 +45,8 @@
   'use strict';
 
   var Heap = require('../../data-structures/heap').Heap;
-
-  /**
-   * Graph vertex.
-   *
-   * @constructor
-   * @public
-   * @param {Number} id Id of the vertex.
-   */
-  exports.Vertex = function (id) {
-    this.id = id;
-  };
-
-  /**
-   * Graph edge.
-   *
-   * @constructor
-   * @public
-   * @param {Vertex} e Vertex which this edge connects.
-   * @param {Vertex} v Vertex which this edge connects.
-   * @param {Number} distance Weight of the edge.
-   */
-  exports.Edge = function (e, v, distance) {
-    this.e = e;
-    this.v = v;
-    this.distance = distance;
-  };
+  exports.Vertex = require('../../data-structures/vertex').Vertex;
+  exports.Edge = require('../../data-structures/edge').Edge;
 
   /**
    * Graph.
