修改水颜色

wangjuncheng

2025-06-24 fc3a075f1b5bd3735a47dd93165d33a0e526803b

 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,6 +9,7 @@
 */
export function destoryWaterPrimitive() {
  if (water) {
    enableWaterArrowFlow(false);
    water.destroy();
    water = null;
    console.log("Water simulation destroyed.");
@@ -20,37 +23,78 @@
 * @param {string} options.baseUrl - 仿真服务地址
 * @param {boolean} options.colorRender - 是否启用颜色渲染
 */
export function createWaterPrimitive(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.005; // 最小允许值
  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,
    alphaByDepth: -0.3, //深度衰减，值越大，水面越透明.
    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" },
      { height: 30.0, color: "#4d0a08" },
    ],
    colorRender, // 控制是否启用颜色渲染
    alphaByDepth: -0.3,
    waterHeightLevels,
    colorRender,
    sizeIndex: 0,
  });

  enableWaterArrowFlow(false);
  toggleWaterShadow(false);
  console.log(
    `仿真模拟参数： 请求路径 ${baseUrl}, 帧间间隔 ${interval}ms, 是否开启专题渲染 ${colorRender}`
    `仿真模拟参数：请求路径 ${baseUrl}, 帧间间隔 ${interval}ms, 是否开启专题渲染 ${colorRender}`
  );
}
/**
@@ -72,7 +116,17 @@
  viewer.scene.camera.flyTo(view);
  // console.log("Camera view initialized for water simulation.");
}

/**
 * 更新水颜色
 */
export function updateWaterColor(color) {
  if (water) {
    water.color = Cesium.Color.fromCssColorString(color);
    console.log("切换颜色为：",color);
  } else {
    console.warn("No water simulation to pause.");
  }
}
/**
 * 暂停水体模拟
 */
@@ -131,8 +185,46 @@
    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 - 当前时间戳
 */