From ebcd177bf5817b516e129de066d777f149e2a899 Mon Sep 17 00:00:00 2001
From: ethan <zhufeng_solo@163.com>
Date: Fri, 1 Sep 2023 18:27:01 +0800
Subject: [PATCH] add curve

---
 examples/index.ts |  3 ++
 index.html        | 20 +-----------
 src/index.ts      |  2 ++
 src/line/curve.ts | 80 +++++++++++++++++++++++++++++++++++++++++++++++
 4 files changed, 86 insertions(+), 19 deletions(-)
 create mode 100644 src/line/curve.ts

diff --git a/examples/index.ts b/examples/index.ts
index 0fa4c75..8d50bf1 100644
--- a/examples/index.ts
+++ b/examples/index.ts
@@ -120,6 +120,9 @@ buttonGroup.onclick = (evt) => {
     case 'drawFreehandLine':
       geometry = new CesiumPlot.FreehandLine(Cesium, viewer);
       break;
+    case 'drawCurve':
+      geometry = new CesiumPlot.Curve(Cesium, viewer);
+      break;
     case 'drawFreehandPolygon':
       geometry = new CesiumPlot.FreehandPolygon(Cesium, viewer, {
         material: Cesium.Color.GREEN,
diff --git a/index.html b/index.html
index 167459e..5a3907b 100644
--- a/index.html
+++ b/index.html
@@ -56,30 +56,12 @@
       <button id="drawDoubleArrow">双箭头</button>
       <button id="drawFreehandLine">自由线</button>
       <button id="drawFreehandPolygon">自由面</button>
+      <button id="drawCurve">曲线</button>
       <button id="hide">隐藏</button>
       <button id="show">显示</button>
       <button id="remove">删除</button>
       <button id="addEvent">绑定事件</button>
       <button id="removeEvent">解绑事件</button>
-      <button id="removeEvent">解绑事件</button>
-      <button id="removeEvent">解绑事件</button>
-      <button id="removeEvent">解绑事件</button>
-      <button id="removeEvent">解绑事件</button>
-      <button id="removeEvent">解绑事件</button>
-      <button id="removeEvent">解绑事件</button>
-      <button id="removeEvent">解绑事件</button>
-      <button id="removeEvent">解绑事件</button>
-      <button id="removeEvent">解绑事件</button>
-      <button id="removeEvent">解绑事件</button>
-      <button id="removeEvent">解绑事件</button>
-      <button id="removeEvent">解绑事件</button>
-      <button id="removeEvent">解绑事件</button>
-      <button id="removeEvent">解绑事件</button>
-      <button id="removeEvent">解绑事件</button>
-      <button id="removeEvent">解绑事件</button>
-      <button id="removeEvent">解绑事件</button>
-      <button id="removeEvent">解绑事件</button>
-      <button id="removeEvent">解绑事件</button>
     </div>
     <script>
       window.CESIUM_BASE_URL = './examples/cesium';
diff --git a/src/index.ts b/src/index.ts
index 6c5acfb..01125c6 100644
--- a/src/index.ts
+++ b/src/index.ts
@@ -9,6 +9,7 @@ import AssaultDirection from './arrow/assault-direction';
 import DoubleArrow from './arrow/double-arrow';
 import FreehandLine from './line/freehand-line';
 import FreehandPolygon from './polygon/freehand-polygon';
+import Curve from './line/curve';
 
 const CesiumPlot = {
   FineArrow,
@@ -22,6 +23,7 @@ const CesiumPlot = {
   DoubleArrow,
   FreehandLine,
   FreehandPolygon,
+  Curve,
 };
 
 export default CesiumPlot;
diff --git a/src/line/curve.ts b/src/line/curve.ts
new file mode 100644
index 0000000..c79e2c2
--- /dev/null
+++ b/src/line/curve.ts
@@ -0,0 +1,80 @@
+import * as Utils from '../utils';
+import Base from '../base';
+// @ts-ignore
+import { Cartesian3 } from 'kmap-3d-engine';
+import { LineStyle } from '../interface';
+
+export default class Curve extends Base {
+	points: Cartesian3[] = [];
+	arrowLengthScale: number = 5;
+	maxArrowLength: number = 3000000;
+	t: number;
+
+	constructor(cesium: any, viewer: any, style?: LineStyle) {
+		super(cesium, viewer, style);
+		this.cesium = cesium;
+		this.t = 0.3;
+		this.setState('drawing');
+		this.onDoubleClick();
+	}
+
+	getType(): 'polygon' | 'line' {
+		return 'line';
+	}
+
+	/**
+	 * 仅点击事件会添加点
+	 */
+	addPoint(cartesian: Cartesian3) {
+		this.points.push(cartesian);
+		if (this.points.length < 2) {
+			this.onMouseMove();
+		} else if (this.points.length === 2) {
+			this.setGeometryPoints(this.points);
+			this.drawLine();
+		}
+	}
+
+	/**
+	 * 初次绘制时根据鼠标移动的点位绘制图形.
+	 */
+	updateMovingPoint(cartesian: Cartesian3) {
+		const tempPoints = [...this.points, cartesian];
+		let geometryPoints = [];
+		if (tempPoints.length === 2) {
+			this.setGeometryPoints(tempPoints);
+			this.drawLine();
+		} else {
+			geometryPoints = this.createLine(tempPoints);
+			this.setGeometryPoints(geometryPoints);
+		}
+	}
+
+	/**
+	 * 编辑状态下，拖拽关键点位，更新对应关键点位数据
+	 */
+	updateDraggingPoint(cartesian: Cartesian3, index: number) {
+		this.points[index] = cartesian;
+		const geometryPoints = this.createLine(this.points);
+		this.setGeometryPoints(geometryPoints);
+		this.drawLine();
+	}
+
+	/**
+	 * 根据关键点生成几何图形点位.
+	 */
+	createLine(positions: Cartesian3[]) {
+		const lnglatPoints = positions.map(pnt => {
+			return this.cartesianToLnglat(pnt);
+		});
+
+		const curvePoints = Utils.getCurvePoints(this.t, lnglatPoints);
+		const temp = [].concat(...curvePoints);
+		const cartesianPoints = this.cesium.Cartesian3.fromDegreesArray(temp);
+		return cartesianPoints;
+	}
+
+	getPoints() {
+		return this.points;
+	}
+}