File: merge_fasta_btab.pl

package info (click to toggle)
fasta3 36.3.8i.14-Nov-2020-3
  • links: PTS, VCS
  • area: main
  • in suites: sid, trixie
  • size: 7,016 kB
  • sloc: ansic: 77,269; perl: 10,677; python: 2,461; sh: 428; csh: 86; sql: 55; makefile: 40
file content (380 lines) | stat: -rwxr-xr-x 9,415 bytes parent folder | download | duplicates (3)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
#!/usr/bin/perl

################################################################
# copyright (c) 2018 by William R. Pearson and The Rector &
# Visitors of the University of Virginia */
################################################################
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing,
# software distributed under this License is distributed on an "AS
# IS" BASIS, WITHOUT WRRANTIES OR CONDITIONS OF ANY KIND, either
# express or implied.  See the License for the specific language
# governing permissions and limitations under the License.
################################################################

################################################################
# merge_fasta_btab.pl --btab .btab file html_file
################################################################

################################################################
# takes a standard (or <html> output FASTA file and converts (or adds) labels using .btab information
################################################################


use warnings;
use strict;
use Getopt::Long;
use Pod::Usage;
use URI::Encode qw(uri_encode);
use URI::Escape qw(uri_escape);

my ($btab_file, $have_qslen, $help, $shelp, $dom_info) = ("", 0, 0, 0, 0);
my ($plot_url) = ("");

GetOptions(
    "btab_file|btab=s" => \$btab_file,
    "have_qslen|have_sqlen" => \$have_qslen,
    "have_qslen|have_sqlen!" => \$have_qslen,
    "domain_info|dom_info!" => \$dom_info,
    "plot_url=s"=> \$plot_url,
    "h|?" => \$shelp,
    "help" => \$help,
     );

pod2usage(1) if $shelp;
pod2usage(exitstatus => 0, verbose => 2) if $help;
unless (-f STDIN || -p STDIN || @ARGV) {
 pod2usage(1);
}

# require a btab file

# read it in, save structure as list/hash on accession (list more robust)
# what happens with multiple hits for same library -- need to add code
#

my @bl_fields = qw(q_seqid s_seqid percid alen mismatch gopen q_start q_end s_start s_end evalue bits score annot);

if ($have_qslen) {
  @bl_fields = qw(q_seqid q_len s_seqid s_len percid alen mismatch gopen q_start q_end s_start s_end evalue bits score annot);
}

my %pgm_names= ('FASTA'=>'fap', 'FASTX'=>'fx', 'FASTY'=>'fy', 'FASTS'=>'fs', 'FASTM'=>'fm',
		'SSEARCH' => 'gsw',  'GGSEARCH'=>'gnw', 'GLSEARCH'=>'lnw',
		'TFASTX' => 'tfx', 'TFASTY'=>'tfx', 'TFASTS'=>'tfs', 'TFASTM'=>'tfm',
		'BLASTP'=>'bp', 'BLASTN'=>'bn', 'TBLASTN'=>'tbn' );

if ($dom_info) {
  push @bl_fields, "dom_info";
}

my $pgm_name = '';
my %tab_data = ();
my @sseq_ids = ();

unless ($btab_file) {
  die "--btab_file required"
}
else {
  # read in btab file
  open(my $fd, $btab_file) || die "cannot open $btab_file";

  while (my $line = <$fd>) {
    if ($line =~ m/^#/) { # check for program name
      if (!$pgm_name) {
	my ($name) = ($line =~ m/^# (\w+) /);
	if ($name && $pgm_names{$name}) {
	  $pgm_name = $pgm_names{$name};
	}
      }
      next;
    }
    chomp($line);

    my %a_data = ();
    @a_data{@bl_fields} = split(/\t/,$line);

    # here we should confirm that the sseqid is new.  If it is not, then add to a list.
    my $sseqid = $a_data{'s_seqid'};

    if (defined($tab_data{$sseqid})) {
      push @{$tab_data{$sseqid}}, \%a_data
    }
    else {
      $tab_data{$sseqid} =  [\%a_data ];
      push @sseq_ids, $sseqid;
    }
  }
}

# have the annotation data in %tab_data{} and @seq_ids
# read in the blastp html file and annotate it

my ($in_best, $in_align, $in_annot) = (0,0,0);
my ($annot_id) = ("");
my ($best_ix, $align_ix, $hsp_ix) = (0,0,0);

while (my $line = <>) {
  chomp($line);
  unless ($line) {
    print "\n";
    next;
  }
  if ($line =~ m/^The best scores are:/) {
    $in_best = 1;
    $best_ix = 0;
    print "$line\n";
    next;
  }

  if ($in_best) {
    if ($line =~ /<pre>>>/) {
      $in_best = 0;
      $in_align = 1;
      $in_annot = 0;
      $align_ix = 0;
      $hsp_ix = 0;
      # print out the first line
      print "$line\n";
      next;
    }
    else {
      if (scalar(@sseq_ids) && $sseq_ids[$best_ix]) {
	$line = add_best($line, $tab_data{$sseq_ids[$best_ix]}->[0]);
	$best_ix++;
      }
    }
  }

  if ($in_align) {
    if ($line =~ m/^<!\-\- ANNOT_START "([^"]+)" \-\->/) {
      $annot_id = $1;
      my $regions_str = regions_to_str($tab_data{$sseq_ids[$align_ix]}->[$hsp_ix]);
      print qq(<!-- ANNOT_START "$annot_id" -->);
      print $regions_str;

      if ($plot_url) {
	my $raw_dom_str = "";
	if ($dom_info) {
	  $raw_dom_str = dom_info_str($tab_data{$sseq_ids[$align_ix]}->[$hsp_ix]{'dom_info'});
	}

	my $plot_tag = plot_tag_str($plot_url, $pgm_name, $tab_data{$sseq_ids[$align_ix]}->[$hsp_ix], $regions_str, $raw_dom_str);
	if ($plot_tag) {print $plot_tag,"\n";}
      }

      $hsp_ix++;

      # remove the old domain information */
      while ($line = <> ) {
	chomp($line);
	if ($line !~ m/^\s*q?Region:/ && $line !~ /ANNOT_STOP/) {
	  print "$line\n";
	}
	if ($line =~ m/^<!\-\- ANNOT_STOP \-\->/) {
	  last;
	}
      }
    }
    elsif ($line =~ m/<pre>>>/) {
      $align_ix++;
      $hsp_ix=0;
    }
  }

  print "$line\n";
}

sub parse_annots {
  my ($annot_str) = @_;

  my @annot_list = ();

  unless ($annot_str && $annot_str =~ m/^\|/) {
    return \@annot_list;
  }

  my @annots = split('\|',$annot_str);
  shift @annots;

  for my $annot ( @annots ) {
    my %annot_data = ();
    next unless ($annot =~ m/^[XR][RX]/);
    my @a_fields = split(/;/,$annot);
    for my $f (@a_fields) {
      if ($f =~ m/^[XR][XR]/) {
	my @a2_f = split(':',$f);
	if ($a2_f[0] =~ m/^XR/) {
	  $annot_data{target} = 'subj';
	}
	else {
	  $annot_data{target} = 'query';
	}
	$annot_data{coord} = "$a2_f[1]:$a2_f[2]";
	$annot_data{score} = (split('=',$a2_f[3]))[1]
      }
      elsif ($f =~ m/(\w)=(.+)/) {
	$annot_data{$1} = $2;
      }
    }
    $annot_data{name} = $a_fields[-1];
    $annot_data{name} =~ s/^C=//;

    push @annot_list, \%annot_data;
  }
  return \@annot_list;
}

sub print_regions {
  my ($annot_id, $annot_ref) = @_;

  my $region_str = "";

  print qq(<!-- ANNOT_START "$annot_id" -->);

  for my $annot ( @{$annot_ref}) {
    if ($annot->{target} =~ m/^q/) {
      $region_str = "qRegion";
    }
    else {
      $region_str = " Region";
    }

    printf "%s: %s : score=%d; bits=%.1f; Id=%.3f; Q=%.1f : %s\n", $region_str,
      @{$annot}{qw(coord score b I Q name)};
  }
}

sub regions_to_str {
  my ($a_data_r) = @_;

  my $annot_ref = parse_annots($a_data_r->{annot});

  my $region_str = "";
  my $annot_str = "";

  for my $annot ( @{$annot_ref}) {
    if ($annot->{target} =~ m/^q/) {
      $region_str = "qRegion";
    }
    else {
      $region_str = " Region";
    }

    $annot_str .= sprintf "%s: %s : score=%d; bits=%.1f; Id=%.3f; Q=%.1f : %s\n", $region_str,
      @{$annot}{qw(coord score b I Q name)};
  }
  return $annot_str;
}

sub add_best {
  my ($line, $a_data) = @_;

  my $annot_str = '';

  my $annot_refs = parse_annots($a_data->{annot});

  # remove old annotation if present
  my @line_words = split(/\s/,$line);
  if ($line_words[-1] =~ m/~\d/) {
    $line = join(' ',@line_words[0 .. $#line_words-1]);
  }

  for my $annot ( @$annot_refs) {
    if ($annot->{target} !~ m/^q/) {
      $annot_str .= $annot->{name} . ";"
    }
  }

  if ($annot_str) {
    return "$line  $annot_str";
  }
  else {
    return $line;
  }
}

sub plot_tag_str {

  my ($plot_script, $pgm_name, $align_data_r, $regions_str, $doms_str) = @_;

  my $svg_pref =  q(<object type="image/svg+xml" );
  my $svg_post =  q( width="660" height="76" ></object>);

  #build argument string
  my %plt_args = ();
  @plt_args{qw(pgm q_cstart l_cstart)} = ($pgm_name, 1, 1);
  @plt_args{qw(q_name q_cstop q_astart q_astop l_name l_cstop l_astart l_astop)} =
    @{$align_data_r}{qw(q_seqid q_len q_start q_end s_seqid s_len s_start s_end)};
  $plt_args{'regions'}= uri_escape(uri_encode($regions_str));
  if ($doms_str) {
    $plt_args{'doms'} = uri_encode($doms_str);
  }

  my $dom_info = ();

  my @args = map {"$_=$plt_args{$_}"} keys(%plt_args);

  return $svg_pref . qq( data="$plot_url?) . join('&amp;',@args) . '"' . $svg_post;
}

sub dom_info_str {
  my ($raw_dom_info) = @_;

  my $dom_str = "";
  
  unless ($raw_dom_info) { return "";}

  my @raw_doms = split('\|',$raw_dom_info);
  shift(@raw_doms);

  for my $dom ( @raw_doms ) {
    my $tmp_dom = $dom;
    $tmp_dom =~ s/^DX:/qDomain:\t/g;
    $tmp_dom =~ s/^XD:/lDomain:\t/g;
    $tmp_dom =~ s/;C=/\t/g;

    $dom_str .= "$tmp_dom\n";
  }

  return $dom_str;
}

__END__

=pod

=head1 NAME

merge_blast_btab.pl

=head1 SYNOPSIS

merge_blast_btab.pl --btab_file=result.b_tab result.html

=head1 OPTIONS

 -h	short help
 --help include description

 --btab_file|--btab file_name -- blast tabular output file with
   sub-alignment scoring

=head1 DESCRIPTION

C<merge_blast_btab.pl> merges the domain annotations and sub-alignment scoring from C<annot_blast_btop2.pl> blast tabular output file with a conventional blast result file.

The tab file is read and parsed, and then the subject/query seqid is used to
capture domain locations in the subject/query sequence.  If the domains
overlap the aligned region, the domain names are appended to the output.

=head1 AUTHOR

William R. Pearson, wrp@virginia.edu

=cut