data-raw/toydata.R

In fionarhuang/TreeHeatmap: A package to plot heatmaps at different levels of a tree

## code to prepare `toydata` dataset goes here

# a random tree: 10 leaves
set.seed(1)
tinyTree <- ape::rtree(10)
tinyTree$node.label <- paste0("N", 11:19)
tinyTree$tip.label <- paste0("N", 1:10)

# a random count matrix:
p1 <- c(rep(0.1/3, 3), rep(0.1, 4), rep(0.4/2, 2), 0.1)
p2 <- c(rep(0.4/3, 3), rep(0.1, 4), rep(0.1/2, 2), 0.1)

aa <- cbind(rmultinom(n = 4, size = 50, prob = p1),
            rmultinom(n = 4, size =50, prob = p2))
colnames(aa) <- paste("S", 1:8, sep = "")
rownames(aa) <- tinyTree$tip.label
aa <- aa[, sample(1:ncol(aa), ncol(aa), replace = FALSE)]
# # a new count matrix: aggregate some rows
# bb <- rbind(colSums(aa[1:3, ]),
#             aa[4:7, ],
#             colSums(aa[8:9, ]),
#             aa[10, ])
# rownames(bb) <- c("N14", paste0("N", 4:7), "N19", "N10")
bb <- rbind(count[1, ],
            colSums(count[2:3, ]),
            count[4:6, ],
            colSums(count[7:9, ]),
            count[10, ])
rownames(bb) <- paste0("N", c(1, 15, 4:6, 18, 10))
toydata <- list(toytree = tinyTree,
                count = aa,
                count_agg = bb)


usethis::use_data(toydata, overwrite = TRUE)