feat: 目标检测功能基本搭建

src/api/interface/demo/TargetDetection.ts

@@ -0,0 +1,239 @@
+import {PageQuery, BaseEntity} from '@/api/interface/index'
+
+export interface TargetDetectionVO extends BaseEntity {
+  /**
+   * ID
+   */
+  id: string | number;
+
+  /**
+   * 任务名称
+   */
+  name: string;
+
+  /**
+   * 状态
+   0：未开始
+   1：进行中
+   2：完成
+   3：失败
+   4：中断
+   */
+  status: string;
+
+  /**
+   * 开始时间
+   */
+  startTime: string;
+
+  /**
+   * 结束时间
+   */
+  endTime: string;
+
+  /**
+   * 耗时
+   */
+  costSecond: number;
+
+  /**
+   * 日志
+   */
+  log: string;
+
+  /**
+   * 备注
+   */
+  remarks: string;
+
+  /**
+   * $column.columnComment
+   */
+  url: string;
+
+  /**
+   * $column.columnComment
+   */
+  inputOssId: string | number;
+
+  /**
+   * 输入路径
+   */
+  inputPath: string;
+
+  /**
+   * 输出路径
+   */
+  outputPath: string;
+
+  /**
+   * zip文件输出路径
+   */
+  zipFilePath: string;
+
+  /**
+   * 模型的id
+   */
+  algorithmModelId: string | number;
+
+}
+
+export interface TargetDetectionForm {
+  /**
+   * ID
+   */
+  id?: string | number;
+
+  /**
+   * 任务名称
+   */
+  name?: string;
+
+  /**
+   * 状态
+   0：未开始
+   1：进行中
+   2：完成
+   3：失败
+   4：中断
+   */
+  status?: string;
+
+  /**
+   * 开始时间
+   */
+  startTime?: string;
+
+  /**
+   * 结束时间
+   */
+  endTime?: string;
+
+  /**
+   * 耗时
+   */
+  costSecond?: number;
+
+  /**
+   * 日志
+   */
+  log?: string;
+
+  /**
+   * 备注
+   */
+  remarks?: string;
+
+  /**
+   * $column.columnComment
+   */
+  version?: number;
+
+  /**
+   * $column.columnComment
+   */
+  url?: string;
+
+  /**
+   * $column.columnComment
+   */
+  inputOssId?: string | number;
+
+  /**
+   * 输入路径
+   */
+  inputPath?: string;
+
+  /**
+   * 输出路径
+   */
+  outputPath?: string;
+
+  /**
+   * zip文件输出路径
+   */
+  zipFilePath?: string;
+
+  /**
+   * 模型的id
+   */
+  algorithmModelId?: string | number;
+
+}
+
+export interface TargetDetectionQuery extends PageQuery {
+  /**
+   * 任务名称
+   */
+  name?: string;
+
+  /**
+   * 状态
+   0：未开始
+   1：进行中
+   2：完成
+   3：失败
+   4：中断
+   */
+  status?: string;
+
+  /**
+   * 开始时间
+   */
+  startTime?: string;
+
+  /**
+   * 结束时间
+   */
+  endTime?: string;
+
+  /**
+   * 耗时
+   */
+  costSecond?: number;
+
+  /**
+   * 日志
+   */
+  log?: string;
+
+  /**
+   * 备注
+   */
+  remarks?: string;
+
+  /**
+   * $column.columnComment
+   */
+  url?: string;
+
+  /**
+   * $column.columnComment
+   */
+  inputOssId?: string | number;
+
+  /**
+   * 输入路径
+   */
+  inputPath?: string;
+
+  /**
+   * 输出路径
+   */
+  outputPath?: string;
+
+  /**
+   * zip文件输出路径
+   */
+  zipFilePath?: string;
+
+  /**
+   * 模型的id
+   */
+  algorithmModelId?: string | number;
+
+  /**
+   * 日期范围参数
+   */
+  params?: any;
+}

src/api/modules/demo/TargetDetection.ts

@@ -0,0 +1,76 @@
+import http from '@/api'
+import {TargetDetectionVO, TargetDetectionForm, TargetDetectionQuery} from '@/api/interface/demo/TargetDetection'
+
+/**
+ * @name 查询目标检测列表
+ * @param query 参数
+ * @returns 返回列表
+ */
+export const listTargetDetectionApi = (query: TargetDetectionQuery) => {
+  return http.get<TargetDetectionVO[]>('/demo/TargetDetection/list', query, {loading: true})
+}
+
+/**
+ * @name 查询目标检测详细
+ * @param id id
+ * @returns returns
+ */
+export const getTargetDetectionApi = (id: string | number) => {
+  return http.get<TargetDetectionVO>(`/demo/TargetDetection/${id}`)
+}
+
+/**
+ * @name 新增目标检测
+ * @param data data
+ * @returns returns
+ */
+export const addTargetDetectionApi = (data: TargetDetectionForm) => {
+  return http.post<any>('/demo/TargetDetection', data, {loading: false})
+}
+
+/**
+ * @name 修改目标检测
+ * @param data data
+ * @returns returns
+ */
+export const updateTargetDetectionApi = (data: TargetDetectionForm) => {
+  return http.put<any>('/demo/TargetDetection', data, {loading: false})
+}
+
+/**
+ * @name 删除目标检测
+ * @param id id
+ * @returns returns
+ */
+export const delTargetDetectionApi = (id: string | number | Array<string | number>) => {
+  return http.delete<any>(`/demo/TargetDetection/${id}`)
+}
+
+/**
+ * @name 下载模板
+ * @returns returns
+ */
+export const importTemplateApi = () => {
+  return http.downloadPost('/demo/TargetDetection/importTemplate', {})
+}
+
+/**
+ * @name 导入数据
+ * @returns returns
+ */
+export const importTargetDetectionDataApi = (data: any) => {
+  return http.post('/demo/TargetDetection/importData', data)
+}
+
+/**
+ * @name 导出数据
+ * @returns returns
+ */
+export const exportTargetDetectionApi = (data: any) => {
+  return http.downloadPost('/demo/TargetDetection/export', data)
+}
+
+
+export const startTargetDetectionApi = (id: string | number) => {
+  return http.get<any>(`/demo/TargetDetection/start/${id}`)
+}

src/views/demo/TargetDetection/index.vue

@@ -0,0 +1,395 @@
+<template>
+  <div class="table-box">
+    <ProTable ref="proTable" :columns="columns" row-key="id" :request-api="listTargetDetectionApi">
+      <!-- 表格 header 按钮 -->
+      <template #tableHeader="scope">
+        <el-button type="primary" v-auth="['demo:TargetDetection:add']" icon="CirclePlus" @click="openDialog(1, '目标检测新增')">
+          新增
+        </el-button>
+        <!--        <el-button type="primary" v-auth="['demo:TargetDetection:import']" icon="Upload" plain @click="batchAdd">-->
+        <!--          导入-->
+        <!--        </el-button>-->
+        <!--        <el-button type="primary" v-auth="['demo:TargetDetection:export']" icon="Download" plain @click="downloadFile">-->
+        <!--          导出-->
+        <!--        </el-button>-->
+        <el-button
+          type="danger"
+          v-auth="['demo:TargetDetection: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:trackSequence:start']"
+          v-if="scope.row.status == '0' || scope.row.status == '3' || scope.row.status == '4'"
+          @click="startTargetDetection(scope.row)"
+        >
+          开始
+        </el-button>
+        <el-popconfirm title="确定终止此任务吗?" 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:trackSequence:download']"
+          v-if="scope.row.status == '2'"
+        >
+          <!--          @confirm="stopTrackSequence(scope.row)"-->
+          <!--          @click="dowloadtrackSequence(scope.row)"-->
+          下载
+        </el-button>
+        <!--        <el-button-->
+        <!--          type="primary"-->
+        <!--          link-->
+        <!--          icon="View"-->
+        <!--          v-auth="['demo:TargetDetection:query']"-->
+        <!--          @click="openDialog(3, '目标检测查看', scope.row)"-->
+        <!--        >-->
+        <!--          查看-->
+        <!--        </el-button>-->
+        <!--        <el-button-->
+        <!--          type="primary"-->
+        <!--          link-->
+        <!--          icon="EditPen"-->
+        <!--          v-auth="['demo:TargetDetection:edit']"-->
+        <!--          @click="openDialog(2, '目标检测编辑', scope.row)"-->
+        <!--        >-->
+        <!--          编辑-->
+        <!--        </el-button>-->
+        <el-button type="primary" link icon="Delete" v-auth="['demo:TargetDetection:remove']"
+                   @click="deleteTargetDetection(scope.row)">
+          删除
+        </el-button>
+      </template>
+    </ProTable>
+    <FormDialog ref="formDialogRef"/>
+    <ImportExcel ref="dialogRef"/>
+  </div>
+</template>
+
+<script setup lang="tsx" name="TargetDetection">
+import {ref, reactive, onMounted} 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 {
+  listTargetDetectionApi,
+  delTargetDetectionApi,
+  addTargetDetectionApi,
+  updateTargetDetectionApi,
+  importTemplateApi,
+  importTargetDetectionDataApi,
+  exportTargetDetectionApi,
+  getTargetDetectionApi,
+  startTargetDetectionApi
+} from '@/api/modules/demo/TargetDetection'
+import statusEnums from "@/utils/status";
+import {enumAlgorithmModelTrackApi, getAlgorithmModelTrackApi} from "@/api/modules/demo/AlgorithmModelTrack";
+import {AlgorithmType, SubSystem} from "@/views/demo/utils";
+import {enumAlgorithmConfigTrackApi} from "@/api/modules/demo/AlgorithmConfigTrack";
+import {updateTrackSequenceApi} from "@/api/modules/demo/trackSequence";
+import {startToInfraredApi} from "@/api/modules/demo/toInfrared";
+
+const startTargetDetection = async (params: any) => {
+  const res: any = await startTargetDetectionApi(params.id)
+  if (res.code === 200) {
+    ElMessage.success('任务已开始，请等待完成！')
+  } else {
+    ElMessage.error('任务开始失败，请检查！')
+  }
+  proTable.value?.getTableList()
+}
+
+
+const enumsAlgorithmConfigTrack = ref<any>([])
+onMounted(async () => {
+  const result = await enumAlgorithmConfigTrackApi()
+  // console.log(result)
+  // console.log(result['data'])
+  enumsAlgorithmConfigTrack.value = result['data']
+  return result['data']
+})
+
+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: updateTrackSequenceApi,
+    getTableList: proTable.value?.getTableList
+  }
+  formDialogRef.value?.openDialog(params)
+}
+
+// ProTable 实例
+const proTable = ref<ProTableInstance>()
+
+// 删除目标检测信息
+const deleteTargetDetection = async (params: any) => {
+  await useHandleData(delTargetDetectionApi, params.id, '删除【' + params.id + '】目标检测')
+  proTable.value?.getTableList()
+}
+
+// 批量删除目标检测信息
+const batchDelete = async (ids: string[]) => {
+  await useHandleData(delTargetDetectionApi, ids, '删除所选目标检测信息')
+  proTable.value?.clearSelection()
+  proTable.value?.getTableList()
+}
+
+// 导出目标检测列表
+const downloadFile = async () => {
+  ElMessageBox.confirm('确认导出目标检测数据?', '温馨提示', {type: 'warning'}).then(() =>
+    useDownload(exportTargetDetectionApi, '目标检测列表', proTable.value?.searchParam)
+  )
+}
+
+// 批量添加目标检测
+const dialogRef = ref<InstanceType<typeof ImportExcel> | null>(null)
+const batchAdd = () => {
+  const params = {
+    title: '目标检测',
+    tempApi: importTemplateApi,
+    importApi: importTargetDetectionDataApi,
+    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 getTargetDetectionApi(row?.id || null)
+  }
+  // 重置表单
+  setItemsOptions()
+  const params = {
+    title,
+    width: 580,
+    isEdit: type !== 3,
+    itemsOptions: itemsOptions,
+    model: type == 1 ? {} : res.data,
+    api: type == 1 ? addTargetDetectionApi : updateTargetDetectionApi,
+    getTableList: proTable.value?.getTableList
+  }
+  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: 'algorithmModelId',
+    label: '模型id',
+    width: 120
+  },
+  {
+    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'}
+])
+
+const enumsAlgorithmModelTrack = ref<any>([])
+
+onMounted(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)
+    }
+  }
+})
+
+// 表单配置项
+let itemsOptions: ProForm.ItemsOptions[] = []
+const setItemsOptions = () => {
+  itemsOptions = [
+    {
+      label: '任务名称',
+      prop: 'name',
+      rules: [{required: true, message: '任务名称不能为空', trigger: 'blur'}],
+      compOptions: {
+        placeholder: '请输入任务名称'
+      }
+    },
+    {
+      label: '上传图片集',
+      prop: 'inputOssId',
+      rules: [{required: true, 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 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>