performance-test/src/views/performance/components/chartsCom/3DDrawingChart.vue

<template>
  <div>
    <!-- 配色方案选择和图表类型切换 -->
    <div style="display: flex; align-items: center; padding-top: 20px">
      <div style="margin-right: 20px; display: flex; align-items: center">
        <el-select
          size="small"
          v-model="color1"
          @change="updateChartColor"
          placeholder="选择配色方案"
          style="width: 200px"
        >
          <el-option
            v-for="(scheme, index) in colorSchemes"
            :key="index"
            :label="scheme.label"
            :value="scheme.colors"
            :style="getOptionStyle(scheme.colors)"
          ></el-option>
        </el-select>
      </div>
    </div>
    <!-- 点大小控制 -->
    <div style="display: flex; align-items: center">
      <el-slider
        v-model="pointSize"
        :min="1"
        :max="15"
        :step="1"
        label="点的大小"
        show-stops
        style="width: 150px"
        @change="updateChartColor"
      ></el-slider>
    </div>

    <!-- 图表展示区域 -->
    <div style="height: 500px">
      <div
        v-loading="loading"
        :id="`plotly-3d-chart-` + index"
        ref="plotlyChart"
        style="height: 500px; background-color: #e5ecf6"
      >
        <el-empty v-if="isError" description="请求失败"></el-empty>
      </div>
    </div>
  </div>
</template>

<script>
import Plotly from "plotly.js-dist";
import axios from "axios";
import { myMixin } from "@/mixins/chartRequestMixin";
import { colorSchemes } from "@/views/overview/js/colors";
import { mapState } from "vuex";
export default {
  props: {
    fileAddr: {
      default: "",
      type: String,
    },
    index: {
      default: "",
      type: String,
    },
    setUpImgData: {
      default: () => [],
      type: Array,
    },
  },
  data() {
    return {
      color1: [
        "#DBEEBC",
        "#A8D7BE",
        "#8ECAC1",
        "#77BDC2",
        "#64ADC2",
        "#559ABE",
        "#4884B7",
        "#406DAB",
        "#3856A0",
        "#314291",
        "#28357A",
        "#1A285E",
        "#FF7F50",
        "#FFD700",
        "#90EE90",
      ], // 默认颜色
      chartData: {},
      chartType: "scatter", // 当前图表类型（默认是散点图）
      colorSchemes: [...colorSchemes],
      pointSize: 2, // 默认点大小
    };
  },
  mixins: [myMixin],
  async mounted() {
    this.getData();
    this.color1 = this.themeColor;
  },
  computed: {
    ...mapState("themes", {
      themeColor: "themeColor",
    }),
  },
  watch: {
    themeColor: {
      handler() {
        this.color1 = this.themeColor;
        this.updateChartColor();
      },
      deep: true,
    },
    setUpImgData: {
      handler(newType) {
        this.renderChart();
      },
      deep: true,
    },
  },
  methods: {
    async getData() {
      if (this.fileAddr !== "") {
        try {
          this.loading = true;
          this.cancelToken = axios.CancelToken.source();

          const resultChartsData = await axios.get(this.fileAddr, {
            cancelToken: this.cancelToken.token,
          });
          if (typeof resultChartsData.data === "string") {
            let dataString = resultChartsData.data;
            dataString = dataString.trim(); // 去除前后空格
            dataString = dataString.replace(/Infinity/g, '"Infinity"'); // 替换 Infinity 为 "Infinity"
            try {
              const parsedData = JSON.parse(dataString);
              this.chartData = parsedData;
            } catch (error) {
              console.error("JSON 解析失败:", error);
            }
          } else {
            this.chartData = resultChartsData.data;
          }
          this.renderChart();
          this.isError = false;
          this.loading = false;
        } catch (error) {
          this.isError = true;
          this.loading = false;
        }
      }
    },

    // 更新配色方案
    updateChartColor() {
      this.renderChart(); // 当配色方案或点大小发生变化时重新渲染图表
    },

    // 切换图表类型
    setChartType(type) {
      this.chartType = type;
      this.renderChart(); // 切换图表类型时重新渲染图表
    },

    // 获取配色选项样式
    getOptionStyle(scheme) {
      return {
        background: `linear-gradient(to right, ${scheme
          .slice(0, 8)
          .join(", ")})`,
        color: "#fff",
        height: "30px",
        lineHeight: "30px",
        borderRadius: "0px",
      };
    },

    renderChart() {
      console.log(this.setUpImgData, "setUpImgData");
      const uniqueColors = [...new Set(this.chartData.data[0].color)];
      if (!this.color1) {
        this.color1 = colorSchemes[0].colors;
      }
      const traces = uniqueColors.map((color, idx) => {
        const colorData = this.chartData.data[0].color.map((c) =>
          c === color ? 1 : 0
        );

        const trace = {
          x: this.chartData.data[0].xData.filter((_, i) => colorData[i] === 1),
          y: this.chartData.data[0].yData.filter((_, i) => colorData[i] === 1),
          z: this.chartData.data[0].zData.filter((_, i) => colorData[i] === 1),
          mode: this.chartType === "scatter" ? "markers" : "lines", // 根据选择的图表类型来设置模式
          type: "scatter3d",
          marker: {
            size: this.pointSize, // 使用动态点大小
            color: this.color1[idx], // 使用配色方案
            colorscale: "YlGnBu",
          },
          name: ` ${color}`,
          legendgroup: `group-${idx}`,
          hovertemplate:
            `${this.chartData.xaixs}:` +
            ` %{x} <br> ` +
            `${this.chartData.yaixs}:` +
            "%{y} <br>" +
            `${this.chartData.zaixs}:` +
            "%{z} <br>",
        };
        return trace;
      });

      const layout = {
        title: this.chartData.data[0].title,
        scene: {
          xaxis: {
            title: this.chartData.xaixs,
            gridcolor: "rgb(255,255,255)",
            tickcolor: "rgb(255,255,255)",
            backgroundcolor: "#e5ecf6",
            showbackground: true,
          },
          yaxis: {
            title: this.chartData.yaixs,
            tickvals: [...new Set(this.chartData.data[0].yData)],
            ticktext: [...new Set(this.chartData.data[0].yData)],
            gridcolor: "rgb(255,255,255)",
            tickcolor: "rgb(255,255,255)",

            backgroundcolor: "#e5ecf6",
            showbackground: true,
          },
          zaxis: {
            title: this.chartData.zaixs,
            gridcolor: "rgb(255,255,255)",
            tickcolor: "rgb(255,255,255)",
            backgroundcolor: "#e5ecf6",
            showbackground: true,
          },
          aspectratio: {
            x: 3,
            y: 1,
            z: 1,
          },
          // plot_bgcolor: "#e5ecf6",
          // gridcolor: "#fff",
          camera: {
            center: { x: -0.233946, y: -0.319396, z: 0.483055 },
            eye: { x: -1.729102, y: -3.506569, z: -1.296184 },
            up: { x: 0.12359, y: 0.43886022, z: -0.89001518 },
            projection: {
              type: "perspective",
            },
          },
        },
        margin: { t: 50, b: 50, l: 50, r: 50 },
        staticPlot: false,
        showlegend: true,
        legend: {
          marker: {
            size: 10, // 图例中点的大小
          },
        },
      };
      // 获取x轴和y轴的设置
      const getChartSetUp = (axisTitle) => {
        return this.setUpImgData.find((item) => item.text.includes(axisTitle));
      };

      // 更新x轴和y轴的范围与步长
      const xChartSetUp = getChartSetUp(layout.scene.xaxis.title);
      if (xChartSetUp) {
        layout.scene.xaxis.dtick = xChartSetUp.dtick;
        layout.scene.xaxis.range = [xChartSetUp.min, xChartSetUp.max];
      }
      const yChartSetUp = getChartSetUp(layout.scene.yaxis.title);
      if (yChartSetUp) {
        layout.scene.yaxis.dtick = yChartSetUp.dtick;
        layout.scene.yaxis.range = [yChartSetUp.min, yChartSetUp.max];
      }
      const zChartSetUp = getChartSetUp(layout.scene.zaxis.title);
      if (zChartSetUp) {
        layout.scene.zaxis.dtick = zChartSetUp.dtick;
        layout.scene.zaxis.range = [zChartSetUp.min, zChartSetUp.max];
      }

      try {
        // 假设这里是 WebGL 的相关初始化代码
        Plotly.react(`plotly-3d-chart-` + this.index, traces, layout).catch(
          (err) => {
            console.error("WebGL 错误: ", err);
            // 你可以根据错误类型做更多处理
            if (err.message.includes("shaderSource")) {
              // alert("着色器编译失败！");
            }
          }
        );
        // 监听图表的 relayout 事件，获取并输出相机视角
        const plotElement = document.getElementById(
          `plotly-3d-chart-` + this.index
        );

        plotElement.on("plotly_relayout", function (eventData) {
          // 在每次布局变更时，打印当前相机视角
          if (eventData["scene.camera"]) {
            console.log(
              "当前相机视角:",
              eventData["scene.camera"],
              eventData["scene.aspectratio"]
            );
          }
        });
      } catch (e) {
        console.error("捕获到 WebGL 错误:", e);
        // alert("图表渲染失败！");
      }
      // Plotly.newPlot(`plotly-3d-chart-` + this.index, traces, layout);
    },
  },
};
</script>

<style scoped>
#scene {
  background: #e5ecf6 !important;
}
.js-plotly-plot .plotly,
.js-plotly-plot .plotly div {
  background: #e5ecf6 !important;
}
/* 样式可以根据需求自定义 */
#plotly-3d-chart {
  width: 100%;
  height: 500px;
}

::v-deep canvas {
  /* height: 400px !important; */
}
</style>