@reearth/core 0.0.7-alpha.14 → 0.0.7-alpha.15

package/src/engines/Cesium/Sketch/SurfaceControlPoints.tsx

@@ -1,62 +1,152 @@
-import { Cartesian3, type Color } from "@cesium/engine";
+import { Cartesian3, Color } from "@cesium/engine";
 import { memo, type FC } from "react";
 
+import { DEFAULT_EDIT_COLOR } from "./constants";
 import { ControlPoint } from "./ControlPoint";
 import { type GeometryOptions } from "./createGeometry";
 import { SurfaceMeasurement } from "./SurfaceMeasurement";
 
+import { ControlPointMouseEventHandler } from ".";
+
 export interface SurfaceControlPointsProps {
   geometryOptions: GeometryOptions;
   color?: Color;
+  isEditing?: boolean;
+  selectedControlPointIndex?: number;
+  catchedControlPointIndex?: number;
+  handleControlPointMouseEvent?: ControlPointMouseEventHandler;
 }
 
 const cartesianScratch1 = new Cartesian3();
-const cartesianScratch2 = new Cartesian3();
 
 const SurfaceControlPoints: FC<SurfaceControlPointsProps> = memo(
-  ({ geometryOptions: { type, controlPoints: controlPointsProp }, color }) => {
-    let controlPoints = [...controlPointsProp];
-    let measurementPoints: [Cartesian3, Cartesian3] | undefined;
-    let showLine = false;
+  ({
+    geometryOptions: { type, controlPoints },
+    color,
+    isEditing,
+    catchedControlPointIndex,
+    selectedControlPointIndex,
+    handleControlPointMouseEvent,
+  }) => {
+    const measurements: { points: [Cartesian3, Cartesian3]; showLine: boolean }[] = [];
 
-    if (type === "rectangle" && controlPoints.length === 3) {
-      const [p1, p2, p3] = controlPoints;
-      const projection = Cartesian3.projectVector(
-        Cartesian3.subtract(p3, p1, cartesianScratch1),
-        Cartesian3.subtract(p2, p1, cartesianScratch2),
-        cartesianScratch1,
-      );
-      const offset = Cartesian3.subtract(
-        p3,
-        Cartesian3.add(p1, projection, cartesianScratch1),
-        cartesianScratch2,
-      );
-      const p4 = Cartesian3.midpoint(p1, p2, cartesianScratch1);
-      const p5 = Cartesian3.add(p4, offset, cartesianScratch2);
-      controlPoints = [p1, p2, p5];
-      measurementPoints = [p4, p5];
-      showLine = true;
-    } else if (type === "marker") {
-      measurementPoints = undefined;
-      controlPoints = [];
-    } else if (controlPoints.length >= 2) {
-      measurementPoints = controlPoints.slice(-2) as [Cartesian3, Cartesian3];
-      showLine = type === "circle";
+    if (controlPoints.length >= 2) {
+      if (isEditing) {
+        if (catchedControlPointIndex !== undefined && catchedControlPointIndex !== -1) {
+          switch (type) {
+            case "polyline":
+              if (catchedControlPointIndex > 0) {
+                measurements.push({
+                  points: [
+                    controlPoints[catchedControlPointIndex - 1],
+                    controlPoints[catchedControlPointIndex],
+                  ],
+                  showLine: false,
+                });
+              }
+              if (catchedControlPointIndex < controlPoints.length - 1) {
+                measurements.push({
+                  points: [
+                    controlPoints[catchedControlPointIndex],
+                    controlPoints[catchedControlPointIndex + 1],
+                  ],
+                  showLine: false,
+                });
+              }
+              break;
+            case "circle":
+            case "extrudedCircle":
+              if (catchedControlPointIndex !== 2) {
+                measurements.push({
+                  points: [controlPoints[0], controlPoints[1]],
+                  showLine: true,
+                });
+              }
+              break;
+            case "rectangle":
+            case "extrudedRectangle":
+              if (catchedControlPointIndex <= 1) {
+                measurements.push({
+                  points: [controlPoints[0], controlPoints[1]],
+                  showLine: false,
+                });
+              } else if (catchedControlPointIndex === 2) {
+                const [p1, p2, p3] = controlPoints;
+                const p4 = Cartesian3.midpoint(p1, p2, cartesianScratch1);
+                measurements.push({ points: [p4, p3], showLine: true });
+              }
+              break;
+            case "polygon":
+            case "extrudedPolygon":
+              measurements.push({
+                points: [
+                  controlPoints[catchedControlPointIndex],
+                  catchedControlPointIndex === 0
+                    ? controlPoints[controlPoints.length - 1]
+                    : controlPoints[catchedControlPointIndex - 1],
+                ],
+                showLine: false,
+              });
+              measurements.push({
+                points: [
+                  controlPoints[catchedControlPointIndex],
+                  catchedControlPointIndex === controlPoints.length - 1
+                    ? controlPoints[0]
+                    : controlPoints[catchedControlPointIndex + 1],
+                ],
+                showLine: false,
+              });
+              break;
+            default:
+              break;
+          }
+        }
+      } else {
+        switch (type) {
+          case "rectangle" || "extrudedRectangle":
+            if (controlPoints.length === 2) {
+              measurements.push({
+                points: controlPoints as [Cartesian3, Cartesian3],
+                showLine: true,
+              });
+            } else if (controlPoints.length === 3) {
+              const [p1, p2, p3] = controlPoints;
+              const p4 = Cartesian3.midpoint(p1, p2, cartesianScratch1);
+              measurements.push({ points: [p4, p3], showLine: true });
+            }
+            break;
+          default:
+            measurements.push({
+              points: controlPoints.slice(-2) as [Cartesian3, Cartesian3],
+              showLine: type === "circle",
+            });
+            break;
+        }
+      }
     }
 
     return (
       <>
         {controlPoints.map((controlPoint, index) => (
-          <ControlPoint key={index} position={controlPoint} clampToGround />
+          <ControlPoint
+            key={index}
+            position={controlPoint}
+            index={index}
+            isSelected={selectedControlPointIndex === index}
+            clampToGround
+            isEditing={isEditing}
+            handleControlPointMouseEvent={handleControlPointMouseEvent}
+          />
         ))}
-        {measurementPoints != undefined && (
+        {measurements.map(({ points, showLine }, index) => (
           <SurfaceMeasurement
-            a={measurementPoints[0]}
-            b={measurementPoints[1]}
-            color={color}
+            key={index}
+            a={points[0]}
+            b={points[1]}
+            color={isEditing ? Color.fromCssColorString(DEFAULT_EDIT_COLOR) : color}
             showLine={showLine}
           />
-        )}
+        ))}
       </>
     );
   },

package/src/engines/Cesium/Sketch/constants.ts

@@ -0,0 +1,5 @@
+export const DEFAULT_SKETCH_COLOR = "#00bebe";
+
+export const DEFAULT_EDIT_COLOR = "#3B3CD0";
+export const SELECTED_EDIT_COLOR = "#FF9900";
+export const ADDING_POINT_COLOR = "#3B3CD0";

package/src/engines/Cesium/Sketch/index.tsx

@@ -2,46 +2,61 @@
 
 import { Color } from "@cesium/engine";
 import { Cartesian3 } from "cesium";
-import { type LineString, type MultiPolygon, type Polygon } from "geojson";
 import { memo, useMemo, type FC } from "react";
-import { type RequireExactlyOne } from "type-fest";
 
 import { SketchType } from "../../../Map/Sketch/types";
 import { Position3d } from "../../../types";
 import { convertGeometryToPositionsArray, convertPolygonToHierarchyArray } from "../utils/polygon";
 
+import { DEFAULT_SKETCH_COLOR } from "./constants";
 import { createGeometry, GeometryOptions } from "./createGeometry";
 import ExtrudedControlPoints from "./ExtrudedControlPoints";
 import { ExtrudedPolygonEntity } from "./ExtrudedPolygonEntity";
 import { PolygonEntity } from "./PolygonEntity";
 import { PolylineEntity } from "./PolylineEntity";
+import SurfaceAddingPoints from "./SurfaceAddingPoints";
 import SurfaceControlPoints from "./SurfaceControlPoints";
 
-export type SketchComponentProps = RequireExactlyOne<
-  {
-    geometry?: LineString | Polygon | MultiPolygon | null;
-    geometryOptions?: {
-      type: SketchType;
-      controlPoints: readonly Position3d[];
-    } | null;
-    extrudedHeight?: number;
-    disableShadow?: boolean;
-    color?: string;
-    enableRelativeHeight?: boolean;
-  },
-  "geometry" | "geometryOptions"
->;
+export type SketchComponentProps = {
+  geometryOptions?: {
+    type: SketchType;
+    controlPoints: readonly Position3d[];
+  } | null;
+  extrudedHeight?: number;
+  extrudedPoint?: Position3d;
+  centroidBasePoint?: Position3d;
+  centroidExtrudedPoint?: Position3d;
+  disableShadow?: boolean;
+  color?: string;
+  isEditing?: boolean;
+  catchedControlPointIndex?: number;
+  catchedExtrudedPoint?: boolean;
+  selectedControlPointIndex?: number;
+  handleControlPointMouseEvent?: ControlPointMouseEventHandler;
+  handleAddControlPoint?: (position: Position3d, index: number) => void;
+};
 
-const DEFAULT_SKETCH_COLOR = "#00bebe";
+export type ControlPointMouseEventHandler = (
+  index: number,
+  isExtrudedPoint: boolean,
+  type: "mousedown" | "click",
+) => void;
 
 const SketchComponent: FC<SketchComponentProps> = memo(
   ({
-    geometry,
     geometryOptions,
     extrudedHeight,
     disableShadow,
     color: stringColor,
-    enableRelativeHeight,
+    isEditing,
+    extrudedPoint,
+    centroidBasePoint,
+    centroidExtrudedPoint,
+    catchedControlPointIndex,
+    catchedExtrudedPoint,
+    selectedControlPointIndex,
+    handleControlPointMouseEvent,
+    handleAddControlPoint,
   }) => {
     const cartesianGeometryOptions: GeometryOptions | null = useMemo(
       () =>
@@ -55,9 +70,8 @@ const SketchComponent: FC<SketchComponentProps> = memo(
     );
 
     const g = useMemo(
-      () =>
-        geometry ?? (cartesianGeometryOptions ? createGeometry(cartesianGeometryOptions) : null),
-      [geometry, cartesianGeometryOptions],
+      () => (cartesianGeometryOptions ? createGeometry(cartesianGeometryOptions) : null),
+      [cartesianGeometryOptions],
     );
 
     const { positionsArray, hierarchyArray } = useMemo(() => {
@@ -82,31 +96,56 @@ const SketchComponent: FC<SketchComponentProps> = memo(
     return (
       <>
         {positionsArray?.map((positions, index) => (
-          <PolylineEntity key={index} dynamic positions={positions} color={color} />
+          <PolylineEntity
+            key={index}
+            dynamic
+            positions={positions}
+            color={color}
+            isEditing={isEditing}
+          />
         ))}
         {hierarchyArray?.map((hierarchy, index) => (
           <PolygonEntity key={index} dynamic hierarchy={hierarchy} color={color} />
         ))}
-        {cartesianGeometryOptions != null && extrudedHeight == null && (
-          <SurfaceControlPoints geometryOptions={cartesianGeometryOptions} color={color} />
+        {cartesianGeometryOptions != null && (!extrudedHeight || isEditing) && (
+          <SurfaceControlPoints
+            geometryOptions={cartesianGeometryOptions}
+            color={color}
+            isEditing={isEditing}
+            catchedControlPointIndex={catchedControlPointIndex}
+            selectedControlPointIndex={selectedControlPointIndex}
+            handleControlPointMouseEvent={handleControlPointMouseEvent}
+          />
+        )}
+        {cartesianGeometryOptions != null && isEditing && (
+          <SurfaceAddingPoints
+            geometryOptions={cartesianGeometryOptions}
+            isEditing={isEditing}
+            handleAddControlPoint={handleAddControlPoint}
+          />
         )}
-        {cartesianGeometryOptions != null && extrudedHeight != null && (
+        {cartesianGeometryOptions != null && extrudedHeight && (
           <ExtrudedControlPoints
             geometryOptions={cartesianGeometryOptions}
             extrudedHeight={extrudedHeight}
+            extrudedPoint={extrudedPoint}
+            centroidBasePoint={centroidBasePoint}
+            centroidExtrudedPoint={centroidExtrudedPoint}
+            catchedExtrudedPoint={catchedExtrudedPoint}
             color={color}
+            isEditing={isEditing}
+            handleControlPointMouseEvent={handleControlPointMouseEvent}
           />
         )}
-        {extrudedHeight != null &&
+        {extrudedHeight &&
           hierarchyArray?.map((hierarchy, index) => (
             <ExtrudedPolygonEntity
               key={index}
-              dynamic
               hierarchy={hierarchy}
               extrudedHeight={extrudedHeight}
               disableShadow={disableShadow}
               color={color}
-              enableRelativeHeight={enableRelativeHeight}
+              isEditing={isEditing}
             />
           ))}
       </>

package/src/engines/Cesium/hooks/useEngineRef.ts

@@ -268,6 +268,21 @@ export default function useEngineRef(
           new Cesium.Cartesian2(windowPosition[0], windowPosition[1]),
         );
       },
+      getExtrudedPoint: (position, extrudedHeight) => {
+        if (!position || !extrudedHeight) return;
+        const viewer = cesium.current?.cesiumElement;
+        if (!viewer || viewer.isDestroyed()) return;
+        const point = new Cesium.Cartesian3(position[0], position[1], position[2]);
+        const cartesianScratch = new Cesium.Cartesian3();
+        const normal = viewer.scene?.globe.ellipsoid.geodeticSurfaceNormal(point, cartesianScratch);
+        if (!normal) return;
+        const extrudedPoint = Cesium.Cartesian3.add(
+          point,
+          Cesium.Cartesian3.multiplyByScalar(normal, extrudedHeight, cartesianScratch),
+          cartesianScratch,
+        );
+        return [extrudedPoint.x, extrudedPoint.y, extrudedPoint.z];
+      },
       getSurfaceDistance: (point1, point2) => {
         const viewer = cesium.current?.cesiumElement;
         if (!viewer || viewer.isDestroyed()) return;
@@ -938,6 +953,27 @@ export default function useEngineRef(
         tickEventCallback.current = tickEventCallback.current.filter(c => c !== cb) || [];
       },
       tickEventCallback,
+      calcRectangleControlPoint: (p1: Position3d, p2: Position3d, p3: Position3d) => {
+        const pp1 = new Cesium.Cartesian3(...p1);
+        const pp2 = new Cesium.Cartesian3(...p2);
+        const pp3 = new Cesium.Cartesian3(...p3);
+        const cartesianScratch1 = new Cesium.Cartesian3();
+        const cartesianScratch2 = new Cesium.Cartesian3();
+        const projection = Cesium.Cartesian3.projectVector(
+          Cesium.Cartesian3.subtract(pp3, pp1, cartesianScratch1),
+          Cesium.Cartesian3.subtract(pp2, pp1, cartesianScratch2),
+          cartesianScratch1,
+        );
+        const offset = Cesium.Cartesian3.subtract(
+          pp3,
+          Cesium.Cartesian3.add(pp1, projection, cartesianScratch1),
+          cartesianScratch2,
+        );
+        const pp4 = Cesium.Cartesian3.midpoint(pp1, pp2, cartesianScratch1);
+        const pp5 = Cesium.Cartesian3.add(pp4, offset, cartesianScratch2);
+        const p5 = [pp5.x, pp5.y, pp5.z] as Position3d;
+        return [p1, p2, p5];
+      },
     };
   }, [cesium]);