From 5a93ff9c70a25e09d77aff8e9175022b63b3060f Mon Sep 17 00:00:00 2001
From: wangjuncheng <1>
Date: 星期五, 18 七月 2025 10:54:23 +0800
Subject: [PATCH] 1

---
 src/utils/water.js |  166 +++++++++++++++++++++++++++++++++++++++++++++---------
 1 files changed, 137 insertions(+), 29 deletions(-)

diff --git a/src/utils/water.js b/src/utils/water.js
index 22723c3..3d7de0b 100644
--- a/src/utils/water.js
+++ b/src/utils/water.js
@@ -1,5 +1,7 @@
-import { cartesianToXY } from "@/utils/map";
-
+import { useSimStore } from "@/store/simulation";
+import { storeToRefs } from "pinia";
+const simStore = useSimStore();
+const { waterLegendData } = storeToRefs(simStore);
 let water = null;
 
 /**
@@ -7,9 +9,10 @@
  */
 export function destoryWaterPrimitive() {
   if (water) {
+    enableWaterArrowFlow(false);
     water.destroy();
     water = null;
-    console.log("Water simulation destroyed.");
+    // console.log("Water simulation destroyed.");
   }
 }
 
@@ -20,30 +23,86 @@
  * @param {string} options.baseUrl - 浠跨湡鏈嶅姟鍦板潃
  * @param {boolean} options.colorRender - 鏄惁鍚敤棰滆壊娓叉煋
  */
-export function createWaterPrimitive(options = {}) {
-  const { baseUrl = "/simu/c2h1dc", interval = 1000, colorRender = true } = options;
+export async function createWaterPrimitive(options = {}) {
+  const {
+    baseUrl = "/simu/c2h1dc",
+    interval = 1000,
+    colorRender = true,
+    minFlowRate = 0.1, // 鏂板鍙傛暟
+    maxFlowRate = 12, // 鏂板鍙傛暟
+  } = options;
 
-  water = earthCtrl.simulate.createWaterSimulateLayer({
+  // 瀹氫箟姘存繁棰滆壊鏄犲皠鐨勮壊鏍�
+  const colorStops = [
+    "#09a2dc",
+    "#58c196",
+    "#bedf74",
+    "#d7f06e",
+    "#ffe930",
+    "#fdd10a",
+    "#feb652",
+    "#fd7f06",
+    "#fe2b07",
+    "#4d0a08",
+  ];
+
+  const levelCount = colorStops.length;
+  const minAllowed = 0.05; // 鏈�灏忓厑璁稿��
+  const threshold = 1; // 灏忓�间笌澶у�煎垎鐣岀偣
+
+  let effectiveMin = Math.max(minFlowRate, minAllowed); // 鏈�灏忎笉鑳藉皬浜� 0.01
+
+  const waterHeightLevels = [];
+
+  // 鍒嗕袱娈垫瀯閫犻珮搴︽暟缁�
+  for (let i = 0; i < levelCount; i++) {
+    let ratio = i / (levelCount - 1); // 0 ~ 1
+
+    let height;
+    if (ratio <= 0.5) {
+      // 鍓嶅崐娈碉細浣庡�煎尯鍩燂紝浣跨敤寮烘寚鏁板闀匡紝浠� effectiveMin 鍒� threshold
+      const localRatio = ratio * 2; // 鏄犲皠鍒� 0~1
+      const expRatio = Math.pow(localRatio, 2); // 鏇村己璋冧綆鍊煎尯鍩熷瘑搴�
+      height = effectiveMin + (threshold - effectiveMin) * expRatio;
+    } else {
+      // 鍚庡崐娈碉細楂樺�煎尯鍩燂紝浠� threshold 鍒� maxFlowRate锛屼娇鐢ㄦ寚鏁板闀�
+      const localRatio = (ratio - 0.5) * 2; // 鏄犲皠鍒� 0~1
+      const expBase = Math.exp(Math.log(maxFlowRate / threshold) / 1);
+      height = threshold * Math.pow(expBase, localRatio);
+    }
+
+    waterHeightLevels.push({
+      height: parseFloat(height.toFixed(2)), // 淇濈暀涓や綅灏忔暟
+      color: colorStops[i],
+    });
+  }
+
+  waterLegendData.value = waterHeightLevels;
+  // console.log(waterLegendData.value, "鍥句緥鏁版嵁");
+  water = await earthCtrl.simulate.createWaterSimulateLayer({
     baseUrl,
     interval,
-    color: new SmartEarth.Cesium.Color.fromCssColorString("#D4F2E7"),
+    color: SmartEarth.Cesium.Color.fromCssColorString("#D4F2E7"),
     loop: false,
     callback: timeCallback,
-    waterHeightLevels: [
-      { height: 0.5, color: "#09a2dc" },
-      { height: 1.0, color: "#58c196" },
-      { height: 1.5, color: "#bedf74" },
-      { height: 2.0, color: "#d7f06e" },
-      { height: 2.5, color: "#ffe930" },
-      { height: 3.0, color: "#fdd10a" },
-      { height: 4.0, color: "#feb652" },
-      { height: 5.0, color: "#fd7f06" },
-      { height: 10.0, color: "#fe2b07" },
-    ],
-    colorRender, // 鎺у埗鏄惁鍚敤棰滆壊娓叉煋
+    alphaByDepth: -0.3,
+    waterHeightLevels,
+    colorRender,
+    sizeIndex: 0,
   });
+  //闃叉缂╂斁瀵艰嚧鍦板舰鍙樺姩鍘嬬洊姘撮潰锛屾按闈㈠鍔犺缃�
+  water.clampMinHeight = 0; //鐩告満鐩稿浜庢按闈㈡渶灏忛珮搴�
+  water.clampMaxHeight = 1000; //鐩告満鐩稿浜庢按闈㈡渶澶ч珮搴�
+  water.offsetMinHeight = 0; //姘村亸绉绘渶灏忛珮搴�
+  water.offsetMaxHeight = 100; //姘村亸绉绘渶澶ч珮搴�
+  // 鏄惁寮�鍚澶�
+  enableWaterArrowFlow(false); 
+  // 鏄惁寮�鍚按闈㈤槾褰�
+  toggleWaterShadow(false);
 
-  console.log(`Water simulation started with baseUrl: ${baseUrl}, interval: ${interval}ms, colorRender: ${colorRender}`);
+  // console.log(
+  //   `浠跨湡妯℃嫙鍙傛暟锛氳姹傝矾寰� ${baseUrl}, 甯ч棿闂撮殧 ${interval}ms, 鏄惁寮�鍚笓棰樻覆鏌� ${colorRender}`
+  // );
 }
 /**
  * 鍒濆鍖栨按浣撴ā鎷熻鍥�
@@ -62,16 +121,26 @@
     },
   };
   viewer.scene.camera.flyTo(view);
-  console.log("Camera view initialized for water simulation.");
+  // console.log("Camera view initialized for water simulation.");
 }
-
+/**
+ * 鏇存崲姘撮�忔槑搴�
+ */
+export function updateWaterColor(color, alpha) {
+  if (water) {
+    water.color = Cesium.Color.fromCssColorString(color);
+    water.alphaByDepth = alpha;
+  } else {
+    console.warn("No water simulation to pause.");
+  }
+}
 /**
  * 鏆傚仠姘翠綋妯℃嫙
  */
 export function pauseWaterSimulation() {
   if (water) {
     water.pause();
-    console.log("Water simulation paused.");
+    // console.log("鏆傚仠浠跨湡");
   } else {
     console.warn("No water simulation to pause.");
   }
@@ -83,7 +152,7 @@
 export function resumeWaterSimulation() {
   if (water) {
     water.resume();
-    console.log("Water simulation resumed.");
+    // console.log("缁х画浠跨湡");
   } else {
     console.warn("No water simulation to resume.");
   }
@@ -101,9 +170,10 @@
       console.warn("No timestamps available for water simulation.");
       return;
     }
-
-    const idx = Math.floor(Math.random() * imageList.length);
-    console.log(`Jumping to timestamp: count:[${imageList.length}], index:[${closestIndex}]`);
+    // const idx = Math.floor(Math.random() * imageList.length); //闅忔満绱㈠紩璺宠浆锛屽疄闄呬腑鐢ㄤ笉鍒帮紝鍙敤浣滄紨绀�
+    // console.log(
+    //   `Jumping to timestamp: count:[${imageList.length}], index:[${closestIndex}]`
+    // );
     water.setTime(imageList[closestIndex]);
   } else {
     console.warn("No water simulation to set time for.");
@@ -117,16 +187,54 @@
 export function toggleWaterColorRender(enabled) {
   if (water) {
     water.colorRender = enabled;
-    console.log(`Water color render set to: ${enabled}`);
+    // console.log(`鏄惁寮�鍚笓棰樻覆鏌� ${enabled}`);
   } else {
     console.warn("No water simulation to toggle color rendering.");
   }
 }
+// ==================銆怱DK 鏂板鍔熻兘 - 绠ご娴佸悜 & 闃村奖鎺у埗銆�==================
 
+/**
+ * 寮�鍚�/鍏抽棴绠ご娴佸悜鍔ㄧ敾
+ * @param {boolean} enabled - 鏄惁鍚敤绠ご鍔ㄧ敾
+ */
+export function enableWaterArrowFlow(enabled) {
+  if (water) {
+    // 榛樿鍏抽棴鐘舵��
+    water.flowEnabled = enabled; // 鍋囪 SDK 鏀寔姝ゅ睘鎬�
+    // console.log(`绠ご娴佸悜鍔ㄧ敾宸�${enabled ? "寮�鍚�" : "鍏抽棴"}`);
+  } else {
+    console.warn("鏈壘鍒版按浣撴ā鎷熷浘灞傦紝璇峰厛鍚姩娲按妯℃嫙");
+  }
+}
+
+/**
+ * 寮�鍚�/鍏抽棴鍏ㄥ眬闃村奖鏁堟灉
+ * @param {boolean} enabled - 鏄惁鍚敤闃村奖
+ */
+export function toggleWaterShadow(enabled) {
+  if (!viewer) {
+    console.warn("Cesium Viewer 鏈垵濮嬪寲锛屾棤娉曡缃槾褰�");
+    return;
+  }
+
+  try {
+    earthCtrl.shadows = enabled;
+    if (enabled) {
+      earthCtrl.shadowMap.maximumDistance = 10000.0; //鏈�澶ц窛绂�
+      earthCtrl.shadowMap.pointLightRadius = 50.0; //鐐瑰厜婧愬崐寰�
+    }
+    // console.log(`闃村奖鏁堟灉宸�${enabled ? "寮�鍚�" : "鍏抽棴"}`);
+  } catch (error) {
+    console.error("璁剧疆闃村奖澶辫触:", error);
+  }
+}
+
+// ==================================================
 /**
  * 鏃堕棿鎴冲洖璋冨嚱鏁�
  * @param {number} timeStamp - 褰撳墠鏃堕棿鎴�
  */
 function timeCallback(timeStamp) {
   // console.log(`Current timestamp: ${timeStamp}`);
-}
\ No newline at end of file
+}

--
Gitblit v1.9.3