R/count_pos.R

In OTrecod: Data Fusion using Optimal Transportation Theory

count_pos <- function(n, n_grp) {

  if ((round(n) != n) || (round(n_grp) != n_grp)) {
    stop("No decimal values allowed for n and n_grp")
  } else {}


  max_grp <- n - (n_grp - 1)

  BDD2 <- rep(rep(1:max_grp, each = max_grp^(n_grp - 1)), max_grp^0)

  if (n_grp > 1) {
    for (k in 1:(n_grp - 1)) {
      BDD2 <- cbind(BDD2, rep(rep(1:max_grp, each = max_grp^(n_grp - 1 - k)), max_grp^k))
    }

    recup <- list()
    for (i in 1:nrow(BDD2)) {
      recup[[i]] <- as.vector(BDD2[i, ])
    }

    recup2 <- unique(lapply(recup, sort))
    indic <- sapply(recup2, sum)
    recup3 <- recup2[indic == n]
  } else {
    recup3 <- BDD2[BDD2 == n]
  }

  return(recup3)
}