multidimensional/main.js

Generate self aware multidimensional arrays of anything.

function MultiDimensional(dimensions, cb) {

    var initialPosition = createInitialArray([], dimensions.length);

    this.dimensions = dimensions;
    this.matrix = createNDimensionArray(this, dimensions, initialPosition, cb);

    return this;
}

MultiDimensional.prototype.position = function(position, value) {

    var returnValue;

    if (value === undefined) {
        returnValue = getElement(this.matrix, position);
    } else {
        returnValue = setElement(this.matrix, position, value);
    }

    return returnValue;
};


// MultiDimensional Array Algorithms
// from: http://stackoverflow.com/questions/12588618/javascript-n-dimensional-array-creation/12588826#12588826

function createNDimensionArray(multidimensional, dimensions, position, cb) {

    var width;
    var rest;
    var newArray;
    var i;
    var currentDimension;
    var returnValue;

    if (dimensions.length > 0) {

        width = dimensions[0];
        rest = dimensions.slice(1);
        newArray = [];
        currentDimension = position.length - 1 - rest.length;
        position[currentDimension] = 0;

        for (i = 0; i < width; i++) {
            newArray[i] = createNDimensionArray(multidimensional, rest, position, cb);
            position[currentDimension]++;
        }

        returnValue = newArray;

    } else {

        if (typeof cb === 'function') {
            returnValue = cb(position.slice(), multidimensional);
        } else if (cb !== undefined) {
            returnValue = cb;
        } else {
            returnValue = null;
        }
    }

    return returnValue;
}

function getElement(array, indices) {

    var returnValue;

    if (indices.length === 0) {
        returnValue = array;
    } else {
        returnValue = getElement(array[indices[0]], indices.slice(1));
    }

    return returnValue;
}

function setElement(array, indices, value) {

    var returnValue;

    if (indices.length === 1) {
        array[indices[0]] = value;
        returnValue = value;
    } else {
        returnValue = setElement(array[indices[0]], indices.slice(1), value);
    }

    return returnValue;
}

function createInitialArray(arr, length) {

    var i;
    for (i = 0; i < length; i++) {
        arr[i] = 0;
    }

    return arr;
}


module.exports = MultiDimensional;