annotate population_structure.r @ 0:c3ba43704319 draft default tip

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