修改项目树问题 绘制稳态治理分析页面

src/components/tree/point.vue

@@ -1,24 +1,19 @@
 <!-- 设备监控使用折叠面板渲染多个tree -->
 <template>
     <div :style="{ width: menuCollapse ? '40px' : props.width }" style="display: flex; overflow: hidden">
-        <Icon
+        <!-- <Icon
             v-show="menuCollapse"
             @click="onMenuCollapse"
             :name="menuCollapse ? 'el-icon-Expand' : 'el-icon-Fold'"
             :class="menuCollapse ? 'unfold' : ''"
-            size="18"
+            size="18px"
             class="fold ml10 mt20 menu-collapse"
             style="cursor: pointer"
             v-if="route.path != '/admin/govern/reportCore/statistics/index'"
-        />
+        /> -->
         <div class="cn-tree" :style="{ opacity: menuCollapse ? 0 : 1, display: menuCollapse ? 'none' : '' }">
             <div style="display: flex; align-items: center" class="mb10">
-                <!-- <el-input maxlength="32"  v-model.trim="filterText" placeholder="请输入内容" clearable>
-                    <template #prefix>
-                        <Icon name="el-icon-Search" style="font-size: 16px" />
-                    </template>
-                </el-input> -->
-                <el-input maxlength="32"  v-model.trim="filterText" placeholder="请输入内容" clearable>
+                <el-input maxlength="32" v-model.trim="filterText" placeholder="请输入内容" clearable>
                     <template #prepend>
                         <el-select v-model="treeType" @change="changeTreeType" style="min-width: 75px">
                             <el-option
@@ -33,35 +28,33 @@
                         <Icon name="el-icon-Search" style="font-size: 16px" />
                     </template>
                 </el-input>
-                <!--  -->
-                <Icon
+                <!-- <Icon
                     @click="onMenuCollapse"
                     :name="menuCollapse ? 'el-icon-Expand' : 'el-icon-Fold'"
                     :class="menuCollapse ? 'unfold' : ''"
-                    size="18"
+                    size="18px"
                     class="fold ml10 menu-collapse"
                     style="cursor: pointer"
                     v-if="props.canExpand && route.path != '/admin/govern/reportCore/statistics/index'"
-                />
+                /> -->
             </div>
 
             <el-collapse
                 :accordion="true"
-                v-model.trim="activeName"
+                v-model="activeName"
                 style="flex: 1; height: 100%"
                 @change="changeDevice"
                 v-if="treeType == '1'"
                 v-loading="loading"
             >
-                <el-collapse-item title="治理设备" name="0" v-if="zlDeviceData.length != 0">
-                    <el-select v-model.trim="process" clearable placeholder="请选择状态" class="mb10">
-                        <el-option label="功能调试" value="2"></el-option>
-                        <el-option label="出厂调试" value="3"></el-option>
-                        <el-option label="正式投运" value="4"></el-option>
+                <el-collapse-item title="治理设备" name="0" v-if="zlDeviceData.length">
+                    <el-select v-model="process" clearable placeholder="请选择状态" class="mb10">
+                        <el-option label="功能调试" value="2" />
+                        <el-option label="出厂调试" value="3" />
+                        <el-option label="正式投运" value="4" />
                     </el-select>
-
                     <el-tree
-                        :style="{ height: treeType.length != 0 ? 'calc(100vh - 380px)' : 'calc(100vh - 278px)' }"
+                        :style="{ height: governTreeHeight }"
                         ref="treeRef1"
                         :props="defaultProps"
                         highlight-current
@@ -72,22 +65,22 @@
                         style="overflow: auto"
                         :default-expand-all="false"
                     >
-                        <template #default="{ node, data }">
+                        <template #default="{ node, data: nodeData }">
                             <span class="custom-tree-node">
                                 <Icon
-                                    :name="data.icon"
+                                    :name="nodeData.icon"
                                     style="font-size: 16px"
-                                    :style="{ color: data.color }"
-                                    v-if="data.icon"
+                                    :style="{ color: nodeData.color }"
+                                    v-if="nodeData.icon"
                                 />
                                 <span style="margin-left: 4px">{{ node.label }}</span>
                             </span>
                         </template>
                     </el-tree>
                 </el-collapse-item>
-                <el-collapse-item title="便携式设备" name="1" v-if="bxsDeviceData.length != 0">
+                <el-collapse-item title="便携式设备" name="1" v-if="bxsDeviceData.length">
                     <el-tree
-                        :style="{ height: zlDeviceData.length != 0 ? 'calc(100vh - 340px)' : 'calc(100vh - 238px)' }"
+                        :style="{ height: otherTreeHeight }"
                         ref="treeRef2"
                         :props="defaultProps"
                         highlight-current
@@ -98,22 +91,22 @@
                         v-bind="$attrs"
                         style="overflow: auto"
                     >
-                        <template #default="{ node, data }">
+                        <template #default="{ node, data: nodeData }">
                             <span class="custom-tree-node">
                                 <Icon
-                                    :name="data.icon"
+                                    :name="nodeData.icon"
                                     style="font-size: 16px"
-                                    :style="{ color: data.color }"
-                                    v-if="data.icon"
+                                    :style="{ color: nodeData.color }"
+                                    v-if="nodeData.icon"
                                 />
                                 <span style="margin-left: 4px">{{ node.label }}</span>
                             </span>
                         </template>
                     </el-tree>
                 </el-collapse-item>
-                <el-collapse-item title="监测设备" name="2" v-if="yqfDeviceData.length != 0">
+                <el-collapse-item title="监测设备" name="2" v-if="yqfDeviceData.length">
                     <el-tree
-                        :style="{ height: zlDeviceData.length != 0 ? 'calc(100vh - 340px)' : 'calc(100vh - 238px)' }"
+                        :style="{ height: otherTreeHeight }"
                         ref="treeRef3"
                         :props="defaultProps"
                         highlight-current
@@ -124,13 +117,13 @@
                         v-bind="$attrs"
                         style="overflow: auto"
                     >
-                        <template #default="{ node, data }">
+                        <template #default="{ node, data: nodeData }">
                             <span class="custom-tree-node">
                                 <Icon
-                                    :name="data.icon"
+                                    :name="nodeData.icon"
                                     style="font-size: 16px"
-                                    :style="{ color: data.color }"
-                                    v-if="data.icon"
+                                    :style="{ color: nodeData.color }"
+                                    v-if="nodeData.icon"
                                 />
                                 <span style="margin-left: 4px">{{ node.label }}</span>
                             </span>
@@ -140,7 +133,7 @@
             </el-collapse>
             <div v-if="treeType == '2'" v-loading="loading">
                 <el-tree
-                    :style="{ height: 'calc(100vh - 188px)' }"
+                    :style="{ height: engineeringTreeHeight }"
                     class="pt10"
                     ref="treeRef4"
                     :props="defaultProps"
@@ -148,17 +141,17 @@
                     :filter-node-method="filterNode"
                     node-key="id"
                     v-bind="$attrs"
-                    :data="data"
+                    :data="props.data"
                     style="overflow: auto"
                     :default-expand-all="false"
                 >
-                    <template #default="{ node, data }">
+                    <template #default="{ node, data: nodeData }">
                         <span class="custom-tree-node">
                             <Icon
-                                :name="data.icon"
+                                :name="nodeData.icon"
                                 style="font-size: 16px"
-                                :style="{ color: data.color }"
-                                v-if="data.icon"
+                                :style="{ color: nodeData.color }"
+                                v-if="nodeData.icon"
                             />
                             <span style="margin-left: 4px">{{ node.label }}</span>
                         </span>
@@ -171,308 +164,217 @@
 
 <script lang="ts" setup>
 import useCurrentInstance from '@/utils/useCurrentInstance'
-import { ElTree } from 'element-plus'
-import { el, fa } from 'element-plus/es/locale'
-import { ref, watch, defineEmits, onMounted, nextTick, computed } from 'vue'
+import { ElTree, type CollapseModelValue } from 'element-plus'
+import { ref, watch, nextTick, computed } from 'vue'
 import { useRoute } from 'vue-router'
+import { collectDeviceLeaves } from './govern/lineTreeUtils'
+
 defineOptions({
-    name: 'govern/tree'
+    name: 'govern/point',
+    inheritAttrs: false
 })
+
 const emit = defineEmits(['changePointType', 'changeTreeType'])
+
 interface Props {
     width?: string
     canExpand?: boolean
     type?: string
-    data?: any
+    data?: any[]
 }
 
 const props = withDefaults(defineProps<Props>(), {
     width: '100%',
     canExpand: true,
     type: '',
-    data: []
+    data: () => []
 })
-const process = ref('')
+
 const route = useRoute()
 const { proxy } = useCurrentInstance()
+
+const treeType = ref('1')
+const options = [
+    { label: '设备', value: '1' },
+    { label: '工程', value: '2' }
+]
+
 const menuCollapse = ref(false)
 const activeName = ref('0')
 const filterText = ref('')
-const defaultProps = {
-    label: 'name',
-    value: 'id'
-}
-const treeType = ref('1')
-const options = [
-    {
-        label: '设备',
-        value: '1'
-    },
-    {
-        label: '工程',
-        value: '2'
-    }
-]
-//治理设备数据
-const zlDeviceData = ref<any>([])
-const zlDevList = ref<any>([])
-//便携式设备数据
-const bxsDeviceData = ref<any>([])
-//监测设备数据
-const yqfDeviceData = ref<any>([])
-watch(
-    () => props.data,
-    (val, oldVal) => {
-        if (val && val.length != 0) {
-            val.map((item: any) => {
-                if (item.name == '治理设备') {
-                    zlDeviceData.value = []
-                    item.children.map((vv: any) => {
-                        zlDeviceData.value.push(vv)
-                    })
-                    zlDevList.value = JSON.parse(JSON.stringify(zlDeviceData.value))
-                } else if (item.name == '便携式设备') {
-                    bxsDeviceData.value = []
-                    item.children.map((vv: any) => {
-                        bxsDeviceData.value.push(vv)
-                    })
-                } else if (item.name == '监测设备') {
-                    yqfDeviceData.value = []
-                    item.children.map((vv: any) => {
-                        yqfDeviceData.value.push(vv)
-                    })
-                }
-            })
-        }
-    },
-    {
-        immediate: true,
-        deep: true
-    }
+const process = ref('')
+const loading = ref(false)
+
+const defaultProps = { label: 'name', value: 'id' }
+
+const zlDeviceData = ref<any[]>([])
+const zlDevList = ref<any[]>([])
+const bxsDeviceData = ref<any[]>([])
+const yqfDeviceData = ref<any[]>([])
+
+const governTreeHeight = computed(() => 'calc(100vh - 380px)')
+const otherTreeHeight = computed(() =>
+    zlDeviceData.value.length ? 'calc(100vh - 340px)' : 'calc(100vh - 238px)'
 )
+const engineeringTreeHeight = computed(() => 'calc(100vh - 188px)')
 
-watch(filterText, val => {
-    if (treeType.value == '2') {
-        treeRef4.value!.filter(val)
-    } else if (activeName.value == '0') {
-        treeRef1.value!.filter(val)
-    } else if (activeName.value == '1') {
-        treeRef2.value!.filter(val)
-    } else {
-        treeRef3.value!.filter(val)
-    }
-})
-watch(process, val => {
-    if (val == '' || val == undefined) {
-        zlDevList.value = JSON.parse(JSON.stringify(zlDeviceData.value))
-    } else {
-        zlDevList.value = filterProcess(JSON.parse(JSON.stringify(zlDeviceData.value)))
-    }
+const treeRef1 = ref<InstanceType<typeof ElTree>>()
+const treeRef2 = ref<InstanceType<typeof ElTree>>()
+const treeRef3 = ref<InstanceType<typeof ElTree>>()
+const treeRef4 = ref<InstanceType<typeof ElTree>>()
 
-    setTimeout(() => {
-        changeDevice(activeName.value)
-    }, 0)
-})
+defineExpose({ treeRef1, treeRef2, treeRef3, treeRef4 })
 
-const changeDevice = (val: any) => {
-    let arr1: any = []
-    //zlDeviceData
-    zlDevList.value.forEach((item: any) => {
-        item.children.forEach((item2: any) => {
-            item2.children.forEach((item3: any) => {
-                item3.children.forEach((item4: any) => {
-                    arr1.push(item4)
-                })
-            })
-        })
+function splitTreeData(val: any[]) {
+    zlDeviceData.value = []
+    bxsDeviceData.value = []
+    yqfDeviceData.value = []
+
+    val.forEach(item => {
+        if (item.name === '治理设备') {
+            zlDeviceData.value = item.children ?? []
+        } else if (item.name === '便携式设备') {
+            bxsDeviceData.value = item.children ?? []
+        } else if (item.name === '监测设备') {
+            yqfDeviceData.value = item.children ?? []
+        }
     })
-    let arr2: any = []
-    bxsDeviceData.value.forEach((item: any) => {
-        item.children.forEach((item2: any) => {
-            arr2.push(item2)
-        })
-    })
-    let arr3: any = []
-    yqfDeviceData.value.forEach((item: any) => {
-        item.children.forEach((item2: any) => {
-            item2.children.forEach((item3: any) => {
-                item3.children.forEach((item4: any) => {
-                    arr3.push(item4)
-                })
-            })
-        })
-    })
-    activeName.value = val
-    if (val == '0') {
-        arr2.map((item: any) => {
-            item.checked = false
-        })
-        emit('changePointType', activeName.value, arr1[0])
-        setTimeout(() => {
-            treeRef1.value?.setCurrentKey(arr1[0]?.id)
-        }, 100)
-    }
-    if (val == '1') {
-        arr1.map((item: any) => {
-            item.checked = false
-        })
-        emit('changePointType', activeName.value, arr2[0])
-        setTimeout(() => {
-            treeRef2.value?.setCurrentKey(arr2[0]?.id)
-        }, 100)
-    }
-    if (val == '2') {
-        arr3.map((item: any) => {
-            item.checked = false
-        })
-        emit('changePointType', activeName.value, arr3[0])
-        setTimeout(() => {
-            treeRef3.value?.setCurrentKey(arr3[0]?.id)
-        }, 100)
-    }
-    // if(activeName.value){
-    //    emit('changePointType', activeName.value)
-    // }
-}
-const onMenuCollapse = () => {
-    menuCollapse.value = !menuCollapse.value
-    proxy.eventBus.emit('cnTreeCollapse', menuCollapse)
-}
-const filterNode = (value: string, data: any, node: any) => {
-    if (!value) return true
-    // return data.name.includes(value)
-    if (data.name) {
-        return chooseNode(value, data, node)
-    }
+
+    zlDevList.value = filterProcess(JSON.parse(JSON.stringify(zlDeviceData.value)))
 }
 
-function filterProcess(nodes: any) {
-    if (process.value == '') {
-        return nodes
-    }
+/** 测点树专用：level=2 按 process 过滤，其余层级保留有子节点或匹配的节点 */
+function filterProcess(nodes: any[]): any[] {
+    if (!process.value) return nodes
+
     return nodes
         .map(node => {
-            // 递归处理子节点
             const children = node.children ? filterProcess(node.children) : []
 
-            // 对于设备层级（level=2），只保留 process 值匹配的节点
             if (node.level === 2) {
                 if (node.process == process.value) {
-                    return {
-                        ...node,
-                        children: children
-                    }
+                    return { ...node, children }
                 }
                 return null
             }
 
-            // 对于其他节点：
-            // 1. 如果有满足条件的子节点则保留
-            // 2. 如果本身 process 值匹配则保留
-            // 3. 如果是叶子节点也保留（监测点通常没有子节点）
-            if (children.length > 0 || node.process == process.value || !node.children || node.children.length === 0) {
-                return {
-                    ...node,
-                    children: children
-                }
+            if (children.length > 0 || node.process == process.value || !node.children?.length) {
+                return { ...node, children }
             }
 
             return null
         })
-        .filter(Boolean) // 移除null节点
+        .filter(Boolean)
 }
 
-// function filterProcess(nodes: any) {
-//     if (process.value == '') {
-//         return nodes
-//     }
-//     return nodes
-//         .map(node => {
-//             // 递归处理子节点
-//             const children = node.children ? filterProcess(node.children) : []
-
-//             // 如果当前节点的process=4，或者有子节点满足条件，则保留当前节点
-
-//             if (node.process == process.value || children.length > 0) {
-//                 return {
-//                     ...node,
-//                     children: node.children
-//                 }
-//             }
-
-//             // 否则过滤掉当前节点
-//             return null
-//         })
-//         .filter(Boolean) // 移除null节点
-// }
-// 过滤父节点 / 子节点 (如果输入的参数是父节点且能匹配，则返回该节点以及其下的所有子节点；如果参数是子节点，则返回该节点的父节点。name是中文字符，enName是英文字符.
-const chooseNode = (value: string, data: any, node: any) => {
-    if (data.name.indexOf(value) !== -1) {
-        return true
-    }
-    const level = node.level
-    // 如果传入的节点本身就是一级节点就不用校验了
-    if (level === 1) {
-        return false
-    }
-    // 先取当前节点的父节点
-    let parentData = node.parent
-    // 遍历当前节点的父节点
-    let index = 0
-    while (index < level - 1) {
-        // 如果匹配到直接返回，此处name值是中文字符，enName是英文字符。判断匹配中英文过滤
-        if (parentData.data.name.indexOf(value) !== -1) {
-            return true
-        }
-        // 否则的话再往上一层做匹配
-        parentData = parentData.parent
-        index++
-    }
-    // 没匹配到返回false
-    return false
+function getActiveTreeRef() {
+    if (treeType.value === '2') return treeRef4.value
+    if (activeName.value === '0') return treeRef1.value
+    if (activeName.value === '1') return treeRef2.value
+    return treeRef3.value
 }
-//治理
-const treeRef1 = ref<InstanceType<typeof ElTree>>()
-//便携式
-const treeRef2 = ref<InstanceType<typeof ElTree>>()
-//在线
-const treeRef3 = ref<InstanceType<typeof ElTree>>()
-// 工程
-const treeRef4 = ref<InstanceType<typeof ElTree>>()
-defineExpose({ treeRef1, treeRef2, treeRef3, treeRef4 })
-onMounted(() => {
-    setTimeout(() => {
-        setActiveName()
-    }, 500)
-})
-const setActiveName = () => {
-    if (zlDeviceData.value.length != 0) {
-        zlDevList.value = filterProcess(JSON.parse(JSON.stringify(zlDeviceData.value)))
-        activeName.value = '0'
-    }
-    if (zlDeviceData.value.length === 0 && bxsDeviceData.value.length != 0) {
-        activeName.value = '1'
-    }
-    if (zlDeviceData.value.length === 0 && bxsDeviceData.value.length === 0) {
-        activeName.value = '2'
-    }
-    if (!zlDeviceData.value && !bxsDeviceData.value) {
-        activeName.value = '2'
-    }
+
+function resolveActiveName() {
+    if (zlDeviceData.value.length) return '0'
+    if (bxsDeviceData.value.length) return '1'
+    if (yqfDeviceData.value.length) return '2'
+    return ''
+}
+
+function selectPointPanel(panelName: string, node?: any) {
+    if (!node) return
+    emit('changePointType', panelName, node)
     nextTick(() => {
-        changeDevice(activeName.value)
+        getActiveTreeRef()?.setCurrentKey(node.id)
     })
 }
-const loading = ref(false)
+
+const changeDevice = (val: CollapseModelValue) => {
+    if (Array.isArray(val) || val == null || val === '') return
+    const panelName = String(val)
+    const { govern, portable, monitor } = collectDeviceLeaves(
+        zlDevList.value,
+        bxsDeviceData.value,
+        yqfDeviceData.value
+    )
+
+    const panelMap: Record<string, { nodes: any[]; clearOthers: any[][] }> = {
+        '0': { nodes: govern, clearOthers: [portable, monitor] },
+        '1': { nodes: portable, clearOthers: [govern, monitor] },
+        '2': { nodes: monitor, clearOthers: [govern, portable] }
+    }
+
+    const panel = panelMap[panelName]
+    if (!panel) return
+
+    panel.clearOthers.forEach(list => list.forEach(item => (item.checked = false)))
+    selectPointPanel(panelName, panel.nodes[0])
+}
+
+const setActiveName = () => {
+    activeName.value = resolveActiveName()
+    if (activeName.value) {
+        nextTick(() => changeDevice(activeName.value))
+    }
+}
+
+watch(
+    () => props.data,
+    val => {
+        if (!val?.length) return
+        splitTreeData(val)
+        if (treeType.value === '1') {
+            nextTick(() => setActiveName())
+        }
+    },
+    { immediate: true, deep: true }
+)
+
+watch(filterText, val => {
+    getActiveTreeRef()?.filter(val)
+})
+
+watch(process, () => {
+    zlDevList.value = filterProcess(JSON.parse(JSON.stringify(zlDeviceData.value)))
+    if (activeName.value === '0') {
+        nextTick(() => changeDevice(activeName.value))
+    }
+})
+
+const onMenuCollapse = () => {
+    menuCollapse.value = !menuCollapse.value
+    proxy.eventBus.emit('cnTreeCollapse', menuCollapse)
+}
+
+const filterNode = (value: string, data: any, node: any): boolean => {
+    if (!value) return true
+    if (!data.name) return false
+    return chooseNode(value, data, node)
+}
+
+const chooseNode = (value: string, data: any, node: any): boolean => {
+    if (data.name.indexOf(value) !== -1) return true
+
+    const level = node.level
+    if (level === 1) return false
+
+    let parentData = node.parent
+    for (let i = 0; i < level - 1; i++) {
+        if (parentData?.data?.name?.indexOf(value) !== -1) return true
+        parentData = parentData.parent
+    }
+    return false
+}
+
 const changeTreeType = (val: string) => {
     loading.value = true
     emit('changeTreeType', val)
-    if (val == '1') {
-        setActiveName()
+    if (val === '1') {
+        nextTick(() => setActiveName())
     }
     setTimeout(() => {
         loading.value = false
-    }, 1000)
+    }, 300)
 }
 </script>