annotate ideas_preprocessor.R @ 3:e97851e8951b draft

Uploaded
author greg
date Tue, 23 Jan 2018 10:48:38 -0500
parents 91c5dbb14a13
children 2b4e1bd725f7
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
1
91c5dbb14a13 Uploaded
greg
parents:
diff changeset
1 #!/usr/bin/env Rscript
91c5dbb14a13 Uploaded
greg
parents:
diff changeset
2
91c5dbb14a13 Uploaded
greg
parents:
diff changeset
3 suppressPackageStartupMessages(library("data.table"))
91c5dbb14a13 Uploaded
greg
parents:
diff changeset
4 suppressPackageStartupMessages(library("optparse"))
91c5dbb14a13 Uploaded
greg
parents:
diff changeset
5
91c5dbb14a13 Uploaded
greg
parents:
diff changeset
6 option_list <- list(
3
e97851e8951b Uploaded
greg
parents: 1
diff changeset
7 make_option(c("--bed_input"), action="store", dest="bed_input", defaul=NULL, help="Chromosome windows positions file"),
e97851e8951b Uploaded
greg
parents: 1
diff changeset
8 make_option(c("--exclude_input"), action="store", dest="exclude_input", defaul=NULL, help="File(s) containing regions to exclude"),
e97851e8951b Uploaded
greg
parents: 1
diff changeset
9 make_option(c("--bychr"), action="store_true", dest="bychr", defaul=FALSE, help="Separate files by chromosome"),
e97851e8951b Uploaded
greg
parents: 1
diff changeset
10 make_option(c("--chrom_len_file"), action="store", dest="chrom_len_file", default=NULL, help="Chromosome lengths file"),
e97851e8951b Uploaded
greg
parents: 1
diff changeset
11 make_option(c("--output"), action="store", dest="output", help="Primary output dataset"),
e97851e8951b Uploaded
greg
parents: 1
diff changeset
12 make_option(c("--output_files_path"), action="store", dest="output_files_path", help="Primary output dataset extra files path"),
e97851e8951b Uploaded
greg
parents: 1
diff changeset
13 make_option(c("--prep_input_config"), action="store", dest="prep_input_config", help="Preprocessing input config file"),
e97851e8951b Uploaded
greg
parents: 1
diff changeset
14 make_option(c("--prep_output_config"), action="store", dest="prep_output_config", help="Preprocessing output config file"),
e97851e8951b Uploaded
greg
parents: 1
diff changeset
15 make_option(c("--reads_per_bp"), action="store", dest="reads_per_bp", type="integer", help="Calculate the signal in each genomic window"),
e97851e8951b Uploaded
greg
parents: 1
diff changeset
16 make_option(c("--restrict_to_chroms"), action="store", dest="restrict_to_chroms", default=NULL, help="Restrict processing to specified chromosomes"),
e97851e8951b Uploaded
greg
parents: 1
diff changeset
17 make_option(c("--standardize_datasets"), action="store_true", dest="standardize_datasets", default=FALSE, help="Standardize all datasets"),
e97851e8951b Uploaded
greg
parents: 1
diff changeset
18 make_option(c("--windows_positions_by_chroms_config"), action="store", dest="windows_positions_by_chroms_config", default=NULL, help="Windows positions by chroms config file"),
e97851e8951b Uploaded
greg
parents: 1
diff changeset
19 make_option(c("--window_size"), action="store", dest="window_size", type="integer", default=NULL, help="Window size in base pairs")
1
91c5dbb14a13 Uploaded
greg
parents:
diff changeset
20 )
91c5dbb14a13 Uploaded
greg
parents:
diff changeset
21
91c5dbb14a13 Uploaded
greg
parents:
diff changeset
22 parser <- OptionParser(usage="%prog [options] file", option_list=option_list)
91c5dbb14a13 Uploaded
greg
parents:
diff changeset
23 args <- parse_args(parser, positional_arguments=TRUE)
91c5dbb14a13 Uploaded
greg
parents:
diff changeset
24 opt <- args$options
91c5dbb14a13 Uploaded
greg
parents:
diff changeset
25
91c5dbb14a13 Uploaded
greg
parents:
diff changeset
26 create_primary_html = function(output, output_files_path) {
3
e97851e8951b Uploaded
greg
parents: 1
diff changeset
27 files = list.files(path=output_files_path, pattern="\\.txt");
1
91c5dbb14a13 Uploaded
greg
parents:
diff changeset
28 s <- paste('<html><head></head><body>', sep="\n");
91c5dbb14a13 Uploaded
greg
parents:
diff changeset
29 s <- paste(s, '<h3>Contents of directory required by IDEAS</h3>\n', sep="");
91c5dbb14a13 Uploaded
greg
parents:
diff changeset
30 s <- paste(s, '<ul>\n', sep="");
3
e97851e8951b Uploaded
greg
parents: 1
diff changeset
31 for (i in 1:length(files)) {
e97851e8951b Uploaded
greg
parents: 1
diff changeset
32 s <- paste(s, '<li><a href="', files[i], '">', files[i], '</a></li>\n', sep="");
1
91c5dbb14a13 Uploaded
greg
parents:
diff changeset
33 }
3
e97851e8951b Uploaded
greg
parents: 1
diff changeset
34 s <- paste(s, '<li><a href="tmp">tmp</a></li>\n', sep="");
1
91c5dbb14a13 Uploaded
greg
parents:
diff changeset
35 s <- paste(s, '</ul>\n</body>\n</html>', sep="");
91c5dbb14a13 Uploaded
greg
parents:
diff changeset
36 cat(s, file=output);
91c5dbb14a13 Uploaded
greg
parents:
diff changeset
37 }
91c5dbb14a13 Uploaded
greg
parents:
diff changeset
38
3
e97851e8951b Uploaded
greg
parents: 1
diff changeset
39 tmp_dir = "tmp";
e97851e8951b Uploaded
greg
parents: 1
diff changeset
40 output_tmp_dir = paste(opt$output_files_path, tmp_dir, sep="/");
e97851e8951b Uploaded
greg
parents: 1
diff changeset
41 dir.create(output_tmp_dir, showWarnings=FALSE);
1
91c5dbb14a13 Uploaded
greg
parents:
diff changeset
42
91c5dbb14a13 Uploaded
greg
parents:
diff changeset
43 # Read the prep_input_config text file which has this format:
91c5dbb14a13 Uploaded
greg
parents:
diff changeset
44 # "cell type name" "epigenetic factor name" "file path" "file name" "datatype"
91c5dbb14a13 Uploaded
greg
parents:
diff changeset
45 prep_input_config_matrix = as.matrix(read.table(opt$prep_input_config));
91c5dbb14a13 Uploaded
greg
parents:
diff changeset
46 # Process data to windows mean.
91c5dbb14a13 Uploaded
greg
parents:
diff changeset
47 for (i in 1:dim(prep_input_config_matrix)[1]) {
91c5dbb14a13 Uploaded
greg
parents:
diff changeset
48 file_path = prep_input_config_matrix[i, 3]
91c5dbb14a13 Uploaded
greg
parents:
diff changeset
49 file_name = prep_input_config_matrix[i, 4]
91c5dbb14a13 Uploaded
greg
parents:
diff changeset
50 datatype = prep_input_config_matrix[i, 5]
91c5dbb14a13 Uploaded
greg
parents:
diff changeset
51 if (datatype == "bam") {
3
e97851e8951b Uploaded
greg
parents: 1
diff changeset
52 cmd = paste("samtools index", file_path);
e97851e8951b Uploaded
greg
parents: 1
diff changeset
53 system(cmd);
e97851e8951b Uploaded
greg
parents: 1
diff changeset
54 bigwig_file_name = paste(file_name, "bw", sep=".");
e97851e8951b Uploaded
greg
parents: 1
diff changeset
55 cmd = paste("bamCoverage --bam", file_path, "-o", bigwig_file_name, "--binSize", opt$window_size);
e97851e8951b Uploaded
greg
parents: 1
diff changeset
56 system(cmd);
1
91c5dbb14a13 Uploaded
greg
parents:
diff changeset
57 } else {
3
e97851e8951b Uploaded
greg
parents: 1
diff changeset
58 bigwig_file_name = file_path;
1
91c5dbb14a13 Uploaded
greg
parents:
diff changeset
59 }
3
e97851e8951b Uploaded
greg
parents: 1
diff changeset
60 bed_file_name = paste(file_name, "bed", sep=".");
e97851e8951b Uploaded
greg
parents: 1
diff changeset
61 bed_file_path = paste("tmp", bed_file_name, sep="/");
e97851e8951b Uploaded
greg
parents: 1
diff changeset
62 cmd = paste("bigWigAverageOverBed", bigwig_file_name, opt$bed_input, "stdout | cut -f5 >", bed_file_path);
e97851e8951b Uploaded
greg
parents: 1
diff changeset
63 system(cmd);
e97851e8951b Uploaded
greg
parents: 1
diff changeset
64 cmd = paste("gzip -f", bed_file_path);
e97851e8951b Uploaded
greg
parents: 1
diff changeset
65 system(cmd);
1
91c5dbb14a13 Uploaded
greg
parents:
diff changeset
66 }
91c5dbb14a13 Uploaded
greg
parents:
diff changeset
67
3
e97851e8951b Uploaded
greg
parents: 1
diff changeset
68 # Create file1.txt.
e97851e8951b Uploaded
greg
parents: 1
diff changeset
69 cmd = paste("cut -d' '", opt$prep_input_config, "-f1,2 > file1.txt", sep=" ");
e97851e8951b Uploaded
greg
parents: 1
diff changeset
70 system(cmd);
e97851e8951b Uploaded
greg
parents: 1
diff changeset
71 # Compress the bed files in the tmp directory.
e97851e8951b Uploaded
greg
parents: 1
diff changeset
72 tmp_gzipped_files = paste(tmp_dir, "*.bed.gz", sep="/");
e97851e8951b Uploaded
greg
parents: 1
diff changeset
73 # Create file2.txt.
e97851e8951b Uploaded
greg
parents: 1
diff changeset
74 cmd = paste("ls", tmp_gzipped_files, "> file2.txt", sep=" ");
e97851e8951b Uploaded
greg
parents: 1
diff changeset
75 system(cmd);
e97851e8951b Uploaded
greg
parents: 1
diff changeset
76 # Create the prep_output_config with the format required by IDEAS.
e97851e8951b Uploaded
greg
parents: 1
diff changeset
77 cmd = paste("paste -d' ' file1.txt file2.txt >", opt$prep_output_config, sep=" ");
e97851e8951b Uploaded
greg
parents: 1
diff changeset
78 system(cmd);
e97851e8951b Uploaded
greg
parents: 1
diff changeset
79 # Move the prep_output_config to the output directory.
e97851e8951b Uploaded
greg
parents: 1
diff changeset
80 to_path = paste(opt$output_files_path, opt$prep_output_config, sep="/");
e97851e8951b Uploaded
greg
parents: 1
diff changeset
81 file.rename(opt$prep_output_config, to_path);
e97851e8951b Uploaded
greg
parents: 1
diff changeset
82 # Move the compressed bed files in the tmp
e97851e8951b Uploaded
greg
parents: 1
diff changeset
83 # directory to the output tmp directory.
e97851e8951b Uploaded
greg
parents: 1
diff changeset
84 tmp_files = list.files(path=tmp_dir);
e97851e8951b Uploaded
greg
parents: 1
diff changeset
85 for (i in 1:length(tmp_files)) {
e97851e8951b Uploaded
greg
parents: 1
diff changeset
86 from_path = paste(tmp_dir, tmp_files[i], sep="/");
e97851e8951b Uploaded
greg
parents: 1
diff changeset
87 to_path = paste(output_tmp_dir, tmp_files[i], sep="/");
e97851e8951b Uploaded
greg
parents: 1
diff changeset
88 file.rename(from_path, to_path);
e97851e8951b Uploaded
greg
parents: 1
diff changeset
89 }
e97851e8951b Uploaded
greg
parents: 1
diff changeset
90 if (!is.null(opt$windows_positions_by_chroms_config)) {
e97851e8951b Uploaded
greg
parents: 1
diff changeset
91 # Move the windows_positions_by_chroms_config to the output directory.
e97851e8951b Uploaded
greg
parents: 1
diff changeset
92 to_path = paste(opt$output_files_path, opt$windows_positions_by_chroms_config, sep="/");
e97851e8951b Uploaded
greg
parents: 1
diff changeset
93 file.rename(opt$windows_positions_by_chroms_config, to_path);
e97851e8951b Uploaded
greg
parents: 1
diff changeset
94 }
1
91c5dbb14a13 Uploaded
greg
parents:
diff changeset
95 # Create the primary HTML dataset.
91c5dbb14a13 Uploaded
greg
parents:
diff changeset
96 create_primary_html(opt$output, opt$output_files_path);