From e8a242f8d1c40b495ecfc3293898bf18ef40a2e5 Mon Sep 17 00:00:00 2001
From: maxmon <541182180@qq.com>
Date: Fri, 9 Jul 2021 22:48:54 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E4=BC=98=E5=85=88=E4=BB=8E10=E7=A7=8D?=
 =?UTF-8?q?=E5=88=9D=E5=A7=8B=E9=A2=9C=E8=89=B2=E4=B8=AD=E8=8E=B7=E5=8F=96?=
 =?UTF-8?q?=E9=9A=8F=E6=9C=BA=E9=A2=9C=E8=89=B2?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 fe/src/components/NER/index.vue  | 12 +++-------
 fe/src/components/home/index.vue | 12 +++-------
 fe/src/js/color.js               | 40 ++++++++++++++++++++++++++++++++
 3 files changed, 46 insertions(+), 18 deletions(-)
 create mode 100644 fe/src/js/color.js

diff --git a/fe/src/components/NER/index.vue b/fe/src/components/NER/index.vue
index 4507780..638f301 100644
--- a/fe/src/components/NER/index.vue
+++ b/fe/src/components/NER/index.vue
@@ -101,16 +101,10 @@
 </template>
 
 <script>
+import { getColor } from '../../js/color.js'
+
 // 是否是单机版
 const isLocal = false
-function getColor () {
-  const idxs = '0123456789abcdef'
-  let color = '#'
-  for (let i = 0; i < 6; i += 1) {
-    color += idxs[Math.random() * idxs.length | 0]
-  }
-  return color
-}
 
 function get (url, cb) {
   query('GET', url, '', cb)
@@ -376,7 +370,7 @@ export default {
       if (!newType) return false
       if (that.types[newType]) return false
       that.$set(that.types, newType, {
-        color: getColor()
+        color: getColor(that.types)
       })
       that.typeList.push(newType)
       updateType2Server(that.projectName, that.typeList, that.types)
diff --git a/fe/src/components/home/index.vue b/fe/src/components/home/index.vue
index 578f0ac..d2a40d2 100644
--- a/fe/src/components/home/index.vue
+++ b/fe/src/components/home/index.vue
@@ -61,14 +61,8 @@
 </template>
 
 <script>
-function getColor () {
-  const idxs = '0123456789abcdef'
-  let color = '#'
-  for (let i = 0; i < 6; i += 1) {
-    color += idxs[Math.random() * idxs.length | 0]
-  }
-  return color
-}
+import { getColor } from '../../js/color'
+
 function get (url, cb) {
   query('GET', url, '', cb)
 }
@@ -197,7 +191,7 @@ export default {
       if (!newType) return false
       if (that.types[newType]) return false
       that.$set(that.types, newType, {
-        color: getColor()
+        color: getColor(that.types)
       })
       that.typeList.push(newType)
       console.log(that.typeList)
diff --git a/fe/src/js/color.js b/fe/src/js/color.js
new file mode 100644
index 0000000..92b0525
--- /dev/null
+++ b/fe/src/js/color.js
@@ -0,0 +1,40 @@
+const COLORS = [
+  '#e12d2d', // 红
+  '#f5a314', // 橙
+  '#fff838', // 黄
+  '#72ca2b', // 绿
+  '#529dff', // 蓝
+  '#647cf7', // 靛
+  '#a855ec', // 紫
+  '#eb98de', // 粉
+  '#c08c8e', // 褐
+  '#d2ff8f' // 青
+]
+
+function getColor (types = {}) {
+  const alreadyColorDic = {}
+  Object.keys(types).forEach(type => {
+    const { color } = types[type]
+    alreadyColorDic[color] = true
+  })
+  const remainingColors = []
+  COLORS.forEach(color => {
+    if (!alreadyColorDic[color]) remainingColors.push(color)
+  })
+  if (remainingColors.length) {
+    // 优先选用未选择的初始颜色
+    const idx = Math.random() * remainingColors.length | 0
+    return remainingColors[idx]
+  } else {
+    const idxs = '0123456789abcdef'
+    let color = '#'
+    for (let i = 0; i < 6; i += 1) {
+      color += idxs[Math.random() * idxs.length | 0]
+    }
+    return color
+  }
+}
+
+export {
+  getColor
+}