☰

Predecision Label

Decision label. Try to take over the world!

目前為 2024-10-16 提交的版本，檢視 最新版本。

  1. // ==UserScript==
  2. // @name         Predecision Label
  3. // @namespace    http://baidu.com/
  4. // @version      0.4.4
  5. // @description  Decision label. Try to take over the world!
  6. // @author       You
  7. // @match        http://ov.baidu-int.com/*
  8. // @match        http://yf.baidu-int.com/*
  9. // @icon         https://www.google.com/s2/favicons?sz=64&domain=baidu-int.com
  10. // @grant        none
  11. // @license MIT
  12. // ==/UserScript==
  13.  
  14. (function() {
  15.     'use strict';
  16.     let label_list = [];
  17.     let cur_label = {};
  18.     let ads_id = -1;
  19.     let start_pre_seq = -1;
  20.     let end_pre_seq = -1;
  21.     let cur_pre_seq = -1;
  22.     let cur_obs_id = -1
  23.     let is_label_panel_show = false;
  24.     let old_info;
  25.     var initialMouseX, initialMouseY, initialElementX, initialElementY;
  26.  
  27.     function set_cur_info() {
  28.         var _ads_id = parseInt($("section").children('p').children('span')[1].textContent);
  29.         var _cur_pre_seq = parseInt($(".seq-wrap").text().replace('PlanSeq:', ""));
  30.         var _cur_obs_id = parseInt($("div.viz-detail-text-wrapper > span").text().replace('id:', ""));
  31.         if (window.location.host == 'yf.baidu-int.com') {
  32.             _ads_id = getUrlParam("ads_id");
  33.             if (_ads_id != null) {
  34.                 _ads_id = parseInt(_ads_id);
  35.             } else {
  36.                 _ads_id = -1;
  37.             }
  38.             var s = $('#dreamview-container > div > div > div.pannel.react-draggable > div.context > div.dashboard.inner-box.scrollHeightL > table > tbody > tr:nth-child(3) > td:nth-child(2) > p:nth-child(3) > span')[0];
  39.             _cur_pre_seq = parseInt(s.textContent.replace('PRE:', ""));
  40.         }
  41.         ads_id = isNaN(_ads_id)? ads_id: _ads_id;
  42.         cur_pre_seq = isNaN(_cur_pre_seq)? cur_pre_seq: _cur_pre_seq;
  43.         cur_obs_id = isNaN(_cur_obs_id)? cur_obs_id: _cur_obs_id;
  44.         $('#ads_id').text(ads_id);
  45.         $('#seq_num').text(cur_pre_seq);
  46.         $('#obs_id').text(cur_obs_id);
  47.         cur_label.ads_id = ads_id;
  48.         console.log(ads_id);
  49.         console.log(cur_pre_seq);
  50.         console.log(cur_obs_id);
  51.     }
  52.  
  53.     function getUrlParam(name)
  54.     {
  55.         var reg = new RegExp("(^|&)"+ name +"=([^&]*)(&|$)");
  56.         var value = window.location.search.substr(1).match(reg);
  57.         if (value != null) return unescape(value[2]); return null;
  58.     }
  59.  
  60.     function setStartSeq() {
  61.         set_cur_info();
  62.         cur_label.start_seq = cur_pre_seq;
  63.         console.log("set start seq", cur_label.start_seq);
  64.     }
  65.     function setEndSeq() {
  66.         set_cur_info();
  67.         cur_label.end_seq = cur_pre_seq;
  68.         console.log("set end seq", cur_label.end_seq);
  69.     }
  70. //<textarea id="decision_label_show_list" placeholder="" rows="6" style="width:100%;background-color: #292f42;padding: 10px;max-width: 100%;max-height: 100%;line-height: 1.5;border-radius: 5px;border: 1px solid #ccc;"></textarea>
  71.     function showLabelList() {
  72.         $('#decision_label_show_list').empty();
  73.         for (let i = 0; i < label_list.length; i++){
  74.             var label = label_list[i];
  75.             var str = label.ads_id + '\t' + label.obs_id + '\t' + label.start_seq + '\t' + label.end_seq + '\t' + label.dec_label;
  76.             var text_ele = $('<textarea>', {
  77.                 id: `label${i}`,
  78.                 text: str,
  79.                 placeholder: "",
  80.                 rows: "1",
  81.                 style: "width:80%;background-color: #008844;margin-left: 5px;margin-right:5px;margin-top: 5px;"
  82.             });
  83.             $(`label${i}`).on('input', function(){
  84.                 this.styel.height = 'auto';
  85.                 this.style.height = (this.scrollHeight) + 'px';
  86.             });
  87.             var button_ele = $("<button>", {
  88.                 text: "删除",
  89.                 style: "background-color: #008844;margin-left: 10px;margin-right:5px;margin-top: 5px;"
  90.             });
  91.             button_ele.click(function(){
  92.                 if (i >= 0 && i < label_list.length) {
  93.                     label_list.splice(i, 1);
  94.                     console.log("删除" + toString(i));
  95.                     showLabelList();
  96.                 }
  97.             });
  98.             var div = $('<div>', {
  99.                 style: "display: flex;"
  100.             })
  101.             // div.append(text_ele);
  102.             div.append(text_ele).append(button_ele);
  103.             $('#decision_label_show_list').append(div);
  104.             console.log("add show string", str);
  105.         }
  106.     }
  107.  
  108.     function checkOverlaps(labels, cur_label) {
  109.         for (let i = 0; i < labels.length; i++) {
  110.           var label = labels[i];
  111.           if (cur_label.obs_id != label.obs_id){
  112.             continue;
  113.           }
  114.             // 检查重叠：如果一个区间的开始在另一个区间的开始和结束之间，或者反之亦然
  115.             if ((cur_label.start_seq < label.end_seq && cur_label.end_seq > label.start_seq) ||
  116.                 (label.start_seq < cur_label.end_seq && label.end_seq > cur_label.start_seq)) {
  117.                     if (cur_label.label != label.dec_label) {
  118.                         console.log(`障碍物${cur_label.obs_id}与第${i}个标签冲突`);
  119.                         alert(`障碍物${cur_label.obs_id}与第${i}个标签冲突`);
  120.                         return true;
  121.                     } else {
  122.                         console.log(`障碍物${label.obs_id}区间重复标注`);
  123.                         console.warn();(`障碍物${label.obs_id}区间重复标注`);
  124.                         return false;
  125.                     }
  126.             }
  127.         }
  128.         return false; // 没有重叠
  129.       }
  130.  
  131.     function onLabel(dec_label){
  132.         $('#label_btn_' + dec_label.toString()).click(function() {
  133.             // 添加类以实现按钮变亮效果
  134.             $('#label_btn_' + dec_label.toString()).addClass('clicked');
  135.  
  136.             // 在 300 毫秒后移除类以恢复按钮原始状态
  137.             setTimeout(function() {
  138.                 $('#label_btn_' + dec_label.toString()).removeClass('clicked');
  139.             }, 300);
  140.         });
  141.         if (isNaN(ads_id) || isNaN(cur_obs_id) || isNaN(cur_label.start_seq) || isNaN(cur_label.end_seq) || isNaN(dec_label)) {
  142.             return;
  143.         }
  144.         if (ads_id <= 0 && getUrlParam("ads_id") != null) {
  145.             alert("ads id 不合法:" + ads_id.toString());
  146.             return;
  147.         }
  148.         if (cur_obs_id <= 0) {
  149.             alert("obs id 不合法:" + cur_obs_id.toString());
  150.             return;
  151.         }
  152.         if (dec_label < 0) {
  153.             alert("label 不合法" + dec_label.toString());
  154.             return;
  155.         }
  156.         cur_label.ads_id = ads_id;
  157.         cur_label.obs_id = cur_obs_id;
  158.         cur_label.dec_label = dec_label;
  159.         console.log('add decision label to list', cur_label);
  160.         if (checkOverlaps(label_list, cur_label)) {
  161.             return;
  162.         }
  163.         label_list.push(cur_label);
  164.         showLabelList();
  165.         cur_label = {};
  166.     }
  167.  
  168.     function showDecisionLabelPanel() {
  169.         $("#decision_label_panel").css('z-index', -1 * $("#decision_label_panel").css('z-index'));
  170.         console.log("move decision label panel to z index ", $("#decision_label_panel").css('z-index'));
  171.     }
  172.     // Your code here...
  173.     var label_tab = `
  174.         <button type="button" onclick="showDecisionLabelPanel()" class="ant-btn ant-btn-default ant-btn-sm" style="margin-right: 10px;">
  175.             <div class="ant-space ant-space-horizontal ant-space-align-center" style="gap: 8px;">
  176.                 <div class="ant-space-item">
  177.                     决策标注
  178.                 </div>
  179.             </div>
  180.         </button>
  181.         `;
  182.  
  183.     var label_panel_html =`
  184. <div class="container-fluid" id="decision_label_panel" style="position: absolute; top: 55%; right: 20px;width: 500px;height:30%;background-color: #292f42;z-index: -9999;">
  185.     <div class="col-md-2 column" style="float: center;">
  186.         </div>
  187. <div class="row clearfix">
  188.     <div class="col-md-2 column" style="float: center;">
  189.         <div class="row clearfix">
  190.             <label>
  191.             ads_id:
  192.                 </label>
  193. <label id='ads_id'>
  194.     -1
  195. </label>
  196. <label>
  197.     obs_id:
  198.         </label>
  199. <label id='obs_id'>
  200.     -1
  201. </label>
  202. <label>
  203.     seq_num:
  204.         </label>
  205. <label id='seq_num'>
  206.     -1
  207. </label>
  208. </div>
  209. </div>
  210. <div class="col-md-2 column">
  211.     <div class="row clearfix" style="float: center;">
  212.         <button style="width:45%;margin-left: 3%;background-color: red;transition: background-color 0.3s ease;" onclick="setStartSeq()">
  213.             seq_start
  214. </button>
  215. <button style="width:45%;margin-left: 2%;background-color: green;" onclick="setEndSeq()">
  216.     seq_end
  217. </button>
  218. </div>
  219. <br>
  220.     <div class="btn-group" style="margin-left: 2%">
  221. <button style="background-color: #8C0044;margin-left: 5px;margin-top: 5px;" id="label_btn_5" type="button" value="5" onclick="onLabel(5)">
  222.     Yield(5)
  223. </button>
  224. <button style="background-color: #886600;margin-left: 5px;margin-top: 5px;" id="label_btn_2" type="button" value="2" onclick="onLabel(2)">
  225.     Overtake(2)
  226. </button>
  227. </div>
  228. <br>
  229.     <div>
  230.     <div id="decision_label_show_list" style="width:100%">
  231.     </div>
  232.     </div>
  233. </div>
  234. </div>
  235. </div>
  236. </div>
  237.  
  238. `;
  239.     function clearLabel() {
  240.         return;
  241.         $('#decision_label_show_list').val('');
  242.         label_list = [];
  243.     }
  244.  
  245.  
  246.       function onMouseMove(event) {
  247.         // 计算鼠标偏移量
  248.         var offsetX = event.clientX - initialMouseX;
  249.         var offsetY = event.clientY - initialMouseY;
  250.  
  251.         // 将偏移量应用于元素的位置
  252.         $('#decision_label_panel').css({
  253.           left: initialElementX + offsetX + 'px',
  254.           top: initialElementY + offsetY + 'px'
  255.         });
  256.       }
  257.  
  258.       // 鼠标释放事件处理程序
  259.       function onMouseUp() {
  260.         // 移除事件监听器
  261.         $(document).off('mousemove', onMouseMove);
  262.         $(document).off('mouseup', onMouseUp);
  263.       }
  264.  
  265.     window.onMouseMove = onMouseMove;
  266.     window.onMouseUp = onMouseUp
  267.     window.set_cur_info = set_cur_info;
  268.     window.showDecisionLabelPanel = showDecisionLabelPanel;
  269.     window.setStartSeq = setStartSeq;
  270.     window.setEndSeq = setEndSeq;
  271.     window.onLabel = onLabel;
  272.     window.clearLabel = clearLabel;
  273.     window.showLabelList = showLabelList;
  274.  
  275.     setTimeout(function() {
  276.         console.log("3s time out");
  277.         $(".ov-header-operations").prepend(label_tab);
  278.         $("body").append(label_panel_html);
  279.         $('#decision_label_panel').mousedown(function(event) {
  280.             var elementTop = $('#decision_label_panel').offset().top;
  281.             var mouseY = event.clientY;
  282.             console.log('elementTop', elementTop, 'mouseY', mouseY);
  283.             if (mouseY > elementTop + 10) {
  284.                 return; // 如果不在顶部，则忽略拖动
  285.             }
  286.             // 记录鼠标位置和元素初始位置
  287.             initialMouseX = event.clientX;
  288.             initialMouseY = event.clientY;
  289.             initialElementX = $('#decision_label_panel').offset().left;
  290.             initialElementY = $('#decision_label_panel').offset().top;
  291.             console.log("initialMouseX", initialMouseX);
  292.  
  293.             // 添加事件监听器
  294.             $(document).mousemove(onMouseMove);
  295.             $(document).mouseup(onMouseUp);
  296.         });
  297.     }, 3000); // 等待3秒
  298.  
  299. })();