0
|
1 #!/usr/bin/env Rscript
|
|
2
|
|
3 suppressPackageStartupMessages(library("adegenet"))
|
|
4 suppressPackageStartupMessages(library("ape"))
|
|
5 suppressPackageStartupMessages(library("data.table"))
|
|
6 suppressPackageStartupMessages(library("dbplyr"))
|
|
7 suppressPackageStartupMessages(library("dplyr"))
|
|
8 suppressPackageStartupMessages(library("ggplot2"))
|
|
9 suppressPackageStartupMessages(library("knitr"))
|
|
10 suppressPackageStartupMessages(library("maps"))
|
|
11 suppressPackageStartupMessages(library("mapproj"))
|
|
12 suppressPackageStartupMessages(library("optparse"))
|
|
13 suppressPackageStartupMessages(library("poppr"))
|
|
14 suppressPackageStartupMessages(library("RColorBrewer"))
|
|
15 suppressPackageStartupMessages(library("RPostgres"))
|
|
16 suppressPackageStartupMessages(library("SNPRelate"))
|
|
17 suppressPackageStartupMessages(library("tidyr"))
|
|
18 suppressPackageStartupMessages(library("vcfR"))
|
|
19 suppressPackageStartupMessages(library("vegan"))
|
|
20 suppressPackageStartupMessages(library("yarrr"))
|
|
21 theme_set(theme_bw())
|
|
22
|
|
23 DEFAULT_MISSING_NUMERIC_VALUE <- -9.000000;
|
|
24
|
|
25 option_list <- list(
|
|
26 make_option(c("--database_connection_string"), action="store", dest="database_connection_string", help="Corals (stag) database connection string"),
|
|
27 make_option(c("--input_affy_metadata"), action="store", dest="input_affy_metadata", help="Affymetrix 96 well plate input file"),
|
|
28 make_option(c("--input_pop_info"), action="store", dest="input_pop_info", help="Population information input file"),
|
|
29 make_option(c("--input_vcf"), action="store", dest="input_vcf", help="VCF input file"),
|
|
30 make_option(c("--output_nj_phylogeny_tree"), action="store", dest="output_nj_phylogeny_tree", default=NULL, help="Flag to plot neighbor-joining phylogeny tree"),
|
|
31 make_option(c("--output_stag_db_report"), action="store", dest="output_stag_db_report", help="Flag to output stag db report file")
|
|
32 )
|
|
33
|
|
34 parser <- OptionParser(usage="%prog [options] file", option_list=option_list);
|
|
35 args <- parse_args(parser, positional_arguments=TRUE);
|
|
36 opt <- args$options;
|
|
37
|
|
38 get_file_path = function(dir, file_name) {
|
|
39 file_path = paste(dir, file_name, sep="/");
|
|
40 return(file_path);
|
|
41 }
|
|
42
|
|
43 get_database_connection <- function(db_conn_string) {
|
|
44 # Instantiate database connection.
|
|
45 # The connection string has this format:
|
|
46 # postgresql://user:password@host/dbname
|
|
47 conn_items <- strsplit(db_conn_string, "://")[[1]];
|
|
48 string_needed <- conn_items[2];
|
|
49 items_needed <- strsplit(string_needed, "@")[[1]];
|
|
50 user_pass_string <- items_needed[1];
|
|
51 host_dbname_string <- items_needed[2];
|
|
52 user_pass_items <- strsplit(user_pass_string, ":")[[1]];
|
|
53 host_dbname_items <- strsplit(host_dbname_string, "/")[[1]];
|
|
54 user <- user_pass_items[1];
|
|
55 pass <- user_pass_items[2];
|
|
56 host <- host_dbname_items[1];
|
|
57 dbname <- host_dbname_items[2];
|
|
58 conn <- DBI::dbConnect(RPostgres::Postgres(), host=host, port="5432", dbname=dbname, user=user, password=pass);
|
|
59 return (conn);
|
|
60 }
|
|
61
|
4
|
62 log_data_frame <- function(name, df) {
|
|
63 cat("\n", name, ":\n");
|
|
64 show(df);
|
|
65 cat("\n\n");
|
|
66 }
|
|
67
|
0
|
68 time_elapsed <- function(start_time) {
|
|
69 cat("Elapsed time: ", proc.time() - start_time, "\n\n");
|
|
70 }
|
|
71
|
|
72 time_start <- function(msg) {
|
|
73 start_time <- proc.time();
|
|
74 cat(msg, "...\n");
|
|
75 return(start_time);
|
|
76 }
|
|
77
|
|
78 write_data_frame <- function(dir, file_name, data_frame) {
|
|
79 cat("\nWriting file: ", file_name, "\n");
|
|
80 file_path <- get_file_path(dir, file_name);
|
|
81 write.table(data_frame, file=file_path, quote=FALSE, row.names=FALSE, sep="\t");
|
|
82 }
|
|
83
|
|
84 # Prepare for processing.
|
|
85 output_data_dir = "output_data_dir";
|
|
86 output_plots_dir = "output_plots_dir";
|
|
87 # Read in VCF input file.
|
|
88 start_time <- time_start("Reading VCF input");
|
|
89 vcf <- read.vcfR(opt$input_vcf);
|
|
90 time_elapsed(start_time);
|
|
91
|
|
92 # Convert VCF file into a genind for the Poppr package.
|
|
93 start_time <- time_start("Converting VCF data to a genind object");
|
|
94 genind_obj <- vcfR2genind(vcf);
|
4
|
95 cat("\ngenind_obj:\n");
|
|
96 genind_obj
|
|
97 cat("\n\n");
|
0
|
98 time_elapsed(start_time);
|
|
99
|
|
100 # Add population information to the genind object.
|
|
101 population_info_data_table <- read.table(opt$input_pop_info, check.names=FALSE, header=F, na.strings=c("", "NA"), stringsAsFactors=FALSE, sep="\t", quote="");
|
|
102 colnames(population_info_data_table) <- c("row_id", "affy_id", "user_specimen_id", "region");
|
4
|
103 cat("\npopulation_info_data_table:\n");
|
|
104 population_info_data_table
|
|
105 cat("\n\n");
|
0
|
106 #write_data_frame(output_data_dir, "population_info_data_table", population_info_data_table);
|
|
107 genind_obj@pop <- as.factor(population_info_data_table$region);
|
|
108 strata(genind_obj) <- data.frame(pop(genind_obj));
|
|
109
|
|
110 # Convert genind object to a genclone object.
|
|
111 start_time <- time_start("Converting the genind object to a genclone object");
|
|
112 genind_clone <- as.genclone(genind_obj);
|
4
|
113 cat("\ngenind_clone:\n");
|
|
114 genind_clone
|
|
115 cat("\n\n");
|
0
|
116 time_elapsed(start_time);
|
7
|
117 # Remove genind object from memory.
|
|
118 rm(genind_obj);
|
0
|
119
|
|
120 # Calculate the bitwise distance between individuals.
|
|
121 start_time <- time_start("Calculating the bitwise distance between individuals");
|
|
122 bitwise_distance <- bitwise.dist(genind_clone);
|
|
123 time_elapsed(start_time);
|
|
124
|
|
125 # Multilocus genotypes (threshold of 3.2%).
|
4
|
126 cat("\nFiltering multilocus genotypes with threshold of 3.2%...\n\n");
|
0
|
127 mlg.filter(genind_clone, distance=bitwise_distance) <- 0.032;
|
|
128
|
|
129 # Create list of MLGs.
|
4
|
130 cat("\nCreating list of mlg_ids...\n\n");
|
0
|
131 mlg_ids <- mlg.id(genind_clone);
|
|
132
|
|
133 # Read user's Affymetrix 96 well plate tabular file.
|
|
134 affy_metadata_data_frame <- read.table(opt$input_affy_metadata, header=FALSE, stringsAsFactors=FALSE, sep="\t", na.strings=c("", "NA"), quote="");
|
|
135 colnames(affy_metadata_data_frame) <- c("user_specimen_id", "field_call", "bcoral_genet_id", "bsym_genet_id", "reef",
|
|
136 "region", "latitude", "longitude", "geographic_origin", "colony_location",
|
|
137 "depth", "disease_resist", "bleach_resist", "mortality","tle",
|
|
138 "spawning", "collector_last_name", "collector_first_name", "organization", "collection_date",
|
|
139 "email", "seq_facility", "array_version", "public", "public_after_date",
|
|
140 "sperm_motility", "healing_time", "dna_extraction_method", "dna_concentration", "registry_id",
|
|
141 "result_folder_name", "plate_barcode");
|
|
142 affy_metadata_data_frame$user_specimen_id <- as.character(affy_metadata_data_frame$user_specimen_id);
|
4
|
143 log_data_frame("affy_metadata_data_frame", affy_metadata_data_frame);
|
0
|
144 user_specimen_ids <- as.character(affy_metadata_data_frame$user_specimen_id);
|
4
|
145 cat("\nuser_specimen_ids:\n", toString(user_specimen_ids), "\n\n");
|
0
|
146 # The specimen_id_field_call_data_table looks like this:
|
|
147 # user_specimen_ids V2
|
|
148 # 1090 prolifera
|
|
149 # 1091 prolifera
|
4
|
150 cat("\nCreating specimen_id_field_call_data_table...\n");
|
0
|
151 specimen_id_field_call_data_table <- data.table(user_specimen_ids, affy_metadata_data_frame$field_call);
|
|
152 # Rename the user_specimen_ids column.
|
|
153 setnames(specimen_id_field_call_data_table, c("user_specimen_ids"), c("user_specimen_id"));
|
|
154 # Rename the V2 column.
|
|
155 setnames(specimen_id_field_call_data_table, c("V2"), c("field_call"));
|
|
156
|
|
157 # Connect to database.
|
|
158 conn <- get_database_connection(opt$database_connection_string);
|
|
159 # Import the sample table.
|
|
160 sample_table <- tbl(conn, "sample");
|
|
161 # Import the genotype table.
|
|
162 genotype_table <- tbl(conn, "genotype");
|
2
|
163 # Import the probe_annotation table.
|
|
164 probe_annotation_table <- tbl(conn, "probe_annotation");
|
0
|
165 # Select columns from the sample table and the
|
|
166 # genotype table joined by genotype_id.
|
|
167 sample_table_columns <- sample_table %>% select(user_specimen_id, affy_id, bcoral_genet_id, genotype_id);
|
|
168 smlg <- sample_table_columns %>%
|
|
169 left_join(genotype_table %>%
|
|
170 select("id", "coral_mlg_clonal_id", "coral_mlg_rep_sample_id", "genetic_coral_species_call"),
|
|
171 by=c("genotype_id"="id"));
|
|
172 # Name the columns.
|
|
173 smlg_data_frame <- as.data.frame(smlg);
|
|
174 colnames(smlg_data_frame) <- c("user_specimen_id", "affy_id", "bcoral_genet_id", "genotype_id",
|
2
|
175 "coral_mlg_clonal_id", "coral_mlg_rep_sample_id", "genetic_coral_species_call");
|
4
|
176 log_data_frame("smlg_data_frame", smlg_data_frame);
|
0
|
177 # Missing GT in samples submitted.
|
|
178 start_time <- time_start("Discovering missing GT in samples");
|
|
179 gt <- extract.gt(vcf, element="GT", as.numeric=FALSE);
|
|
180 missing_gt <- apply(gt, MARGIN=2, function(x){ sum(is.na(x))});
|
|
181 missing_gt <- (missing_gt / nrow(vcf)) * 100;
|
|
182 missing_gt_data_frame <- data.frame(missing_gt);
|
4
|
183 log_data_frame("missing_gt_data_frame", missing_gt_data_frame);
|
0
|
184 # The specimen_id_field_call_data_table looks like this:
|
|
185 # rn missing_gt
|
|
186 # a100000-4368120-060520-256_I07.CEL 0.06092608
|
|
187 # a100000-4368120-060520-256_K07.CEL 0.05077173
|
4
|
188 cat("\nCreating missing_gt_data_table...\n");
|
|
189 missing_gt_data_table <- setDT(missing_gt_data_frame, keep.rownames=TRUE)[];
|
0
|
190 # Rename the rn column.
|
|
191 setnames(missing_gt_data_table, c("rn"), c("affy_id"));
|
|
192 # Rename the missing_gt column.
|
|
193 setnames(missing_gt_data_table, c("missing_gt"), c("percent_missing_data_coral"));
|
|
194 # Round data to two digits.
|
|
195 missing_gt_data_table$percent_missing_data_coral <- round(missing_gt_data_table$percent_missing_data_coral, digits=2);
|
|
196 time_elapsed(start_time);
|
|
197
|
2
|
198 # Subset genotypes for the fixed SNPs by probe id.
|
|
199 # Select columns from the probe_annotation table.
|
|
200 probe_annotation_table_columns <- probe_annotation_table %>% select(probe_set_id, custid, fixed_status, acerv_allele);
|
|
201 # Convert to data frame.
|
|
202 fixed_snp_data_frame <- as.data.frame(probe_annotation_table_columns);
|
|
203 # Name the columns.
|
|
204 colnames(fixed_snp_data_frame) <- c("probe_set_id", "custid", "fixed_status", "acerv_allele");
|
|
205 # Filter unwanted rows.
|
|
206 fixed_snp_data_frame <- subset(fixed_snp_data_frame, fixed_snp_data_frame$fixed_status=="keep");
|
4
|
207 log_data_frame("fixed_snp_data_frame", fixed_snp_data_frame);
|
2
|
208 gt_fixed <- gt[rownames(gt) %in% fixed_snp_data_frame$probe_set_id, ];
|
4
|
209 log_data_frame("gt_fixed", gt_fixed);
|
2
|
210
|
|
211 # Missing GT in fixed SNPs.
|
|
212 missing_gt_fixed <- apply(gt_fixed, MARGIN=2, function(x){ sum(is.na(x))});
|
|
213 missing_gt_fixed <- (missing_gt_fixed / nrow(gt_fixed)) * 100;
|
|
214 missing_gt_fixed_data_frame <- data.frame(missing_gt_fixed);
|
4
|
215 log_data_frame("missing_gt_fixed_data_frame", missing_gt_fixed_data_frame);
|
|
216 cat("\nCreating missing_gt_fixed_data_table...\n");
|
2
|
217 missing_gt_fixed_data_table <- setDT(missing_gt_fixed_data_frame, keep.rownames=TRUE)[];
|
|
218 # Rename the rn column.
|
|
219 setnames(missing_gt_fixed_data_table, c("rn"), c("affy_id"));
|
|
220 # Rename the missing_gt column.
|
|
221 setnames(missing_gt_fixed_data_table, c("missing_gt_fixed"), c("percent_missing_data_fixed"));
|
|
222 # Round data to two digits.
|
|
223 missing_gt_fixed_data_table$percent_missing_data_fixed <- round(missing_gt_fixed_data_table$percent_missing_data_fixed, digits=2);
|
|
224
|
|
225 # Heterozygous alleles for fixed SNPs.
|
0
|
226 start_time <- time_start("Discovering heterozygous alleles");
|
2
|
227 heterozygous_alleles <- apply(gt_fixed, MARGIN=2, function(x) {sum(lengths(regmatches(x, gregexpr("0/1", x))))});
|
|
228 heterozygous_alleles <- (heterozygous_alleles / nrow(gt_fixed)) * 100;
|
0
|
229 heterozygous_alleles_data_frame <- data.frame(heterozygous_alleles);
|
4
|
230 log_data_frame("heterozygous_alleles_data_frame", heterozygous_alleles_data_frame);
|
0
|
231 # The heterozygous_alleles_data_table looks like this:
|
|
232 # rn heterozygous_alleles
|
|
233 # a100000-4368120-060520-256_I07.CEL 73.94903
|
|
234 # a100000-4368120-060520-256_K07.CEL 74.40089
|
4
|
235 cat("\nCreating heterozygous_alleles_data_table...\n");
|
0
|
236 heterozygous_alleles_data_table <- setDT(heterozygous_alleles_data_frame, keep.rownames=TRUE)[];
|
|
237 # Rename the rn column.
|
|
238 setnames(heterozygous_alleles_data_table, c("rn"), c("affy_id"));
|
|
239 # Rename the heterozygous_alleles column.
|
|
240 setnames(heterozygous_alleles_data_table, c("heterozygous_alleles"), c("percent_heterozygous_coral"));
|
|
241 # Round data to two digits.
|
|
242 heterozygous_alleles_data_table$percent_heterozygous_coral <- round(heterozygous_alleles_data_table$percent_heterozygous_coral, digits=2);
|
|
243 time_elapsed(start_time);
|
|
244
|
2
|
245 # Create list of Acerv reference and alternative probes.
|
|
246 rAC <- subset(fixed_snp_data_frame, fixed_snp_data_frame$acerv_allele=="reference");
|
|
247 aAC <- subset(fixed_snp_data_frame, fixed_snp_data_frame$acerv_allele=="alternative");
|
|
248
|
|
249 # Subset probes for the reference and alternative SNPs in Acerv.
|
|
250 ref_ac <- gt_fixed[rownames(gt_fixed) %in% rAC$probe,];
|
|
251 alt_ac <- gt_fixed[rownames(gt_fixed) %in% aAC$probe,];
|
|
252
|
|
253 # Reference alleles for each species.
|
|
254 reference_alleles_ac <- apply(ref_ac, MARGIN=2, function(x) {sum(lengths(regmatches(x, gregexpr("0/0", x))))});
|
|
255 reference_alleles_ap <- apply(alt_ac, MARGIN=2, function(x) {sum(lengths(regmatches(x, gregexpr("0/0", x))))});
|
|
256
|
|
257 # Alternative alleles for each species.
|
|
258 alternative_alleles_ac <- apply(alt_ac, MARGIN=2, function(x) {sum(lengths(regmatches(x, gregexpr("1/1", x))))});
|
|
259 alternative_alleles_ap <- apply(ref_ac, MARGIN=2, function(x) {sum(lengths(regmatches(x, gregexpr("1/1", x))))});
|
|
260
|
|
261 # Apalm alleles.
|
0
|
262 start_time <- time_start("Discovering reference alleles");
|
2
|
263 ap_sum <- rowSums(cbind(reference_alleles_ap,alternative_alleles_ap));
|
|
264 ap_alleles <- (ap_sum / nrow(gt_fixed)) * 100;
|
|
265 ap_alleles_data_frame <- data.frame(ap_alleles);
|
4
|
266 log_data_frame("ap_alleles_data_frame", ap_alleles_data_frame);
|
0
|
267 # The reference_alleles_data_table looks like this:
|
|
268 # rn reference_alleles
|
|
269 # a100000-4368120-060520-256_I07.CEL 11.60642
|
|
270 # a100000-4368120-060520-256_K07.CEL 11.45918
|
4
|
271 cat("\nCreating ap_alleles_data_table...\n");
|
2
|
272 ap_alleles_data_table <- setDT(ap_alleles_data_frame, keep.rownames=TRUE)[];
|
0
|
273 # Rename the rn column.
|
2
|
274 setnames(ap_alleles_data_table, c("rn"), c("affy_id"));
|
0
|
275 # Rename the reference_alleles column.
|
2
|
276 setnames(ap_alleles_data_table, c("ap_alleles"), c("percent_apalm_coral"));
|
0
|
277 # Round data to two digits.
|
2
|
278 ap_alleles_data_table$percent_apalm_coral <- round(ap_alleles_data_table$percent_apalm_coral, digits=2);
|
0
|
279 time_elapsed(start_time);
|
|
280
|
2
|
281 # Acerv alleles.
|
0
|
282 start_time <- time_start("Discovering alternative alleles");
|
2
|
283 ac_sum <- rowSums(cbind(reference_alleles_ac,alternative_alleles_ac));
|
|
284 ac_alleles <- (ac_sum / nrow(gt_fixed)) * 100;
|
|
285 ac_alleles_data_frame <- data.frame(ac_alleles);
|
4
|
286 log_data_frame("ac_alleles_data_frame", ac_alleles_data_frame);
|
0
|
287 # The alternative_alleles_data_table looks like this:
|
|
288 # rn alternative_alleles
|
|
289 # a100000-4368120-060520-256_I07.CEL 14.38363
|
|
290 # a100000-4368120-060520-256_K07.CEL 14.08916
|
4
|
291 cat("\nCreating ac_alleles_data_table...\n");
|
2
|
292 ac_alleles_data_table <- setDT(ac_alleles_data_frame, keep.rownames=TRUE)[];
|
0
|
293 # Rename the rn column.
|
2
|
294 setnames(ac_alleles_data_table, c("rn"), c("affy_id"));
|
0
|
295 # Rename the alternative_alleles column.
|
2
|
296 setnames(ac_alleles_data_table, c("ac_alleles"), c("percent_acerv_coral"));
|
0
|
297 # Round data to two digits.
|
2
|
298 ac_alleles_data_table$percent_acerv_coral <- round(ac_alleles_data_table$percent_acerv_coral, digits=2);
|
0
|
299 time_elapsed(start_time);
|
|
300
|
|
301 # The mlg_ids_data_table looks like this:
|
|
302 # mlg_ids
|
|
303 # a550962-4368120-060520-500_M23.CEL
|
|
304 # a550962-4368120-060520-256_A19.CEL
|
4
|
305 cat("\nCreating mlg_ids_data_table...\n");
|
0
|
306 mlg_ids_data_table <- data.table(mlg_ids, keep.rownames=TRUE);
|
|
307 # Rename the mlg_ids column.
|
|
308 setnames(mlg_ids_data_table, c("mlg_ids"), c("affy_id"));
|
|
309
|
|
310 # sample_mlg_tibble looks like this:
|
|
311 # A tibble: 262 x 3
|
|
312 # Groups: group [?]
|
|
313 # group affy_id coral_mlg_clonal_id coral_mlg_rep_sample_id
|
|
314 # <int> <chr> <chr> <chr>
|
|
315 # 1 a550962-4368.CEL NA 13905
|
|
316 sample_mlg_tibble <- mlg_ids_data_table %>%
|
|
317 group_by(row_number()) %>%
|
|
318 dplyr::rename(group="row_number()") %>%
|
|
319 unnest (affy_id) %>%
|
|
320 # Join with mlg table.
|
|
321 left_join(smlg_data_frame %>%
|
|
322 select("affy_id","coral_mlg_clonal_id", "coral_mlg_rep_sample_id"),
|
|
323 by="affy_id");
|
|
324
|
|
325 # If found in database, group members on previous mlg id.
|
|
326 uniques <- unique(sample_mlg_tibble[c("group", "coral_mlg_clonal_id")]);
|
|
327 uniques <- uniques[!is.na(uniques$coral_mlg_clonal_id),];
|
|
328 na.mlg <- which(is.na(sample_mlg_tibble$coral_mlg_clonal_id));
|
|
329 na.group <- sample_mlg_tibble$group[na.mlg];
|
|
330 sample_mlg_tibble$coral_mlg_clonal_id[na.mlg] <- uniques$coral_mlg_clonal_id[match(na.group, uniques$group)];
|
|
331
|
|
332 # Find out if the sample mlg matched a previous genotyped sample.
|
|
333 # sample_mlg_match_tibble looks like this:
|
|
334 # A tibble: 262 x 4
|
|
335 # Groups: group [230]
|
|
336 # group affy_id coral_mlg_clonal_id db_match
|
|
337 # <int> <chr> <chr> <chr>
|
|
338 # 1 a550962-436.CEL NA no_match
|
|
339 sample_mlg_match_tibble <- sample_mlg_tibble %>%
|
|
340 group_by(group) %>%
|
|
341 mutate(db_match = ifelse(is.na(coral_mlg_clonal_id), "no_match", "match"));
|
|
342
|
|
343 # Create new mlg id for samples with no matches in the database.
|
|
344 none <- unique(sample_mlg_match_tibble[c("group", "coral_mlg_clonal_id")]);
|
|
345 none <- none[is.na(none$coral_mlg_clonal_id),];
|
|
346 na.mlg2 <- which(is.na(sample_mlg_match_tibble$coral_mlg_clonal_id));
|
|
347 n.g <- sample_mlg_match_tibble$group[na.mlg2];
|
|
348 ct <- length(unique(n.g));
|
|
349
|
|
350 # List of new group ids, the sequence starts at the number of
|
|
351 # ids present in sample_mlg_match_tibble$coral_mlg_clonal_ids
|
|
352 # plus 1.
|
|
353 n.g_ids <- sprintf("HG%04d", seq((sum(!is.na(unique(sample_mlg_match_tibble["coral_mlg_clonal_id"]))) + 1), by=1, length=ct));
|
|
354
|
|
355 # Assign the new id iteratively for all that have NA.
|
|
356 for (i in 1:length(na.mlg2)) {
|
|
357 sample_mlg_match_tibble$coral_mlg_clonal_id[na.mlg2[i]] <- n.g_ids[match(sample_mlg_match_tibble$group[na.mlg2[i]], unique(n.g))];
|
|
358 }
|
|
359
|
|
360 # Subset population_info_data_table for all samples.
|
|
361 # affy_id_user_specimen_id_vector looks like this:
|
|
362 # affy_id user_specimen_id
|
|
363 # a100000-432.CEL 13704
|
|
364 affy_id_user_specimen_id_vector <- population_info_data_table[c(2, 3)];
|
|
365
|
|
366 # Merge data frames for final table.
|
|
367 start_time <- time_start("Merging data frames");
|
|
368 stag_db_report <- specimen_id_field_call_data_table %>%
|
|
369 left_join(affy_id_user_specimen_id_vector %>%
|
|
370 select("affy_id", "user_specimen_id"),
|
|
371 by="user_specimen_id") %>%
|
|
372 mutate(db_record = ifelse(affy_id %in% smlg_data_frame$affy_id, "genotyped", "new")) %>%
|
|
373 filter(db_record=="new") %>%
|
|
374 left_join(sample_mlg_match_tibble %>%
|
|
375 select("affy_id", "coral_mlg_clonal_id", "db_match"),
|
|
376 by="affy_id") %>%
|
|
377 left_join(missing_gt_data_table %>%
|
|
378 select("affy_id", "percent_missing_data_coral"),
|
|
379 by="affy_id") %>%
|
2
|
380 left_join(missing_gt_fixed_data_table %>%
|
|
381 select("affy_id", "percent_missing_data_fixed"),
|
|
382 by="affy_id") %>%
|
0
|
383 left_join(heterozygous_alleles_data_table %>%
|
|
384 select("affy_id", "percent_heterozygous_coral"),
|
|
385 by="affy_id") %>%
|
2
|
386 left_join(ac_alleles_data_table %>%
|
1
|
387 select("affy_id", "percent_acerv_coral"),
|
0
|
388 by="affy_id") %>%
|
2
|
389 left_join(ap_alleles_data_table %>%
|
1
|
390 select("affy_id", "percent_apalm_coral"),
|
0
|
391 by="affy_id") %>%
|
|
392 mutate(db_match = ifelse(is.na(db_match), "failed", db_match))%>%
|
|
393 mutate(coral_mlg_clonal_id = ifelse(is.na(coral_mlg_clonal_id), "failed", coral_mlg_clonal_id)) %>%
|
2
|
394 mutate(genetic_coral_species_call = ifelse(percent_apalm_coral >= 85 & percent_acerv_coral <= 10, "A.palmata", "other")) %>%
|
|
395 mutate(genetic_coral_species_call = ifelse(percent_acerv_coral >= 85 & percent_apalm_coral <= 10, "A.cervicornis", genetic_coral_species_call)) %>%
|
0
|
396 mutate(genetic_coral_species_call = ifelse(percent_heterozygous_coral > 40, "A.prolifera", genetic_coral_species_call)) %>%
|
|
397 ungroup() %>%
|
2
|
398 select(-group, -db_record);
|
0
|
399 time_elapsed(start_time);
|
|
400
|
|
401 start_time <- time_start("Writing csv output");
|
|
402 write.csv(stag_db_report, file=opt$output_stag_db_report, quote=FALSE);
|
|
403 time_elapsed(start_time);
|
|
404
|
|
405 # Representative clone for genotype table.
|
|
406 start_time <- time_start("Creating representative clone for genotype table");
|
2
|
407 no_dup_genotypes_genind <- clonecorrect(genind_clone, strata=~pop.genind_obj.);
|
0
|
408 id_rep <- mlg.id(no_dup_genotypes_genind);
|
4
|
409 cat("\nCreating id_data_table...\n");
|
0
|
410 id_data_table <- data.table(id_rep, keep.rownames=TRUE);
|
|
411 # Rename the id_rep column.
|
|
412 setnames(id_data_table, c("id_rep"), c("affy_id"));
|
|
413 time_elapsed(start_time);
|
7
|
414 # Remove clonecorrect genind from memory.
|
|
415 rm(no_dup_genotypes_genind);
|
0
|
416
|
|
417 # Table of alleles for the new samples subset to new plate data.
|
|
418 # Create vector indicating number of individuals desired from
|
|
419 # affy_id column of stag_db_report data table.
|
|
420 i <- ifelse(is.na(stag_db_report[3]), "", stag_db_report[[3]]);
|
2
|
421 i <- i[!apply(i== "", 1, all), ];
|
0
|
422
|
|
423 # Subset VCF to the user samples.
|
|
424 start_time <- time_start("Subsetting vcf to the user samples");
|
6
|
425 affy_list <- append(stag_db_report$affy_id,"FORMAT");
|
|
426 svcf <- vcf[,colnames(vcf@gt) %in% affy_list];
|
0
|
427 write.vcf(svcf, "subset.vcf.gz");
|
7
|
428
|
|
429 # Remove original and subset VCFs written to file from R memory.
|
|
430 rm(svcf);
|
|
431 rm(vcf);
|
|
432
|
|
433 # Load in subset VCF.
|
0
|
434 vcf.fn <- "subset.vcf.gz";
|
|
435 snpgdsVCF2GDS(vcf.fn, "test3.gds", method="biallelic.only");
|
|
436 genofile <- snpgdsOpen(filename="test3.gds", readonly=FALSE);
|
|
437 gds_array <- read.gdsn(index.gdsn(genofile, "sample.id"));
|
|
438 # gds_array looks like this:
|
|
439 # [1] "a550962-4368120-060520-500_A03.CEL" "a550962-4368120-060520-500_A05.CEL"
|
|
440 gds_data_frame <- data.frame(gds_array);
|
4
|
441 log_data_frame("gds_data_frame", gds_data_frame);
|
0
|
442 # gds_data_frame looks like this:
|
|
443 # gds_array
|
|
444 # a550962-4368120-060520-500_A03.CEL
|
|
445 # a550962-4368120-060520-500_A05.CEL
|
4
|
446 cat("\nCreating gds_data_table...\n");
|
0
|
447 gds_data_table <- setDT(gds_data_frame, keep.rownames=FALSE)[];
|
|
448 # Rename the gds_array column.
|
|
449 setnames(gds_data_table, c("gds_array"), c("affy_id"));
|
|
450 # affy_id_region_list looks like this:
|
|
451 # affy_id region
|
|
452 # a100000-4368120-060520-256_I07.CEL USVI
|
|
453 # a100000-4368120-060520-256_K07.CEL USVI
|
|
454 affy_id_region_list <- population_info_data_table[c(2,3,4)];
|
|
455 gds_data_table_join <- gds_data_table %>%
|
|
456 left_join(affy_id_region_list %>%
|
2
|
457 select("affy_id", "user_specimen_id", "region"),
|
0
|
458 by='affy_id')%>%
|
|
459 drop_na();
|
|
460 samp.annot <- data.frame(pop.group=c(gds_data_table_join$region));
|
|
461 add.gdsn(genofile, "sample.annot", samp.annot);
|
|
462 # population_code looks like this:
|
|
463 # [1] 18.361733 18.361733 18.361733 18.361733 18.361733 18.361733
|
|
464 # [7] 25.11844009 25.11844009 25.11844009 25.11844009 25.11844009 25.11844009
|
|
465 population_code <- read.gdsn(index.gdsn(genofile, path="sample.annot/pop.group"));
|
|
466 pop.group <- as.factor(read.gdsn(index.gdsn(genofile, "sample.annot/pop.group")));
|
|
467 # pop.group looks like this:
|
|
468 # [1] 18.361733 18.361733 18.361733 18.361733 18.361733 18.361733
|
|
469 # [7] 25.11844009 25.11844009 25.11844009 25.11844009 25.11844009 25.11844009
|
|
470 time_elapsed(start_time);
|
|
471
|
|
472 # Distance matrix calculation and sample labels change to user specimen ids.
|
|
473 start_time <- time_start("Calculating distance matrix");
|
|
474 ibs <- snpgdsIBS(genofile, num.thread=2, autosome.only=FALSE);
|
|
475 ibs$sample.id <-gds_data_table_join$user_specimen_id;
|
|
476 time_elapsed(start_time);
|
|
477
|
|
478 # Cluster analysis on the genome-wide IBS pairwise distance matrix.
|
|
479 start_time <- time_start("Clustering the genome-wide IBS pairwise distance matrix");
|
|
480 set.seed(100);
|
2
|
481 par(cex=0.6, cex.lab=1, cex.axis=1.5, cex.main=2);
|
0
|
482 ibs.hc <- snpgdsHCluster(ibs);
|
|
483 time_elapsed(start_time);
|
|
484
|
|
485 # cols looks like this:
|
|
486 # blue1 red green pink orange blue2
|
|
487 # "#0C5BB0FF" "#EE0011FF" "#15983DFF" "#EC579AFF" "#FA6B09FF" "#149BEDFF"
|
|
488 # green2 yellow turquoise poop
|
|
489 # "#A1C720FF" "#FEC10BFF" "#16A08CFF" "#9A703EFF"
|
|
490 cols <- piratepal("basel");
|
|
491 set.seed(999);
|
|
492
|
|
493 # Generate plots.
|
|
494 # Default clustering.
|
|
495 start_time <- time_start("Creating ibs_default.pdf");
|
|
496 # Start PDF device driver.
|
|
497 dev.new(width=40, height=20);
|
|
498 file_path = get_file_path(output_plots_dir, "ibs_default.pdf");
|
|
499 pdf(file=file_path, width=40, height=20);
|
|
500 rv <- snpgdsCutTree(ibs.hc, col.list=cols, pch.list=15);
|
|
501 snpgdsDrawTree(rv, main="Color by Cluster", leaflab="perpendicular", yaxis.kinship=FALSE);
|
2
|
502 abline(h=0.032, lty=2);
|
0
|
503 legend("topleft", legend=levels(rv$samp.group), xpd=T, col=cols[1:nlevels(rv$samp.group)], pch=15, ncol=4, cex=1.2);
|
|
504 dev.off()
|
|
505 time_elapsed(start_time);
|
|
506
|
|
507 # Color cluster by region.
|
|
508 start_time <- time_start("Creating ibs_region.pdf");
|
|
509 # Start PDF device driver.
|
|
510 dev.new(width=40, height=20);
|
|
511 file_path = get_file_path(output_plots_dir, "ibs_region.pdf");
|
|
512 pdf(file=file_path, width=40, height=20);
|
|
513 race <- as.factor(population_code);
|
2
|
514 rv2 <- snpgdsCutTree(ibs.hc, samp.group=race, col.list=cols, pch.list=15);
|
0
|
515 snpgdsDrawTree(rv2, main="Color by Region", leaflab="perpendicular", yaxis.kinship=FALSE);
|
|
516 legend("topleft", legend=levels(race), xpd=T, col=cols[1:nlevels(race)], pch=15, ncol=4, cex=1.2);
|
|
517 dev.off()
|
|
518 time_elapsed(start_time);
|
|
519
|
|
520 # Missing data barplot.
|
|
521 start_time <- time_start("Creating missing_data.pdf");
|
|
522 population_info_data_table$miss <- stag_db_report$percent_missing_data_coral[match(missing_gt_data_frame$affy_id, stag_db_report$affy_id)];
|
|
523 test2 <- which(!is.na(population_info_data_table$miss));
|
|
524 miss96 <- population_info_data_table$miss[test2];
|
|
525 name96 <- population_info_data_table$user_specimen_id[test2];
|
|
526 # Start PDF device driver.
|
|
527 dev.new(width=20, height=10);
|
|
528 file_path = get_file_path(output_plots_dir, "missing_data.pdf");
|
|
529 pdf(file=file_path, width=20, height=10);
|
|
530 par(mar = c(8, 4, 4, 2));
|
|
531 x <- barplot(miss96, las=2, col=cols, ylim=c(0, 3), cex.axis=0.8, space=0.8, ylab="Missingness (%)", xaxt="n");
|
|
532 text(cex=0.8, x=x-0.25, y=-.05, name96, xpd=TRUE, srt=60, adj=1);
|
|
533 dev.off()
|
|
534 time_elapsed(start_time);
|
|
535
|
|
536 # Sample MLG on a map.
|
|
537 start_time <- time_start("Creating mlg_map.pdf");
|
|
538 # Get the lattitude and longitude boundaries for rendering
|
|
539 # the map. Tese boundaries will restrict the map to focus
|
|
540 # (i.e., zoom) on the region of the world map from which
|
|
541 # the samples were taken.
|
|
542 max_latitude <- max(affy_metadata_data_frame$latitude, na.rm=TRUE);
|
|
543 min_latitude <- min(affy_metadata_data_frame$latitude, na.rm=TRUE);
|
|
544 latitude_range_vector <- c(min_latitude-3, max_latitude+3);
|
|
545 max_longitude <- max(affy_metadata_data_frame$longitude, na.rm=TRUE);
|
|
546 min_longitude <- min(affy_metadata_data_frame$longitude, na.rm=TRUE);
|
|
547 longitude_range_vector <- c(min_longitude-3, max_longitude+3);
|
|
548 # Get the palette colors for rendering plots.
|
|
549 colors <- length(unique(stag_db_report$coral_mlg_clonal_id));
|
|
550 # Get a color palette.
|
|
551 palette <- colorRampPalette(piratepal("basel"));
|
|
552 # Start PDF device driver.
|
|
553 dev.new(width=20, height=20);
|
|
554 file_path = get_file_path(output_plots_dir, "mlg_map.pdf");
|
|
555 pdf(file=file_path, width=20, height=20);
|
|
556 world_data = map_data("world");
|
|
557 # Add the coral_mlg_clonal_id column from the stag_db_report
|
|
558 # data fram to the affy_metadata_data_frame.
|
|
559 affy_metadata_data_frame$mlg <- stag_db_report$coral_mlg_clonal_id;
|
|
560 # Get the number of colors needed from the palette for plotting
|
|
561 # the sample locations on the world map.
|
|
562 num_colors = length(unique(affy_metadata_data_frame$mlg));
|
|
563 # Get a color palette.
|
|
564 palette = colorRampPalette(piratepal("basel"));
|
|
565 ggplot() +
|
|
566 geom_map(data=world_data, map=world_data, aes(x=long, y=lat, group=group, map_id=region), fill="white", colour="#7f7f7f") +
|
|
567 coord_quickmap(xlim=longitude_range_vector, ylim=latitude_range_vector) +
|
|
568 geom_point(data=affy_metadata_data_frame, aes(x=longitude, y=latitude, group=mlg, colour=mlg), alpha=.7, size=3) +
|
|
569 scale_color_manual(values=palette(num_colors)) +
|
|
570 theme(legend.position="bottom") +
|
|
571 guides(color=guide_legend(nrow=8, byrow=F));
|
|
572
|
|
573 # Sample MLG on a map for each region.
|
|
574 for (i in unique(affy_metadata_data_frame$region)) {
|
2
|
575 m <- i;
|
|
576 num_colors_2 = length(unique(affy_metadata_data_frame$mlg[which(affy_metadata_data_frame$region == m)]));
|
|
577 max_latitude_region <- max(affy_metadata_data_frame$latitude[which(affy_metadata_data_frame$region == m)], na.rm=TRUE);
|
|
578 min_latitude_region <- min(affy_metadata_data_frame$latitude[which(affy_metadata_data_frame$region == m)], na.rm=TRUE);
|
|
579 latitude_range_vector_region <- c(min_latitude_region-0.5, max_latitude_region+0.5);
|
|
580 max_longitude_region <- max(affy_metadata_data_frame$longitude[which(affy_metadata_data_frame$region == m)], na.rm=TRUE);
|
|
581 min_longitude_region <- min(affy_metadata_data_frame$longitude[which(affy_metadata_data_frame$region == m)], na.rm=TRUE);
|
|
582 longitude_range_vector_region <- c(min_longitude_region-0.5, max_longitude_region+0.5);
|
|
583 print(ggplot() +
|
|
584 geom_map(data=world_data, map=world_data, aes(x=long, y=lat, group=group, map_id=region),
|
|
585 fill="grey", colour="#7f7f7f") +
|
|
586 coord_quickmap(xlim=longitude_range_vector_region, ylim=latitude_range_vector_region, clip = "on") +
|
|
587 geom_point(data=affy_metadata_data_frame[which(affy_metadata_data_frame$region == m),],
|
|
588 aes(x=longitude, y=latitude, group=mlg, colour=mlg), alpha=.5, size=3) +
|
|
589 scale_color_manual(values=palette(num_colors_2)) +
|
|
590 theme(legend.position="bottom") + labs(title=paste("MLG assignments for", m)) +
|
|
591 guides(color=guide_legend(nrow=8, byrow=F)));
|
0
|
592 }
|
|
593 dev.off()
|
|
594 time_elapsed(start_time);
|
|
595
|
|
596 if (!is.null(opt$output_nj_phylogeny_tree)) {
|
|
597 # Create a phylogeny tree of samples based on distance matrices.
|
|
598 # Start PDF device driver.
|
|
599 start_time <- time_start("Creating nj_phylogeny_tree.pdf");
|
|
600 # Table of alleles for the new samples subset to new plate data.
|
|
601 # Create vector indicating number of individuals desired from
|
|
602 # affy_id column of stag_db_report data table.
|
|
603 i <- ifelse(is.na(stag_db_report[1]), "", stag_db_report[[1]]);
|
|
604 i <- i[!apply(i== "", 1, all),];
|
|
605 sample_alleles_vector <- genind_clone[i, mlg.reset=FALSE, drop=FALSE];
|
|
606 dev.new(width=40, height=80);
|
|
607 file_path = get_file_path(output_plots_dir, "nj_phylogeny_tree.pdf");
|
|
608 pdf(file=file_path, width=40, height=80);
|
|
609 # Organize branches by clade.
|
|
610 nj_phylogeny_tree <- sample_alleles_vector %>%
|
|
611 aboot(dist=provesti.dist, sample=100, tree="nj", cutoff=50, quiet=TRUE, showtree = FALSE) %>%
|
|
612 ladderize();
|
|
613 nj_phylogeny_tree$tip.label <- stag_db_report$user_specimen_id[match(nj_phylogeny_tree$tip.label, stag_db_report$affy_id)];
|
|
614 plot.phylo(nj_phylogeny_tree, tip.color=cols[sample_alleles_vector$pop], label.offset=0.0025, cex=0.6, font=2, lwd=4, align.tip.label=F, no.margin=T);
|
|
615 # Add a scale bar showing 5% difference.
|
|
616 add.scale.bar(0, 0.95, length=0.05, cex=0.65, lwd=2);
|
|
617 nodelabels(nj_phylogeny_tree$node.label, cex=.5, adj=c(1.5, -0.1), frame="n", font=3, xpd=TRUE);
|
|
618 legend("topright", legend=c(levels(sample_alleles_vector$pop)), text.col=cols, xpd=T, cex=0.8);
|
|
619 dev.off()
|
|
620 time_elapsed(start_time);
|
|
621 }
|
|
622
|
|
623 # Generate a pie chart for each sample with a genotype.
|
|
624 # Store the numerical and user_specimen_id values from
|
|
625 # stag_db_report for the charts (user_specimen_id names
|
|
626 # will be used to label each chart).
|
|
627 start_time <- time_start("Creating percent_breakdown.pdf");
|
3
|
628 stag_db_report_data_table <- stag_db_report[c(-2, -3, -4, -5)];
|
0
|
629 # Remove NA and NaN values.
|
|
630 stag_db_report_data_table <- na.omit(stag_db_report_data_table);
|
|
631 # Translate to N (i.e., number of samples with a genotype)
|
|
632 # columns and 5 rows.
|
|
633 translated_stag_db_report_data_table <- t(stag_db_report_data_table);
|
|
634 translated_stag_db_report_matrix <- as.matrix(translated_stag_db_report_data_table[-1,]);
|
|
635 # Set the storage mode of the matrix to numeric. In some
|
|
636 # cases this could result in the following:
|
|
637 # Warning message:
|
|
638 # In mde(x) : NAs introduced by coercion
|
|
639 mode(translated_stag_db_report_matrix) <- "numeric";
|
|
640 # Remove NA and NaN values that may have been introduced
|
|
641 # by coercion.
|
|
642 translated_stag_db_report_matrix <- na.omit(translated_stag_db_report_matrix);
|
2
|
643 translated_stag_db_report_matrix<-translated_stag_db_report_matrix[-c(1),];
|
|
644 #tsdbrm_row_means <- rowMeans(translated_stag_db_report_matrix, na.rm=TRUE);
|
0
|
645 dev.new(width=10, height=7);
|
|
646 file_path = get_file_path(output_plots_dir, "percent_breakdown.pdf");
|
|
647 pdf(file=file_path, width=10, height=7);
|
|
648 par(mfrow=c(3, 2));
|
2
|
649 col <- c("#A6A6A6","#FFA626","#EB0ACF", "#80FF00" );
|
0
|
650 # Generate a pie chart for each sample with genotypes.
|
|
651 for (i in 1:ncol(translated_stag_db_report_matrix)) {
|
2
|
652 tmp_labels <- paste(c("no call", "heterozygous", "A. cervicornis", "A. palmata"), " (", round(translated_stag_db_report_matrix[,i], 1), "%)", sep="");
|
0
|
653 main <- paste("Breakdown of SNP assignments for", translated_stag_db_report_data_table[1, i]);
|
|
654 pie(translated_stag_db_report_matrix[,i], labels=tmp_labels, radius=0.90, col=col, main=main, cex.main=.85, cex=0.75);
|
|
655 }
|
|
656 dev.off()
|
|
657 time_elapsed(start_time);
|
|
658
|
|
659 # close GDS file.
|
|
660 snpgdsClose(genofile);
|
|
661
|
|
662 # Prepare to output data frames for input to a downstream
|
|
663 # tool that will use them to update the stag database.
|
|
664 start_time <- time_start("Building data frames for insertion into database tables");
|
|
665 # sample_prep_data_frame looks like this (split across comment lines):
|
|
666 # user_specimen_id field_call bcoral_genet_id bsym_genet_id reef
|
|
667 # test_002 prolifera NA NA JohnsonsReef
|
|
668 # region latitude longitude geographic_origin colony_location
|
|
669 # Bahamas 18.36173 -64.77430 Reef NA
|
|
670 # depth disease_resist bleach_resist
|
|
671 # 5 NA N
|
|
672 # mortality tle spawning collector_last_name collector_first_name organization
|
|
673 # NA NA False Kitchen Sheila Penn State
|
|
674 # collection_date email seq_facility array_version public
|
|
675 # 2018-11-08 k89@psu.edu Affymetrix 1 True
|
|
676 # public_after_date sperm_motility healing_time dna_extraction_method
|
|
677 # NA -9 -9 NA
|
|
678 # dna_concentration registry_id result_folder_name plate_barcode mlg
|
|
679 # NA NA PRO100175_PSU175_SAX_b02 P9SR10074 HG0227
|
|
680 # affy_id percent_missing_data_coral percent_heterozygous_coral
|
|
681 # a550962-436.CEL 1.06 19.10
|
1
|
682 # percent_acerv_coral percent_apalm_coral
|
0
|
683 # 40.10459 39.73396
|
|
684 sample_prep_data_frame <- affy_metadata_data_frame %>%
|
|
685 left_join(stag_db_report %>%
|
|
686 select("user_specimen_id", "affy_id", "percent_missing_data_coral", "percent_heterozygous_coral",
|
1
|
687 "percent_acerv_coral", "percent_apalm_coral"),
|
0
|
688 by='user_specimen_id');
|
|
689 # Get the number of rows for all data frames.
|
|
690 num_rows <- nrow(sample_prep_data_frame);
|
|
691 # Set the column names so that we can extract only those columns
|
|
692 # needed for the sample table.
|
|
693 colnames(sample_prep_data_frame) <- c("user_specimen_id", "field_call", "bcoral_genet_id", "bsym_genet_id", "reef",
|
|
694 "region", "latitude", "longitude", "geographic_origin", "colony_location",
|
|
695 "depth", "disease_resist", "bleach_resist", "mortality", "tle",
|
|
696 "spawning", "collector_last_name", "collector_first_name", "organization",
|
|
697 "collection_date", "email", "seq_facility", "array_version", "public",
|
|
698 "public_after_date", "sperm_motility", "healing_time", "dna_extraction_method",
|
|
699 "dna_concentration", "registry_id", "result_folder_name", "plate_barcode",
|
|
700 "mlg", "affy_id", "percent_missing_data_coral", "percent_heterozygous_coral",
|
1
|
701 "percent_acerv_coral", "percent_apalm_coral");
|
0
|
702
|
|
703 # Output the data frame for updating the alleles table.
|
|
704 # Subset to only the new plate data.
|
|
705 i <- ifelse(is.na(stag_db_report[3]), "", stag_db_report[[3]]);
|
|
706 # Create a vector indicating the number of individuals desired
|
|
707 # from the affy_id collumn in the report_user data table.
|
|
708 i <- i[!apply(i=="", 1, all),];
|
|
709 # Subset the genclone object to the user data.
|
|
710 allele_vector <- genind_clone[i, mlg.reset=FALSE, drop=FALSE];
|
|
711 # Convert the subset genclone to a data frame.
|
|
712 allele_data_frame <- genind2df(allele_vector, sep="");
|
|
713 allele_data_frame <- allele_data_frame %>%
|
|
714 select(-pop);
|
|
715 # Allele string for Allele.table in database.
|
|
716 allele_table_data_frame <- unite(allele_data_frame, alleles, 1:19696, sep=" ", remove=TRUE);
|
|
717 allele_table_data_frame <- setDT(allele_table_data_frame, keep.rownames=TRUE)[];
|
|
718 setnames(allele_table_data_frame, c("rn"), c("affy_id"));
|
|
719 # write.csv(concat_sample_alleles,file=paste("Seed_genotype_alleles.csv",sep = ""),quote=FALSE,row.names=FALSE);
|
|
720 write_data_frame(output_data_dir, "allele.tabular", allele_table_data_frame);
|
|
721
|
|
722 # Output the data frame for updating the experiment table.
|
|
723 experiment_table_data_frame <- data.frame(matrix(ncol=4, nrow=num_rows));
|
|
724 colnames(experiment_table_data_frame) <- c("seq_facility", "array_version", "result_folder_name", "plate_barcode");
|
|
725 for (i in 1:num_rows) {
|
|
726 experiment_table_data_frame$seq_facility[i] <- sample_prep_data_frame$seq_facility[i];
|
|
727 experiment_table_data_frame$array_version[i] <- sample_prep_data_frame$array_version[i];
|
|
728 experiment_table_data_frame$result_folder_name[i] <- sample_prep_data_frame$result_folder_name[i];
|
|
729 experiment_table_data_frame$plate_barcode[i] <- sample_prep_data_frame$plate_barcode[i];
|
|
730 }
|
|
731 write_data_frame(output_data_dir, "experiment.tabular", experiment_table_data_frame);
|
|
732
|
|
733 # Output the data frame for updating the colony table.
|
|
734 # The geographic_origin value is used for deciding into which table
|
|
735 # to insert the latitude and longitude values. If the geographic_origin
|
|
736 # is "reef", the values will be inserted into the reef table, and if it is
|
|
737 # "colony", the values will be inserted into the colony table. We insert
|
|
738 # these values in both data frames so that the downstream tool that parses
|
|
739 # them can determine the appropriate table.
|
|
740 colony_table_data_frame <- data.frame(matrix(ncol=4, nrow=num_rows));
|
|
741 colnames(colony_table_data_frame) <- c("latitude", "longitude", "depth", "geographic_origin");
|
|
742 for (i in 1:num_rows) {
|
|
743 colony_table_data_frame$latitude[i] <- sample_prep_data_frame$latitude[i];
|
|
744 colony_table_data_frame$longitude[i] <- sample_prep_data_frame$longitude[i];
|
|
745 colony_table_data_frame$depth[i] <- sample_prep_data_frame$depth[i];
|
|
746 colony_table_data_frame$geographic_origin[i] <- sample_prep_data_frame$geographic_origin[i];
|
|
747 }
|
|
748 write_data_frame(output_data_dir, "colony.tabular", colony_table_data_frame);
|
|
749
|
|
750 # Output the data frame for populating the genotype table.
|
|
751 # Combine with previously genotyped samples.
|
|
752 # prep_genotype_tibble looks like this:
|
|
753 # A tibble: 220 x 7
|
|
754 # Groups: group [?]
|
|
755 # group affy_id coral_mlg_clona… user_specimen_id db_match
|
|
756 # <int> <chr> <chr> <chr> <chr>
|
|
757 # 1 a10000… 13905 HG0048 match
|
|
758 # genetic_coral_species_call coral_mlg_rep_sample_id
|
|
759 # <chr> <chr>
|
|
760 # A.palmata 1104
|
|
761 prep_genotype_tibble <- id_data_table %>%
|
|
762 group_by(row_number()) %>%
|
|
763 dplyr::rename(group='row_number()') %>%
|
|
764 unnest(affy_id) %>%
|
|
765 left_join(smlg_data_frame %>%
|
|
766 select("affy_id", "coral_mlg_rep_sample_id", "coral_mlg_clonal_id", "user_specimen_id",
|
|
767 "genetic_coral_species_call", "bcoral_genet_id"),
|
|
768 by='affy_id');
|
|
769 # Confirm that the representative mlg is the same between runs.
|
|
770 uniques2 <- unique(prep_genotype_tibble[c("group", "coral_mlg_rep_sample_id")]);
|
|
771 uniques2 <- uniques2[!is.na(uniques2$coral_mlg_rep_sample_id),];
|
|
772 na.mlg3 <- which(is.na(prep_genotype_tibble$coral_mlg_rep_sample_id));
|
|
773 na.group2 <- prep_genotype_tibble$group[na.mlg3];
|
|
774 prep_genotype_tibble$coral_mlg_rep_sample_id[na.mlg3] <- uniques2$coral_mlg_rep_sample_id[match(na.group2, uniques2$group)];
|
|
775 # Transform the representative mlg column with new genotyped samples.
|
|
776 # representative_mlg_tibble looks like this:
|
|
777 # A tibble: 220 x 5
|
|
778 # affy_id coral_mlg_rep_sa… coral_mlg_clona… user_specimen_id
|
|
779 # <chr> <chr> <chr> <chr>
|
|
780 # a100000-… 13905 HG0048 13905
|
|
781 # genetic_coral_species_call bcoral_genet_id
|
|
782 # <chr> <chr>
|
|
783 # A.palmata C1651
|
|
784 representative_mlg_tibble <- prep_genotype_tibble %>%
|
|
785 mutate(coral_mlg_rep_sample_id=ifelse(is.na(coral_mlg_rep_sample_id), affy_id, coral_mlg_rep_sample_id)) %>%
|
|
786 ungroup() %>%
|
|
787 select(-group);
|
|
788 # prep_genotype_table_tibble looks like this:
|
|
789 # affy_id coral_mlg_clonal_id user_specimen_id db_match
|
|
790 # a550962...CEL HG0120 1090 match
|
|
791 # genetic_coral_species_call coral_mlg_rep_sample_id
|
|
792 # A.palmata 1104
|
|
793 prep_genotype_table_tibble <- stag_db_report %>%
|
|
794 select("affy_id", "coral_mlg_clonal_id", "user_specimen_id", "db_match", "genetic_coral_species_call") %>%
|
|
795 left_join(representative_mlg_tibble %>%
|
|
796 select("affy_id", "coral_mlg_rep_sample_id"),
|
|
797 by='affy_id');
|
|
798 # genotype_table_tibble looks like this:
|
|
799 # affy_id coral_mlg_clonal_id user_specimen_id db_match
|
|
800 # a550962-436.CEL HG0120 1090 match
|
|
801 # genetic_coral_species_call coral_mlg_rep_sample_id bcoral_genet_id
|
|
802 # A.palmata 1104 <NA>
|
|
803 genotype_table_tibble <- prep_genotype_table_tibble %>%
|
|
804 left_join(affy_metadata_data_frame %>%
|
|
805 select("user_specimen_id", "bcoral_genet_id"),
|
|
806 by='user_specimen_id');
|
|
807 write_data_frame(output_data_dir, "genotype.tabular", genotype_table_tibble);
|
|
808
|
|
809 # Output the file needed for populating the person table.
|
|
810 person_table_data_frame <- data.frame(matrix(ncol=4, nrow=num_rows));
|
|
811 colnames(person_table_data_frame) <- c("last_name", "first_name", "organization", "email");
|
|
812 # person_table_data_frame looks like this:
|
|
813 # last_name first_name organization email
|
|
814 # Kitchen Sheila Penn State s89@psu.edu
|
|
815 for (i in 1:num_rows) {
|
|
816 person_table_data_frame$last_name[i] <- sample_prep_data_frame$collector_last_name[i];
|
|
817 person_table_data_frame$first_name[i] <- sample_prep_data_frame$collector_first_name[i];
|
|
818 person_table_data_frame$organization[i] <- sample_prep_data_frame$organization[i];
|
|
819 person_table_data_frame$email[i] <- sample_prep_data_frame$email[i];
|
|
820 }
|
|
821 write_data_frame(output_data_dir, "person.tabular", person_table_data_frame);
|
|
822
|
|
823 # Output the file needed for populating the phenotype table.
|
|
824 phenotype_table_data_frame <- data.frame(matrix(ncol=7, nrow=num_rows));
|
|
825 colnames(phenotype_table_data_frame) <- c("disease_resist", "bleach_resist", "mortality", "tle",
|
|
826 "spawning", "sperm_motility", "healing_time");
|
|
827 # phenotype_table_data_frame looks like this:
|
|
828 # disease_resist bleach_resist mortality tle spawning sperm_motility healing_time
|
|
829 # NA NA NA NA False NA NA
|
|
830 for (i in 1:num_rows) {
|
|
831 phenotype_table_data_frame$disease_resist[i] <- sample_prep_data_frame$disease_resist[i];
|
|
832 phenotype_table_data_frame$bleach_resist[i] <- sample_prep_data_frame$bleach_resist[i];
|
|
833 phenotype_table_data_frame$mortality[i] <- sample_prep_data_frame$mortality[i];
|
|
834 phenotype_table_data_frame$tle[i] <- sample_prep_data_frame$tle[i];
|
|
835 phenotype_table_data_frame$spawning[i] <- sample_prep_data_frame$spawning[i];
|
|
836 phenotype_table_data_frame$sperm_motility[i] <- sample_prep_data_frame$sperm_motility[i];
|
|
837 phenotype_table_data_frame$healing_time[i] <- sample_prep_data_frame$healing_time[i];
|
|
838 }
|
|
839 write_data_frame(output_data_dir, "phenotype.tabular", phenotype_table_data_frame);
|
|
840
|
|
841 # Output the file needed for populating the reef table.
|
|
842 reef_table_data_frame <- data.frame(matrix(ncol=5, nrow=num_rows));
|
|
843 colnames(reef_table_data_frame) <- c("name", "region", "latitude", "longitude", "geographic_origin");
|
|
844 # The geographic_origin value is used for deciding into which table
|
|
845 # to insert the latitude and longitude values. If the geographic_origin
|
|
846 # is "reef", the values will be inserted into the reef table, and if it is
|
|
847 # "colony", the values will be inserted into the colony table. We insert
|
|
848 # these values in both data frames so that the downstream tool that parses
|
|
849 # them can determine the appropriate table.
|
|
850 # reef_table_data_frame looks like this:
|
|
851 # name region latitude longitude geographic_origin
|
|
852 # JohnsonsReef Bahamas 18.361733 -64.7743 Reef
|
|
853 for (i in 1:num_rows) {
|
|
854 reef_table_data_frame$name[i] <- sample_prep_data_frame$reef[i];
|
|
855 reef_table_data_frame$region[i] <- sample_prep_data_frame$region[i];
|
|
856 reef_table_data_frame$latitude[i] <- sample_prep_data_frame$latitude[i];
|
|
857 reef_table_data_frame$longitude[i] <- sample_prep_data_frame$longitude[i];
|
|
858 reef_table_data_frame$geographic_origin[i] <- sample_prep_data_frame$geographic_origin[i];
|
|
859 }
|
|
860 write_data_frame(output_data_dir, "reef.tabular", reef_table_data_frame);
|
|
861
|
|
862 # Output the file needed for populating the sample table.
|
|
863 sample_table_data_frame <- data.frame(matrix(ncol=20, nrow=num_rows));
|
|
864 colnames(sample_table_data_frame) <- c("affy_id", "colony_location", "collection_date", "user_specimen_id",
|
|
865 "registry_id", "depth", "dna_extraction_method", "dna_concentration",
|
|
866 "public", "public_after_date", "percent_missing_data_coral",
|
1
|
867 "percent_missing_data_sym", "percent_acerv_coral",
|
|
868 "percent_reference_sym", "percent_apalm_coral",
|
0
|
869 "percent_alternative_sym", "percent_heterozygous_coral",
|
|
870 "percent_heterozygous_sym", "field_call", "bcoral_genet_id");
|
|
871 for (i in 1:num_rows) {
|
|
872 sample_table_data_frame$affy_id[i] <- sample_prep_data_frame$affy_id[i];
|
|
873 sample_table_data_frame$colony_location[i] <- sample_prep_data_frame$colony_location[i];
|
|
874 sample_table_data_frame$collection_date[i] <- sample_prep_data_frame$collection_date[i];
|
|
875 sample_table_data_frame$user_specimen_id[i] <- sample_prep_data_frame$user_specimen_id[i];
|
|
876 sample_table_data_frame$registry_id[i] <- sample_prep_data_frame$registry_id[i];
|
|
877 sample_table_data_frame$depth[i] <- sample_prep_data_frame$depth[i];
|
|
878 sample_table_data_frame$dna_extraction_method[i] <- sample_prep_data_frame$dna_extraction_method[i];
|
|
879 sample_table_data_frame$dna_concentration[i] <- sample_prep_data_frame$dna_concentration[i];
|
|
880 sample_table_data_frame$public[i] <- sample_prep_data_frame$public[i];
|
|
881 sample_table_data_frame$public_after_date[i] <- sample_prep_data_frame$public_after_date[i];
|
|
882 sample_table_data_frame$percent_missing_data_coral[i] <- sample_prep_data_frame$percent_missing_data_coral[i];
|
|
883 sample_table_data_frame$percent_missing_data_sym[i] <- DEFAULT_MISSING_NUMERIC_VALUE;
|
1
|
884 sample_table_data_frame$percent_acerv_coral[i] <- sample_prep_data_frame$percent_acerv_coral[i];
|
0
|
885 sample_table_data_frame$percent_reference_sym[i] <- DEFAULT_MISSING_NUMERIC_VALUE;
|
1
|
886 sample_table_data_frame$percent_apalm_coral[i] <- sample_prep_data_frame$percent_apalm_coral[i];
|
0
|
887 sample_table_data_frame$percent_alternative_sym[i] <- DEFAULT_MISSING_NUMERIC_VALUE;
|
|
888 sample_table_data_frame$percent_heterozygous_coral[i] <- sample_prep_data_frame$percent_heterozygous_coral[i];
|
|
889 sample_table_data_frame$percent_heterozygous_sym[i] <- DEFAULT_MISSING_NUMERIC_VALUE;
|
|
890 sample_table_data_frame$field_call[i] <- sample_prep_data_frame$field_call[i];
|
|
891 sample_table_data_frame$bcoral_genet_id[i] <- sample_prep_data_frame$bcoral_genet_id[i];
|
|
892 }
|
|
893 write_data_frame(output_data_dir, "sample.tabular", sample_table_data_frame);
|
|
894
|
|
895 # Output the file needed for populating the taxonomy table.
|
|
896 # taxonomy_table_data_frame looks like this:
|
|
897 # genetic_coral_species_call affy_id genus_name species_name
|
|
898 # A.palmata a550962-4368120-060520-500_A05.CEL Acropora palmata
|
|
899 taxonomy_table_data_frame <- stag_db_report %>%
|
|
900 select(genetic_coral_species_call, affy_id) %>%
|
2
|
901 mutate(genus_name = ifelse(grepl("^A.*", genetic_coral_species_call), "Acropora",ifelse(!is.na(genetic_coral_species_call), "other", NA))) %>%
|
0
|
902 mutate(species_name = ifelse(genetic_coral_species_call == "A.palmata", "palmata", "other")) %>%
|
|
903 mutate(species_name = ifelse(genetic_coral_species_call == "A.cervicornis", "cervicornis", species_name)) %>%
|
|
904 mutate(species_name = ifelse(genetic_coral_species_call == "A.prolifera", "prolifera", species_name));
|
|
905 colnames(taxonomy_table_data_frame) <- c("genetic_coral_species_call", "affy_id", "genus_name", "species_name");
|
|
906 write_data_frame(output_data_dir, "taxonomy.tabular", taxonomy_table_data_frame);
|
|
907 time_elapsed(start_time);
|