mathjs/lib/function/matrix/squeeze.js

module.exports = function (math) {
  var util = require('../../util/index.js'),

      Matrix = require('../../type/Matrix.js'),

      object = util.object,
      isArray = Array.isArray;

  /**
   * Remove singleton dimensions from a matrix
   *
   *     squeeze(x)
   *
   * @param {Matrix | Array} x
   * @return {Matrix | Array} res
   */
  math.squeeze = function squeeze (x) {
    if (arguments.length != 1) {
      throw new util.error.ArgumentsError('squeeze', arguments.length, 1);
    }

    if (isArray(x)) {
      return _squeezeArray(object.clone(x));
    }
    else if (x instanceof Matrix) {
      return new Matrix(_squeezeArray(x.toArray()));
    }
    else if (isArray(x.valueOf())) {
      return _squeezeArray(object.clone(x.valueOf()));
    }
    else {
      // scalar
      return object.clone(x);
    }
  };

  /**
   * Recursively squeeze a multi dimensional array
   * @param {Array} array
   * @return {Array} array
   * @private
   */
  function _squeezeArray(array) {
    if (array.length == 1) {
      // squeeze this array
      return _squeezeArray(array[0]);
    }
    else {
      // process all childs
      for (var i = 0, len = array.length; i < len; i++) {
        var child = array[i];
        if (isArray(child)) {
          array[i] = _squeezeArray(child);
        }
      }
      return array;
    }
  }
};