@cocorof/graphier 1.4.0 → 1.4.1

package/dist/index.cjs

@@ -2049,6 +2049,45 @@ const _right = new THREE__namespace.Vector3();
 const _q = new THREE__namespace.Quaternion();
 const _dir = new THREE__namespace.Vector3();
 const _worldUp = new THREE__namespace.Vector3(0, 1, 0);
+const PAN_SPEED = 0.02;
+const PAN_ZOOM_FACTOR = 0.02;
+function createPanUpdate(camera, controls, keys) {
+  const _panRight = new THREE__namespace.Vector3();
+  const _panUp = new THREE__namespace.Vector3();
+  const _panOffset = new THREE__namespace.Vector3();
+  return function update() {
+    let dx = 0;
+    let dy = 0;
+    if (keys["ArrowLeft"] || keys["a"] || keys["A"]) dx -= 1;
+    if (keys["ArrowRight"] || keys["d"] || keys["D"]) dx += 1;
+    if (keys["ArrowUp"] || keys["w"] || keys["W"]) dy += 1;
+    if (keys["ArrowDown"] || keys["s"] || keys["S"]) dy -= 1;
+    const zoomIn = keys["z"] || keys["Z"] || keys["+"] || keys["="];
+    const zoomOut = keys["x"] || keys["X"] || keys["-"] || keys["_"];
+    if (!dx && !dy && !zoomIn && !zoomOut) return false;
+    const target = controls.target;
+    const dist = camera.position.distanceTo(target);
+    if (dx || dy) {
+      const step = Math.max(dist, 50) * PAN_SPEED;
+      _panRight.setFromMatrixColumn(camera.matrix, 0);
+      _panUp.setFromMatrixColumn(camera.matrix, 1);
+      _panOffset.set(0, 0, 0).addScaledVector(_panRight, dx * step).addScaledVector(_panUp, dy * step);
+      camera.position.add(_panOffset);
+      target.add(_panOffset);
+    }
+    if (zoomIn || zoomOut) {
+      const factor = zoomIn ? -PAN_ZOOM_FACTOR : PAN_ZOOM_FACTOR;
+      _panOffset.copy(camera.position).sub(target);
+      const newLen = Math.max(
+        controls.minDistance,
+        Math.min(controls.maxDistance, _panOffset.length() * (1 + factor))
+      );
+      _panOffset.setLength(newLen);
+      camera.position.copy(target).add(_panOffset);
+    }
+    return true;
+  };
+}
 function createOrbitUpdate(camera, controls, keys) {
   return function update() {
     let zoom = 0;
@@ -2148,8 +2187,13 @@ function setupKeyboardControls(camera, controls, container, mode = "fly", flySpe
   container.addEventListener("keyup", onKeyUp);
   container.addEventListener("blur", onBlur);
   let enabled = true;
-  const innerUpdate = mode === "fly" ? createFlyUpdate(camera, controls, keys, state) : createOrbitUpdate(camera, controls, keys);
-  const updateFn = () => enabled ? innerUpdate() : false;
+  let currentMode = mode;
+  const updaters = {
+    fly: createFlyUpdate(camera, controls, keys, state),
+    orbit: createOrbitUpdate(camera, controls, keys),
+    pan: createPanUpdate(camera, controls, keys)
+  };
+  const updateFn = () => enabled ? updaters[currentMode]() : false;
   function cleanup() {
     container.removeEventListener("keydown", onKeyDown2);
     container.removeEventListener("keyup", onKeyUp);
@@ -2162,7 +2206,11 @@ function setupKeyboardControls(camera, controls, container, mode = "fly", flySpe
     enabled = on;
     if (!on) for (const k of Object.keys(keys)) keys[k] = false;
   }
-  return { update: updateFn, cleanup, setFlySpeed, setEnabled };
+  function setMode(m2) {
+    currentMode = m2;
+    for (const k of Object.keys(keys)) keys[k] = false;
+  }
+  return { update: updateFn, cleanup, setFlySpeed, setEnabled, setMode };
 }
 function createScene(container, backgroundColor, style, rendererConfig) {
   const scene = new THREE__namespace.Scene();
@@ -2202,7 +2250,10 @@ function createScene(container, backgroundColor, style, rendererConfig) {
   const cameraMode = (rendererConfig == null ? void 0 : rendererConfig.cameraMode) ?? "fly";
   const keyboard = setupKeyboardControls(camera, controls, container, cameraMode, style.flySpeed ?? 1);
   controls.enableZoom = false;
-  const flags = { is2D: false };
+  const flags = {
+    is2D: false,
+    keyboardMode3D: cameraMode
+  };
   const _wheelDir = new THREE__namespace.Vector3();
   const onWheel = (e) => {
     if (flags.is2D) return;
@@ -2236,15 +2287,16 @@ function createScene(container, backgroundColor, style, rendererConfig) {
 function applyCameraMode2D(state, is2D) {
   const { controls, camera, keyboard } = state;
   state.flags.is2D = is2D;
-  controls.enableRotate = !is2D;
+  controls.enableRotate = true;
   controls.enableZoom = is2D;
   controls.screenSpacePanning = is2D;
-  keyboard.setEnabled(!is2D);
+  keyboard.setEnabled(true);
+  keyboard.setMode(is2D ? "pan" : state.flags.keyboardMode3D);
   if (is2D) {
     controls.mouseButtons = {
       LEFT: THREE__namespace.MOUSE.PAN,
       MIDDLE: THREE__namespace.MOUSE.DOLLY,
-      RIGHT: THREE__namespace.MOUSE.PAN
+      RIGHT: THREE__namespace.MOUSE.ROTATE
     };
     controls.touches = { ONE: THREE__namespace.TOUCH.PAN, TWO: THREE__namespace.TOUCH.DOLLY_PAN };
     const t = controls.target;
@@ -2622,11 +2674,12 @@ function setupInteraction(canvas, camera, controls, getNodesMesh, getNodes, call
   const dragPlane = new THREE__namespace.Plane();
   const dragIntersect = new THREE__namespace.Vector3();
   function onPointerDown2(e) {
+    var _a;
     mouseDownPos = { x: e.clientX, y: e.clientY };
     if (container && document.activeElement !== container) {
       container.focus({ preventScroll: true });
     }
-    if (callbacks.onNodeDrag) {
+    if (callbacks.onNodeDrag && (((_a = options == null ? void 0 : options.getNodeDragEnabled) == null ? void 0 : _a.call(options)) ?? true)) {
       const nodesMesh = getNodesMesh();
       if (nodesMesh) {
         const rect = canvas.getBoundingClientRect();
@@ -2938,6 +2991,7 @@ function NetworkGraph3DInner(props, ref) {
     nodeValueAccessor,
     visibleNodeIds,
     linkVisibility,
+    enableNodeDrag = true,
     clickToFocus = true,
     hoverHighlight = false,
     hoverHighlightHops = 1
@@ -2978,6 +3032,7 @@ function NetworkGraph3DInner(props, ref) {
   const labelFormatterRef = react.useRef(labelFormatter);
   const highlightSetRef = react.useRef(null);
   const clickToFocusRef = react.useRef(clickToFocus);
+  const enableNodeDragRef = react.useRef(enableNodeDrag);
   const hoverHighlightRef = react.useRef(hoverHighlight);
   const hoverHighlightHopsRef = react.useRef(hoverHighlightHops);
   const adjacencyMapRef = react.useRef(/* @__PURE__ */ new Map());
@@ -2999,6 +3054,7 @@ function NetworkGraph3DInner(props, ref) {
   selectedNodeIdRef.current = selectedNodeId;
   labelFormatterRef.current = labelFormatter;
   clickToFocusRef.current = clickToFocus;
+  enableNodeDragRef.current = enableNodeDrag;
   hoverHighlightRef.current = hoverHighlight;
   hoverHighlightHopsRef.current = hoverHighlightHops;
   const is2D = (layoutProp == null ? void 0 : layoutProp.dimensions) === 2;
@@ -3181,7 +3237,8 @@ function NetworkGraph3DInner(props, ref) {
       },
       {
         getClickToFocus: () => clickToFocusRef.current,
-        getIs2D: () => is2DRef.current
+        getIs2D: () => is2DRef.current,
+        getNodeDragEnabled: () => enableNodeDragRef.current
       }
     );
     return () => {