Mercurial > repos > lsong10 > psiclass
diff PsiCLASS-1.0.2/samtools-0.1.19/misc/ace2sam.c @ 0:903fc43d6227 draft default tip
Uploaded
author | lsong10 |
---|---|
date | Fri, 26 Mar 2021 16:52:45 +0000 |
parents | |
children |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/PsiCLASS-1.0.2/samtools-0.1.19/misc/ace2sam.c Fri Mar 26 16:52:45 2021 +0000 @@ -0,0 +1,249 @@ +/* The MIT License + + Copyright (c) 2011 Heng Li <lh3@live.co.uk> + + Permission is hereby granted, free of charge, to any person obtaining + a copy of this software and associated documentation files (the + "Software"), to deal in the Software without restriction, including + without limitation the rights to use, copy, modify, merge, publish, + distribute, sublicense, and/or sell copies of the Software, and to + permit persons to whom the Software is furnished to do so, subject to + the following conditions: + + The above copyright notice and this permission notice shall be + included in all copies or substantial portions of the Software. + + THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, + EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF + MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND + NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS + BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN + ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN + CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE + SOFTWARE. +*/ + +#include <stdio.h> +#include <stdlib.h> +#include <string.h> +#include <unistd.h> +#include <zlib.h> +#include "kstring.h" +#include "kseq.h" +KSTREAM_INIT(gzFile, gzread, 16384) + +#define N_TMPSTR 5 +#define LINE_LEN 60 + +// append a CIGAR operation plus length +#define write_cigar(_c, _n, _m, _v) do { \ + if (_n == _m) { \ + _m = _m? _m<<1 : 4; \ + _c = realloc(_c, _m * sizeof(unsigned)); \ + } \ + _c[_n++] = (_v); \ + } while (0) + +// a fatal error +static void fatal(const char *msg) +{ + fprintf(stderr, "E %s\n", msg); + exit(1); +} +// remove pads +static void remove_pads(const kstring_t *src, kstring_t *dst) +{ + int i, j; + dst->l = 0; + kputsn(src->s, src->l, dst); + for (i = j = 0; i < dst->l; ++i) + if (dst->s[i] != '*') dst->s[j++] = dst->s[i]; + dst->s[j] = 0; + dst->l = j; +} + +int main(int argc, char *argv[]) +{ + gzFile fp; + kstream_t *ks; + kstring_t s, t[N_TMPSTR]; + int dret, i, k, af_n, af_max, af_i, c, is_padded = 0, write_cns = 0, *p2u = 0; + long m_cigar = 0, n_cigar = 0; + unsigned *af, *cigar = 0; + + while ((c = getopt(argc, argv, "pc")) >= 0) { + switch (c) { + case 'p': is_padded = 1; break; + case 'c': write_cns = 1; break; + } + } + if (argc == optind) { + fprintf(stderr, "\nUsage: ace2sam [-pc] <in.ace>\n\n"); + fprintf(stderr, "Options: -p output padded SAM\n"); + fprintf(stderr, " -c write the contig sequence in SAM\n\n"); + fprintf(stderr, "Notes: 1. Fields must appear in the following order: (CO->[BQ]->(AF)->(RD->QA))\n"); + fprintf(stderr, " 2. The order of reads in AF and in RD must be identical\n"); + fprintf(stderr, " 3. Except in BQ, words and numbers must be separated by a single SPACE or TAB\n"); + fprintf(stderr, " 4. This program writes the headerless SAM to stdout and header to stderr\n\n"); + return 1; + } + + s.l = s.m = 0; s.s = 0; + af_n = af_max = af_i = 0; af = 0; + for (i = 0; i < N_TMPSTR; ++i) t[i].l = t[i].m = 0, t[i].s = 0; + fp = strcmp(argv[1], "-")? gzopen(argv[optind], "r") : gzdopen(fileno(stdin), "r"); + ks = ks_init(fp); + while (ks_getuntil(ks, 0, &s, &dret) >= 0) { + if (strcmp(s.s, "CO") == 0) { // contig sequence + kstring_t *cns; + t[0].l = t[1].l = t[2].l = t[3].l = t[4].l = 0; // 0: name; 1: padded ctg; 2: unpadded ctg/padded read; 3: unpadded read; 4: SAM line + af_n = af_i = 0; // reset the af array + ks_getuntil(ks, 0, &s, &dret); kputs(s.s, &t[0]); // contig name + ks_getuntil(ks, '\n', &s, &dret); // read the whole line + while (ks_getuntil(ks, '\n', &s, &dret) >= 0 && s.l > 0) kputsn(s.s, s.l, &t[1]); // read the padded consensus sequence + remove_pads(&t[1], &t[2]); // construct the unpadded sequence + // compute the array for mapping padded positions to unpadded positions + p2u = realloc(p2u, t[1].m * sizeof(int)); + for (i = k = 0; i < t[1].l; ++i) { + p2u[i] = k; + if (t[1].s[i] != '*') ++k; + } + // write out the SAM header and contig sequences + fprintf(stderr, "H @SQ\tSN:%s\tLN:%ld\n", t[0].s, t[is_padded?1:2].l); // The SAM header line + cns = &t[is_padded?1:2]; + fprintf(stderr, "S >%s\n", t[0].s); + for (i = 0; i < cns->l; i += LINE_LEN) { + fputs("S ", stderr); + for (k = 0; k < LINE_LEN && i + k < cns->l; ++k) + fputc(cns->s[i + k], stderr); + fputc('\n', stderr); + } + +#define __padded2cigar(sp) do { \ + int i, l_M = 0, l_D = 0; \ + for (i = 0; i < sp.l; ++i) { \ + if (sp.s[i] == '*') { \ + if (l_M) write_cigar(cigar, n_cigar, m_cigar, l_M<<4); \ + ++l_D; l_M = 0; \ + } else { \ + if (l_D) write_cigar(cigar, n_cigar, m_cigar, l_D<<4 | 2); \ + ++l_M; l_D = 0; \ + } \ + } \ + if (l_M) write_cigar(cigar, n_cigar, m_cigar, l_M<<4); \ + else write_cigar(cigar, n_cigar, m_cigar, l_D<<4 | 2); \ + } while (0) + + if (write_cns) { // write the consensus SAM line (dummy read) + n_cigar = 0; + if (is_padded) __padded2cigar(t[1]); + else write_cigar(cigar, n_cigar, m_cigar, t[2].l<<4); + kputsn(t[0].s, t[0].l, &t[4]); kputs("\t516\t", &t[4]); kputsn(t[0].s, t[0].l, &t[4]); kputs("\t1\t60\t", &t[4]); + for (i = 0; i < n_cigar; ++i) { + kputw(cigar[i]>>4, &t[4]); kputc("MIDNSHP=X"[cigar[i]&0xf], &t[4]); + } + kputs("\t*\t0\t0\t", &t[4]); kputsn(t[2].s, t[2].l, &t[4]); kputs("\t*", &t[4]); + } + } else if (strcmp(s.s, "BQ") == 0) { // contig quality + if (t[0].l == 0) fatal("come to 'BQ' before reading 'CO'"); + if (dret != '\n') ks_getuntil(ks, '\n', &s, &dret); // read the entire "BQ" line + if (write_cns) t[4].s[--t[4].l] = 0; // remove the trailing "*" + for (i = 0; i < t[2].l; ++i) { // read the consensus quality + int q; + if (ks_getuntil(ks, 0, &s, &dret) < 0) fprintf(stderr, "E truncated contig quality\n"); + if (s.l) { + q = atoi(s.s) + 33; + if (q > 126) q = 126; + if (write_cns) kputc(q, &t[4]); + } else --i; + } + if (dret != '\n') ks_getuntil(ks, '\n', &s, &dret); + ks_getuntil(ks, '\n', &s, &dret); // skip the empty line + if (write_cns) puts(t[4].s); t[4].l = 0; + } else if (strcmp(s.s, "AF") == 0) { // padded read position + int reversed, neg, pos; + if (t[0].l == 0) fatal("come to 'AF' before reading 'CO'"); + if (write_cns) { + if (t[4].l) puts(t[4].s); + t[4].l = 0; + } + ks_getuntil(ks, 0, &s, &dret); // read name + ks_getuntil(ks, 0, &s, &dret); reversed = s.s[0] == 'C'? 1 : 0; // strand + ks_getuntil(ks, 0, &s, &dret); pos = atoi(s.s); neg = pos < 0? 1 : 0; pos = pos < 0? -pos : pos; // position + if (af_n == af_max) { // double the af array + af_max = af_max? af_max<<1 : 4; + af = realloc(af, af_max * sizeof(unsigned)); + } + af[af_n++] = pos << 2 | neg << 1 | reversed; // keep the placement information + } else if (strcmp(s.s, "RD") == 0) { // read sequence + if (af_i >= af_n) fatal("more 'RD' records than 'AF'"); + t[2].l = t[3].l = t[4].l = 0; + ks_getuntil(ks, 0, &t[4], &dret); // QNAME + if (dret != '\n') ks_getuntil(ks, '\n', &s, &dret); // read the entire RD line + while (ks_getuntil(ks, '\n', &s, &dret) >= 0 && s.l > 0) kputs(s.s, &t[2]); // read the read sequence + } else if (strcmp(s.s, "QA") == 0) { // clipping + if (af_i >= af_n) fatal("more 'QA' records than 'AF'"); + int beg, end, pos, op; + ks_getuntil(ks, 0, &s, &dret); ks_getuntil(ks, 0, &s, &dret); // skip quality clipping + ks_getuntil(ks, 0, &s, &dret); beg = atoi(s.s) - 1; // align clipping start + ks_getuntil(ks, 0, &s, &dret); end = atoi(s.s); // clipping end + if (dret != '\n') ks_getuntil(ks, '\n', &s, &dret); + // compute 1-based POS + pos = af[af_i]>>2; // retrieve the position information + if (af[af_i]>>1&1) pos = -pos; + pos += beg; // now pos is the true padded position + // generate CIGAR + remove_pads(&t[2], &t[3]); // backup the unpadded read sequence + n_cigar = 0; + if (beg) write_cigar(cigar, n_cigar, m_cigar, beg<<4|4); + if (is_padded) { + __padded2cigar(t[2]); + if (beg && n_cigar > 1) cigar[1] -= beg<<4; // fix the left-hand CIGAR + if (end < t[2].l && n_cigar) cigar[n_cigar-1] -= (t[2].l - end)<<4; // fix the right-hand CIGAR + } else { + // generate flattened CIGAR string + for (i = beg, k = pos - 1; i < end; ++i, ++k) + t[2].s[i] = t[2].s[i] != '*'? (t[1].s[k] != '*'? 0 : 1) : (t[1].s[k] != '*'? 2 : 6); + // generate the proper CIGAR + for (i = beg + 1, k = 1, op = t[2].s[beg]; i < end; ++i) { + if (op != t[2].s[i]) { + write_cigar(cigar, n_cigar, m_cigar, k<<4|op); + op = t[2].s[i]; k = 1; + } else ++k; + } + write_cigar(cigar, n_cigar, m_cigar, k<<4|op); + // remove unnecessary "P" and possibly merge adjacent operations + for (i = 2; i < n_cigar; ++i) { + if ((cigar[i]&0xf) != 1 && (cigar[i-1]&0xf) == 6 && (cigar[i-2]&0xf) != 1) { + cigar[i-1] = 0; + if ((cigar[i]&0xf) == (cigar[i-2]&0xf)) // merge operations + cigar[i] += cigar[i-2], cigar[i-2] = 0; + } + } + for (i = k = 0; i < n_cigar; ++i) // squeeze out dumb operations + if (cigar[i]) cigar[k++] = cigar[i]; + n_cigar = k; + } + if (end < t[2].l) write_cigar(cigar, n_cigar, m_cigar, (t[2].l - end)<<4|4); + // write the SAM line for the read + kputc('\t', &t[4]); // QNAME has already been written + kputw((af[af_i]&1)? 16 : 0, &t[4]); kputc('\t', &t[4]); // FLAG + kputsn(t[0].s, t[0].l, &t[4]); kputc('\t', &t[4]); // RNAME + kputw(is_padded? pos : p2u[pos-1]+1, &t[4]); // POS + kputs("\t60\t", &t[4]); // MAPQ + for (i = 0; i < n_cigar; ++i) { // CIGAR + kputw(cigar[i]>>4, &t[4]); kputc("MIDNSHP=X"[cigar[i]&0xf], &t[4]); + } + kputs("\t*\t0\t0\t", &t[4]); // empty MRNM, MPOS and TLEN + kputsn(t[3].s, t[3].l, &t[4]); // unpadded SEQ + kputs("\t*", &t[4]); // QUAL + puts(t[4].s); // print to stdout + ++af_i; + } else if (dret != '\n') ks_getuntil(ks, '\n', &s, &dret); + } + ks_destroy(ks); + gzclose(fp); + free(af); free(s.s); free(cigar); free(p2u); + for (i = 0; i < N_TMPSTR; ++i) free(t[i].s); + return 0; +}