// ==UserScript==
// @name         标注校验插件
// @namespace    http://tampermonkey.net/
// @version      7.1
// @description  2D框遮挡校验(Canvas文字) + 3D框可见性校验 + 切帧自动检测
// @match        http://121.37.95.217:8080/pointcloud/*
// @grant        none
// @run-at       document-start
// ==/UserScript==

(function () {
    'use strict';

    const VIEW_NAMES = {
        'front_fisheye': '前鱼眼', 'front_wide': '前广角',
        'back_fisheye': '后鱼眼', 'left_fisheye': '左鱼眼', 'right_fisheye': '右鱼眼'
    };

    let isAutoValidating = false;
    let lastFrameKey = '';
    let autoEnabled = true;
    let panel = null;
    let panelReady = false;

    // ========== 1. Canvas文字拦截（保留hook但已知无效） ==========
    let allCanvasTexts = [];
    let hookInstalled = false;

    function installCanvasHook() {
        if (hookInstalled) return;
        hookInstalled = true;
        const proto = CanvasRenderingContext2D.prototype;
        const origFillText = proto.fillText;
        const origStrokeText = proto.strokeText;

        proto.fillText = function (text, x, y, maxWidth) {
            try {
                if (text && typeof text === 'string' && text.length > 0) {
                    let canvas = this.canvas;
                    let camId = '';
                    if (canvas) {
                        camId = canvas.getAttribute('data-value') || canvas.dataset.value || '';
                        if (!camId) {
                            let parent = canvas.parentElement;
                            let depth = 0;
                            while (parent && !camId && depth < 5) {
                                camId = parent.getAttribute('data-value') || parent.dataset.value || '';
                                parent = parent.parentElement;
                                depth++;
                            }
                        }
                    }
                    allCanvasTexts.push({ text: text, x: Math.round(x), y: Math.round(y), camId: camId || 'unknown' });
                }
            } catch (e) { }
            return origFillText.apply(this, arguments);
        };

        proto.strokeText = function (text, x, y, maxWidth) {
            try {
                if (text && typeof text === 'string' && text.length > 0) {
                    let canvas = this.canvas;
                    let camId = '';
                    if (canvas) {
                        camId = canvas.getAttribute('data-value') || canvas.dataset.value || '';
                        if (!camId) {
                            let parent = canvas.parentElement;
                            let depth = 0;
                            while (parent && !camId && depth < 5) {
                                camId = parent.getAttribute('data-value') || parent.dataset.value || '';
                                parent = parent.parentElement;
                                depth++;
                            }
                        }
                    }
                    allCanvasTexts.push({ text: text, x: Math.round(x), y: Math.round(y), camId: camId || 'unknown' });
                }
            } catch (e) { }
            return origStrokeText.apply(this, arguments);
        };
    }

    installCanvasHook();

    function clearCanvasText() {
        allCanvasTexts = [];
    }

    // ========== 2. Vue Store获取 ==========
    function getVueStore() {
        let el = document.getElementById('app') || document.querySelector('#app');
        if (!el) return null;
        if (el.__vue_app__) {
            try {
                let gp = el.__vue_app__.config.globalProperties;
                if (gp && gp.$store) return gp.$store;
            } catch (e) { }
        }
        if (el.__vue__) {
            try {
                if (el.__vue__.$store) return el.__vue__.$store;
            } catch (e) { }
        }
        return null;
    }

    const VALID_OCCLUSION_CODES = new Set(['10', '11', '12', '13']);

    // ========== 2b. 2D遮挡校验（通过平台getShowAttrs获取显示文字） ==========
    function check2DOcclusion() {
        let store = getVueStore();
        if (!store || !store.state || !store.state.marks) {
            return { errors: [], views: ['前鱼眼', '前广角', '后鱼眼', '左鱼眼', '右鱼眼'] };
        }

        let marks = store.state.marks;
        if (!marks || typeof marks !== 'object') {
            return { errors: [], views: ['前鱼眼', '前广角', '后鱼眼', '左鱼眼', '右鱼眼'] };
        }

        // 获取平台自己的getShowAttrs getter
        let getShowAttrs = null;
        try { getShowAttrs = store.getters['config/getShowAttrs']; } catch (e) { }

        // 获取className getter（获取中文类名）
        let getClassName = null;
        try { getClassName = store.getters['config/className']; } catch (e) { }

        // 获取classId getter
        let getClassId = null;
        try { getClassId = store.getters['marks/classId']; } catch (e) { }

        // 获取attrsDirection
        let attrsDirection = false;
        try { attrsDirection = store.state.status.attrsDirection; } catch (e) { }

        // 找当前帧ID
        let curFrameId = null;
        try {
            curFrameId = store.state.status.curFrameId;
        } catch (e) { }
        if (!curFrameId) {
            // fallback: 找出现次数最多的帧ID
            let frameIds = {};
            for (let catKey in marks) {
                let category = marks[catKey];
                if (!category || typeof category !== 'object') continue;
                for (let instKey in category) {
                    let instance = category[instKey];
                    if (!instance || typeof instance !== 'object') continue;
                    if (instance['3d'] && typeof instance['3d'] === 'object') {
                        for (let fid in instance['3d']) {
                            frameIds[fid] = (frameIds[fid] || 0) + 1;
                        }
                    }
                }
            }
            let maxCount = 0;
            for (let fid in frameIds) {
                if (frameIds[fid] > maxCount) { maxCount = frameIds[fid]; curFrameId = fid; }
            }
        }

        if (!curFrameId) {
            return { errors: [], views: ['前鱼眼', '前广角', '后鱼眼', '左鱼眼', '右鱼眼'] };
        }

        let errors = [];
        let checkedViews = new Set();

        for (let catKey in marks) {
            let category = marks[catKey];
            if (!category || typeof category !== 'object') continue;

            for (let instKey in category) {
                let instance = category[instKey];
                if (!instance || typeof instance !== 'object') continue;

                let d2 = instance['2d'];
                if (!d2 || typeof d2 !== 'object') continue;
                let frame2d = d2[curFrameId];
                if (!frame2d) continue;

                let boxList = [];
                if (Array.isArray(frame2d)) {
                    boxList = frame2d;
                } else if (typeof frame2d === 'object' && frame2d !== null) {
                    for (let k in frame2d) {
                        let v = frame2d[k];
                        if (Array.isArray(v)) boxList = boxList.concat(v);
                        else if (typeof v === 'object' && v !== null) boxList.push(v);
                    }
                }

                for (let i = 0; i < boxList.length; i++) {
                    let box = boxList[i];
                    if (!box || typeof box !== 'object') continue;

                    let camera = box.camera || '';
                    let viewName = VIEW_NAMES[camera] || camera || '未知视角';
                    checkedViews.add(viewName);

                    let class_name = instance.class || instKey;
                    // 尝试获取中文类名
                    let display_name = class_name + '-' + instKey;
                    if (getClassName) {
                        try {
                            let classId = null;
                            if (getClassId) classId = getClassId(instKey);
                            if (!classId && instance.classId) classId = instance.classId;
                            let toolType = box.type || instance.type || 'rect2d';
                            if (classId) {
                                let cn = getClassName(classId, toolType);
                                if (cn) display_name = cn + '-' + instKey;
                            }
                        } catch (e) { }
                    }
                    let attrs = box.attrs;

                    if (!attrs) continue;
                    if (typeof attrs === 'string') { try { attrs = JSON.parse(attrs); } catch (e) { } }

                    // 用平台的getShowAttrs获取显示文字
                    let displayText = null;
                    if (getShowAttrs) {
                        try {
                            let classId = null;
                            if (getClassId) {
                                classId = getClassId(instKey);
                            }
                            let boxType = box.originalType || box.type || instance.type || 'rect2d';
                            let result = getShowAttrs(boxType, classId, attrs, !attrsDirection);
                            if (Array.isArray(result)) {
                                displayText = result.join('\n');
                            } else if (typeof result === 'string') {
                                displayText = result;
                            }
                        } catch (e) {
                            console.log('[校验插件] getShowAttrs调用失败:', e.message, 'instKey:', instKey);
                        }
                    }

                    if (displayText) {
                        // 检查显示文字中"遮挡:"后面是数字还是汉字
                        let occMatch = displayText.match(/遮挡[：:]\s*(.{1,30})/);
                        if (occMatch) {
                            let occValue = occMatch[1].trim();
                            occValue = occValue.split(/截断|运动|车门|可见性/)[0].trim();

                            // 如果遮挡值以数字开头且不含"可见" → 报错
                            if (!occValue.includes('可') && /^\d/.test(occValue)) {
                                errors.push(`[2D] ${viewName} - ${display_name}：遮挡显示为"${occValue}"`);
                            }
                        }
                    } else {
                        // fallback: 直接检查attrs里的遮挡编码
                        let occlusion = null;
                        if (attrs.occlusion) {
                            occlusion = Array.isArray(attrs.occlusion) ? attrs.occlusion[0] : attrs.occlusion;
                        }
                        if (occlusion !== null && occlusion !== undefined && occlusion !== '') {
                            let occlusionStr = String(occlusion);
                            if (!VALID_OCCLUSION_CODES.has(occlusionStr)) {
                                errors.push(`[2D] ${viewName} - ${display_name}：遮挡值"${occlusionStr}"`);
                            }
                        }
                    }
                }
            }
        }

        return { errors, views: Array.from(checkedViews) };
    }

    // ========== 3. 3D可见性校验 ==========
    function check3DVisibility() {
        let labels3D = document.querySelectorAll('.threejs-label');
        let missing = [];
        let count = 0;

        labels3D.forEach(label => {
            let text = (label.textContent || '').trim();
            if (text.length === 0) return;
            if (!text.includes(':') && !text.includes('：')) return;
            count++;
            if (!text.includes('可见性')) {
                missing.push(text.split('\n')[0].trim());
            }
        });

        return { count, missing };
    }

    // ========== 4. 注入CSS ==========
    function injectStyles() {
        if (document.getElementById('chk-styles')) return;
        const style = document.createElement('style');
        style.id = 'chk-styles';
        style.textContent = `
            #chk-panel {
                position: fixed; top: 80px; left: 320px;
                width: 340px; max-height: 70vh;
                background: linear-gradient(145deg, #1a1a2e 0%, #16213e 100%);
                color: #e0e0e0;
                border: 1px solid rgba(0,210,255,0.2);
                border-radius: 12px; z-index: 999999;
                font-family: "Microsoft YaHei", "PingFang SC", sans-serif;
                font-size: 13px;
                box-shadow: 0 8px 32px rgba(0,0,0,0.6), 0 0 0 1px rgba(255,255,255,0.05);
                overflow: hidden; transition: box-shadow 0.2s;
            }
            #chk-panel:hover { box-shadow: 0 8px 32px rgba(0,0,0,0.6), 0 0 20px rgba(0,210,255,0.15); }
            #chk-header {
                background: linear-gradient(135deg, #0f3460 0%, #1a1a40 100%);
                padding: 10px 14px; display: flex; justify-content: space-between;
                align-items: center; cursor: move; user-select: none;
                border-bottom: 1px solid rgba(0,210,255,0.15);
            }
            #chk-header:active { cursor: grabbing; }
            .chk-title {
                font-weight: bold; font-size: 14px; color: #00d2ff;
                text-shadow: 0 0 10px rgba(0,210,255,0.3);
                display: flex; align-items: center; gap: 6px;
            }
            .chk-title-dot {
                width: 8px; height: 8px; background: #00d2ff;
                border-radius: 50%; box-shadow: 0 0 8px #00d2ff;
                animation: chk-pulse 2s ease-in-out infinite;
            }
            @keyframes chk-pulse { 0%, 100% { opacity: 1; } 50% { opacity: 0.4; } }
            .chk-btn-check {
                background: linear-gradient(135deg, #00d2ff 0%, #0288d1 100%);
                color: #fff; border: none; padding: 5px 14px;
                border-radius: 6px; cursor: pointer; font-size: 12px;
                font-weight: bold; transition: all 0.2s;
                box-shadow: 0 2px 8px rgba(0,210,255,0.3);
            }
            .chk-btn-check:hover { transform: translateY(-1px); box-shadow: 0 4px 12px rgba(0,210,255,0.5); }
            .chk-btn-check:active { transform: translateY(0); }
            .chk-btn-check:disabled { opacity: 0.6; cursor: not-allowed; transform: none; }
            .chk-btn-toggle {
                background: rgba(255,255,255,0.08); color: #aaa;
                border: 1px solid rgba(255,255,255,0.1);
                width: 26px; height: 26px; border-radius: 6px;
                cursor: pointer; font-size: 14px;
                display: flex; align-items: center; justify-content: center;
                transition: all 0.2s; margin-left: 6px;
            }
            .chk-btn-toggle:hover { background: rgba(255,255,255,0.15); color: #fff; }
            #chk-content { padding: 12px 14px; max-height: calc(70vh - 50px); overflow-y: auto; }
            #chk-content::-webkit-scrollbar { width: 6px; }
            #chk-content::-webkit-scrollbar-track { background: transparent; }
            #chk-content::-webkit-scrollbar-thumb { background: rgba(0,210,255,0.3); border-radius: 3px; }
            #chk-content::-webkit-scrollbar-thumb:hover { background: rgba(0,210,255,0.5); }
            .chk-stats {
                margin-bottom: 10px; padding: 10px 12px;
                background: rgba(15,52,96,0.4); border-radius: 8px;
                border: 1px solid rgba(255,255,255,0.05);
            }
            .chk-stats-title {
                color: #00d2ff; font-weight: bold; margin-bottom: 6px;
                font-size: 12px; text-transform: uppercase; letter-spacing: 1px;
            }
            .chk-stats-body { color: #8892b0; font-size: 12px; line-height: 1.8; }
            .chk-badge { display: inline-block; padding: 1px 8px; border-radius: 10px; font-size: 11px; font-weight: bold; }
            .chk-badge-ok { background: rgba(78,204,163,0.15); color: #4ecca3; border: 1px solid rgba(78,204,163,0.3); }
            .chk-badge-err { background: rgba(233,69,96,0.15); color: #e94560; border: 1px solid rgba(233,69,96,0.3); }
            .chk-pass { color: #4ecca3; text-align: center; padding: 24px 0; font-size: 15px; font-weight: bold; }
            .chk-warn-header {
                color: #e94560; font-weight: bold; margin: 10px 0 6px;
                font-size: 13px; display: flex; align-items: center; gap: 4px;
            }
            .chk-warn-item {
                background: rgba(233,69,96,0.08); border-left: 3px solid #e94560;
                padding: 8px 10px; margin-bottom: 5px; border-radius: 0 6px 6px 0;
                font-size: 13px; line-height: 1.6; color: #ccc; transition: background 0.2s;
            }
            .chk-warn-item:hover { background: rgba(233,69,96,0.15); }
            .chk-loading { color: #8892b0; text-align: center; padding: 24px 0; }
            @keyframes chk-shake {
                0%, 100% { transform: translateX(0); }
                10%, 30%, 50%, 70%, 90% { transform: translateX(-5px); }
                20%, 40%, 60%, 80% { transform: translateX(5px); }
            }
            .chk-shake { animation: chk-shake 0.5s ease-in-out; }
        `;
        document.head.appendChild(style);
    }

    // ========== 5. 创建面板 ==========
    function createPanel() {
        if (panel && document.body.contains(panel)) return panel;
        injectStyles();

        panel = document.createElement('div');
        panel.id = 'chk-panel';
        panel.innerHTML = `
            <div id="chk-header">
                <span class="chk-title"><span class="chk-title-dot"></span>标注校验</span>
                <div style="display:flex;align-items:center;">
                    <button id="chk-btn-check" class="chk-btn-check">校验当前帧</button>
                    <button id="chk-btn-toggle" class="chk-btn-toggle">&#8722;</button>
                </div>
            </div>
            <div id="chk-content">
                <div class="chk-loading">等待标注数据加载...</div>
            </div>
        `;
        document.body.appendChild(panel);
        document.getElementById('chk-btn-check').addEventListener('click', function () { validate(false); });
        document.getElementById('chk-btn-toggle').addEventListener('click', togglePanel);
        makeDraggable();
        panelReady = true;
        return panel;
    }

    // ========== 6. 拖拽 ==========
    function makeDraggable() {
        const header = document.getElementById('chk-header');
        let isDragging = false, startX = 0, startY = 0, startLeft = 0, startTop = 0;
        header.addEventListener('mousedown', function (e) {
            if (e.target.tagName === 'BUTTON') return;
            isDragging = true; startX = e.clientX; startY = e.clientY;
            const rect = panel.getBoundingClientRect();
            startLeft = rect.left; startTop = rect.top; e.preventDefault();
        });
        document.addEventListener('mousemove', function (e) {
            if (!isDragging) return;
            let newLeft = startLeft + (e.clientX - startX);
            let newTop = startTop + (e.clientY - startY);
            newLeft = Math.max(-260, Math.min(newLeft, window.innerWidth - 80));
            newTop = Math.max(0, Math.min(newTop, window.innerHeight - 50));
            panel.style.left = newLeft + 'px'; panel.style.top = newTop + 'px'; panel.style.right = 'auto';
        });
        document.addEventListener('mouseup', function () { isDragging = false; });
    }

    function togglePanel() {
        const content = document.getElementById('chk-content');
        const btn = document.getElementById('chk-btn-toggle');
        if (content.style.display === 'none') { content.style.display = ''; btn.innerHTML = '&#8722;'; }
        else { content.style.display = 'none'; btn.innerHTML = '&#43;'; }
    }

    // ========== 7. 渲染结果 ==========
    function renderResults(errors, info, isAuto) {
        if (!panelReady) createPanel();
        const content = document.getElementById('chk-content');
        const btn = document.getElementById('chk-btn-check');
        btn.textContent = '校验当前帧'; btn.disabled = false;

        let hasErrors = errors.length > 0;
        let html = '';
        let statsBorder = hasErrors ? 'rgba(233,69,96,0.3)' : 'rgba(78,204,163,0.3)';
        let statsColor = hasErrors ? '#e94560' : '#4ecca3';
        let statusBadge = hasErrors
            ? '<span class="chk-badge chk-badge-err">有异常</span>'
            : '<span class="chk-badge chk-badge-ok">通过</span>';
        if (isAuto) statusBadge += ' <span class="chk-badge chk-badge-ok">自动</span>';

        html += `<div class="chk-stats" style="border-color:${statsBorder};">
            <div class="chk-stats-title" style="color:${statsColor};">校验结果</div>
            <div class="chk-stats-body">状态: ${statusBadge}<br>${info.join('<br>')}</div>
        </div>`;

        if (hasErrors) {
            html += `<div class="chk-warn-header">&#9888; 发现问题（${errors.length}条）</div>`;
            for (let i = 0; i < errors.length; i++) {
                html += `<div class="chk-warn-item">${errors[i]}</div>`;
            }
        } else {
            html += `<div class="chk-pass">&#10003; 校验通过，无问题</div>`;
        }

        content.innerHTML = html;
        if (hasErrors) {
            panel.classList.remove('chk-shake');
            void panel.offsetWidth;
            panel.classList.add('chk-shake');
        }
    }

    // ========== 8. 校验 ==========
    function validate(isAuto) {
        let errors = [];
        let info = [];

        let result2D = check2DOcclusion();
        if (result2D.views.length > 0) {
            info.push(`2D遮挡校验（${result2D.views.join('/')}）`);
        }
        errors = errors.concat(result2D.errors);

        let result3D = check3DVisibility();
        info.push(`3D可见性校验（共${result3D.count}个）`);
        for (let i = 0; i < result3D.missing.length; i++) {
            errors.push(`[3D] ${result3D.missing[i]}：缺少"可见性"字段`);
        }

        renderResults(errors, info, isAuto);
    }

    // ========== 9. 切帧自动检测 ==========
    let observer = new MutationObserver(function () {
        let labels = document.querySelectorAll('.threejs-label');
        let frameKey = '';
        if (labels.length > 0) {
            frameKey = labels.length + '|' + (labels[0].textContent || '').substring(0, 30);
        }
        if (frameKey && frameKey !== lastFrameKey) {
            lastFrameKey = frameKey;
            if (!isAutoValidating && autoEnabled) {
                isAutoValidating = true;
                clearCanvasText();
                setTimeout(function () {
                    validate(true);
                    isAutoValidating = false;
                }, 3000);
            }
        }
    });

    function startObserver() {
        let target = document.getElementById('threejs-label') || document.body;
        observer.observe(target, { childList: true, subtree: true, characterData: true });
        observer.observe(document.body, { childList: true, subtree: true, characterData: true });
    }

    // ========== 10. 初始化 ==========
    function init() {
        let attempts = 0;
        let interval = setInterval(function () {
            attempts++;
            if (document.getElementById('app') || attempts > 60) {
                clearInterval(interval);
                setTimeout(function () {
                    createPanel();
                    startObserver();
                    console.log('[校验插件V7.1] 已加载');
                }, 2000);
            }
        }, 500);
    }

    if (document.readyState === 'loading') {
        document.addEventListener('DOMContentLoaded', init);
    } else {
        init();
    }

    document.addEventListener('keydown', function (e) {
        if (e.ctrlKey && e.shiftKey && e.key === 'V') {
            e.preventDefault();
            clearCanvasText();
            validate(false);
        }
    });

    window.__validate = validate;
    window.__clearCanvasText = clearCanvasText;
    window.__allCanvasTexts = allCanvasTexts;
    window.__dump2D = function () {
        let store = getVueStore();
        if (!store || !store.state || !store.state.marks) { console.log('无Vue Store'); return; }
        let marks = store.state.marks;
        let frameIds = {};
        for (let catKey in marks) {
            let category = marks[catKey];
            for (let instKey in category) {
                let inst = category[instKey];
                if (inst && inst['3d']) {
                    for (let fid in inst['3d']) frameIds[fid] = (frameIds[fid] || 0) + 1;
                }
            }
        }
        let curFrame = Object.keys(frameIds).sort(function (a, b) { return frameIds[b] - frameIds[a]; })[0];
        console.log('当前帧:', curFrame, '帧统计:', frameIds);

        let results = [];
        for (let catKey in marks) {
            let category = marks[catKey];
            for (let instKey in category) {
                let inst = category[instKey];
                if (!inst || !inst['2d']) continue;
                let d2 = inst['2d'];
                let frame2d = d2[curFrame];
                if (!frame2d) continue;
                let boxList = [];
                if (Array.isArray(frame2d)) boxList = frame2d;
                else { for (let k in frame2d) { let v = frame2d[k]; if (Array.isArray(v)) boxList = boxList.concat(v); else if (typeof v === 'object' && v) boxList.push(v); } }
                for (let i = 0; i < boxList.length; i++) {
                    let box = boxList[i];
                    if (!box) continue;
                    let cam = box.camera || '?';
                    let viewName = VIEW_NAMES[cam] || cam;
                    let attrs = box.attrs;
                    let occ = '无attrs';
                    if (attrs) {
                        if (typeof attrs === 'string') { try { attrs = JSON.parse(attrs); } catch (e) { } }
                        if (attrs.occlusion) {
                            occ = Array.isArray(attrs.occlusion) ? attrs.occlusion[0] : attrs.occlusion;
                        } else { occ = '无occlusion字段'; }
                    }
                    results.push({
                        view: viewName,
                        class: inst.class || instKey,
                        inst: instKey,
                        occlusion: occ,
                        hasAttrs: !!box.attrs,
                        allAttrKeys: box.attrs ? Object.keys(box.attrs) : []
                    });
                }
            }
        }
        console.table(results);
        console.log('共' + results.length + '个2D框');
    };
})();