cesium-plot-js/src/arrow/fine-arrow.ts
2023-08-16 16:17:33 +08:00

86 lines
2.8 KiB
TypeScript

import Draw from '../draw';
import * as Utils from '../utils';
import { Cartesian3 } from '@examples/cesium';
export default class FineArrow extends Draw {
points: Cartesian3[] = [];
arrowLengthScale: number = 5;
maxArrowLength: number = 2;
tailWidthFactor: number;
neckWidthFactor: number;
headWidthFactor: number;
headAngle: number;
neckAngle: number;
constructor(cesium: any, viewer: any, style: any) {
super(cesium, viewer);
this.cesium = cesium;
this.tailWidthFactor = 0.1;
this.neckWidthFactor = 0.2;
this.headWidthFactor = 0.25;
this.headAngle = Math.PI / 8.5;
this.neckAngle = Math.PI / 13;
this.setState('drawing');
}
/**
* Add points only on click events
*/
addPoint(cartesian: Cartesian3) {
if (this.points.length < 2) {
this.points.push(cartesian);
this.onMouseMove();
}
if (this.points.length === 2) {
const geometryPoints = this.createPolygon(this.points);
this.setGeometryPoints(geometryPoints);
this.addToMap();
this.finishDrawing();
}
}
/**
* Draw a shape based on mouse movement points during the initial drawing.
*/
updateMovingPoint(cartesian: Cartesian3) {
const tempPoints = [...this.points, cartesian];
const geometryPoints = this.createPolygon(tempPoints);
this.setGeometryPoints(geometryPoints);
this.addToMap();
}
/**
* In edit mode, drag key points to update corresponding key point data.
*/
updateDraggingPoint(cartesian: Cartesian3, index: number) {
this.points[index] = cartesian;
const geometryPoints = this.createPolygon(this.points);
this.setGeometryPoints(geometryPoints);
this.addToMap();
}
/**
* Generate geometric shapes based on key points.
*/
createPolygon(positions: Cartesian3[]) {
const [p1, p2] = positions.map(this.cartesianToLnglat);
const len = Utils.getBaseLength([p1, p2]);
const tailWidth = len * this.tailWidthFactor;
const neckWidth = len * this.neckWidthFactor;
const headWidth = len * this.headWidthFactor;
const tailLeft = Utils.getThirdPoint(p2, p1, Math.PI / 2, tailWidth, true);
const tailRight = Utils.getThirdPoint(p2, p1, Math.PI / 2, tailWidth, false);
const headLeft = Utils.getThirdPoint(p1, p2, this.headAngle, headWidth, false);
const headRight = Utils.getThirdPoint(p1, p2, this.headAngle, headWidth, true);
const neckLeft = Utils.getThirdPoint(p1, p2, this.neckAngle, neckWidth, false);
const neckRight = Utils.getThirdPoint(p1, p2, this.neckAngle, neckWidth, true);
const points = [...tailLeft, ...neckLeft, ...headLeft, ...p2, ...headRight, ...neckRight, ...tailRight, ...p1];
const cartesianPoints = this.cesium.Cartesian3.fromDegreesArray(points);
return cartesianPoints;
}
getPoints() {
return this.points;
}
}