annotate PanExplorer_workflow/R/heatmap.R @ 1:032f6b3806a3 draft

Uploaded
author dereeper
date Thu, 30 May 2024 11:16:08 +0000
parents
children
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
1
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
1 #!/usr/local/R-4.1.2/bin/R
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
2
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
3
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
4 library(dendextend)
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
5 library("optparse")
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
6 library(reshape2)
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
7 library(ape)
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
8
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
9
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
10 #args = commandArgs(trailingOnly=TRUE)
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
11
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
12 option_list = list(
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
13 make_option(c("-f", "--file"), type="character", default=NULL,
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
14 help="dataset file name", metavar="character"),
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
15 make_option(c("-o", "--out"), type="character", default="out.txt",
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
16 help="output file name [default= %default]", metavar="character")
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
17 );
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
18 opt_parser = OptionParser(option_list=option_list);
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
19 opt = parse_args(opt_parser);
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
20
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
21 if (is.null(opt$file)){
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
22 print_help(opt_parser)
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
23 stop("At least one argument must be supplied (input file).\n", call.=FALSE)
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
24 }
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
25
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
26 if (is.null(opt$out)){
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
27 print_help(opt_parser)
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
28 stop("At least one argument must be supplied (out file).\n", call.=FALSE)
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
29 }
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
30
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
31 #svglite(opt$out,width = 31, height = 28)
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
32 pdf(opt$out,width = 31,height = 28)
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
33
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
34 mydata <- read.table(opt$file, header=TRUE,sep="\t", row.names="Gene")
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
35
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
36 iris <- mydata
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
37
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
38 #dend_r <- iris %>% dist(method = "man") %>% hclust(method = "ward.D") %>% as.dendrogram %>% ladderize
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
39 #dend_r <- iris %>% dist(method = "man") %>% hclust(method = "com") %>% as.dendrogram %>% ladderize
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
40
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
41 #dend_c <- t(iris) %>% dist(method = "man") %>% hclust(method = "com") %>% as.dendrogram %>% ladderize
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
42 #dend_c <- t(iris) %>% dist(method = "man") %>% hclust(method = "ward.D") %>% as.dendrogram %>% ladderize
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
43
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
44 #dend_c
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
45
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
46
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
47 distance_matrix = t(iris) %>% dist(method = "man")
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
48 h = distance_matrix %>% hclust(method = "ward.D")
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
49
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
50 tree = as.phylo(h)
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
51
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
52
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
53 dend_c = h %>% as.dendrogram
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
54
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
55
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
56 write.tree(tree, file=paste(opt$out, "distance_matrix.hclust.newick", sep="."))
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
57
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
58 dist_df <- melt(as.matrix(distance_matrix), varnames = c("row", "col"))
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
59 write.table(
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
60 dist_df,
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
61 paste(opt$out, "distance_matrix.txt", sep=".")
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
62 )
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
63
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
64 mat <- as.matrix(t(iris-1))
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
65 out <- gplots::heatmap.2(mat,
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
66 main = "",
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
67 scale="none",
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
68 srtCol=NULL,
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
69 Rowv = dend_c,
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
70 #Colv = dend_r,
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
71 key = FALSE,
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
72 margins =c(20,20),
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
73 trace="row", hline = NA, tracecol = NA
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
74 )
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
75
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
76 write.table(
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
77 data.frame(gene = rownames(mat)[out$rowInd]),
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
78 paste(opt$out, "rows.csv", sep="."),
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
79 row.names = FALSE,
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
80 quote = FALSE,
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
81 sep = ',')
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
82
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
83 write.table(
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
84 data.frame(gene = colnames(mat)[out$colInd]),
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
85 paste(opt$out, "cols.csv", sep="."),
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
86 row.names = FALSE,
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
87 quote = FALSE,
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
88 sep = ',')
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
89
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
90 dev.off()
032f6b3806a3 Uploaded
dereeper
parents:
diff changeset
91