How to keep edge colours after delete_edge() in DiagrammeR

时间:2015-10-30 21:54:49

标签: r graph diagrammer

I've created a graph with DiagrammeR by first setting up a basic graph and then adding nodes and edges. I've set up the edge color attribute in the edge_attrs() inside create_graph().

If I add a new edge without the color attribute it uses, as expected, the pre-defined colour. However, if I delete one of the edges with delete_edge() the general edge attribute color disappears for all edges. Since the graph$edges_df doesn't include any color information the graph defaults to black.

Is there a way to add the colour of the edge to graph$edges_df when using add_node()?

The only way I've thought would work is to add the node without the edge and then add the edge individually with add_edge().

Here a reproducible example:

library(DiagrammeR)
library(magrittr)

nodes <-
  create_nodes(
    nodes = "a",
    label = "A",
    type = "lower",
    style = "filled",
    shape = "circle",
    color = "orange",
    x = 5,
    y = 4
  )

graph_1 <-
  create_graph(
    nodes_df = nodes,
    graph_attrs = c(
      "layout = neato"
      ),
   edge_attrs = c(
      "relationship = requires",
      "arrowhead = normal",
      "color = 'lightgrey'"
      )
  )

render_graph(graph_1)

graph_1 %>%
  add_node(
    node = "b",
    from = "a",
    label = "B",
    style = "filled",
    shape = "circle",
    color = "orange",
    x = 5,
    y = 3
  ) ->
  graph_2

render_graph(graph_2)

new_edges <- create_edges(
  from = "a",
  to = "a"
)

graph_2 %>%
  add_edges(
    edges_df = new_edges
  ) ->
  graph_3

render_graph(graph_3)

graph_3 %>%
  delete_edge(
    to = "a",
    from = "a"
  ) ->
  graph_4

render_graph(graph_4)

1 个答案:

答案 0 :(得分:0)

这是我发现的最佳解决方案。简单地忽略一般属性并依赖于edges_df。

library(DiagrammeR)
library(magrittr)

nodes <-
  create_nodes(
    nodes = "a",
    label = "A",
    type = "lower",
    style = "filled",
    shape = "circle",
    color = "orange",
    x = 5,
    y = 4
  )

graph_1 <-
  create_graph(
    nodes_df = nodes,
    graph_attrs = c(
      "layout = neato"
    )
  )

render_graph(graph_1)

graph_1 %>%
  add_node(
    node = "b",
    label = "B",
    style = "filled",
    shape = "circle",
    color = "orange",
    x = 5,
    y = 3
  ) %>%
  add_edges(
    edges_df = create_edges(
      from = "a",
      to = "b",
      color = "lightgrey"
    )
  ) ->
  graph_2

render_graph(graph_2)

graph_2 %>%
  add_edges(
    edges_df = create_edges(
      from = "a",
      to = "a",
      color = "lightgrey"
    )
  ) ->
  graph_3

render_graph(graph_3)

graph_3 %>%
  delete_edge(
    to = "b",
    from = "a"
  ) ->
  graph_4

render_graph(graph_4)

编辑: 我现在查看来自rich-iannone的delete_edge()代码。它基本上基于原始的edges_df,nodes_df,directed和graph_attrs重新构建图形,但它不包括node_attrs或edge_attrs,因此它们返回到默认值。 简单的解决方案是构建一个考虑到这一点的新函数,但如果与create_graph存在冲突,我会忽略。