From 94900a9ae9f5481e6601d3f09855f6f887549469 Mon Sep 17 00:00:00 2001
From: ethan <zhufeng_solo@163.com>
Date: Tue, 22 Aug 2023 15:40:58 +0800
Subject: [PATCH] add FreehandPolygon

---
 examples/index.ts               |  3 ++
 index.html                      |  1 +
 src/index.ts                    |  2 ++
 src/polygon/freehand-polygon.ts | 52 +++++++++++++++++++++++++++++++++
 4 files changed, 58 insertions(+)
 create mode 100644 src/polygon/freehand-polygon.ts

diff --git a/examples/index.ts b/examples/index.ts
index 611b111..739e066 100644
--- a/examples/index.ts
+++ b/examples/index.ts
@@ -69,6 +69,9 @@ buttonGroup.onclick = (evt) => {
     case 'drawFreehandLine':
       new CesiumPlot.FreehandLine(Cesium, viewer, {});
       break;
+    case 'drawFreehandPolygon':
+      new CesiumPlot.FreehandPolygon(Cesium, viewer, {});
+      break;
 
     default:
       break;
diff --git a/index.html b/index.html
index 5aceb3d..601ca53 100644
--- a/index.html
+++ b/index.html
@@ -49,6 +49,7 @@
       <button id="drawAssaultDirection" class="button">突击方向</button>
       <button id="drawDoubleArrow" class="button">双箭头</button>
       <button id="drawFreehandLine" class="button">自由线</button>
+      <button id="drawFreehandPolygon" class="button">自由面</button>
     </div>
     <script>
       window.CESIUM_BASE_URL = './examples/cesium';
diff --git a/src/index.ts b/src/index.ts
index 509d18c..6c5acfb 100644
--- a/src/index.ts
+++ b/src/index.ts
@@ -8,6 +8,7 @@ import CurvedArrow from './arrow/curved-arrow';
 import AssaultDirection from './arrow/assault-direction';
 import DoubleArrow from './arrow/double-arrow';
 import FreehandLine from './line/freehand-line';
+import FreehandPolygon from './polygon/freehand-polygon';
 
 const CesiumPlot = {
   FineArrow,
@@ -20,6 +21,7 @@ const CesiumPlot = {
   AssaultDirection,
   DoubleArrow,
   FreehandLine,
+  FreehandPolygon,
 };
 
 export default CesiumPlot;
diff --git a/src/polygon/freehand-polygon.ts b/src/polygon/freehand-polygon.ts
new file mode 100644
index 0000000..a858cd5
--- /dev/null
+++ b/src/polygon/freehand-polygon.ts
@@ -0,0 +1,52 @@
+import Draw from '../draw';
+import { Cartesian3 } from '@examples/cesium';
+
+export default class FreehandPolygon extends Draw {
+  points: Cartesian3[] = [];
+  type: 'polygon' | 'line';
+  freehand: boolean;
+
+  constructor(cesium: any, viewer: any, style: any) {
+    super(cesium, viewer);
+    this.cesium = cesium;
+    this.type = 'polygon';
+    this.freehand = true;
+    this.setState('drawing');
+  }
+
+  /**
+   * Add points only on click events
+   */
+  addPoint(cartesian: Cartesian3) {
+    this.points.push(cartesian);
+    if (this.points.length === 1) {
+      this.onMouseMove();
+    } else if (this.points.length > 2) {
+      this.finishDrawing();
+    }
+  }
+
+  /**
+   * Draw a shape based on mouse movement points during the initial drawing.
+   */
+  updateMovingPoint(cartesian: Cartesian3) {
+    this.points.push(cartesian);
+    if (this.points.length > 2) {
+      this.setGeometryPoints(this.points);
+      this.drawPolygon();
+    }
+  }
+
+  /**
+   * In edit mode, drag key points to update corresponding key point data.
+   */
+  updateDraggingPoint(cartesian: Cartesian3, index: number) {
+    this.points[index] = cartesian;
+    this.setGeometryPoints(this.points);
+    this.drawPolygon();
+  }
+
+  getPoints() {
+    return this.points;
+  }
+}