From 84c78cf0d40a8e167471b2e5ec53121560e412d6 Mon Sep 17 00:00:00 2001
From: zhandouxiaojiji <zhandouxiaojiji@gmail.com>
Date: Tue, 20 Oct 2020 20:15:22 +0800
Subject: [PATCH] draw TreeNode

---
 render-process/Editor/index.tsx |  2 +-
 render-process/index.tsx        | 45 ++++++++++++++++++++++++++++++---
 2 files changed, 42 insertions(+), 5 deletions(-)

diff --git a/render-process/Editor/index.tsx b/render-process/Editor/index.tsx
index b130768..03ccb30 100644
--- a/render-process/Editor/index.tsx
+++ b/render-process/Editor/index.tsx
@@ -91,7 +91,7 @@ export default class Editor extends React.Component<EditorProps, EditorState> {
         type: 'dendrogram', // 布局类型
         direction: 'LR',    // 自左至右布局，可选的有 H / V / LR / RL / TB / BT
         nodeSep: 50,        // 节点之间间距
-        rankSep: 200        // 每个层级之间的间距
+        rankSep: 250        // 每个层级之间的间距
       }
     });
 
diff --git a/render-process/index.tsx b/render-process/index.tsx
index 728e49b..0d6ed55 100644
--- a/render-process/index.tsx
+++ b/render-process/index.tsx
@@ -14,6 +14,7 @@ import G6 from '@antv/g6';
 
 import 'antd/dist/antd.dark.css';
 import './index.css';
+import { ModelConfig } from "@antv/g6/lib/types";
 
 const { Header, Sider, Content, Footer } = Layout;
 const { TabPane } = Tabs;
@@ -24,6 +25,11 @@ interface MainState {
   curPath?: string;
 }
 
+const ICON_MAP = {
+  a: 'https://gw.alipayobjects.com/mdn/rms_8fd2eb/afts/img/A*0HC-SawWYUoAAAAAAAAAAABkARQnAQ',
+  b: 'https://gw.alipayobjects.com/mdn/rms_8fd2eb/afts/img/A*sxK0RJ1UhNkAAAAAAAAAAABkARQnAQ',
+};
+
 export default class Main extends Component {
   state: MainState = {
     workspace: '',
@@ -60,12 +66,43 @@ export default class Main extends Component {
             }
           },
         },
-        // draw(cfg, group) { },
+        draw(cfg, group) {
+          const color = cfg.error ? '#F4664A' : '#30BF78';
+          var size = cfg.size ? cfg.size as number[] : [150, 40];
+          const w = size[0];
+          const h = size[1];
+          const r = 2;
+          const shape = group.addShape('rect', {
+            attrs: {
+              x: -w / 2,
+              y: -h / 2,
+              width: w,
+              height: h,
+              stroke: color,
+              fill: 'white',
+              radius: r,
+            },
+            name: 'main-box',
+            draggable: true,
+          });
+
+          // title text
+          group.addShape('text', {
+            attrs: {
+              textBaseline: 'top',
+              x: -w / 2 + 8,
+              y: -h / 2 + 2,
+              lineHeight: 20,
+              text: cfg.id,
+              fill: 'black',
+            },
+            name: 'title',
+          });
+          return shape;
+        },
 
       },
-      // 继承内置节点类型的名字，例如基类 'single-node'，或 'circle', 'rect' 等
-      // 当不指定该参数则代表不继承任何内置节点类型
-      'rect',
+      'single-node',
     );
   }