mapcomplete/Logic/FeatureSource/Sources/TouchesBboxFeatureSource.ts

import { FeatureSource, FeatureSourceForLayer } from "../FeatureSource"
import StaticFeatureSource from "./StaticFeatureSource"
import { BBox } from "../../BBox"
import FilteredLayer from "../../../Models/FilteredLayer"
import { Store } from "../../UIEventSource"
import { Feature } from "geojson"

/**
 * Results in a feature source which has all the elements that touch the given features
 */
export default class BBoxFeatureSource<T extends Feature = Feature> extends StaticFeatureSource<T> {
    constructor(features: FeatureSource<T>, mustTouch: Store<BBox>) {
        super(
            features.features.mapD(
                (features) => {
                    if (mustTouch.data === undefined) {
                        return features
                    }
                    const box = mustTouch.data
                    return features.filter((feature) => {
                        if (feature.geometry.type === "Point") {
                            return box.contains(<[number, number]>feature.geometry.coordinates)
                        }
                        return box.overlapsWith(BBox.get(feature))
                    })
                },
                [mustTouch]
            )
        )
    }
}

export class BBoxFeatureSourceForLayer<T extends Feature = Feature>
    extends BBoxFeatureSource<T>
    implements FeatureSourceForLayer
{
    readonly layer: FilteredLayer

    constructor(features: FeatureSourceForLayer<T>, mustTouch: Store<BBox>) {
        super(features, mustTouch)
        this.layer = features.layer
    }
}