forked from MapComplete/MapComplete
		
	
		
			
				
	
	
		
			93 lines
		
	
	
	
		
			3.3 KiB
		
	
	
	
		
			TypeScript
		
	
	
	
	
	
			
		
		
	
	
			93 lines
		
	
	
	
		
			3.3 KiB
		
	
	
	
		
			TypeScript
		
	
	
	
	
	
import TileHierarchy from "./TileHierarchy";
 | 
						|
import FeatureSource, {FeatureSourceForLayer, Tiled} from "../FeatureSource";
 | 
						|
import {OsmNode, OsmObject, OsmWay} from "../../Osm/OsmObject";
 | 
						|
import SimpleFeatureSource from "../Sources/SimpleFeatureSource";
 | 
						|
import FilteredLayer from "../../../Models/FilteredLayer";
 | 
						|
import {UIEventSource} from "../../UIEventSource";
 | 
						|
 | 
						|
 | 
						|
export default class FullNodeDatabaseSource implements TileHierarchy<FeatureSource & Tiled> {
 | 
						|
    public readonly loadedTiles = new Map<number, FeatureSource & Tiled>()
 | 
						|
    private readonly onTileLoaded: (tile: (Tiled & FeatureSourceForLayer)) => void;
 | 
						|
    private readonly layer: FilteredLayer
 | 
						|
    private readonly nodeByIds = new Map<number, OsmNode>();
 | 
						|
    private readonly parentWays = new Map<number, UIEventSource<OsmWay[]>>()
 | 
						|
 | 
						|
    constructor(
 | 
						|
        layer: FilteredLayer,
 | 
						|
        onTileLoaded: ((tile: Tiled & FeatureSourceForLayer) => void)) {
 | 
						|
        this.onTileLoaded = onTileLoaded
 | 
						|
        this.layer = layer;
 | 
						|
        if (this.layer === undefined) {
 | 
						|
            throw "Layer is undefined"
 | 
						|
        }
 | 
						|
    }
 | 
						|
 | 
						|
    public handleOsmJson(osmJson: any, tileId: number) {
 | 
						|
 | 
						|
        const allObjects = OsmObject.ParseObjects(osmJson.elements)
 | 
						|
        const nodesById = new Map<number, OsmNode>()
 | 
						|
 | 
						|
        for (const osmObj of allObjects) {
 | 
						|
            if (osmObj.type !== "node") {
 | 
						|
                continue
 | 
						|
            }
 | 
						|
            const osmNode = <OsmNode>osmObj;
 | 
						|
            nodesById.set(osmNode.id, osmNode)
 | 
						|
            this.nodeByIds.set(osmNode.id, osmNode)
 | 
						|
        }
 | 
						|
 | 
						|
        for (const osmObj of allObjects) {
 | 
						|
            if (osmObj.type !== "way") {
 | 
						|
                continue
 | 
						|
            }
 | 
						|
            const osmWay = <OsmWay>osmObj;
 | 
						|
            for (const nodeId of osmWay.nodes) {
 | 
						|
 | 
						|
                if (!this.parentWays.has(nodeId)) {
 | 
						|
                    const src =  new UIEventSource<OsmWay[]>([])
 | 
						|
                    this.parentWays.set(nodeId,src)
 | 
						|
                    src.addCallback(parentWays => {
 | 
						|
                        const tgs = nodesById.get(nodeId).tags
 | 
						|
                        tgs    ["parent_ways"] = JSON.stringify(parentWays.map(w => w.tags))
 | 
						|
                        tgs["parent_way_ids"] = JSON.stringify(parentWays.map(w => w.id))
 | 
						|
                    })
 | 
						|
                }
 | 
						|
                const src = this.parentWays.get(nodeId)
 | 
						|
                src.data.push(osmWay)
 | 
						|
                src.ping();
 | 
						|
            }
 | 
						|
        }
 | 
						|
        const now = new Date()
 | 
						|
        const asGeojsonFeatures = Array.from(nodesById.values()).map(osmNode => ({
 | 
						|
            feature: osmNode.asGeoJson(), freshness: now
 | 
						|
        }))
 | 
						|
 | 
						|
        const featureSource = new SimpleFeatureSource(this.layer, tileId)
 | 
						|
        featureSource.features.setData(asGeojsonFeatures)
 | 
						|
        this.loadedTiles.set(tileId, featureSource)
 | 
						|
        this.onTileLoaded(featureSource)
 | 
						|
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Returns the OsmNode with the corresponding id (undefined if not found)
 | 
						|
     * Note that this OsmNode will have a calculated tag 'parent_ways' and 'parent_way_ids', which are resp. stringified lists of parent way tags and ids
 | 
						|
     * @param id
 | 
						|
     * @constructor
 | 
						|
     */
 | 
						|
    public GetNode(id: number): OsmNode {
 | 
						|
        return this.nodeByIds.get(id)
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Gets the parent way list
 | 
						|
     * @param nodeId
 | 
						|
     * @constructor
 | 
						|
     */
 | 
						|
    public GetParentWays(nodeId: number): UIEventSource<OsmWay[]> {
 | 
						|
        return this.parentWays.get(nodeId)
 | 
						|
    }
 | 
						|
 | 
						|
}
 | 
						|
 |