annotate population_structure.r @ 0:580da578c5e6 default tip

Uploaded
author rico
date Thu, 05 Apr 2012 15:56:36 -0400
parents
children
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
0
580da578c5e6 Uploaded
rico
parents:
diff changeset
1 library(RColorBrewer)
580da578c5e6 Uploaded
rico
parents:
diff changeset
2
580da578c5e6 Uploaded
rico
parents:
diff changeset
3 args = commandArgs(trailingOnly=TRUE)
580da578c5e6 Uploaded
rico
parents:
diff changeset
4 q_file = args[[1]]
580da578c5e6 Uploaded
rico
parents:
diff changeset
5 output_file = args[[2]]
580da578c5e6 Uploaded
rico
parents:
diff changeset
6 populations = args[[3]]
580da578c5e6 Uploaded
rico
parents:
diff changeset
7
580da578c5e6 Uploaded
rico
parents:
diff changeset
8 tbl <- read.table(q_file)
580da578c5e6 Uploaded
rico
parents:
diff changeset
9
580da578c5e6 Uploaded
rico
parents:
diff changeset
10 if ( populations >= 3 && populations <= 12 ) {
580da578c5e6 Uploaded
rico
parents:
diff changeset
11 colors = brewer.pal(populations, 'Paired')
580da578c5e6 Uploaded
rico
parents:
diff changeset
12 } else {
580da578c5e6 Uploaded
rico
parents:
diff changeset
13 colors = rainbow(populations)
580da578c5e6 Uploaded
rico
parents:
diff changeset
14 }
580da578c5e6 Uploaded
rico
parents:
diff changeset
15
580da578c5e6 Uploaded
rico
parents:
diff changeset
16 pdf(file=output_file, onefile=TRUE, width=7, height=3)
580da578c5e6 Uploaded
rico
parents:
diff changeset
17 barplot(t(as.matrix(tbl)), col=colors, xlab="Individual #", ylab="Ancestry", border=NA)
580da578c5e6 Uploaded
rico
parents:
diff changeset
18
580da578c5e6 Uploaded
rico
parents:
diff changeset
19 dev.off()