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 |  191 ++++++++++++++++++++++++++++++++++++++++-------
 1 files changed, 162 insertions(+), 29 deletions(-)

diff --git a/src/utils/water.js b/src/utils/water.js
index 7acbb37..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.");
   }
 }
 
@@ -17,20 +20,90 @@
  * 鍒涘缓姘翠綋妯℃嫙灞�
  * @param {Object} options - 鍙�夊弬鏁�
  * @param {number} options.interval - 姘翠綋妯℃嫙鐨勬椂闂撮棿闅旓紙鍗曚綅锛氭绉掞級
+ * @param {string} options.baseUrl - 浠跨湡鏈嶅姟鍦板潃
+ * @param {boolean} options.colorRender - 鏄惁鍚敤棰滆壊娓叉煋
  */
-export function createWaterPrimitive(options = {}) {
-  const { interval = 1000 } = options; // 榛樿 interval 涓� 1000
+export async function createWaterPrimitive(options = {}) {
+  const {
+    baseUrl = "/simu/c2h1dc",
+    interval = 1000,
+    colorRender = true,
+    minFlowRate = 0.1, // 鏂板鍙傛暟
+    maxFlowRate = 12, // 鏂板鍙傛暟
+  } = options;
 
-  water = earthCtrl.simulate.createWaterSimulateLayer({
-    baseUrl: "/simu/c2h1dc", // 浠跨湡鏈嶅姟 URL
-    interval, // 鍔ㄦ�佽缃� interval
-    color: new SmartEarth.Cesium.Color.fromCssColorString("#D4F2E7"),
-    loop: false, // 鏄惁寰幆鎾斁
-    callback: timeCallback, // 鍥炶皟鍑芥暟
+  // 瀹氫箟姘存繁棰滆壊鏄犲皠鐨勮壊鏍�
+  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: SmartEarth.Cesium.Color.fromCssColorString("#D4F2E7"),
+    loop: false,
+    callback: timeCallback,
+    alphaByDepth: -0.3,
+    waterHeightLevels,
+    colorRender,
+    sizeIndex: 0,
   });
-  console.log(`Water simulation started with interval: ${interval}ms`);
-}
+  //闃叉缂╂斁瀵艰嚧鍦板舰鍙樺姩鍘嬬洊姘撮潰锛屾按闈㈠鍔犺缃�
+  water.clampMinHeight = 0; //鐩告満鐩稿浜庢按闈㈡渶灏忛珮搴�
+  water.clampMaxHeight = 1000; //鐩告満鐩稿浜庢按闈㈡渶澶ч珮搴�
+  water.offsetMinHeight = 0; //姘村亸绉绘渶灏忛珮搴�
+  water.offsetMaxHeight = 100; //姘村亸绉绘渶澶ч珮搴�
+  // 鏄惁寮�鍚澶�
+  enableWaterArrowFlow(false); 
+  // 鏄惁寮�鍚按闈㈤槾褰�
+  toggleWaterShadow(false);
 
+  // console.log(
+  //   `浠跨湡妯℃嫙鍙傛暟锛氳姹傝矾寰� ${baseUrl}, 甯ч棿闂撮殧 ${interval}ms, 鏄惁寮�鍚笓棰樻覆鏌� ${colorRender}`
+  // );
+}
 /**
  * 鍒濆鍖栨按浣撴ā鎷熻鍥�
  */
@@ -48,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.");
   }
@@ -69,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.");
   }
@@ -77,31 +160,81 @@
 
 /**
  * 璺宠浆鍒版煇涓椂闂寸偣鐨勬按闈㈢姸鎬�
+ * @param {number} closestIndex - 鐩爣鏃堕棿鎴崇储寮�
  */
 export function setTimeForWaterSimulation(closestIndex) {
-  console.log(closestIndex,'index');
-  
   if (water) {
-    const imageList = water.getTimeList(); // 鑾峰彇鎵�鏈夊彲鐢ㄦ椂闂存埑
-    if (imageList.length === 0) {
+    const imageList = water.getTimeList();
+
+    if (!imageList.length) {
       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:[${idx}]`
-    );
-    
-    water.setTime(imageList[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.");
   }
 }
 
 /**
+ * 璁剧疆鎴栧叧闂鑹叉覆鏌�
+ * @param {boolean} enabled
+ */
+export function toggleWaterColorRender(enabled) {
+  if (water) {
+    water.colorRender = 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
+  // console.log(`Current timestamp: ${timeStamp}`);
+}

--
Gitblit v1.9.3