R/BetweenessAttack.R

In JonnoB/PowerGridNetworking: Power-grid network simulation in R

BetweenessAttack <-function(g, Target = "Nodes", Number = 1){
  #Attacks the nodes based on their betweenness, as long as the the graph has not weighted edges called "weights"
    #the betweeness is unweighted
  #g: a network an igraph object
  #Target: Either nodes or edges
  #Number the total number of nodes/edges to remove

  if(Target == "Nodes"){
    df <- igraph::as_data_frame(g, what = "vertices") %>%
      dplyr::mutate(metric = igraph::betweenness(g)) %>%
      dplyr::arrange(dplyr::desc(metric))
    
    Out <- df$name[1:Number]
    
  } else if (Target =="Edges") {
    df <- igraph::as_data_frame(g, what = "edges") %>%
      dplyr::mutate(metric = igraph::edge_betweenness(g)) %>%
      dplyr::arrange(dplyr::desc(metric))
    
    Out <- df$name[1:Number]
  } else {
    
    stop("Target must be either 'Nodes' or 'Edges'")
  }
  
  return(Out)
}