bamtk.c

   1 #include <stdio.h>
   2 #include <unistd.h>
   3 #include "bam.h"
   4
   5 #ifndef PACKAGE_VERSION
   6 #define PACKAGE_VERSION "0.1.1-19"
   7 #endif
   8
   9 int bam_taf2baf(int argc, char *argv[]);
  10 int bam_pileup(int argc, char *argv[]);
  11 int bam_merge(int argc, char *argv[]);
  12 int bam_index(int argc, char *argv[]);
  13 int bam_sort(int argc, char *argv[]);
  14 int bam_tview_main(int argc, char *argv[]);
  15 int bam_mating(int argc, char *argv[]);
  16 int bam_rmdup(int argc, char *argv[]);
  17
  18 int faidx_main(int argc, char *argv[]);
  19 int glf_view_main(int argc, char *argv[]);
  20
  21 static int view_aux(const bam1_t *b, void *data)
  22 {
  23         bam_view1((bam_header_t*)data, b);
  24         return 0;
  25 }
  26 static int view_auxb(const bam1_t *b, void *data)
  27 {
  28         bam_write1((bamFile)data, b);
  29         return 0;
  30 }
  31
  32 int bam_view(int argc, char *argv[])
  33 {
  34         bamFile fp, fpout = 0;
  35         bam_header_t *header;
  36         bam1_t *b;
  37         int ret, c, is_bam = 0;
  38         while ((c = getopt(argc, argv, "b")) >= 0) {
  39                 switch (c) {
  40                 case 'b': is_bam = 1; break;
  41                 default: fprintf(stderr, "Unrecognized option: -%c\n", c); return 1;
  42                 }
  43         }
  44         if (argc == optind) {
  45                 fprintf(stderr, "Usage: samtools view [-b] <in.bam> [<region> [...]]\n");
  46                 return 1;
  47         }
  48         fp = strcmp(argv[optind], "-")? bam_open(argv[optind], "r") : bam_dopen(fileno(stdin), "r");
  49         assert(fp);
  50         header = bam_header_read(fp);
  51         if (is_bam) {
  52                 assert(fpout = bam_dopen(fileno(stdout), "w"));
  53                 bam_header_write(fpout, header);
  54         }
  55         if (optind + 1 == argc) {
  56                 b = (bam1_t*)calloc(1, sizeof(bam1_t));
  57                 while ((ret = bam_read1(fp, b)) >= 0) bam_view1(header, b);
  58                 if (ret < -1) fprintf(stderr, "[bam_view] truncated file? Continue anyway. (%d)\n", ret);
  59                 free(b->data); free(b);
  60         } else {
  61                 int i;
  62                 bam_index_t *idx;
  63                 idx = bam_index_load(argv[optind]);
  64                 for (i = optind + 1; i < argc; ++i) {
  65                         int tid, beg, end;
  66                         bam_parse_region(header, argv[i], &tid, &beg, &end);
  67                         if (is_bam) bam_fetch(fp, idx, tid, beg, end, fpout, view_auxb);
  68                         else bam_fetch(fp, idx, tid, beg, end, header, view_aux);
  69                 }
  70                 bam_index_destroy(idx);
  71         }
  72         bam_header_destroy(header);
  73         bam_close(fp);
  74         if (is_bam) bam_close(fpout);
  75         return 0;
  76 }
  77
  78 static int usage()
  79 {
  80         fprintf(stderr, "\n");
  81         fprintf(stderr, "Program: samtools (Tools for alignments in the SAM format)\n");
  82         fprintf(stderr, "Version: %s\n\n", PACKAGE_VERSION);
  83         fprintf(stderr, "Usage:   samtools <command> [options]\n\n");
  84         fprintf(stderr, "Command: import      import from the text format\n");
  85         fprintf(stderr, "         view        export to the text format\n");
  86         fprintf(stderr, "         sort        sort alignment file\n");
  87         fprintf(stderr, "         merge       merge multiple sorted alignment files\n");
  88         fprintf(stderr, "         pileup      generate pileup output\n");
  89         fprintf(stderr, "         faidx       index/extract FASTA\n");
  90 #ifndef _NO_CURSES
  91         fprintf(stderr, "         tview       text alignment viewer\n");
  92 #endif
  93         fprintf(stderr, "         index       index alignment\n");
  94         fprintf(stderr, "         fixmate     fix mate information\n");
  95         fprintf(stderr, "         rmdup       remove PCR duplicates\n");
  96         fprintf(stderr, "         glfview     print GLFv2 file\n");
  97         fprintf(stderr, "\n");
  98         return 1;
  99 }
 100
 101 int main(int argc, char *argv[])
 102 {
 103         if (argc < 2) return usage();
 104         if (strcmp(argv[1], "view") == 0) return bam_view(argc-1, argv+1);
 105         else if (strcmp(argv[1], "import") == 0) return bam_taf2baf(argc-1, argv+1);
 106         else if (strcmp(argv[1], "pileup") == 0) return bam_pileup(argc-1, argv+1);
 107         else if (strcmp(argv[1], "merge") == 0) return bam_merge(argc-1, argv+1);
 108         else if (strcmp(argv[1], "sort") == 0) return bam_sort(argc-1, argv+1);
 109         else if (strcmp(argv[1], "index") == 0) return bam_index(argc-1, argv+1);
 110         else if (strcmp(argv[1], "faidx") == 0) return faidx_main(argc-1, argv+1);
 111         else if (strcmp(argv[1], "fixmate") == 0) return bam_mating(argc-1, argv+1);
 112         else if (strcmp(argv[1], "rmdup") == 0) return bam_rmdup(argc-1, argv+1);
 113         else if (strcmp(argv[1], "glfview") == 0) return glf_view_main(argc-1, argv+1);
 114 #ifndef _NO_CURSES
 115         else if (strcmp(argv[1], "tview") == 0) return bam_tview_main(argc-1, argv+1);
 116 #endif
 117         else {
 118                 fprintf(stderr, "[main] unrecognized command '%s'\n", argv[1]);
 119                 return 1;
 120         }
 121         return 0;
 122 }