AnalysisSystemForRadionuclide_vue/src/views/spectrumAnalysis/components/Modals/AnalyzeInteractiveToolModal/index.vue

<template>
  <custom-modal v-model="visible" :width="1280" title="Interactive Analyse Tools" :footer="null" destroy-on-close>
    <a-spin :spinning="isLoading">
      <div class="interactive-analysis-tools">
        <div class="interactive-analysis-tools-left">
          <div class="chart">
            <CustomChart
              ref="chartRef"
              :option="option"
              :opts="opts"
              @zr:mousedown="handleMouseDown"
              @zr:mouseup="handleMouseUp"
              @brushEnd="handleBrushEnd"
              @zr:click="handleChartClick"
            />
          </div>
          <!-- 缩略图 -->
          <div class="thumbnail">
            <CustomChart :option="thumbnailOption" />
          </div>
          <!-- 缩略图结束 -->

          <!-- 表格 -->
          <div class="table">
            <p class="title">
              <span @click="handleChangeMarkLine('prev')">&lt; </span>
              6 Peaks with Anthro.Nuclides
              <span @click="handleChangeMarkLine('next')">&gt;</span>
            </p>
            <custom-table
              :class="list.length ? 'has-data' : ''"
              :list="list"
              :columns="columns"
              :scroll="{ y: 288 }"
              :selectedRowKeys.sync="selectedKeys"
              rowKey="index"
              :canDeselect="false"
              @rowClick="handleTableRowClick"
            >
            </custom-table>
            <div class="operators">
              <a-button type="primary" @click="nuclideReviewModalVisible = true">Nuclide Review Window</a-button>
              <a-button type="primary" @click="handleAddPeakComment()">Add Peak Comment</a-button>
              <a-button type="primary" @click="handleAddGeneralComment()">Add General Comment</a-button>
            </div>
          </div>
          <!-- 表格结束 -->
        </div>

        <!-- 右侧 -->
        <div class="interactive-analysis-tools-right">
          <title-over-border :title="btnGroupType == 1 ? 'Peak' : 'Baseline Control Points'">
            <div class="peak-box">
              <!-- 按钮组1 -->
              <template v-if="btnGroupType == 1">
                <div class="peak-box-item">
                  <a-button type="primary" @click="handleInsert">Insert</a-button>
                </div>
                <div class="peak-box-item">
                  <a-button type="primary" @click="handleDel">Delete</a-button>
                </div>
                <div class="peak-box-item">
                  <a-button type="primary" :class="{ 'is-fitting': isFitting }" @click="handleFit">Fit</a-button>
                </div>

                <div class="peak-box-item symbol" :key="4">
                  <a-button type="primary" @click="handleChangeMarkLine('prev')">&lt;</a-button>
                  <a-button type="primary" @click="handleChangeMarkLine('next')">&gt;</a-button>
                </div>

                <div class="peak-box-item base-line">
                  <a-button type="primary" @click="handleSwitchOperation">BaseLine</a-button>
                </div>
              </template>
              <!-- 按钮组2 -->
              <template v-if="btnGroupType == 2">
                <div class="peak-box-item">
                  <a-button type="primary" @click="handleAddCP">(A)dd CP</a-button>
                </div>
                <div class="peak-box-item">
                  <a-button type="primary" @click="handleRemoveCP">(R)emove CP</a-button>
                </div>
                <div class="peak-box-item">
                  <a-button
                    type="primary"
                    key="modify-btn"
                    :class="{ 'is-modify': isModifying }"
                    @click="handleModifyCP"
                    >(M)odify CP</a-button
                  >
                </div>
                <div class="peak-box-item">
                  <a-button type="primary" @click="handleEditSlope">Edit (S)lope</a-button>
                </div>
                <div class="peak-box-item">
                  <a-button type="primary" :disabled="isOperationStackEmpty" @click="handleUndo">Undo</a-button>
                </div>
                <div class="peak-box-item">
                  <a-button type="primary" @click="handleReplot">Replot</a-button>
                </div>
                <div class="peak-box-item">
                  <a-button type="primary" @click="handleAccept">Accept</a-button>
                </div>
                <div class="peak-box-item">
                  <a-button type="primary" @click="handleSwitchOperation">Cancel</a-button>
                </div>
              </template>
            </div>
          </title-over-border>
          <div class="reset-btn-box">
            <a-button type="primary" @click="handleResetChart">Reset Chart</a-button>
          </div>
          <div class="identify-box">
            <title-over-border title="Nuclide Identify">
              <a-form-model class="tolerance">
                <a-form-model-item label="Tolerance">
                  <a-input-number v-model="model.tolerance"></a-input-number>
                </a-form-model-item>
              </a-form-model>
              <div class="identify-item">
                <div class="title">
                  Possible Nuclide
                </div>
                <a-spin :spinning="!!(selectedTableItem && selectedTableItem._loading)">
                  <div class="content">
                    <template v-if="selectedTableItem && selectedTableItem._possible">
                      <div
                        class="item"
                        :class="{ active: possible == model.possibleNuclide }"
                        v-for="(possible, index) in selectedTableItem._possible"
                        :key="index"
                        @click="model.possibleNuclide = possible"
                      >
                        {{ possible }}
                      </div>
                    </template>
                  </div>
                </a-spin>
              </div>
              <div class="identify-item">
                <div class="title">
                  Nuclide Identified
                </div>
                <div class="content">
                  <template v-if="selectedTableItem">
                    <div
                      class="item"
                      :class="{ active: identified == model.identifiedNuclide }"
                      v-for="(identified, index) in selectedTableItem.nuclides"
                      :key="index"
                      @click="model.identifiedNuclide = identified"
                    >
                      {{ identified }}
                    </div>
                  </template>
                </div>
              </div>
              <div class="identify-operators">
                <span class="text">{{ model.possibleNuclide }}</span>
                <a-button type="primary" :disabled="!model.possibleNuclide" @click="handleAddNuclide">Add</a-button>
                <a-button type="primary" @click="handleDelNuclide">Del</a-button>
              </div>
            </title-over-border>
          </div>
        </div>
        <!-- 右侧结束 -->
      </div>
    </a-spin>
    <!-- Peak Comment弹窗 开始 -->
    <peak-comment-modal v-model="peakCommentModalVisible" :curRow="curRow" />
    <!-- Peak Comment弹窗 结束 -->

    <!-- General Comment弹窗 开始 -->
    <general-comment-modal v-model="generalCommentModalVisible" />
    <!-- General Comment弹窗 结束 -->

    <!-- Fit Peaks and Baseline弹窗 开始 -->
    <fit-peaks-and-base-line-modal
      v-model="fitPeaksAndBaselineModalVisible"
      :curChan="currChannel"
      @result="handleInsertSuccess"
      @cancel="handleCancelSuccess"
    />
    <!-- Fit Peaks and Baseline弹窗 结束 -->
    <!-- Nuclide Review 弹窗开始 -->
    <nuclide-review-modal v-model="nuclideReviewModalVisible" :sampleId="sampleId" :channel="currChannel" />
    <!-- Nuclide Review 弹窗结束 -->

    <!-- Edit Slope 弹窗 -->
    <edit-slope-modal ref="editSlopeModal" @change="handleSlopeChange" />
    <!-- Edit Slope 结束 -->
  </custom-modal>
</template>

<script>
import CustomChart from '@/components/CustomChart/index.vue'
import TitleOverBorder from '../../TitleOverBorder.vue'
import PeakCommentModal from './components/PeakCommentModal.vue'
import FitPeaksAndBaseLineModal from './components/FitPeaksAndBaselineModal.vue'
import NuclideReviewModal from './components/NuclideReviewModal.vue'
import ModalMixin from '@/mixins/ModalMixin'
import { getAction, postAction } from '@/api/manage'
import { cloneDeep } from 'lodash'
import { buildLineSeries, findSeriesByName, getXAxisAndYAxisByPosition, rangeNumber } from '@/utils/chartHelper'
import SampleDataMixin from '@/views/spectrumAnalysis/SampleDataMixin'
import GeneralCommentModal from './components/GeneralCommentModal.vue'
import EditSlopeModal from './components/EditSlopeModal.vue'
import Response from './Response.json'

// 初始配置
const initialOption = {
  grid: {
    top: 40,
    left: 40,
    right: 30,
    bottom: 30,
    containLabel: true
  },
  title: {
    text: '',
    left: 'center',
    bottom: 10,
    textStyle: {
      color: '#8FD4F8',
      rich: {
        a: {
          padding: [0, 20, 0, 0],
          fontSize: 16
        }
      }
    }
  },
  tooltip: {
    trigger: 'axis',
    axisPointer: {
      lineStyle: {
        color: '#3CAEBB',
        width: 1
      }
    },
    formatter: undefined,
    className: 'figure-chart-option-tooltip'
  },
  xAxis: {
    axisLine: {
      lineStyle: {
        color: '#ade6ee'
      }
    },
    splitLine: {
      show: false
    },
    axisLabel: {
      textStyle: {
        color: '#ade6ee'
      }
    },
    min: 1,
    max: 'dataMax',
    animation: false
  },
  yAxis: {
    type: 'log',
    name: 'Counts',
    nameLocation: 'center',
    nameGap: 40,
    nameTextStyle: {
      color: '#8FD4F8',
      fontSize: 16
    },
    axisLine: {
      show: true,
      lineStyle: {
        color: '#ade6ee'
      }
    },
    splitLine: {
      show: true,
      lineStyle: {
        color: 'rgba(173, 230, 238, .2)'
      }
    },
    axisLabel: {
      textStyle: {
        color: '#ade6ee'
      }
    },
    min: 0.1,
    max: 'dataMax',
    animation: false
  },
  series: [],
  brush: {},
  graphic: [
    {
      type: 'group',
      id: 1,
      children: []
    },
    {
      type: 'group',
      id: 2,
      children: []
    }
  ]
}

const columns = [
  {
    title: 'ID',
    customRender: (_, __, index) => {
      return index + 1
    },
    width: 60
  },
  {
    title: 'Energy (keV)',
    dataIndex: 'energy',
    width: 120,
    customRender: text => {
      return text.toFixed(3)
    }
  },
  {
    title: 'Centroid (C)',
    dataIndex: 'peakCentroid',
    width: 120,
    customRender: text => {
      return text.toFixed(3)
    }
  },
  {
    title: 'FWHM (keV)',
    dataIndex: 'fwhm',
    width: 120,
    customRender: text => {
      return text.toFixed(3)
    }
  },
  {
    title: 'Area',
    dataIndex: 'area',
    width: 120,
    customRender: text => {
      return text.toFixed(3)
    }
  },
  {
    title: 'Detectability',
    dataIndex: 'significance',
    width: 120,
    customRender: text => {
      return text.toFixed(3)
    }
  },
  {
    title: '#Cmnt',
    dataIndex: 'comments',
    width: 120
  },
  {
    title: 'Nuclides',
    dataIndex: 'nuclides',
    width: 120,
    customRender: text => {
      return text && text.join(';')
    }
  }
]

// 缩略图配置
const thumbnailOption = {
  grid: {
    top: 0,
    left: 5,
    right: 5,
    bottom: 0
  },
  xAxis: {
    type: 'category',
    axisLine: {
      show: false
    },
    splitLine: {
      show: false
    },
    axisLabel: {
      show: false
    },
    min: 1,
    max: 'dataMax'
  },
  yAxis: {
    type: 'value',
    axisLine: {
      show: false
    },
    splitLine: {
      show: false
    },
    axisLabel: {
      show: false
    },
    min: 0.1,
    max: 'dataMax'
  },
  series: null
}

const nuclideIdentifyModal = {
  possibleNuclide: '',
  tolerance: 0.5,
  identifiedNuclide: ''
}

// 操作类型
export const Operators = {
  ADD: 1, // 新增
  REMOVE: 2, // 移除
  MODIFY: 3, // 改变
  SLOPE_CHANGE: 4 // 改变slope
}

export default {
  mixins: [ModalMixin, SampleDataMixin],
  components: {
    CustomChart,
    TitleOverBorder,
    PeakCommentModal,
    FitPeaksAndBaseLineModal,
    NuclideReviewModal,
    GeneralCommentModal,
    EditSlopeModal
  },
  data() {
    this.columns = columns
    return {
      option: cloneDeep(initialOption),
      opts: { notMerge: false },
      thumbnailOption: cloneDeep(thumbnailOption),

      isLoading: false,

      channelBaseCPChart: [],
      channelBaseLineChart: [],
      channelCountChart: [],
      channelPeakChart: [],
      energy: [],
      list: [],
      BaseCtrls: {},
      FitBaseLine: '#fff',
      sampleId: -1,

      peakCommentModalVisible: false, // Comment 弹窗是否显示
      generalCommentModalVisible: false, // Comment 弹窗是否显示

      btnGroupType: 1, // 右侧 Peak 中的按钮组切换

      selectedKeys: [], // 选中的列表

      fitPeaksAndBaselineModalVisible: false, // Fit Peaks And Base Line 弹窗
      nuclideReviewModalVisible: false, // Nuclide Review 弹窗

      model: cloneDeep(nuclideIdentifyModal),

      currChannel: undefined, // 当currChannel前选中的channel
      selectedTableItem: undefined, // 当前选中的表格项

      isModifying: false, // 正在修改控制点
      isFitting: false, // 正在进行Fit操作
      firstFittingChannel: null, // Fit操作时点击的第一个channel

      operationStack: [] // 操作记录
    }
  },
  created() {
    this.option.tooltip.formatter = params => {
      const channel = parseInt(params[0].value[0])
      const energy = this.energy[channel - 1]
      return `<div class="channel">Channel: ${channel}</div>
              <div class="energy">Energy: ${energy.toFixed(2)}</div>`
    }
  },
  methods: {
    async getInfo() {
      try {
        this.isLoading = true
        this.option.series = []
        this.thumbnailOption.series = []
        this.list = []
        this.model = cloneDeep(nuclideIdentifyModal)

        const { success, result, message } = await getAction('/gamma/InteractiveTool', {
          sampleId: this.sampleId,
          fileName: this.fileName
        })

        // const { success, result, message } = Response

        if (success) {
          this.isLoading = false
          const {
            barChart,
            channelBaseCPChart,
            channelBaseLineChart,
            channelCountChart,
            channelPeakChart,
            energy,
            table,
            BaseCtrls,
            FitBaseLine
          } = result

          console.log('%c [  ]-374', 'font-size:13px; background:pink; color:#bf2c9f;', result)

          this.channelBaseCPChart = channelBaseCPChart
          this.channelBaseLineChart = channelBaseLineChart
          this.channelCountChart = channelCountChart
          this.channelPeakChart = channelPeakChart
          this.energy = energy
          this.BaseCtrls = BaseCtrls
          this.FitBaseLine = FitBaseLine

          const series = []

          // 推入BaseLine
          series.push(
            buildLineSeries(
              'BaseLine',
              channelBaseLineChart.pointlist.map(({ x, y }) => [x, y]),
              channelBaseLineChart.color,
              {
                markLine: {
                  silent: true,
                  symbol: 'none',
                  label: {
                    show: false
                  },
                  lineStyle: {
                    color: 'red',
                    width: 1
                  },
                  data: [{ xAxis: -1 }]
                },
                zlevel: 10
              }
            )
          )

          // 推入Count
          series.push(
            buildLineSeries(
              'CountChart',
              channelCountChart.pointlist.map(({ x, y }) => [x, y]),
              channelCountChart.color
            )
          )

          // 推入Peak
          const peakSeries = []
          channelPeakChart.forEach((item, index) => {
            peakSeries.push(
              buildLineSeries(
                'Peak_' + (index + 1),
                item.pointlist.map(({ x, y }) => [x, y]),
                item.color
              )
            )
          })

          series.push(...peakSeries)

          // 推入基线控制点
          series.push({
            name: 'BaseLine_Ctrl_Point',
            type: 'scatter',
            data: this.buildCPPointData(channelBaseCPChart),
            silent: true,
            animation: false,
            zlevel: 20
          })

          this.thumbnailOption.series = buildLineSeries(
            'BarChart',
            barChart.map(({ x, y }) => [x, y]),
            '#fff',
            {
              silent: true
            }
          )

          this.list = table
          this.option.series = series
        } else {
          this.$message.error(message)
        }
      } catch (error) {
        console.error(error)
      }
    },

    reset() {
      this.currChannel = undefined
      this.btnGroupType = 1
      this.opts.notMerge = false
      this.isFitting = false
      this.$nextTick(() => {
        this.option.brush = { toolbox: [] }
        this.selectedKeys = []
      })
      this.removeGraphics()
    },

    // 清空graphic（小方块和加号）
    removeGraphics() {
      this.option.graphic[0].children = []
      this.option.graphic[1].children = []
    },

    beforeModalOpen() {
      const { sampleId, inputFileName } = this.sampleData
      this.sampleId = sampleId
      this.fileName = inputFileName

      this.getInfo()
      this.reset()
    },

    // 点击图表，设置红线，改变表格的选中项
    handleChartClick(param) {
      const { offsetX, offsetY } = param
      const point = getXAxisAndYAxisByPosition(this.$refs.chartRef.getChartInstance(), offsetX, offsetY)
      if (point) {
        const xAxis = parseInt(point[0].toFixed())
        this.option.series[0].markLine.data[0].xAxis = xAxis

        this.currChannel = xAxis

        // 获取每一段 Channel 中的最大值
        const maxXAxises = this.getPeakMaxValues()

        let index = 0
        // 计算当前选中的xAxis跟哪条 peak的最大值 最近
        if (xAxis >= maxXAxises[maxXAxises.length - 1]) {
          index = maxXAxises.length - 1
        } else if (xAxis <= maxXAxises[0]) {
          index = 0
        } else {
          for (let i = 1; i < maxXAxises.length; i++) {
            const prev = maxXAxises[i - 1]
            const curr = maxXAxises[i]

            if (xAxis >= prev && xAxis <= curr) {
              index = xAxis - prev < curr - xAxis ? i - 1 : i
              break
            }
            continue
          }
        }

        const selectedRow = this.list[index]

        this.selectedKeys = [selectedRow.index]

        this.getSelPosNuclide(selectedRow)

        this.selectedTableItem = selectedRow

        // 如果点击了Fit按钮
        if (this.isFitting) {
          this.handleFittingChannel(xAxis)
        }
      }
    },

    // 处理Fit中的两个channel
    handleFittingChannel(channel) {
      // 第一个channel存在，又点了一次，则是第二次点击
      if (this.firstFittingChannel !== null) {
        // 查找两个channel之间的Peak峰

        let left = this.firstFittingChannel
        let right = channel
        if (left > right) {
          right = left
          left = channel
        }

        const peaksBetweenChannel = this.list.filter(peak => {
          const centroidId = peak.peakCentroid
          return centroidId >= left && centroidId <= right
        })

        if (!peaksBetweenChannel.length) {
          this.$message.warn(`There are 0 peak between channel ${left} and ${right}`)
          return
        }

        this.fitPeaksAndBaselineModalVisible = true

        this.isFitting = false
        this.firstFittingChannel = null
      } else {
        this.firstFittingChannel = channel
      }
    },

    // 切换图表上的红色竖线及表格选中
    handleChangeMarkLine(direction) {
      const markLineOption = this.option.series[0].markLine.data[0]
      const prevAxis = markLineOption.xAxis

      // 获取每一段 Channel 中的最大值
      const maxXAxises = this.getPeakMaxValues()
      if (direction == 'next') {
        // 找到第一个比prevAxis大的xAxis
        const find = maxXAxises.find(xAxis => xAxis > prevAxis)
        if (find) {
          markLineOption.xAxis = find
        }
      } else if (direction == 'prev') {
        // 找到第一个比prevAxis小的xAxis
        const find = cloneDeep(maxXAxises)
          .reverse()
          .find(xAxis => xAxis < prevAxis)
        if (find) {
          markLineOption.xAxis = find
        }
      }

      const xAxis = markLineOption.xAxis
      if (xAxis >= 0) {
        const index = maxXAxises.findIndex(item => item == xAxis)
        if (index !== -1) {
          this.selectedKeys = [this.list[index].index]
        }
      }
    },

    // 获取右下角possible nuclide 和 identified nuclide
    async getSelPosNuclide(row) {
      this.model.possibleNuclide = ''
      this.model.identifiedNuclide = ''

      if (!row._possible) {
        this.$set(row, '_loading', true)
        try {
          const { success, result, message } = await getAction('/gamma/getSelPosNuclide', {
            sampleId: this.sampleId,
            channel: parseInt(row.peakCentroid),
            fileName: this.fileName
          })
          if (success) {
            const { possible } = result
            this.$set(row, '_possible', possible)
          } else {
            this.$message.error(message)
          }
        } catch (error) {
          console.error(error)
        } finally {
          row._loading = false
        }
      }
    },

    // 获取每一段 Channel 中的最大值
    getPeakMaxValues() {
      const maxXAxises = this.channelPeakChart.map(item => {
        const allY = item.pointlist.map(point => point.y)
        const max = item.pointlist.find(point => point.y == Math.max(...allY))
        return max.x
      })
      return maxXAxises
    },

    // 显示peak comment弹窗
    handleAddPeakComment() {
      if (!this.selectedKeys.length) {
        this.$message.warn('Please Select a Peak that You Want to Add Comment!')
        return
      }
      this.peakCommentModalVisible = true
    },

    // 显示general comment弹窗
    handleAddGeneralComment() {
      this.generalCommentModalVisible = true
    },

    // Insert按钮
    handleInsert() {
      this.isFitting = false

      const xAxises = this.channelBaseCPChart.map(({ point: { x } }) => x)
      const min = xAxises[0]
      const max = xAxises[xAxises.length - 1]
      if (!this.currChannel || this.currChannel < min || this.currChannel > max) {
        this.$message.warn("Couldn't insert peak, maybe out of range")
        return
      }

      this.fitPeaksAndBaselineModalVisible = true
    },

    // 点击 Fit Peak XXX 弹窗中的 Peaks 按钮
    handleInsertSuccess(result) {
      const {
        allData,
        barChart,
        channelBaseLineChart,
        channelPeakChart,
        shadowChannelChart,
        shadowEnergyChart,
        shapeChannelData,
        shapeEnergyData,
        table
      } = result

      this.$bus.$emit('refresh', {
        allData,
        channelPeakChart,
        shadowChannelChart,
        shadowEnergyChart,
        shapeChannelData,
        shapeEnergyData
      })

      this.channelPeakChart = channelPeakChart
      this.channelBaseLineChart = channelBaseLineChart

      const series = []

      // 推入BaseLine
      series.push(
        buildLineSeries(
          'BaseLine',
          channelBaseLineChart.pointlist.map(({ x, y }) => [x, y]),
          channelBaseLineChart.color,
          {
            markLine: {
              silent: true,
              symbol: 'none',
              label: {
                show: false
              },
              lineStyle: {
                color: 'red',
                width: 1
              },
              data: [{ xAxis: -1 }]
            },
            zlevel: 10
          }
        )
      )

      // 推入旧的Count
      series.push(
        buildLineSeries(
          'CountChart',
          this.channelCountChart.pointlist.map(({ x, y }) => [x, y]),
          this.channelCountChart.color
        )
      )

      // 推入Peak
      const peakSeries = []
      channelPeakChart.forEach((item, index) => {
        peakSeries.push(
          buildLineSeries(
            'Peak_' + (index + 1),
            item.pointlist.map(({ x, y }) => [x, y]),
            item.color
          )
        )
      })

      series.push(...peakSeries)

      // 推入旧的基线控制点
      series.push({
        name: 'BaseLine_Ctrl_Point',
        type: 'scatter',
        data: this.channelBaseCPChart.map(({ size, color, point: { x, y } }) => {
          return {
            value: [x, y],
            itemStyle: {
              color: 'transparent',
              borderColor: color,
              borderWidth: size / 2
            }
          }
        }),
        silent: true,
        animation: false,
        zlevel: 20
      })

      this.thumbnailOption.series = buildLineSeries(
        'BarChart',
        barChart.map(({ x, y }) => [x, y]),
        '#fff',
        {
          silent: true
        }
      )

      this.list = table
      this.option.series = series
    },

    // 点击 Fit Peak XXX 弹窗中的 Cancel 按钮
    handleCancelSuccess(result) {
      const { channelPeakChart, table } = result
      this.channelPeakChart = channelPeakChart
      const series = []

      // 推入旧的BaseLine
      series.push(
        buildLineSeries(
          'BaseLine',
          this.channelBaseLineChart.pointlist.map(({ x, y }) => [x, y]),
          this.channelBaseLineChart.color,
          {
            markLine: {
              silent: true,
              symbol: 'none',
              label: {
                show: false
              },
              lineStyle: {
                color: 'red',
                width: 1
              },
              data: [{ xAxis: -1 }]
            },
            zlevel: 10
          }
        )
      )

      // 推入旧的Count
      series.push(
        buildLineSeries(
          'CountChart',
          this.channelCountChart.pointlist.map(({ x, y }) => [x, y]),
          this.channelCountChart.color
        )
      )

      // 推入Peak
      const peakSeries = []
      channelPeakChart.forEach((item, index) => {
        peakSeries.push(
          buildLineSeries(
            'Peak_' + (index + 1),
            item.pointlist.map(({ x, y }) => [x, y]),
            item.color
          )
        )
      })

      series.push(...peakSeries)

      // 推入旧的基线控制点
      series.push({
        name: 'BaseLine_Ctrl_Point',
        type: 'scatter',
        data: this.channelBaseCPChart.map(({ size, color, point: { x, y } }) => {
          return {
            value: [x, y],
            itemStyle: {
              color: 'transparent',
              borderColor: color,
              borderWidth: size / 2
            }
          }
        }),
        silent: true,
        animation: false,
        zlevel: 20
      })

      this.list = table
      this.option.series = series
    },

    // 删除
    handleDel() {
      this.isFitting = false

      if (!this.selectedKeys.length) {
        this.$message.warn('No peak to delete.')
        return
      }

      this.$confirm({
        title: 'Warning',
        content: 'Are you sure to delete this peak?',
        cancelButtonProps: {
          props: {
            type: 'warn'
          }
        },
        onOk: async () => {
          // this.list.splice(findIndex, 1)
          // this.selectedKeys = []

          // const seriesIndex = this.option.series.findIndex(item => {
          //   return item.name == 'Peak_' + willDelKey
          // })

          // this.opts.notMerge = true
          // this.option.series.splice(seriesIndex, 1)
          // this.channelPeakChart.splice(findIndex, 1)

          // this.$nextTick(() => {
          //   this.resetChartOpts()
          // })
          try {
            const { inputFileName: fileName } = this.sampleData
            const { success, result, message } = await getAction('/gamma/deletePeak', {
              fileName,
              curRow: this.curRow
            })
            if (success) {
              const {
                allData,
                channelPeakChart,
                shadowChannelChart,
                shadowEnergyChart,
                shapeChannelData,
                shapeEnergyData,
                table
              } = result

              this.$bus.$emit('refresh', {
                allData,
                channelPeakChart,
                shadowChannelChart,
                shadowEnergyChart,
                shapeChannelData,
                shapeEnergyData
              })

              this.channelPeakChart = channelPeakChart
              const series = []
              // 推入旧的BaseLine
              series.push(
                buildLineSeries(
                  'BaseLine',
                  this.channelBaseLineChart.pointlist.map(({ x, y }) => [x, y]),
                  this.channelBaseLineChart.color,
                  {
                    markLine: {
                      silent: true,
                      symbol: 'none',
                      label: {
                        show: false
                      },
                      lineStyle: {
                        color: 'red',
                        width: 1
                      },
                      data: [{ xAxis: -1 }]
                    },
                    zlevel: 10
                  }
                )
              )

              // 推入旧的Count
              series.push(
                buildLineSeries(
                  'CountChart',
                  this.channelCountChart.pointlist.map(({ x, y }) => [x, y]),
                  this.channelCountChart.color
                )
              )

              // 推入Peak
              const peakSeries = []
              channelPeakChart.forEach((item, index) => {
                peakSeries.push(
                  buildLineSeries(
                    'Peak_' + (index + 1),
                    item.pointlist.map(({ x, y }) => [x, y]),
                    item.color
                  )
                )
              })

              series.push(...peakSeries)

              // 推入旧的基线控制点
              series.push({
                name: 'BaseLine_Ctrl_Point',
                type: 'scatter',
                data: this.buildCPPointData(this.channelBaseCPChart),
                silent: true,
                animation: false,
                zlevel: 20
              })

              this.list = table
              this.option.series = series

              this.selectedKeys = []
            } else {
              this.$message.error(message)
            }
          } catch (error) {
            console.error(error)
          }
        }
      })
    },

    // 重置图表配置
    resetChartOpts() {
      this.opts.notMerge = false
      this.option.brush = { toolbox: [] }
    },

    // 匹配
    handleFit() {
      if (!this.channelPeakChart || !this.channelPeakChart.length) {
        this.$message.warn('No peak to fit.')
        return
      }

      this.isFitting = true
      this.firstFittingChannel = null
    },

    // 表格的行点击
    handleTableRowClick(row) {
      if (this.selectedTableItem == row) {
        return
      }

      const channel = row.peakCentroid
      this.currChannel = channel

      this.option.series[0].markLine.data[0].xAxis = channel

      this.getSelPosNuclide(row)

      this.selectedTableItem = row
    },

    // 鼠标按下时开启可刷选状态
    handleMouseDown() {
      if (this.isModifying) {
        return
      }

      const chart = this.$refs.chartRef.getChartInstance()
      chart.dispatchAction({
        type: 'takeGlobalCursor',
        // 如果想变为“可刷选状态”，必须设置。不设置则会关闭“可刷选状态”。
        key: 'brush',
        brushOption: {
          // 参见 brush 组件的 brushType。如果设置为 false 则关闭“可刷选状态”。
          brushType: 'rect'
        }
      })
    },

    handleMouseUp() {
      setTimeout(() => {
        const chart = this.$refs.chartRef.getChartInstance()
        this.clearBrush(chart)
      }, 0)
    },

    clearBrush(chart) {
      // 清理刷选的范围
      chart.dispatchAction({
        type: 'brush',
        areas: []
      })

      // 改为不可刷选状态
      chart.dispatchAction({
        type: 'takeGlobalCursor'
      })
    },

    // 刷选完毕时
    handleBrushEnd(param) {
      const chart = this.$refs.chartRef.getChartInstance()
      const areas = param.areas[0]
      if (areas) {
        const range = areas.range
        const [[minX, maxX], [minY, maxY]] = range
        const point1 = chart.convertFromPixel({ seriesIndex: 0 }, [minX, minY]).map(num => parseInt(num.toFixed()))
        const point2 = chart.convertFromPixel({ seriesIndex: 0 }, [maxX, maxY]).map(num => parseInt(num.toFixed()))
        const xAxisMax = chart.getModel().getComponent('xAxis').axis.scale._extent[1]
        const yAxisMax = this.option.yAxis.max
        let [x1, y2, x2, y1] = [...point1, ...point2] // 根据解析出的数据确定真实的范围
        const xAxisLimit = rangeNumber(1, xAxisMax)
        const yAxisLimit = rangeNumber(0.1, yAxisMax)
        x1 = xAxisLimit(x1)
        x2 = xAxisLimit(x2)
        y1 = yAxisLimit(y1)
        y2 = yAxisLimit(y2)
        this.option.xAxis.min = x1
        this.option.xAxis.max = x2
        this.option.yAxis.min = y1
        this.option.yAxis.max = y2

        this.thumbnailOption.xAxis.min = x1
        this.thumbnailOption.xAxis.max = x2
        this.thumbnailOption.yAxis.min = y1
        this.thumbnailOption.yAxis.max = y2

        if (this.btnGroupType == 2) {
          this.buildGraphicRectByData()
          this.buildGraphicPlusByData()
        }
      }
      this.clearBrush(chart)
    },

    handleResetChart() {
      this.option.xAxis.min = 1
      this.option.xAxis.max = 'dataMax'
      this.option.yAxis.min = 0.1
      this.option.yAxis.max = 'dataMax'

      this.thumbnailOption.xAxis.min = 1
      this.thumbnailOption.xAxis.max = 'dataMax'
      this.thumbnailOption.yAxis.min = 0.1
      this.thumbnailOption.yAxis.max = 'dataMax'

      if (this.btnGroupType == 2) {
        this.buildGraphicRectByData()
        this.buildGraphicPlusByData()
      }
    },

    // 切换操作
    handleSwitchOperation() {
      // 切换到Base Line 和 Control Point 操作
      if (this.btnGroupType == 1) {
        this.btnGroupType = 2
        this._channelBaseCPChart = cloneDeep(this.channelBaseCPChart)
        this._channelBaseLineChart = cloneDeep(this.channelBaseLineChart)
        this._baseCtrls = cloneDeep(this.BaseCtrls)

        // 供编辑的白色基线
        const baseLineEditSeries = buildLineSeries(
          'BaseLine_Edit',
          this._channelBaseLineChart.pointlist.map(({ x, y }) => [x, y]),
          this.FitBaseLine,
          {
            zlevel: 21
          }
        )
        this.option.series.push(baseLineEditSeries)

        this.buildGraphicRectByData()
        this.buildGraphicPlusByData()
      }
      // 切换回 Peak 操作
      else {
        this.btnGroupType = 1
        this.opts.notMerge = true
        this.option.series.splice(this.option.series.length - 1, 1) // 去掉白色的基线副本
        this.removeGraphics()

        const baseLineSeries = findSeriesByName(this.option.series, 'BaseLine')
        baseLineSeries.data = this.channelBaseLineChart.pointlist.map(({ x, y }) => [x, y]) // 恢复基线

        const baseLineCP = findSeriesByName(this.option.series, 'BaseLine_Ctrl_Point')
        baseLineCP.data = this.buildCPPointData(this.channelBaseCPChart)

        this.$nextTick(() => {
          this.resetChartOpts()
        })
      }

      this.isModifying = false
      this.isFitting = false
      this.clearOperationStack()
    },

    // 根据数据绘制小方块
    buildGraphicRectByData() {
      this.option.graphic[0].children.forEach(item => {
        item.$action = 'remove'
      })

      this.$nextTick(() => {
        this.option.graphic[0].children = this._channelBaseCPChart.map(({ point: { x, y } }) => {
          return this.buildGraphicRect(x, y)
        })
      })
    },

    // 绘制单个红色小方块
    buildGraphicRect(xAxis, yAxis) {
      const chart = this.$refs.chartRef.getChartInstance()
      const [xPix, yPix] = chart.convertToPixel('grid', [xAxis, yAxis])
      const that = this

      return {
        type: 'rect',
        id: Math.random().toString(),
        $action: 'replace',
        position: [xPix, yPix],
        shape: {
          x: -4,
          y: -4,
          width: 8,
          height: 8
        },
        style: {
          stroke: 'red',
          fill: 'transparent',
          lineWidth: 2
        },
        draggable: this.isModifying,
        ondrag: function() {
          const [xPixel] = chart.convertToPixel('grid', [xAxis, yAxis])
          // 保持x轴不变
          this.position[0] = xPixel
          that.redrawBaseLine()
        },
        ondragend: ({ offsetY }) => {
          this.setGraphicDraggable(false)

          const _channelBaseCPChart = this._channelBaseCPChart
          const rectIndex = _channelBaseCPChart.findIndex(item => item.point.x == xAxis)

          // 拖动到的位置
          const [, rectYAxis] = getXAxisAndYAxisByPosition(chart, xPix, offsetY)

          const willModifyPoint = this._channelBaseCPChart[rectIndex].point
          this.pushOperationStack(Operators.MODIFY, {
            index: rectIndex,
            xAxis: willModifyPoint.x,
            yAxis: willModifyPoint.y
          })

          // 改变控制点位置
          willModifyPoint.y = rectYAxis
          this.buildGraphicRectByData()

          // 改变基线位置
          const baseLineEditSeries = findSeriesByName(this.option.series, 'BaseLine_Edit')
          baseLineEditSeries.data[xAxis - 1][1] = rectYAxis

          this._channelBaseLineChart.pointlist[xAxis - 1].y = rectYAxis
        },
        zlevel: 100
      }
    },

    // 重新生成基线
    redrawBaseLine() {
      console.log('%c [ 重新生成基线 ]-1355', 'font-size:13px; background:pink; color:#bf2c9f;')
    },

    // 根据数据绘制 + 号
    buildGraphicPlusByData() {
      this.option.graphic[1].children.forEach(item => {
        item.$action = 'remove'
      })

      this.$nextTick(() => {
        const { xctrl: xctrlList, yctrl: yctrlList, yslope: yslopeList } = this._baseCtrls
        const plusGraphic = []

        xctrlList.forEach((xctrl, index) => {
          const yctrl = yctrlList[index]
          const yslope = yslopeList[index]
          if (yslope != 0) {
            plusGraphic.push(this.buildGraphicPlus(xctrl + 2, yctrl + 2 * yslope))
            plusGraphic.push(this.buildGraphicPlus(xctrl - 2, yctrl - 2 * yslope))
          }
        })
        this.option.graphic[1].children = plusGraphic
      })
    },

    // 绘制单个绿色加号
    buildGraphicPlus(x, y) {
      const chart = this.$refs.chartRef.getChartInstance()
      const [xPix, yPix] = chart.convertToPixel('grid', [x, y])
      return {
        type: 'text',
        id: Math.random().toString(),
        $action: 'replace',
        position: [xPix, yPix],
        style: {
          x: -10,
          y: -10,
          text: '+',
          fill: '#0f0',
          font: 'bolder 20px "Microsoft YaHei", sans-serif'
        },
        zlevel: 101
      }
    },

    /**
     * 设置小方块可拖拽
     */
    setGraphicDraggable(draggable) {
      this.option.graphic[0].children.forEach(item => {
        item.draggable = draggable
      })
      this.isModifying = draggable
    },

    // 在当前选中的红线位置新增控制点
    handleAddCP() {
      this.setGraphicDraggable(false)
      const { rg_high, rg_low } = this.BaseCtrls
      if (!this.currChannel || this.currChannel < rg_low || this.currChannel > rg_high) {
        this.$message.warn("Can't insert Control Point out of range")
        return
      }

      const chart = this.$refs.chartRef.getChartInstance()

      const controlPointList = this.option.graphic[0].children

      let i = 0 // 记录新控制点在列表中的位置
      const [xPix] = chart.convertToPixel('grid', [this.currChannel, 0])
      for (; i < controlPointList.length; ++i) {
        const currCP = controlPointList[i].position[0]
        if (currCP >= xPix) {
          if (currCP == xPix) {
            this.$message.warn(`The new control point in channel ${this.currChannel} exists, can't introduce twice`)
            return
          }
          break
        }
      }

      // 新增的控制点跟基线的值平齐
      const yAxis = this._channelBaseLineChart.pointlist[this.currChannel - 1].y

      this._channelBaseCPChart.splice(i, 0, { point: { x: this.currChannel, y: yAxis } })
      this.buildGraphicRectByData()

      this.pushOperationStack(Operators.ADD, { index: i })
    },

    // 移除控制点
    handleRemoveCP() {
      this.setGraphicDraggable(false)

      // find nearest control-point
      const controlPointList = this._channelBaseCPChart
      let i = 1
      for (; i < controlPointList.length; ++i) {
        const { x: currXAxis } = controlPointList[i].point
        if (currXAxis >= this.currChannel) {
          const { x: prevX } = controlPointList[i - 1].point
          if (currXAxis - this.currChannel > this.currChannel - prevX) --i
          break
        }
      }

      if (i == 0 || i >= controlPointList.length - 1) {
        this.$message.warn("Can't remove first/last control point")
        return
      }

      const [point] = controlPointList.splice(i, 1)
      this.buildGraphicRectByData()
      this.pushOperationStack(Operators.REMOVE, {
        index: i,
        point: point.point
      })
    },

    // 修改控制点
    handleModifyCP() {
      this.setGraphicDraggable(!this.isModifying)
    },

    // 编辑斜率
    handleEditSlope() {
      this.setGraphicDraggable(false)

      const { xctrl, yslope } = this._baseCtrls
      if (!xctrl.length) {
        this.$message.warn('No control points to be edited')
        return
      }
      // find nearest control-point
      let i = 1,
        n = xctrl.length
      for (; i < n; ++i) {
        const currXCtrl = xctrl[i]
        const prevXCtrl = xctrl[i - 1]
        if (currXCtrl >= this.currChannel) {
          if (currXCtrl - this.currChannel > this.currChannel - prevXCtrl) {
            --i
          }
          break
        }
      }
      if (i == n) i = n - 1

      this.$refs.editSlopeModal.open({
        index: i,
        value: yslope[i],
        allowNaN: !(i == 0 || i == n - 1)
      })
    },

    // 确认编辑斜率
    handleSlopeChange(slope, index, prevSlope) {
      if (slope === prevSlope) {
        return
      }

      const { yslope } = this._baseCtrls
      yslope[index] = slope
      this.pushOperationStack(Operators.SLOPE_CHANGE, {
        index,
        slope: prevSlope
      })

      this.buildGraphicPlusByData()
    },

    // 撤销
    handleUndo() {
      this.setGraphicDraggable(false)
      this.popOperationStack()
    },

    // 将原先的基线和控制点移动到新位置
    handleReplot() {
      const baseLineSeries = findSeriesByName(this.option.series, 'BaseLine')
      baseLineSeries.data = this._channelBaseLineChart.pointlist.map(({ x, y }) => [x, y])

      const baseLineCP = findSeriesByName(this.option.series, 'BaseLine_Ctrl_Point')
      baseLineCP.data = this.buildCPPointData(this._channelBaseCPChart)
    },

    /**
     * 构建基线控制点数据
     * @param {Array} controlPointList
     **/
    buildCPPointData(controlPointList) {
      return controlPointList.map(({ size, color, point: { x, y } }) => {
        return {
          value: [x, y],
          itemStyle: {
            color: 'transparent',
            borderColor: color,
            borderWidth: size / 2
          }
        }
      })
    },

    // 确定对Baseline Control Points 的操作
    handleAccept() {
      this.channelBaseLineChart = this._channelBaseLineChart
      this.channelBaseCPChart = this._channelBaseCPChart

      this.handleSwitchOperation()

      this.$bus.$emit('accept')
    },

    // 右下角添加当前选中的nuclide
    async handleAddNuclide() {
      const nuclides = this.selectedTableItem.nuclides
      const possibleNuclide = this.model.possibleNuclide
      if (!nuclides.includes(possibleNuclide)) {
        if (this.selectedTableItem._adding) {
          return
        }
        try {
          this.$set(this.selectedTableItem, '_adding', true)
          const { inputFileName: fileName } = this.sampleData
          const { success, message } = await postAction('/gamma/addNuclide', {
            curRow: this.curRow,
            nuclideName: possibleNuclide,
            fileName,
            list_identify: nuclides
          })
          if (success) {
            nuclides.push(possibleNuclide)
          } else {
            this.$message.error(message)
          }
        } catch (error) {
          console.error(error)
        } finally {
          this.selectedTableItem._adding = false
        }
      }
    },
    // 右下角删除当前选中的nuclide
    async handleDelNuclide() {
      if (!this.model.identifiedNuclide) {
        return
      }
      const nuclides = this.selectedTableItem.nuclides
      if (this.selectedTableItem._deleting) {
        return
      }
      const findIndex = nuclides.findIndex(nuclide => nuclide == this.model.identifiedNuclide)
      if (-1 !== findIndex) {
        try {
          this.$set(this.selectedTableItem, '_deleting', true)
          const { inputFileName: fileName } = this.sampleData
          const { success, message } = await postAction('/gamma/deleteNuclide', {
            curRow: this.curRow,
            nuclideName: this.model.identifiedNuclide,
            fileName,
            list_identify: nuclides
          })
          if (success) {
            nuclides.splice(findIndex, 1)
          } else {
            this.$message.error(message)
          }
        } catch (error) {
          console.error(error)
        } finally {
          this.selectedTableItem._deleting = false
        }
      }
    },

    /**
     * 推入操作
     * @param {*} operator 操作符
     * @param {*} operand 操作数
     */
    pushOperationStack(operator, operand) {
      this.operationStack.push({
        operator,
        operand
      })
    },

    /**
     * 弹出操作
     */
    popOperationStack() {
      const { operator, operand } = this.operationStack.pop()
      const { index } = operand
      switch (operator) {
        case Operators.ADD:
          this._channelBaseCPChart.splice(index, 1)
          this.buildGraphicRectByData()
          break
        case Operators.MODIFY:
          const { xAxis, yAxis } = operand

          // 恢复点的y轴位置
          this._channelBaseCPChart[index].point.y = yAxis
          this.buildGraphicRectByData()

          // 恢复基线位置
          const baseLineEditSeries = findSeriesByName(this.option.series, 'BaseLine_Edit')
          baseLineEditSeries.data[xAxis - 1] = [xAxis, yAxis]

          this._channelBaseLineChart.pointlist[xAxis - 1].y = yAxis
          break
        case Operators.REMOVE:
          const {
            point: { x, y }
          } = operand
          this._channelBaseCPChart.splice(index, 0, {
            point: {
              x,
              y
            }
          })
          this.buildGraphicRectByData()
          break
        case Operators.SLOPE_CHANGE:
          const { slope } = operand
          const { yslope } = this._baseCtrls
          yslope[index] = slope
          break
      }
    },

    /**
     * 清理操作栈
     */
    clearOperationStack() {
      this.operationStack = []
    }
  },
  computed: {
    curRow() {
      const [selectedKey] = this.selectedKeys
      const findIndex = this.list.findIndex(item => item.index == selectedKey)
      return findIndex
    },

    isOperationStackEmpty() {
      return this.operationStack.length == 0
    }
  }
}
</script>

<style lang="less" scoped>
.interactive-analysis-tools {
  display: flex;
  margin-top: 5px;

  &-left {
    width: 75%;
    margin-right: 20px;

    .chart {
      height: 331px;
    }

    .thumbnail {
      height: 50px;
      margin: 10px 10px 35px 40px;
      background-color: #255369;
    }

    .table {
      .title {
        color: #0cebc9;
        font-size: 20px;
        text-align: center;
        margin-bottom: 10px;
        user-select: none;

        span {
          cursor: pointer;
          margin: 0 5px;
        }
      }

      .custom-table {
        &.has-data {
          ::v-deep {
            .ant-table-body {
              height: 288px;
              background-color: #06282a;
            }
          }
        }

        ::v-deep {
          .ant-table-placeholder {
            height: 289px;
            display: flex;
            justify-content: center;
            align-items: center;
          }
        }
      }

      .operators {
        display: flex;
        margin-top: 10px;
        gap: 10px;

        .ant-btn {
          flex: 1;
        }
      }
    }
  }

  &-right {
    .peak-box {
      height: 326px;

      &-item:not(:last-child) {
        margin-bottom: 10px;
      }

      .symbol {
        display: flex;

        .ant-btn {
          flex: 1;

          &:first-child {
            margin-right: 10px;
          }
        }
      }
    }

    .base-line {
      margin-top: 136px;
    }

    .reset-btn-box {
      margin-top: 20px;
      margin-bottom: 30px;
    }

    .identify-box {
      .tolerance {
        ::v-deep {
          .ant-form-item {
            margin-bottom: 10px;

            &-control-wrapper {
              flex: 1;
            }

            &-control {
              width: 100%;
            }
          }
        }

        .ant-input-number {
          width: 100%;
        }
      }

      .identify-item {
        .title {
          background-color: #497e9d;
          height: 30px;
          line-height: 30px;
          text-align: center;
          font-size: 16px;
        }

        .content {
          height: 80px;
          background-color: #275466;
          margin: 10px 0;
          padding: 5px;
          overflow: auto;

          .item {
            cursor: pointer;
            line-height: 26px;
            padding: 0 5px;

            &.active {
              background: #296d81;
            }
          }
        }
      }

      .identify-operators {
        display: flex;
        gap: 10px;
        .text {
          flex: 1;
          line-height: 32px;
          background-color: #285366;
          padding: 0 10px;
        }

        .ant-btn {
          width: 50px;
          padding-left: 5px;
          padding-right: 5px;
        }
      }
    }
  }

  .ant-btn {
    width: 100%;
  }
}

.is-modify,
.is-fitting {
  color: #f00;
}
</style>