Vector in Javascript - Node.js Data Structure

Node.js examples for Data Structure:Vector

  1. HOME
  2. Node.js
  3. Data Structure
  4. Vector

Description

Vector in Javascript

Demo Code


Object.prototype.isArray = function(value) {
  return value &&
    typeof value === 'object' &&
    typeof value.length === 'number' &&
    typeof value.splice === 'function' &&
    !(value.propertyIsEnumerable('length'));
}

Number.prototype.add = function(other) { return this + other; };
Number.prototype.subtract = function(other) { return this - other; };
Number.prototype.multiply = function(other) { return this * other; };
Number.prototype.divide = function(other) { return this / other; };
Number.prototype.invert = function() { return this * -1; };
Number.prototype.copy = function() { return this; };
Number.prototype.root = function(degree) { return Math.sqrt(this); };
Number.prototype.square = function() { return this * this; };
Number.prototype.zero = 0;

var vector = function(spec) {
  var that = {};/*from  ww  w .  j a  v  a  2s. c o  m*/

  that.zero = spec.zero || 0;
  that.dim = spec.dim || spec.elements.length || 0;
  that.elements = spec.elements || $R(0, that.dim).map(function(el){return that.zero;});

  that.toString = function() {
    return that.elements.join('|');
  };

  that.set = function(other) {
    other.elements.each(function(el, index) {
      that.elements[index] = el;
    });
  };

  that.copy = function() {
    return vector({elements: that.elements.map(function(el) {
      return el.copy();
    })});
  };

  that.magnitude = function() {
  return that.elements.map(function(el) {
    return el.square();
  }).inject(that.zero, function(s, el) {return s.add(el);}).root(2);
  };

  that.sum = function() {
    return that.elements.inject(that.zero, function(s, el) {return s.add(el);});
  };

  that.ply = function(other, f) {
    return vector({elements: that.elements.zip(other.elements).map(function(el) {
      return f(el[0], el[1]);
    })});
  };

  that.add = function(other) {
    return that.ply(other, function(a, b) {return a.add(b);});
  };

  that.subtract = function(other) {
    return that.ply(other, function(a, b) {return a.subtract(b);});
  };

  that.multiply = function(other) {
    return that.ply(other, function(a, b) {return a.multiply(b);});
  };

  that.invert = function() {
    return vector({elements: that.elements.map(function(el) {
      return el.invert();
    })});
  };

  that.distance = function(other) {
  return that.ply(other, function(a, b) {
    return a.subtract(b).square();
  }).sum().root(2);
  };

  that.normalize = function() {
    var distance = that.distance(vector({dim: that.dim}));
  return vector({elements: that.elements.map(function(el) {
    return el.divide(distance);
  })});
  };

  that.scale = function(factor) {
  return vector({elements: that.elements.map(function(el) {
    return el.multiply(factor);
  })});
  };

  that.dot = function(other) {
  return that.ply(other, function(a, b) {
    return a.multiply(b);
  }).sum();
  };

  return that;
};


var vectortest = function(){
  var ya = vector({elements: [3, 5, 2]});
  var yo = vector({elements: [7, 7, 6]});

  var basicTest = [ya.sum(),
           ya.add(yo),
           ya.subtract(yo),
           ya.invert(),
           ya.distance(yo),
           yo.normalize(),
           yo.scale(33),
           yo.dot(ya),
           yo.magnitude(),
           ya.magnitude(),
           yo.dot(yo),
           yo.magnitude().square()];

  var ik = vector({elements: [vector({elements: [3, 2, 8]}), vector({elements: [-2, 3, -1]}), vector({elements: [4, 4, -8]})]});
  var ok = vector({elements: [vector({elements: [1, -4, 12]}), vector({elements: [3, 3, 3]}), vector({elements: [-1, -4, -13]})]});

  var furtherTest = [ik.dot(ok)];

  console.log(furtherTest.join(' --- '));
};

Related Tutorials