From 15a9ae1e95077ce3950e0f04e36776faf9a0d927 Mon Sep 17 00:00:00 2001
From: ethan <zhufeng_solo@163.com>
Date: Thu, 24 Aug 2023 15:55:04 +0800
Subject: [PATCH] Add Lifecycle Events

---
 examples/index.ts               | 25 +++++++++++++++++++++-
 index.html                      | 31 ++++++++++++++-------------
 src/base.ts                     | 38 +++++++++++++++++++++++++++++++--
 src/events.ts                   | 37 ++++++++++++++++++++++++++++++++
 src/interface.ts                |  3 +++
 src/line/freehand-line.ts       |  1 +
 src/polygon/freehand-polygon.ts |  1 +
 7 files changed, 118 insertions(+), 18 deletions(-)
 create mode 100644 src/events.ts

diff --git a/examples/index.ts b/examples/index.ts
index f6d1831..9fa6614 100644
--- a/examples/index.ts
+++ b/examples/index.ts
@@ -103,7 +103,11 @@ buttonGroup.onclick = (evt) => {
       geometry = new CesiumPlot.FreehandLine(Cesium, viewer);
       break;
     case 'drawFreehandPolygon':
-      geometry = new CesiumPlot.FreehandPolygon(Cesium, viewer);
+      geometry = new CesiumPlot.FreehandPolygon(Cesium, viewer, {
+        material: Cesium.Color.GREEN,
+        outlineMaterial: Cesium.Color.fromCssColorString('rgba(59, 178, 208, 1)'),
+        outlineWidth: 2,
+      });
       break;
     case 'hide':
       geometry && geometry.hide();
@@ -115,6 +119,25 @@ buttonGroup.onclick = (evt) => {
       geometry && geometry.remove();
       geometry = null;
       break;
+    case 'addEvent':
+      if (geometry) {
+        geometry.on('drawStart', () => {
+          console.error('start');
+        });
+        geometry.on('drawUpdate', (cartesian: Cesium.Cartesian3) => {
+          console.error('update', cartesian);
+        });
+        geometry.on('drawEnd', (geometryPoints: any) => {
+          console.error('drawEnd', geometryPoints);
+        });
+        geometry.on('editStart', () => {
+          console.error('editStart');
+        });
+        geometry.on('editEnd', (geometryPoints: any) => {
+          console.error('editEnd', geometryPoints);
+        });
+      }
+      break;
     default:
       break;
   }
diff --git a/index.html b/index.html
index d6d0f7e..c27425d 100644
--- a/index.html
+++ b/index.html
@@ -24,7 +24,7 @@
         justify-content: center;
       }
 
-      .button {
+      .button-container button {
         margin: 0 10px;
         padding: 6px 10px;
         background-color: #3498db;
@@ -39,20 +39,21 @@
     <div id="root"></div>
     <div id="cesiumContainer"></div>
     <div class="button-container" id="button-group">
-      <button id="drawStraightArrow" class="button">细直箭头</button>
-      <button id="drawCurvedArrow" class="button">曲线箭头</button>
-      <button id="drawFineArrow" class="button">直箭头</button>
-      <button id="drawAttackArrow" class="button">进攻方向箭头</button>
-      <button id="drawSwallowtailAttackArrow" class="button">进攻方向箭头(燕尾)</button>
-      <button id="drawSquadCombat" class="button">分队战斗方向</button>
-      <button id="drawSwallowtailSquadCombat" class="button">分队战斗方向(燕尾)</button>
-      <button id="drawAssaultDirection" class="button">突击方向</button>
-      <button id="drawDoubleArrow" class="button">双箭头</button>
-      <button id="drawFreehandLine" class="button">自由线</button>
-      <button id="drawFreehandPolygon" class="button">自由面</button>
-      <button id="hide" class="button">隐藏</button>
-      <button id="show" class="button">显示</button>
-      <button id="remove" class="button">删除</button>
+      <button id="drawStraightArrow">细直箭头</button>
+      <button id="drawCurvedArrow">曲线箭头</button>
+      <button id="drawFineArrow">直箭头</button>
+      <button id="drawAttackArrow">进攻方向箭头</button>
+      <button id="drawSwallowtailAttackArrow">进攻方向箭头(燕尾)</button>
+      <button id="drawSquadCombat">分队战斗方向</button>
+      <button id="drawSwallowtailSquadCombat">分队战斗方向(燕尾)</button>
+      <button id="drawAssaultDirection">突击方向</button>
+      <button id="drawDoubleArrow">双箭头</button>
+      <button id="drawFreehandLine">自由线</button>
+      <button id="drawFreehandPolygon">自由面</button>
+      <button id="hide">隐藏</button>
+      <button id="show">显示</button>
+      <button id="remove">删除</button>
+      <button id="addEvent">绑定事件</button>
     </div>
     <script>
       window.CESIUM_BASE_URL = './examples/cesium';
diff --git a/src/base.ts b/src/base.ts
index 47ba240..6caa519 100644
--- a/src/base.ts
+++ b/src/base.ts
@@ -1,6 +1,7 @@
 // @ts-ignore
 import * as CesiumTypeOnly from '@examples/cesium';
-import { State, GeometryStyle, PolygonStyle, LineStyle } from './interface';
+import { State, GeometryStyle, PolygonStyle, LineStyle, EventType, EventListener } from './interface';
+import EventDispatcher from './events';
 
 export default class Base {
   cesium: typeof CesiumTypeOnly;
@@ -16,6 +17,7 @@ export default class Base {
   freehand!: boolean;
   style: GeometryStyle | undefined;
   outlineEntity: CesiumTypeOnly.Entity;
+  eventDispatcher: EventDispatcher;
 
   constructor(cesium: CesiumTypeOnly, viewer: CesiumTypeOnly.Viewer, style?: GeometryStyle) {
     this.cesium = cesium;
@@ -24,6 +26,7 @@ export default class Base {
     this.mergeStyle(style);
     this.cartesianToLnglat = this.cartesianToLnglat.bind(this);
     this.pixelToCartesian = this.pixelToCartesian.bind(this);
+    this.eventDispatcher = new EventDispatcher();
     this.onClick();
   }
 
@@ -89,11 +92,18 @@ export default class Base {
           return;
         }
         this.addPoint(cartesian);
+        // Trigger 'drawStart' when the first point is being drawn.
+        if (this.getPoints().length === 1) {
+          this.eventDispatcher.dispatchEvent('drawStart');
+        }
+        this.eventDispatcher.dispatchEvent('drawUpdate', cartesian);
       } else if (this.state === 'edit') {
         //In edit mode, exit the editing state and delete control points when clicking outside the currently edited shape.
         if (!hitEntities || activeEntity.id !== pickedObject.id.id) {
           this.setState('static');
           this.removeControlPoints();
+          // Trigger 'drawEnd' and return the geometry shape points when exiting the edit mode.
+          this.eventDispatcher.dispatchEvent('drawEnd', this.getPoints());
         }
       } else if (this.state === 'static') {
         //When drawing multiple shapes, the click events for all shapes are triggered. Only when hitting a completed shape should it enter editing mode.
@@ -103,6 +113,7 @@ export default class Base {
             // Hit Geometry Shape.
             this.setState('edit');
             this.addControlPoints();
+            this.eventDispatcher.dispatchEvent('editStart');
           }
         }
       }
@@ -140,6 +151,7 @@ export default class Base {
   finishDrawing() {
     this.removeMoveListener();
     this.setState('static');
+    this.eventDispatcher.dispatchEvent('drawEnd', this.getPoints());
   }
 
   removeClickListener() {
@@ -150,6 +162,10 @@ export default class Base {
     this.eventHandler.removeInputAction(this.cesium.ScreenSpaceEventType.MOUSE_MOVE);
   }
 
+  removeDoubleClickListener() {
+    this.eventHandler.removeInputAction(this.cesium.ScreenSpaceEventType.LEFT_DOUBLE_CLICK);
+  }
+
   setGeometryPoints(geometryPoints: CesiumTypeOnly.Cartesian3[]) {
     this.geometryPoints = geometryPoints;
   }
@@ -255,6 +271,7 @@ export default class Base {
 
     let isDragging = false;
     let draggedIcon: CesiumTypeOnly.Entity = null;
+    let dragStartPosition: CesiumTypeOnly.Cartesian3;
 
     this.controlPointsEventHandler = new this.cesium.ScreenSpaceEventHandler(this.viewer.canvas);
     // Listen for left mouse button press events
@@ -266,7 +283,9 @@ export default class Base {
           if (pickedObject.id === this.controlPoints[i]) {
             isDragging = true;
             draggedIcon = this.controlPoints[i];
-            draggedIcon.index = i; //Save the index of dragged points for dynamic updates during movement
+            dragStartPosition = draggedIcon.position._value;
+            //Save the index of dragged points for dynamic updates during movement
+            draggedIcon.index = i;
             break;
           }
         }
@@ -288,6 +307,10 @@ export default class Base {
 
     // Listen for left mouse button release events
     this.controlPointsEventHandler.setInputAction(() => {
+      // Trigger 'drawUpdate' when there is a change in coordinates before and after dragging.
+      if (!this.cesium.Cartesian3.equals(dragStartPosition, draggedIcon.position._value)) {
+        this.eventDispatcher.dispatchEvent('drawUpdate', draggedIcon.position._value);
+      }
       isDragging = false;
       draggedIcon = null;
       this.viewer.scene.screenSpaceCameraController.enableRotate = true;
@@ -342,6 +365,17 @@ export default class Base {
     } else if (this.type === 'line') {
       this.viewer.entities.remove(this.lineEntity);
     }
+    this.removeClickListener();
+    this.removeMoveListener();
+    this.removeDoubleClickListener();
+  }
+
+  on(eventType: EventType, listener: EventListener) {
+    this.eventDispatcher.on(eventType, listener);
+  }
+
+  off(eventType: EventType, listener: EventListener) {
+    this.eventDispatcher.off(eventType, listener);
   }
 
   addPoint(cartesian: CesiumTypeOnly.Cartesian3) {
diff --git a/src/events.ts b/src/events.ts
new file mode 100644
index 0000000..52b1bc8
--- /dev/null
+++ b/src/events.ts
@@ -0,0 +1,37 @@
+import { EventType, EventListener } from './interface';
+
+export default class EventDispatcher {
+  listeners: Map<EventType, Set<EventListener>>;
+
+  constructor() {
+    this.listeners = new Map([
+      ['drawStart', new Set()],
+      ['drawUpdate', new Set()],
+      ['drawEnd', new Set()],
+      ['editStart', new Set()],
+      ['editEnd', new Set()],
+    ]);
+  }
+
+  on(event: EventType, listener: EventListener) {
+    if (!this.listeners.has(event)) {
+      console.warn("Event binding must be one of 'drawStart', 'drawUpdate', or 'drawEnd'.");
+      return;
+    }
+    this.listeners.get(event)!.add(listener);
+  }
+
+  off(event: EventType, listener: EventListener) {
+    if (this.listeners.has(event)) {
+      this.listeners.get(event)!.delete(listener);
+    }
+  }
+
+  dispatchEvent(event: EventType, eventData?: any) {
+    if (this.listeners.has(event)) {
+      this.listeners.get(event)!.forEach((listener) => {
+        listener(eventData);
+      });
+    }
+  }
+}
\ No newline at end of file
diff --git a/src/interface.ts b/src/interface.ts
index 8f96322..53b758b 100644
--- a/src/interface.ts
+++ b/src/interface.ts
@@ -14,3 +14,6 @@ export type LineStyle = {
 
 export type State = 'drawing' | 'edit' | 'static';
 export type GeometryStyle = PolygonStyle | LineStyle;
+
+export type EventType = 'drawStart' | 'drawUpdate' | 'drawEnd' | 'editEnd' | 'editStart';
+export type EventListener = (eventData?: any) => void;
diff --git a/src/line/freehand-line.ts b/src/line/freehand-line.ts
index c4d2f15..89d3cc0 100644
--- a/src/line/freehand-line.ts
+++ b/src/line/freehand-line.ts
@@ -37,6 +37,7 @@ export default class FreehandLine extends Base {
     this.points.push(cartesian);
     this.setGeometryPoints(this.points);
     this.drawLine();
+    this.eventDispatcher.dispatchEvent('drawUpdate', cartesian);
   }
 
   /**
diff --git a/src/polygon/freehand-polygon.ts b/src/polygon/freehand-polygon.ts
index 995f4d7..0ff10f4 100644
--- a/src/polygon/freehand-polygon.ts
+++ b/src/polygon/freehand-polygon.ts
@@ -38,6 +38,7 @@ export default class FreehandPolygon extends Base {
     if (this.points.length > 2) {
       this.setGeometryPoints(this.points);
       this.drawPolygon();
+      this.eventDispatcher.dispatchEvent('drawUpdate', cartesian);
     }
   }