feat: add some graph options

antvis · Jun 17, 2024 · faefa2b · faefa2b
1 parent 7dc2470
commit faefa2b
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/packages/core/src/components/Graph.tsx b/packages/core/src/components/Graph.tsx
@@ -28,6 +28,9 @@ const XFlowGraph = (props: GraphOptions) => {
     restrict,
     restrictOptions,
     connectionOptions,
+    onPortRendered,
+    onEdgeLabelRendered,
+    createCellView,
     selectOptions,
     keyboardOptions,
     scroller,
@@ -67,6 +70,9 @@ const XFlowGraph = (props: GraphOptions) => {
         magnetAvailable: magnetAvailableHighlightOptions,
         magnetAdsorbed: magnetAdsorbedHighlightOptions,
       },
+      onPortRendered,
+      onEdgeLabelRendered,
+      createCellView,
     });
 
     g.use(new Selection({ enabled: true, ...selectOptions }));

diff --git a/packages/core/src/types/index.ts b/packages/core/src/types/index.ts
@@ -37,6 +37,14 @@ export interface GraphOptions {
 
   // node & edge
   connectionEdgeOptions?: Partial<EdgeOptions>;
+  onPortRendered?: (args: Options.OnPortRenderedArgs) => void;
+  onEdgeLabelRendered?: (args: Options.OnEdgeLabelRenderedArgs) => void;
+
+  // cell
+  createCellView?: (
+    this: Graph,
+    cell: Cell,
+  ) => typeof CellView | (new (...args: any[]) => CellView) | null | undefined;
 
   // interactive
   embedable?: boolean;