1

wangjuncheng

5 天以前 5a93ff9c70a25e09d77aff8e9175022b63b3060f

 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.");
  }
}
// ==================【SDK 新增功能 - 箭头流向 & 阴影控制】==================

/**
 * 开启/关闭箭头流向动画
 * @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}`);
}
}