iD/js/id/graph/graph.js

iD.Graph = function(entities) {
    if (!(this instanceof iD.Graph)) return new iD.Graph(entities);

    if (_.isArray(entities)) {
        this.entities = {};
        for (var i = 0; i < entities.length; i++) {
            this.entities[entities[i].id] = entities[i];
        }
    } else {
        this.entities = entities || {};
    }

    this.transients = {};
    this._parentWays = {};
    this._parentRels = {};
    this._fetches = {};

    if (iD.debug) {
        Object.freeze(this);
        Object.freeze(this.entities);
    }
};

iD.Graph.prototype = {
    entity: function(id) {
        return this.entities[id];
    },

    transient: function(entity, key, fn) {
        var id = entity.id,
            transients = this.transients[id] ||
            (this.transients[id] = {});

        if (transients[key]) {
            return transients[key];
        }

        return transients[key] = fn.call(entity);
    },

    parentWays: function(entity) {
        var ent, id, parents;

        if (!this._parentWays.calculated) {
            for (var i in this.entities) {
                ent = this.entities[i];
                if (ent && ent.type === 'way') {
                    for (var j = 0; j < ent.nodes.length; j++) {
                        id = ent.nodes[j];
                        parents = this._parentWays[id] = this._parentWays[id] || [];
                        if (parents.indexOf(ent) < 0) {
                            parents.push(ent);
                        }
                    }
                }
            }
            this._parentWays.calculated = true;
        }

        return this._parentWays[entity.id] || [];
    },

    parentRelations: function(entity) {
        var ent, id, parents;

        if (!this._parentRels.calculated) {
            for (var i in this.entities) {
                ent = this.entities[i];
                if (ent && ent.type === 'relation') {
                    for (var j = 0; j < ent.members.length; j++) {
                        id = ent.members[j].id;
                        parents = this._parentRels[id] = this._parentRels[id] || [];
                        if (parents.indexOf(ent) < 0) {
                            parents.push(ent);
                        }
                    }
                }
            }
            this._parentRels.calculated = true;
        }

        return this._parentRels[entity.id] || [];
    },

    merge: function(graph) {
        var entities = _.clone(this.entities);
        _.defaults(entities, graph.entities);
        return iD.Graph(entities);
    },

    replace: function(entity) {
        var entities = _.clone(this.entities);
        entities[entity.id] = entity;
        return iD.Graph(entities);
    },

    remove: function(entity) {
        var entities = _.clone(this.entities);

        if (entity.created()) {
            delete entities[entity.id];
        } else {
            entities[entity.id] = undefined;
        }

        return iD.Graph(entities);
    },

    // get all objects that intersect an extent.
    intersects: function(extent) {
        var items = [];
        for (var i in this.entities) {
            var entity = this.entities[i];
            if (entity && entity.intersects(extent, this)) {
                items.push(this.fetch(entity.id));
            }
        }
        return items;
    },

    // Resolve the id references in a way, replacing them with actual objects.
    fetch: function(id) {
        if (this._fetches[id]) return this._fetches[id];
        var entity = this.entities[id], nodes = [];
        if (!entity || !entity.nodes || !entity.nodes.length) return entity;
        for (var i = 0, l = entity.nodes.length; i < l; i++) {
            nodes[i] = this.fetch(entity.nodes[i]);
        }
        return (this._fetches[id] = iD.Entity(entity, {nodes: nodes}));
    },

    difference: function (graph) {
        var result = [], entity, oldentity, id;

        for (id in this.entities) {
            entity = this.entities[id];
            oldentity = graph.entities[id];
            if (entity !== oldentity) {
                if (entity && entity.type === 'way') {
                    result = oldentity ?
                        result
                            .concat(_.difference(entity.nodes, oldentity.nodes))
                            .concat(_.difference(oldentity.nodes, entity.nodes))
                        : result.concat(entity.nodes);
                }
                result.push(id);
            }
        }

        for (id in graph.entities) {
            entity = graph.entities[id];
            if (entity && !this.entities.hasOwnProperty(id)) {
                result.push(id);
                if (entity.type === 'way') result = result.concat(entity.nodes);
            }
        }

        return result.sort();
    },

    modified: function() {
        var result = [];
        _.each(this.entities, function(entity, id) {
            if (entity && entity.modified()) result.push(id);
        });
        return result;
    },

    created: function() {
        var result = [];
        _.each(this.entities, function(entity, id) {
            if (entity && entity.created()) result.push(id);
        });
        return result;
    },

    deleted: function() {
        var result = [];
        _.each(this.entities, function(entity, id) {
            if (!entity) result.push(id);
        });
        return result;
    }
};