R/to_bnlearn.R

In cia: Learn and Apply Directed Acyclic Graphs for Causal Inference

#' Convert to bnlearn object.
#' 
#' @param x An object that represents a DAG.
#' @returns bn_obj A bn object.
#' 
#' @examples
#' adj <- UniformlySampleDAG(c('A', 'B', 'C'))
#' toBNLearn(adj)
#' 
#' @export
toBNLearn <- function(x) UseMethod('toBNLearn')

#' @export
toBNLearn.default <- function(x) { 
  return(toBNLearn.matrix(x)) 
}

#' @export
toBNLearn.matrix <- function(x) {
  
  names <- colnames(x)
  bn_obj <- bnlearn::empty.graph(names)
  bnlearn::amat(bn_obj) <- x
  
  return(bn_obj)
}

#' @export
toBNLearn.bn <- function(x) {
  return(x)
}

#' @export
toBNLearn.igraph <- function(x) {
  bn_obj <- x |>
    methods::as('matrix') |> 
    toBNLearn.matrix()
  
  return(bn_obj)
}