Files
History-user/src/uhm/lib/map/engines/snapUtils.ts
T
taDuc 820e0b5216
Build and Release / release (push) Successful in 1m30s
basic Uer UI
2026-06-06 15:36:15 +07:00

581 lines
21 KiB
TypeScript

import maplibregl from "maplibre-gl";
// SHIFT/ALT snap should be forgiving while drawing quickly.
// Vertices get a larger radius and always win over edges when both are available.
const VERTEX_SNAP_THRESHOLD_PX = 34;
const EDGE_SNAP_THRESHOLD_PX = 24;
const QUERY_THRESHOLD_PX = Math.max(VERTEX_SNAP_THRESHOLD_PX, EDGE_SNAP_THRESHOLD_PX);
const COORDINATE_EPSILON = 1e-10;
const SEGMENT_ENDPOINT_EPSILON = 1e-7;
type Coordinate = [number, number];
type GeometryWithCoordinates = Exclude<GeoJSON.Geometry, GeoJSON.GeometryCollection> & {
coordinates: unknown;
};
export type SnapResult = {
lngLat: maplibregl.LngLat;
type: "vertex" | "edge" | "none";
featureId?: string | number;
ringCoords?: Coordinate[];
vertexIdx?: number;
edgeIdx?: number;
};
export function getSnapVertexCoordinate(snap: SnapResult): [number, number] | null {
if (snap.type !== "vertex" || snap.vertexIdx === undefined || !snap.ringCoords) return null;
const coordinate = snap.ringCoords[snap.vertexIdx];
return coordinate ? [coordinate[0], coordinate[1]] : null;
}
export function snapToNearestGeometry(
map: maplibregl.Map,
lngLat: maplibregl.LngLat,
pointPx: maplibregl.Point,
excludeFeatureId?: string | number | null
): maplibregl.LngLat {
return snapToNearestGeometryDetailed(map, lngLat, pointPx, excludeFeatureId).lngLat;
}
export function snapToNearestGeometryDetailed(
map: maplibregl.Map,
lngLat: maplibregl.LngLat,
pointPx: maplibregl.Point,
excludeFeatureId?: string | number | null,
includeFeatureId?: string | number | null
): SnapResult {
const bbox: [maplibregl.PointLike, maplibregl.PointLike] = [
[pointPx.x - QUERY_THRESHOLD_PX, pointPx.y - QUERY_THRESHOLD_PX],
[pointPx.x + QUERY_THRESHOLD_PX, pointPx.y + QUERY_THRESHOLD_PX],
];
const snapLayerIds = getSnapLayerIds(map);
if (!snapLayerIds.length) return { lngLat, type: "none" };
const features = map.queryRenderedFeatures(bbox, {
layers: snapLayerIds,
});
let nearestVertexDist = Infinity;
let nearestVertexLngLat: maplibregl.LngLat | null = null;
let nearestVertexFeatureId: string | number | undefined = undefined;
let nearestVertexRing: Coordinate[] | null = null;
let nearestVertexIdx: number = -1;
let nearestEdgeDist = Infinity;
let nearestEdgeLngLat: maplibregl.LngLat | null = null;
let nearestEdgeFeatureId: string | number | undefined = undefined;
let nearestEdgeRing: Coordinate[] | null = null;
let nearestEdgeIdx: number = -1;
const getDistSq = (p1: maplibregl.Point, p2: maplibregl.Point) => {
return (p1.x - p2.x) ** 2 + (p1.y - p2.y) ** 2;
};
// Tìm t của điểm gần nhất trên đoạn thẳng [a, b] so với điểm p (tính trên pixel màn hình)
const getClosestTOnSegment = (p: maplibregl.Point, a: maplibregl.Point, b: maplibregl.Point): number => {
const atob = { x: b.x - a.x, y: b.y - a.y };
const atop = { x: p.x - a.x, y: p.y - a.y };
const lenSq = atob.x * atob.x + atob.y * atob.y;
if (lenSq === 0) return 0;
let t = (atop.x * atob.x + atop.y * atob.y) / lenSq;
t = Math.max(0, Math.min(1, t));
return t;
};
const getPointOnSegment = (a: maplibregl.Point, b: maplibregl.Point, t: number): maplibregl.Point => {
return new maplibregl.Point(a.x + (b.x - a.x) * t, a.y + (b.y - a.y) * t);
};
// Nội suy trên Mercator bằng đúng t pixel đã chọn để điểm snap nằm ổn định trên đoạn gốc.
const getPointOnLngLatSegment = (a: Coordinate, b: Coordinate, t: number): maplibregl.LngLat => {
if (t <= SEGMENT_ENDPOINT_EPSILON) return new maplibregl.LngLat(a[0], a[1]);
if (t >= 1 - SEGMENT_ENDPOINT_EPSILON) return new maplibregl.LngLat(b[0], b[1]);
const toMercatorY = (lat: number) => {
if (lat > 85.0511) lat = 85.0511;
if (lat < -85.0511) lat = -85.0511;
return Math.log(Math.tan(Math.PI / 4 + (lat * Math.PI) / 360));
};
const fromMercatorY = (y: number) => {
return (360 / Math.PI) * Math.atan(Math.exp(y)) - 90;
};
const ax = a[0];
const ay = toMercatorY(a[1]);
const bx = b[0];
const by = toMercatorY(b[1]);
const resultLng = ax + (bx - ax) * t;
const resultLat = fromMercatorY(ay + (by - ay) * t);
return new maplibregl.LngLat(resultLng, resultLat);
};
const processVertex = (coordinate: Coordinate, featureId: string | number | undefined, ring: Coordinate[], idx: number) => {
const vertexLngLat = new maplibregl.LngLat(coordinate[0], coordinate[1]);
const vertexPx = map.project(vertexLngLat);
const distSq = getDistSq(pointPx, vertexPx);
if (
distSq < nearestVertexDist &&
distSq <= VERTEX_SNAP_THRESHOLD_PX ** 2
) {
nearestVertexDist = distSq;
nearestVertexLngLat = vertexLngLat;
nearestVertexFeatureId = featureId;
nearestVertexRing = ring;
nearestVertexIdx = idx;
}
};
const processLineString = (line: number[][], featureId: string | number | undefined, forceClosed = false) => {
if (!line || line.length < 2) return;
const parsedCoords = line.map(c => toCoordinate(c)).filter((c): c is Coordinate => c !== null);
const treatAsClosed = forceClosed || isClosedRing(parsedCoords);
const lineCoords = treatAsClosed ? removeClosingCoordinate(parsedCoords) : parsedCoords;
if (lineCoords.length < 2) return;
const ringForSnap = treatAsClosed ? closeRing(lineCoords) : lineCoords;
for (let i = 0; i < lineCoords.length; i++) {
processVertex(lineCoords[i], featureId, ringForSnap, i);
}
const segmentCount = treatAsClosed ? lineCoords.length : lineCoords.length - 1;
for (let i = 0; i < segmentCount; i++) {
const start = lineCoords[i];
const end = lineCoords[(i + 1) % lineCoords.length];
const p1LngLat = new maplibregl.LngLat(start[0], start[1]);
const p2LngLat = new maplibregl.LngLat(end[0], end[1]);
const p1 = map.project(p1LngLat);
const p2 = map.project(p2LngLat);
const closestT = getClosestTOnSegment(pointPx, p1, p2);
const closestPx = getPointOnSegment(p1, p2, closestT);
const distSq = getDistSq(pointPx, closestPx);
if (distSq < nearestEdgeDist && distSq <= EDGE_SNAP_THRESHOLD_PX ** 2) {
nearestEdgeDist = distSq;
nearestEdgeLngLat = getPointOnLngLatSegment(start, end, closestT);
nearestEdgeFeatureId = featureId;
nearestEdgeRing = ringForSnap;
nearestEdgeIdx = i;
}
}
};
const processPoint = (coordinate: unknown, featureId: string | number | undefined) => {
const point = toCoordinate(coordinate);
if (point) processVertex(point, featureId, [point], 0);
};
for (const feature of features) {
if (!feature.geometry) continue;
// Bỏ qua các layer preview hoặc edit để không tự snap vào nét đang vẽ dở.
if (feature.layer.id.includes("preview") || feature.layer.id.includes("edit-")) {
continue;
}
// Bỏ qua chính đối tượng đang được chỉnh sửa để không tự snap vào chính nó
const fId = feature.properties?.id ?? feature.id;
if (excludeFeatureId !== undefined && excludeFeatureId !== null && fId !== undefined && fId !== null) {
if (String(fId) === String(excludeFeatureId)) {
continue;
}
}
if (includeFeatureId !== undefined && includeFeatureId !== null && fId !== undefined && fId !== null) {
if (String(fId) !== String(includeFeatureId)) {
continue;
}
}
let geometry = feature.geometry;
const sourceId = feature.layer.source;
const origFeature = getOriginalFeature(map, sourceId, fId);
if (origFeature && origFeature.geometry) {
geometry = origFeature.geometry;
}
const type = geometry.type;
if (type === "GeometryCollection") continue;
const coords = (geometry as GeometryWithCoordinates).coordinates;
// Xử lý cả Polygon và LineString vì viền bản đồ (border) đôi khi được render dưới dạng LineString
if (type === "Polygon") {
for (const ring of asCoordinateMatrix(coords)) processLineString(ring, fId, true);
} else if (type === "MultiPolygon") {
for (const poly of asCoordinateTensor(coords)) {
for (const ring of poly) processLineString(ring, fId, true);
}
} else if (type === "LineString") {
processLineString(asCoordinateArray(coords), fId);
} else if (type === "MultiLineString") {
for (const line of asCoordinateMatrix(coords)) processLineString(line, fId);
} else if (type === "Point") {
processPoint(coords, fId);
} else if (type === "MultiPoint") {
for (const point of asCoordinateArray(coords)) processPoint(point, fId);
}
}
if (nearestVertexLngLat) {
return {
lngLat: nearestVertexLngLat,
type: "vertex",
featureId: nearestVertexFeatureId,
ringCoords: nearestVertexRing || undefined,
vertexIdx: nearestVertexIdx
};
}
if (nearestEdgeLngLat && nearestEdgeRing) {
const edgeLngLat = nearestEdgeLngLat as maplibregl.LngLat;
const edgeRing = nearestEdgeRing as Coordinate[];
return {
lngLat: edgeLngLat,
type: "edge",
featureId: nearestEdgeFeatureId,
ringCoords: edgeRing,
edgeIdx: nearestEdgeIdx
};
}
return { lngLat, type: "none" };
}
function getSnapLayerIds(map: maplibregl.Map): string[] {
const systemGeometrySources = new Set(["countries", "places"]);
const style = map.getStyle();
if (!style?.layers?.length) return [];
return style.layers
.filter((layer) => {
if (!("source" in layer)) return false;
if (!systemGeometrySources.has(String(layer.source))) return false;
if (layer.id.includes("preview") || layer.id.includes("edit-")) return false;
return true;
})
.map((layer) => layer.id)
.filter((layerId) => Boolean(map.getLayer(layerId)));
}
function toCoordinate(value: unknown): Coordinate | null {
if (!Array.isArray(value) || value.length < 2) return null;
const lng = Number(value[0]);
const lat = Number(value[1]);
if (!Number.isFinite(lng) || !Number.isFinite(lat)) return null;
return [lng, lat];
}
function asCoordinateArray(value: unknown): number[][] {
return Array.isArray(value) ? value as number[][] : [];
}
function asCoordinateMatrix(value: unknown): number[][][] {
return Array.isArray(value) ? value as number[][][] : [];
}
function asCoordinateTensor(value: unknown): number[][][][] {
return Array.isArray(value) ? value as number[][][][] : [];
}
export function getArea(points: [number, number][]): number {
let area = 0;
for (let i = 0; i < points.length; i++) {
const p1 = points[i];
const p2 = points[(i + 1) % points.length];
area += (p1[0] + p2[0]) * (p1[1] - p2[1]);
}
return Math.abs(area / 2);
}
export function tracePathBetweenPoints(
ring: [number, number][],
startIdx: number,
endIdx: number
): [number, number][] {
const isClosed = isClosedRing(ring);
const workingRing = (isClosed ? ring.slice(0, -1) : ring) as [number, number][];
const n = workingRing.length;
const normalizedStartIdx = isClosed && startIdx === ring.length - 1 ? 0 : startIdx;
const normalizedEndIdx = isClosed && endIdx === ring.length - 1 ? 0 : endIdx;
if (normalizedStartIdx < 0 || normalizedStartIdx >= n || normalizedEndIdx < 0 || normalizedEndIdx >= n) {
return [];
}
if (normalizedStartIdx === normalizedEndIdx) {
return [workingRing[normalizedStartIdx]];
}
if (!isClosed) {
// Case LineString
if (normalizedStartIdx <= normalizedEndIdx) {
return workingRing.slice(normalizedStartIdx, normalizedEndIdx + 1);
} else {
return workingRing.slice(normalizedEndIdx, normalizedStartIdx + 1).reverse();
}
}
// Case Closed Polygon
// Path 1: Forward
const path1: [number, number][] = [];
let idx = normalizedStartIdx;
while (idx !== normalizedEndIdx) {
path1.push(workingRing[idx]);
idx = (idx + 1) % n;
}
path1.push(workingRing[normalizedEndIdx]);
// Path 2: Backward
const path2: [number, number][] = [];
idx = normalizedStartIdx;
while (idx !== normalizedEndIdx) {
path2.push(workingRing[idx]);
idx = (idx - 1 + n) % n;
}
path2.push(workingRing[normalizedEndIdx]);
const poly1 = [...path1, workingRing[normalizedStartIdx]];
const poly2 = [...path2, workingRing[normalizedStartIdx]];
const area1 = getArea(poly1);
const area2 = getArea(poly2);
return area1 <= area2 ? path1 : path2;
}
export function getRingWithSnaps(
ring: Coordinate[],
snap1: { type: "vertex" | "edge"; vertexIdx?: number; edgeIdx?: number; lngLat: { lng: number; lat: number } },
snap2: { type: "vertex" | "edge"; vertexIdx?: number; edgeIdx?: number; lngLat: { lng: number; lat: number } }
): { ring: Coordinate[]; idx1: number; idx2: number } {
const closed = isClosedRing(ring);
const sourceRing = removeClosingCoordinate(ring);
const insertionGroups = new Map<number, Array<{ coord: Coordinate; t: number; owners: Set<1 | 2> }>>();
type NormalizedSnap =
| { type: "vertex"; vertexIdx: number }
| { type: "edge"; edgeIdx: number; coord: Coordinate; owner: 1 | 2; t: number };
const normalizeSnap = (
snap: typeof snap1,
owner: 1 | 2
): NormalizedSnap | null => {
const coord: Coordinate = [snap.lngLat.lng, snap.lngLat.lat];
if (snap.type === "vertex") {
const vertexIdx = normalizeVertexIndex(snap.vertexIdx, sourceRing.length, closed);
return vertexIdx === null ? null : { type: "vertex", vertexIdx };
}
const edgeIdx = snap.edgeIdx;
if (edgeIdx === undefined || edgeIdx < 0 || edgeIdx >= sourceRing.length) return null;
if (!closed && edgeIdx >= sourceRing.length - 1) return null;
const startIdx = edgeIdx;
const endIdx = (edgeIdx + 1) % sourceRing.length;
const start = sourceRing[startIdx];
const end = sourceRing[endIdx];
if (coordinatesAlmostEqual(coord, start)) return { type: "vertex", vertexIdx: startIdx };
if (coordinatesAlmostEqual(coord, end)) return { type: "vertex", vertexIdx: endIdx };
return {
type: "edge",
edgeIdx,
coord,
owner,
t: segmentProgress(coord, start, end),
};
};
const normalized1 = normalizeSnap(snap1, 1);
const normalized2 = normalizeSnap(snap2, 2);
for (const normalized of [normalized1, normalized2]) {
if (!normalized || normalized.type !== "edge") continue;
const group = insertionGroups.get(normalized.edgeIdx) || [];
const existing = group.find((item) => coordinatesAlmostEqual(item.coord, normalized.coord));
if (existing) {
existing.owners.add(normalized.owner);
} else {
group.push({ coord: normalized.coord, t: normalized.t, owners: new Set([normalized.owner]) });
}
insertionGroups.set(normalized.edgeIdx, group);
}
const builtRing: Coordinate[] = [];
const vertexIndexMap = new Map<number, number>();
const edgeIndexMap = new Map<1 | 2, number>();
for (let i = 0; i < sourceRing.length; i++) {
vertexIndexMap.set(i, builtRing.length);
builtRing.push(sourceRing[i]);
const group = insertionGroups.get(i);
if (!group) continue;
group
.sort((a, b) => a.t - b.t)
.forEach((item) => {
const existingIdx = builtRing.findIndex((coord) => coordinatesAlmostEqual(coord, item.coord));
const idx = existingIdx >= 0 ? existingIdx : builtRing.length;
if (existingIdx < 0) builtRing.push(item.coord);
for (const owner of item.owners) {
edgeIndexMap.set(owner, idx);
}
});
}
if (closed) {
builtRing.push(builtRing[0]);
}
const resolveIndex = (normalized: NormalizedSnap | null, owner: 1 | 2): number => {
if (!normalized) return -1;
if (normalized.type === "vertex") return vertexIndexMap.get(normalized.vertexIdx) ?? -1;
return edgeIndexMap.get(owner) ?? -1;
};
return {
ring: builtRing,
idx1: resolveIndex(normalized1, 1),
idx2: resolveIndex(normalized2, 2),
};
}
function coordinatesAlmostEqual(a: Coordinate, b: Coordinate, epsilon = COORDINATE_EPSILON): boolean {
return Math.abs(a[0] - b[0]) <= epsilon && Math.abs(a[1] - b[1]) <= epsilon;
}
function isClosedRing(ring: Coordinate[]): boolean {
return ring.length > 2 && coordinatesAlmostEqual(ring[0], ring[ring.length - 1]);
}
function removeClosingCoordinate(ring: Coordinate[]): Coordinate[] {
if (!isClosedRing(ring)) return [...ring];
return ring.slice(0, -1);
}
function closeRing(ring: Coordinate[]): Coordinate[] {
if (ring.length === 0 || isClosedRing(ring)) return [...ring];
return [...ring, ring[0]];
}
function normalizeVertexIndex(idx: number | undefined, uniqueLength: number, closed: boolean): number | null {
if (idx === undefined || uniqueLength <= 0) return null;
if (idx >= 0 && idx < uniqueLength) return idx;
if (closed && idx === uniqueLength) return 0;
return null;
}
function segmentProgress(coord: Coordinate, start: Coordinate, end: Coordinate): number {
const dx = end[0] - start[0];
const dy = end[1] - start[1];
const lenSq = dx * dx + dy * dy;
if (lenSq === 0) return 0;
const t = ((coord[0] - start[0]) * dx + (coord[1] - start[1]) * dy) / lenSq;
return Math.max(0, Math.min(1, t));
}
export function getOriginalFeature(
map: maplibregl.Map,
sourceId: string,
featureId: string | number | undefined
): GeoJSON.Feature | null {
if (featureId === undefined || featureId === null) return null;
// 1. Prefer the exact GeoJSON data for the rendered source. This avoids mixing geometries
// when different sources/layers reuse the same feature id.
const source = map.getSource(sourceId) as SourceWithInternalData | undefined;
if (source && source._data) {
const found = findFeatureInSourceData(source._data, featureId);
if (found) return found;
}
// 2. Fallback to the React/Zustand draft ref attached to the map instance.
const renderDraft = (map as MapWithRenderDraft)._renderDraftRef?.current;
if (renderDraft && Array.isArray(renderDraft.features)) {
const found = renderDraft.features.find((f) => {
const id = f.properties?.id ?? f.id;
return id !== undefined && String(id) === String(featureId);
});
if (found) {
return found;
}
}
return null;
}
function findFeatureInSourceData(
data: unknown,
featureId: string | number
): GeoJSON.Feature | null {
// MapLibre v5 updateable Map lookup
const updateable = getObjectProperty(data, "updateable");
if (updateable instanceof Map) {
const featureMap = updateable as Map<string | number, GeoJSON.Feature>;
const found = featureMap.get(featureId) || featureMap.get(String(featureId)) || featureMap.get(Number(featureId));
if (found) {
return found;
}
}
// Resolve GeoJSON object (MapLibre v5 stores geojson under data.geojson)
const geojson = getObjectProperty(data, "geojson") || data;
if (isFeatureCollection(geojson)) {
const found = geojson.features.find((f: GeoJSON.Feature) => {
const id = f.properties?.id ?? f.id;
return id !== undefined && String(id) === String(featureId);
});
return found || null;
}
if (isFeature(geojson)) {
const id = geojson.properties?.id ?? geojson.id;
const matches = id !== undefined && String(id) === String(featureId);
if (matches) {
return geojson;
}
}
return null;
}
type MapWithRenderDraft = maplibregl.Map & {
_renderDraftRef?: {
current?: {
features?: GeoJSON.Feature[];
};
};
};
type SourceWithInternalData = maplibregl.Source & {
_data?: unknown;
};
function getObjectProperty(value: unknown, key: string): unknown {
if (!value || typeof value !== "object") return undefined;
return (value as Record<string, unknown>)[key];
}
function isFeatureCollection(value: unknown): value is GeoJSON.FeatureCollection {
return Boolean(
value &&
typeof value === "object" &&
(value as { type?: unknown }).type === "FeatureCollection" &&
Array.isArray((value as { features?: unknown }).features)
);
}
function isFeature(value: unknown): value is GeoJSON.Feature {
return Boolean(
value &&
typeof value === "object" &&
(value as { type?: unknown }).type === "Feature"
);
}