<template>
  <ContentWrap>
    <!-- 搜索工作栏 -->
    <el-form
      class="-mb-15px"
      :model="queryParams"
      ref="queryFormRef"
      :inline="true"
      label-width="68px"
    >
      <el-form-item label="项目" class="" prop="name">
        <el-input
          v-model="queryParams.name"
          placeholder="请输入项目名称"
          clearable
          @keyup.enter="handleQuery"
          class="!w-240px"
        />
      </el-form-item>
      <el-form-item label="类型" prop="status">
        <el-select
          v-model="queryParams.status"
          placeholder="请选择类型"
          clearable
          class="!w-240px"
        >
          <el-option
            v-for="dict in getIntDictOptions(DICT_TYPE.TRAINING_STATUS)"
            :key="dict.value"
            :label="dict.label"
            :value="dict.value"
        />
        </el-select>
      </el-form-item>
      <el-form-item label="识别类型" prop="type">
        <el-select
          v-model="queryParams.type"
          placeholder="请选择识别类型"
          clearable
          class="!w-240px"
        >
          
        <el-option
            v-for="dict in getIntDictOptions(DICT_TYPE.VISUAL_TYPE)"
            :key="dict.value"
            :label="dict.label"
            :value="dict.value"
        />
        </el-select>
      </el-form-item>
      <el-form-item label="创建时间" prop="createTime">
        <el-date-picker
          v-model="queryParams.createTime"
          value-format="YYYY-MM-DD HH:mm:ss"
          type="daterange"
          start-placeholder="开始日期"
          end-placeholder="结束日期"
          :default-time="[new Date('1 00:00:00'), new Date('1 23:59:59')]"
          class="!w-240px"
        />
      </el-form-item>
      <el-form-item>
        <el-button @click="handleQuery"><Icon icon="ep:search" class="mr-5px" /> 搜索</el-button>
        <el-button @click="resetQuery"><Icon icon="ep:refresh" class="mr-5px" /> 重置</el-button>
        <el-button
          type="primary"
          plain
          @click="openForm('create')"
          v-hasPermi="['annotation:datas:create']"
        >
          <Icon icon="ep:plus" class="mr-5px" /> 新增
        </el-button>
        <el-button
          type="success"
          plain
          @click="handleExport"
          :loading="exportLoading"
          v-hasPermi="['annotation:datas:export']"
        >
          <Icon icon="ep:download" class="mr-5px" /> 导出
        </el-button>
      </el-form-item>
    </el-form>
  </ContentWrap>

  <!-- 列表 -->
  <ContentWrap>
    <el-table v-loading="loading" :data="list" :stripe="true" :show-overflow-tooltip="true">
      <el-table-column label="项目名称" align="center" prop="name" />
      <el-table-column label="描述" align="center" prop="description" />
      <el-table-column label="状态" align="center" prop="status" >
         <template #default="scope">
          <dict-tag :type="DICT_TYPE.TRAINING_STATUS" :value="scope.row.status" />
        </template>
      </el-table-column>
      <el-table-column label="图片总数" align="center" prop="count" />
      <el-table-column label="识别类型" align="center" prop="type" >
         <template #default="scope">
          <dict-tag :type="DICT_TYPE.VISUAL_TYPE" :value="scope.row.type" />
        </template>
      </el-table-column>
      <el-table-column label="进度" align="center" prop="progress" >
        <template #default="scope">
          {{ scope.row.progress +"%" }}
        </template>
      </el-table-column>
      <el-table-column label="数据集" align="center" prop="datasets" min-width="200">
        <template #default="scope">
          <el-tag
            v-for="(dataset, index) in (scope.row.datasetList || [])"
            :key="index"
            class="mx-1"
            style="margin-right: 4px"
          >
            {{ dataset.name }}
          </el-tag>
          <span v-if="!scope.row.datasetList || scope.row.datasetList.length === 0">-</span>
        </template>
      </el-table-column>
      <el-table-column
        label="创建时间"
        align="center"
        prop="createTime"
        :formatter="dateFormatter"
        width="180px"
      />
      <el-table-column width="400px" label="操作" align="center">
        <template #default="scope">
          <el-button
            link
            type="primary"
            @click="openForm('update', scope.row.id)"
          >
            编辑
          </el-button>
<el-button
            link
            type="warning"
            @click="handleRefresh(scope.row)"
            v-hasPermi="['annotation:dataset:update']"
          >
            刷新
          </el-button>
          <el-button
            link
            type="primary"
            @click="handleEnhance(scope.row)"
          >
            增强
          </el-button>

          <router-link :to="'/annotation/mark?id=' + scope.row.id ">
            <el-button
              link
              type="primary"
            >
              标注
            </el-button>
          </router-link>

          <router-link :to="'/annotation/train?name=' + encodeURIComponent(scope.row.name)">
            <el-button
              link
              type="primary"
              :disabled="scope.row.status < 2"
            >
              训练
            </el-button>
          </router-link>

          <router-link
            v-if="scope.row.status === '4'"
            :to="'/annotation/test/' + scope.row.id"
          >
            <el-button link type="primary">测试</el-button>
          </router-link>

          <el-button
            v-if="scope.row.status === '4'"
            link
            type="primary"
            @click="handleExportModel(scope.row.id)"
            v-hasPermi="['annotation:datas:export']"
          >
            导出
          </el-button>

          <el-button
            link
            type="danger"
            @click="handleDelete(scope.row.id)"
            v-hasPermi="['annotation:datas:delete']"
          >
            删除
          </el-button>
        </template>
      </el-table-column>
    </el-table>
    <!-- 分页 -->
    <Pagination
      :total="total"
      v-model:page="queryParams.pageNo"
      v-model:limit="queryParams.pageSize"
      @pagination="getList"
    />
  </ContentWrap>

  <!-- 表单弹窗：添加/修改 -->
  <DatasForm ref="formRef" @success="getList" />
  <!-- 图像增强对话框 -->
<el-dialog 
  v-model="enhanceDialogVisible" 
  title="图像增强" 
  width="500px"
  @close="selectedEnhancements = []"
>
  <el-checkbox-group v-model="selectedEnhancements" class="enhancement-checkbox-group">
    <el-checkbox 
      v-for="option in enhancementOptions" 
      :key="option.value" 
      :label="option.value"
      class="enhancement-checkbox"
    >
      {{ option.label }}
    </el-checkbox>
  </el-checkbox-group>
  
  <template #footer>
    <el-button @click="enhanceDialogVisible = false">取消</el-button>
    <el-button type="primary" @click="executeEnhancement">确定</el-button>
  </template>
</el-dialog>
</template>

<script setup lang="ts">

import { getIntDictOptions, DICT_TYPE } from '@/utils/dict'
import { dateFormatter } from '@/utils/formatTime'
import download from '@/utils/download'
import { DatasApi, DatasVO } from '@/api/annotation/datas'
// import DatasForm from './DatasForm.vue'
import DatasForm from './DatasForm.vue'
/** 数据集管理 列表 */
defineOptions({ name: 'Datas' })

const message = useMessage() // 消息弹窗
const { t } = useI18n() // 国际化

const loading = ref(true) // 列表的加载中
const list = ref<DatasVO[]>([]) // 列表的数据
const total = ref(0) // 列表的总页数
const queryParams = reactive({
  pageNo: 1,
  pageSize: 10,
  name: undefined as string | undefined,
  description: undefined as string | undefined,
  status: undefined as string | undefined,
  count: undefined as number | undefined,
  type: undefined as string | undefined,
  progress: undefined as number | undefined,
  createTime: [] as string[],
})
const queryFormRef = ref() // 搜索的表单
const exportLoading = ref(false) // 导出的加载中
// 添加增强对话框相关状态
const enhanceDialogVisible = ref(false)
const selectedEnhancements = ref<string[]>([])
const currentEnhanceRow = ref<DatasVO | null>(null)

// 增强选项配置
const enhancementOptions = [
  { label: '亮度调整（Brightness）：随机调整图像的亮度', value: 'brightness' },
  { label: '对比度调整（Contrast）：增强或减少图像的对比度', value: 'contrast' },
  { label: '饱和度调整（Saturation）：改变图像的色彩饱和度', value: 'saturation' },
  { label: '色相调整（Hue）：随机改变图像的色相，改变颜色的基调', value: 'hue' },
  { label: '灰度化（Grayscale）：将图像转换为灰度图，模拟不同的拍摄条件', value: 'grayscale' }
]



// 执行增强操作
const executeEnhancement = async () => {
  if (!currentEnhanceRow.value || selectedEnhancements.value.length === 0) {
    message.warning('请选择至少一种增强方式')
    return
  }

  if (!currentEnhanceRow.value?.id) {
    message.error('数据异常，请刷新后重试')
    return
  }

  try {
    // 调用API执行增强操作
    await DatasApi.enhanceImages({
      id: currentEnhanceRow.value.id,
      enhancements: selectedEnhancements.value
    })
    
    message.success('图像增强任务已启动')
    enhanceDialogVisible.value = false
    // 刷新列表
    await getList()
  } catch (error) {
    message.error('增强操作失败: ' + (error as Error).message)
  }
}


/** 刷新按钮操作 */
const handleRefresh = async (row: any) => {
  try {
    // 刷新的二次确认
    await message.confirm('确定要刷新该项目吗？', '刷新确认')
    await DatasApi.handleRefresh(row.id)
    // 更新列表中对应行的数据
    
    message.success('刷新成功')
  } catch {
    message.error('刷新失败')
  }
}
/** 查询列表 */
const getList = async () => {
  loading.value = true
  try {
    const data = await DatasApi.getDatasPage(queryParams)
    list.value = data.list
    total.value = data.total
  } finally {
    loading.value = false
  }
}

// 修改增强处理函数
const handleEnhance = async (row: DatasVO) => {
  // 打开增强对话框
  currentEnhanceRow.value = row
  enhanceDialogVisible.value = true
  selectedEnhancements.value = []

  // 检查状态是否为标注完成(2)
  if (row.status !== "2") {
    message.warning('请先完成标注，进度达到100%后再进行增强操作')
  }
}

// 导出模型
const handleExportModel = async (id: number) => {
  try {
    // 显示导出确认
    await message.exportConfirm()
    
    // 执行导出
    const data = await DatasApi.exportModel(id)
    // 下载文件
    download.excel(data, '模型导出.zip')
    message.success('导出成功')
  } catch (error) {
    message.error('导出失败: ' + (error as Error).message)
  }
}
/** 搜索按钮操作 */
const handleQuery = () => {
  queryParams.pageNo = 1
  getList()
}

/** 重置按钮操作 */
const resetQuery = () => {
  queryFormRef.value.resetFields()
  handleQuery()
}

/** 添加/修改操作 */
const formRef = ref()
const openForm = (type: string, id?: number) => {
  formRef.value.open(type, id)
}

/** 删除按钮操作 */
const handleDelete = async (id: number) => {
  try {
    // 删除的二次确认
    await message.delConfirm()
    // 发起删除
    await DatasApi.deleteDatas(id)
    message.success(t('common.delSuccess'))
    // 刷新列表
    await getList()
  } catch {}
}

/** 导出按钮操作 */
const handleExport = async () => {
  try {
    // 导出的二次确认
    await message.exportConfirm()
    // 发起导出
    exportLoading.value = true
    const data = await DatasApi.exportDatas(queryParams)
    download.excel(data, '数据集管理.xls')
  } catch {
  } finally {
    exportLoading.value = false
  }
}

/** 初始化 **/
onMounted(() => {
  getList()
})
</script>
<style scoped>
.enhancement-checkbox-group {
  display: flex;
  flex-direction: column;
  gap: 15px;
}

.enhancement-checkbox {
  white-space: normal;
  line-height: 1.5;
  padding: 10px;
  border-radius: 4px;
  
  &:hover {
    background-color: #f5f7fa;
  }
}
</style>