MapComplete/Logic/Osm/ExtractRelations.ts

import State from "../../State";

export interface Relation {
    id: number,
    type: "relation"
    members: {
        type: ("way" | "node" | "relation"),
        ref: number,
        role: string
    }[],
    tags: any,
    // Alias for tags; tags == properties
    properties: any
}

export default class ExtractRelations {

    public static RegisterRelations(overpassJson: any): void {
        const memberships = ExtractRelations.BuildMembershipTable(ExtractRelations.GetRelationElements(overpassJson))
        State.state.knownRelations.setData(memberships)
    }

    /**
     * Gets an overview of the relations - except for multipolygons. We don't care about those
     * @param overpassJson
     * @constructor
     */
    public static GetRelationElements(overpassJson: any): Relation[] {
        const relations = overpassJson.elements
            .filter(element => element.type === "relation" && element.tags.type !== "multipolygon")
        for (const relation of relations) {
            relation.properties = relation.tags
        }
        return relations
    }

    /**
     * Build a mapping of {memberId --> {role in relation, id of relation} }
     * @param relations
     * @constructor
     */
    public static BuildMembershipTable(relations: Relation[]): Map<string, { role: string, relation: Relation }[]> {
        const memberships = new Map<string, { role: string, relation: Relation }[]>()

        for (const relation of relations) {
            for (const member of relation.members) {

                const role = {
                    role: member.role,
                    relation: relation
                }
                const key = member.type + "/" + member.ref
                if (!memberships.has(key)) {
                    memberships.set(key, [])
                }
                memberships.get(key).push(role)
            }
        }

        return memberships
    }

}