R/vis2.R

In recon-icm/q1: Functions and Data Related to Project RECON at ICM UW

#' Get data from downloaded XLSX from IDI2
#' 
#' By default retrieves all sheets from interviews and returns a list of data frames.
#'
#' @param file character, XLSX file with data
#' @param sheet numeric/character, index number of name of the sheet(s) to be loaded
#' @param ... other arguments passed to \code{\link{readWorksheet}}
#'
#' @return Data frame or list thereof.
#' @export
get_data2 <- function(file, sheet=NULL, ...) {
  wb <- XLConnect::loadWorkbook(filename=file)
  sheet_names <- XLConnect::getSheets(wb)
  sh <- grep("^DC|^WF", sheet_names)
  # What to get
  if( is.null(sheet) ) {
    toget <- sh
  } else {
    stopifnot(is.character(sheet))
    stopifnot(all(sheet %in% sheet_names))
    toget <- which(sheet_names %in% sheet)
  }
  # Get
  if( length(toget) == 1) {
    rval <- XLConnect::readWorksheet(wb, sheet=toget, ...)
    attr(rval, "idi_id") <- sheet_names[toget]
    return(rval)
  } else {
    rval <- lapply(toget, function(i)
      structure(
        XLConnect::readWorksheet(wb, sheet=i, ...),
        idi_id = sheet_names[i]
      ) )
    names(rval) <- sheet_names[toget]
    return(rval)
  }
}



#' Create network object
#'
#' @param d data frame
#'
#' @return igraph object
#' @export
make_net2 <- function(d) {
  # Vertex DB
  vdb <- data.frame(
    id=d$Nr.wezła.2,
    grupa = gsub("[^0-9,]", "", d$Przynależność.do.grup..na.podstawie.zdjęcie..grupa.może.być.też.dwuosobowa.),
    stringsAsFactors = FALSE
  )
  # Edge DB
  edb <- rbind(
    data.frame(
      ego = d$Nr.wezła.1,
      alter = d$Nr.wezła.2,
      res = gsub("[^0-9,]", "", d$Kategoria.zasobów.pzekazanych),
      stringsAsFactors = FALSE
    ),
    data.frame(
      ego = d$Nr.wezła.2,
      alter = d$Nr.wezła.1,
      res = gsub("[^0-9,]", "", d$Kategorie.zasobów.otrzymanych),
      stringsAsFactors = FALSE
    )
  )
  edb <- subset(edb, ego != alter)
  
  g <- igraph::graph_from_data_frame(edb, directed = TRUE, vertices = vdb)
  g$idi_id <- attr(d, "idi_id")
  g
}



#' Plot IDI2 network
#'
#' @param g igraph object
#' @param show_groups numeric, ids of groups to display. Alternatively, NULL for all, NA for none, 
#' @param ... passed to plot.igraph
#'
#' @return nothing
#' @export
vis_net2 <- function(g, show_groups=NULL, ...) {
  plot(
    g,
    edge.curved=0.2,
    edge.arrow.size=0.5,
    edge.label=E(g)$res,
    edge.label.cex=0.7,
    main = g$idi_id,
    ...
  )
}

# library(ggnetwork)

# g %>% asNetwork() %>% ggplot(aes(x=x, y=y, xend=xend, yend=yend)) +
#   geom_nodes() +
#   geom_edges(curvature=0.1) +
#   geom_edgetext_repel(aes(label=res), box.padding = unit(1, "lines")) +
#   theme_minimal()

if(FALSE) {
  xlsx <- system.file("exdata", "resources.xlsx", package="recon")
  dlist <- get_data2(xlsx)
  glist <- lapply(dlist, make_net2)
  pdf("sieci2.pdf")
  for(d in dlist) {
    plot(make_net2(d))
  }
  dev.off()
}