taais-web/src/views/demo/toInfrared/index.vue

<!-- 
  @Date: 2024-10-29
  @Author: ©WANGKANG
  @Email: 1686617586@qq.com
-->
<template>
  <div class="table-box">
    <ProTable ref="proTable" :columns="columns" row-key="id" :request-api="listToInfraredApi">
      <!-- 表格 header 按钮 -->
      <template #tableHeader="scope">
        <el-button type="primary" v-auth="['demo:toInfrared:add']" icon="CirclePlus" @click="openDialog(1, '可见光转红外新增')"> 新增 </el-button>
        <!-- <el-button type="primary" v-auth="['demo:toInfrared:import']" icon="Upload" plain @click="batchAdd"> 导入 </el-button>
        <el-button type="primary" v-auth="['demo:toInfrared:export']" icon="Download" plain @click="downloadFile"> 导出 </el-button> -->
        <el-button
          type="danger"
          v-auth="['demo:toInfrared:remove']"
          icon="Delete"
          plain
          :disabled="!scope.isSelected"
          @click="batchDelete(scope.selectedListIds)"
        >
          批量删除
        </el-button>
      </template>
      <!-- 表格操作 -->
      <template #operation="scope">
        <el-button type="primary" link icon="View" v-if="scope.row.algorithmModelId != null" @click="openModelDialog(scope.row)">
          <!--@click="openStartDialog(scope.row)"  -->
          详情
        </el-button>
        <el-button
          type="primary"
          link
          icon="View"
          v-auth="['demo:ToInfrared:start']"
          v-if="scope.row.status == '0' || scope.row.status == '3' || scope.row.status == '4'"
          @click="startToInfrared(scope.row)"
        >
          <!--@click="openStartDialog(scope.row)"  -->
          开始
        </el-button>
        <el-popconfirm title="确定终止此任务吗?" @confirm="stopToInfrared(scope.row)" v-if="scope.row.status == '1'">
          <template #reference>
            <el-button type="primary" link icon="Delete"> 终止</el-button>
          </template>
        </el-popconfirm>
        <el-button
          type="primary"
          link
          icon="View"
          v-auth="['demo:DataSeq:query']"
          v-if="scope.row.status == '2' && scope.row.type == AlgorithmType2['预测/推理']"
          @click="preview(scope.row)"
        >
          预览
        </el-button>
        <el-button
          type="primary"
          link
          icon="View"
          v-auth="['demo:DataSeq:query']"
          v-if="scope.row.status == '2' && scope.row.type == AlgorithmType2['预测/推理']"
          @click="showStatisticResultRef.get_statistics_result(scope.row.id)"
        >
          结果
        </el-button>
        <el-button
          type="primary"
          link
          icon="View"
          v-auth="['demo:ToInfrared:download']"
          v-if="scope.row.status == '2'"
          @click="dowloadToInfrared(scope.row)"
        >
          下载
        </el-button>
        <el-button
          type="primary"
          link
          icon="View"
          v-auth="['demo:ToInfrared:download']"
          v-if="scope.row.status == '2' && scope.row.type == AlgorithmType2['训练']"
          @click="showToInfraredModel(scope.row.id)"
        >
          模型
        </el-button>
        <el-button
          type="primary"
          link
          icon="View"
          v-auth="['demo:toInfrared:query']"
          @click="viewLogRef.handleOpen(scope.row.id)"
          v-if="scope.row.status != '0'"
        >
          日志
        </el-button>
        <!-- <el-button type="primary" link icon="View" v-auth="['demo:toInfrared:query']" @click="openDialog(3, '查看', scope.row)"> 查看 </el-button> -->
        <!-- <el-button type="primary" link icon="EditPen"v-auth="['demo:toInfrared:edit']" @click="openDialog(2, '编辑', scope.row)"> 编辑 </el-button> -->
        <el-button
          type="primary"
          link
          icon="Delete"
          v-auth="['demo:toInfrared:remove']"
          @click="deleteToInfrared(scope.row)"
          :disabled="scope.row.status == '1'"
        >
          删除
        </el-button>
      </template>
    </ProTable>
    <FormDialog ref="formDialogRef" />
    <ImportExcel ref="dialogRef" />
    <PreviewCompareImages ref="previewImagesRef" />
    <ViewLog ref="viewLogRef" :get-log-api="getLogToInfraredApi" />
    <ShowStatisticResult ref="showStatisticResultRef" :api="getStatisticsResultToInfraredApi" />
    <el-dialog v-model="showModelDialogVisible" title="模型列表" width="900">
      <el-scrollbar ref="scrollbarRef" id="scrollbarRef1" height="500px">
        <template v-for="model in model_list" :key="model">
          <el-card style="width: 100%; margin-bottom: 10px">
            <el-form label-width="130px" label-position="left">
              <el-form-item label="Model Name">
                {{ model.name }}
              </el-form-item>
              <el-form-item label="Model Path">
                {{ model.path }}
              </el-form-item>
              <el-form-item label="Model Size">
                {{ model.size }}
              </el-form-item>
              <el-form-item label="操作">
                <el-link :href="model.url" type="primary" icon="Download" :underline="false" target="_blank" style="margin-right: 20px"
                  >下载
                </el-link>
                <el-button type="success" link @click="addToInfraredModel(model.path)">
                  <el-icon>
                    <Plus />
                  </el-icon>
                  添加模型
                </el-button>
              </el-form-item>
            </el-form>
          </el-card>
        </template>
      </el-scrollbar>
    </el-dialog>
  </div>
</template>

<script setup lang="tsx" name="ToInfrared">
import { ref, reactive, onMounted, Ref, nextTick } from 'vue'
import { useHandleData } from '@/hooks/useHandleData'
import { useDownload } from '@/hooks/useDownload'
import { ElMessage, ElMessageBox } from 'element-plus'
import ProTable from '@/components/ProTable/index.vue'
import ImportExcel from '@/components/ImportExcel/index.vue'
import FormDialog from '@/components/FormDialog/index.vue'
import { ProTableInstance, ColumnProps } from '@/components/ProTable/interface'
import {
  listToInfraredApi,
  delToInfraredApi,
  addToInfraredApi,
  updateToInfraredApi,
  importTemplateApi,
  importToInfraredDataApi,
  exportToInfraredApi,
  startToInfraredApi,
  stopToInfraredApi,
  dowloadToInfraredApi,
  getToInfraredApi,
  getLogToInfraredApi,
  showToInfraredModelApi,
  previewPredictResultToInfraredModelApi,
  getStatisticsResultToInfraredApi
} from '@/api/modules/demo/toInfrared'

import { getImagesApi, listDataSeqApi } from '@/api/modules/demo/DataSeq'

import { enumAlgorithmModelTrackApi } from '@/api/modules/demo/AlgorithmModelTrack'
import { getAlgorithmModelTrackApi } from '@/api/modules/demo/AlgorithmModelTrack'
import { enumAlgorithmConfigTrackApi } from '@/api/modules/demo/AlgorithmConfigTrack'
import statusEnums from '@/utils/status'
import { AlgorithmType, SubSystem, SubSystem__, enumsAlgorithmType, enumsSubSystem, AlgorithmType2 } from '@/views/demo/utils'
import PreviewCompareImages from '@/views/demo/components/PreviewCompareImages.vue'
import { addAlgorithmModelTrackApi } from '@/api/modules/demo/AlgorithmModelTrack'
import ViewLog from '@/views/demo/components/ViewLog.vue'
import useWebSocketStore from '@/stores/modules/websocket'
import { resetHeart } from '@/utils/websocket'

import ShowStatisticResult from '@/views/demo/components/ShowStatisticResult.vue'

const showStatisticResultRef = ref()

onMounted(() => {
  const websocketStore = useWebSocketStore()
  websocketStore.websocket.onmessage = (e: any) => {
    if (e.data.indexOf('heartbeat') > 0) {
      resetHeart()
    }
    if (e.data.indexOf('ping') > 0) {
      return
    }
    // console.log(e)
    proTable.value?.getTableList(true)
  }
})

const previewImagesRef = ref()
const dialogVisible = ref(false)
const imageUrls = ref([])
const preview = async row => {
  previewImagesRef.value?.handleOpen(previewPredictResultToInfraredModelApi, row.id)
}

const enumsAlgorithmConfigTrack = ref<any>([])
onMounted(async () => {
  const result = await enumAlgorithmConfigTrackApi()
  enumsAlgorithmConfigTrack.value = []
  const tmp_data: any = result['data']
  for (const item of tmp_data) {
    if (item.subsystem === SubSystem__['可见光转红外'] && item.type === AlgorithmType2['预测/推理']) {
      item['label'] = item['label'] + '-' + SubSystem[item['subsystem']] + '-' + AlgorithmType[item['type']]
      enumsAlgorithmConfigTrack.value.push(item)
    }
  }
})

const setItemsOptions222 = () => {
  itemsOptions = [
    {
      label: '算法',
      prop: 'algorithmId',
      rules: [{ required: true, message: '算法不能为空', trigger: 'blur' }],
      compOptions: {
        elTagName: 'select',
        placeholder: '请输入算法',
        enum: enumsAlgorithmConfigTrack
      }
    },
    {
      label: '子系统',
      prop: 'subSystem',
      rules: [{ required: true, message: '子系统不能为空', trigger: 'blur' }],
      compOptions: {
        disabled: true,
        elTagName: 'select',
        placeholder: '请输入子系统',
        enum: enumsSubSystem
      }
    },
    {
      label: '模型名称',
      prop: 'modelName',
      rules: [{ required: true, message: '模型名称不能为空', trigger: 'blur' }],
      compOptions: {
        placeholder: '请输入模型名称'
      }
    },
    {
      label: '模型路径',
      prop: 'modelPath',
      rules: [{ required: false, message: '模型文件不能为空', trigger: 'blur' }],
      compOptions: {
        placeholder: '请输入模型名称',
        disabled: true
      }
    },
    {
      label: '备注',
      prop: 'remarks',
      rules: [{ required: false, message: '备注不能为空', trigger: 'blur' }],
      compOptions: {
        placeholder: '请输入备注'
      }
    }
  ]
}

const addToInfraredModel = async (modelPath: string) => {
  let res = { data: { modelPath, subSystem: SubSystem__['可见光转红外'], type: AlgorithmType2['预测/推理'] } }
  // 重置表单
  setItemsOptions222()
  const params = {
    title: '添加模型',
    width: 580,
    isEdit: true,
    itemsOptions: itemsOptions,
    model: res.data,
    api: addAlgorithmModelTrackApi,
    getTableList: proTable.value?.getTableList,
    closedEvent: async () => {
      await updateWnumsAlgorithmModelTrack()
    }
  }
  formDialogRef.value?.openDialog(params)
}

const showModelDialogVisible = ref(false)
const model_list: Ref<any> = ref([
  // {
  //   id: 1,
  //   name: '1',
  //   path: '/home/11',
  //   size: '123123MB',
  //   url: 'http://localhost:9090/profile/upload/2024/10/27/1_1729404909511_20241027153840A001.zip'
  // },
  // {
  //   id: 2,
  //   name: '1',
  //   path: '/home/11',
  //   size: '123123MB',
  //   url: 'http://localhost:9090/profile/upload/2024/10/27/1_1729404909511_20241027153840A001.zip'
  // },
  // {
  //   id: 3,
  //   name: '1',
  //   path: '/home/11',
  //   size: '123123MB',
  //   url: 'http://localhost:9090/profile/upload/2024/10/27/1_1729404909511_20241027153840A001.zip'
  // },
  // {
  //   id: 4,
  //   name: '1',
  //   path: '/home/11',
  //   size: '123123MB',
  //   url: 'http://localhost:9090/profile/upload/2024/10/27/1_1729404909511_20241027153840A001.zip'
  // }
])

const showToInfraredModel = async (id: any) => {
  const res: any = await showToInfraredModelApi(id)
  model_list.value = res.data
  showModelDialogVisible.value = true
}

const viewLogRef = ref()

const openModelDialog = async row => {
  const algorithmModelId = row.algorithmModelId
  const result: any = await getAlgorithmModelTrackApi(algorithmModelId)

  // console.log(result.data)
  setItemsOptionsModel()
  const params = {
    title: '模型',
    width: 580,
    isEdit: false,
    itemsOptions: itemsOptions,
    model: result.data,
    api: updateToInfraredApi,
    getTableList: proTable.value?.getTableList
  }
  formDialogRef.value?.openDialog(params)
}

const startToInfrared = async (params: any) => {
  const res: any = await startToInfraredApi(params.id)
  if (res.code === 200) {
    ElMessage.success('任务已开始，请等待完成！')
  } else {
    ElMessage.error('任务开始失败，请检查！')
  }
  proTable.value?.getTableList()
}

const stopToInfrared = async (params: any) => {
  const res: any = await stopToInfraredApi(params.id)
  if (res.code === 200) {
    ElMessage.success('终止任务成功！')
  } else {
    ElMessage.error('终止任务失败，请检查！')
  }
  proTable.value?.getTableList()
}
const dowloadToInfrared = async (params: any) => {
  if (params.type == AlgorithmType2['训练']) {
    await useDownload(dowloadToInfraredApi, params.name, params.id, true, '.pt')
  } else if (params.type == AlgorithmType2['预测/推理']) {
    await useDownload(dowloadToInfraredApi, params.name, params.id, true, '.zip')
  }
}

const openStartDialog = async (row: any) => {
  let res = { data: {} }
  if (row?.id) {
    res = await getToInfraredApi(row?.id || null)
  }
  // 重置表单
  setItemsOptions2()
  const params = {
    title: '开始任务',
    width: 580,
    isEdit: true,
    itemsOptions: itemsOptions,
    model: res.data,
    api: startToInfraredApi,
    getTableList: proTable.value?.getTableList
  }
  formDialogRef.value?.openDialog(params)
}

// ProTable 实例
const proTable = ref<ProTableInstance>()

// 删除可见光转红外信息
const deleteToInfrared = async (params: any) => {
  await useHandleData(delToInfraredApi, params.id, '删除【' + params.id + '】可见光转红外')
  proTable.value?.getTableList()
}

// 批量删除可见光转红外信息
const batchDelete = async (ids: string[]) => {
  await useHandleData(delToInfraredApi, ids, '删除所选可见光转红外信息')
  proTable.value?.clearSelection()
  proTable.value?.getTableList()
}

// 导出可见光转红外列表
const downloadFile = async () => {
  ElMessageBox.confirm('确认导出可见光转红外数据?', '温馨提示', { type: 'warning' }).then(() =>
    useDownload(exportToInfraredApi, '可见光转红外列表', proTable.value?.searchParam)
  )
}

// 批量添加可见光转红外
const dialogRef = ref<InstanceType<typeof ImportExcel> | null>(null)
const batchAdd = () => {
  const params = {
    title: '可见光转红外',
    tempApi: importTemplateApi,
    importApi: importToInfraredDataApi,
    getTableList: proTable.value?.getTableList
  }
  dialogRef.value?.acceptParams(params)
}

const formDialogRef = ref<InstanceType<typeof FormDialog> | null>(null)
// 打开弹框的功能
const openDialog = async (type: number, title: string, row?: any) => {
  let res = { data: {} }
  if (row?.id) {
    res = await getToInfraredApi(row?.id || null)
  }
  // 重置表单
  setItemsOptions()
  const params = {
    title,
    width: 580,
    isEdit: type !== 3,
    itemsOptions: itemsOptions,
    model: type == 1 ? {} : res.data,
    api: type == 1 ? addToInfraredApi : updateToInfraredApi,
    getTableList: proTable.value?.getTableList,
    closedEvent: () => {}
  }
  formDialogRef.value?.openDialog(params)
}

// 表格配置项
const columns = reactive<ColumnProps<any>[]>([
  { type: 'selection', fixed: 'left', width: 70 },
  { prop: 'id', label: '主键ID', width: 180 },
  {
    prop: 'name',
    label: '任务名称',
    search: {
      el: 'input'
    },
    width: 150
  },
  {
    prop: 'status',
    label: '任务状态',
    search: {
      el: 'select'
    },
    tag: true,
    enum: statusEnums,
    width: 150
  },
  {
    prop: 'type',
    label: '类型',
    tag: true,
    enum: enumsAlgorithmType,
    width: 120
  },
  {
    prop: 'subsystem',
    label: '分系统',
    tag: true,
    enum: enumsSubSystem,
    width: 200
  },
  {
    prop: 'algorithmName',
    label: '算法名称',
    width: 200
  },
  {
    prop: 'modelName',
    label: '模型名称',
    width: 200
  },
  // {
  //   prop: 'algorithmModelId',
  //   label: '模型',
  //   search: {
  //     el: 'input'
  //   },
  //   width: 150
  // },
  {
    prop: 'startTime',
    label: '开始时间',
    width: 180
  },
  {
    prop: 'endTime',
    label: '结束时间',
    width: 180
  },
  {
    prop: 'costSecond',
    label: '耗时',
    width: 120
  },
  {
    prop: 'log',
    label: '日志',
    width: 120
  },
  {
    prop: 'outputPath',
    label: '输出路径',
    width: 120
  },
  {
    prop: 'remarks',
    label: '备注',
    search: {
      el: 'input'
    },
    width: 120
  },
  { prop: 'operation', label: '操作', width: 230, fixed: 'right' }
])
// 表单配置项
let itemsOptions: ProForm.ItemsOptions[] = []
const setItemsOptions = () => {
  itemsOptions = [
    {
      label: '任务名称',
      prop: 'name',
      rules: [{ required: true, message: '任务名称不能为空', trigger: 'blur' }],
      compOptions: {
        placeholder: '请输入任务名称'
      }
    },
    {
      label: '选择数据集',
      prop: 'inputOssId',
      rules: [{ required: false, message: '数据集不能为空', trigger: 'blur' }],
      compOptions: {
        elTagName: 'select',
        placeholder: '请选择或者上传数据集',
        enum: getImageDataList,
        clearable: true
      }
    },
    {
      label: '上传数据集',
      prop: 'inputOssId',
      rules: [{ required: false, message: '数据集不能为空', trigger: 'blur' }],
      compOptions: {
        elTagName: 'file-upload',
        fileSize: 4096,
        fileType: ['zip'],
        placeholder: '请上传数据集'
      }
    },
    {
      label: '选择模型',
      prop: 'algorithmModelId',
      rules: [{ required: true, message: '模型不能为空', trigger: 'blur' }],
      compOptions: {
        elTagName: 'select',
        placeholder: '请选择模型',
        enum: enumsAlgorithmModelTrack
      }
    },
    {
      label: '备注',
      prop: 'remarks',
      rules: [],
      compOptions: {
        placeholder: '请输入备注'
      }
    }
  ]
}

const getImageDataList = ref<any>([])
onMounted(async () => {
  const qyery = {
    subsystem: SubSystem__['可见光转红外'],
    pageNum: 1,
    pageSize: 1000
  }
  const result: any = await listDataSeqApi(qyery)
  const data = result['data']['list']
  for (const item of data) {
    getImageDataList.value.push({
      value: item['inputOssId'],
      label: item['name'] + '-' + '可见光转红外'
    })
  }

  const qyery2 = {
    subsystem: SubSystem__['注释轨迹序列'],
    pageNum: 1,
    pageSize: 1000
  }
  const result2: any = await listDataSeqApi(qyery2)
  const data2 = result2['data']['list']
  for (const item of data2) {
    getImageDataList.value.push({
      value: item['inputOssId'],
      label: item['name'] + ' - ' + '注释轨迹序列'
    })
  }
})

const enumsAlgorithmModelTrack = ref<any>([])

const updateWnumsAlgorithmModelTrack = async () => {
  const result: any = await enumAlgorithmModelTrackApi()
  // console.log(result.data);
  enumsAlgorithmModelTrack.value = []
  for (const item of result.data) {
    if (SubSystem[item['subsystem']] === '可见光转红外') {
      item['label'] = item['label'] + '-' + SubSystem[item['subsystem']] + '-' + AlgorithmType[item['type']] + '-' + item['algorithmName']
      enumsAlgorithmModelTrack.value.push(item)
    }
  }
}

onMounted(async () => {
  updateWnumsAlgorithmModelTrack()
})

const setItemsOptions2 = () => {
  itemsOptions = [
    {
      label: '任务名称',
      prop: 'name',
      rules: [{ required: true, message: '任务名称不能为空', trigger: 'blur' }],
      compOptions: {
        disabled: true,
        placeholder: '请输入任务名称'
      }
    },
    {
      label: '选择模型',
      prop: 'algorithmModel',
      rules: [{ required: true, message: '模型不能为空', trigger: 'blur' }],
      compOptions: {
        elTagName: 'select',
        placeholder: '请选择模型',
        enum: enumsAlgorithmModelTrack
      }
    },
    {
      label: '备注',
      prop: 'remarks',
      rules: [],
      compOptions: {
        placeholder: '请输入备注'
      }
    }
  ]
}

const setItemsOptionsModel = () => {
  itemsOptions = [
    {
      label: '算法ID',
      prop: 'algorithmId',
      rules: [{ required: true, message: '算法不能为空', trigger: 'blur' }],
      compOptions: {
        disabled: true,
        placeholder: '请输入算法'
      }
    },
    {
      label: '算法类型',
      prop: 'algorithmType',
      rules: [{ required: true, message: '算法不能为空', trigger: 'blur' }],
      compOptions: {
        disabled: true,
        elTagName: 'select',
        placeholder: '请输入算法',
        enum: enumsAlgorithmConfigTrack
      }
    },
    {
      label: '算法参数',
      prop: 'parameterConfig',
      rules: [{ required: true, message: '模型名称不能为空', trigger: 'blur' }],
      compOptions: {
        placeholder: '请输入模型名称'
      }
    },
    {
      label: '模型ID',
      prop: 'id',
      rules: [{ required: true, message: '模型名称不能为空', trigger: 'blur' }],
      compOptions: {
        placeholder: '请输入模型名称'
      }
    },
    {
      label: '模型名称',
      prop: 'modelName',
      rules: [{ required: true, message: '模型名称不能为空', trigger: 'blur' }],
      compOptions: {
        placeholder: '请输入模型名称'
      }
    },
    {
      label: '模型保存路径',
      prop: 'modelAddress',
      rules: [{ required: true, message: '模型名称不能为空', trigger: 'blur' }],
      compOptions: {
        placeholder: '请输入模型名称'
      }
    },
    {
      label: '备注',
      prop: 'remarks',
      rules: [{ required: false, message: '备注不能为空', trigger: 'blur' }],
      compOptions: {
        placeholder: '请输入备注'
      }
    }
  ]
}
</script>