* samtools-0.1.4-16 (r360)

[samtools.git] / bam_index.c

diff --git a/bam_index.c b/bam_index.c
index 7670c2f3f1058af65086cd40f62f3acbd288025d..347189d63e397d4d433cc42783aea3a07545e9e2 100644 (file)
--- a/bam_index.c
+++ b/bam_index.c
@@ -155,7 +155,8 @@ bam_index_t *bam_index_core(bamFile fp)
                        last_tid = c->tid;
                        last_bin = 0xffffffffu;
                } else if (last_coor > c->pos) {
-                       fprintf(stderr, "[bam_index_core] the alignment is not sorted. Abort!\n");
+                       fprintf(stderr, "[bam_index_core] the alignment is not sorted (%s): %u > %u in %d-th chr\n",
+                                       bam1_qname(b), last_coor, c->pos, c->tid+1);
                        exit(1);
                }
                if (b->core.tid >= 0 && b->core.bin < 4681) insert_offset2(&idx->index2[b->core.tid], b, last_off);
@@ -268,7 +269,16 @@ bam_index_t *bam_index_load(const char *fn)
 
        fnidx = (char*)calloc(strlen(fn) + 5, 1);
        strcpy(fnidx, fn); strcat(fnidx, ".bai");
-       if ((fp = fopen(fnidx, "r")) == 0) {
+       fp = fopen(fnidx, "r");
+       if (fp == 0) { // try "{base}.bai"
+               char *s = strstr(fn, "bam");
+               if (s == fn + strlen(fn) - 3) {
+                       strcpy(fnidx, fn);
+                       fnidx[strlen(fn)-1] = 'i';
+                       fp = fopen(fnidx, "r");
+               }
+       }
+       if (fp == 0) {
                fprintf(stderr, "[bam_index_load] the alignment is not indexed. Please run `index' command first. Abort!\n");
                exit(1);
        }
@@ -327,7 +337,7 @@ bam_index_t *bam_index_load(const char *fn)
        return idx;
 }
 
-int bam_index_build(const char *fn)
+int bam_index_build2(const char *fn, const char *_fnidx)
 {
        char *fnidx;
        FILE *fpidx;
@@ -336,9 +346,16 @@ int bam_index_build(const char *fn)
        assert(fp = bam_open(fn, "r"));
        idx = bam_index_core(fp);
        bam_close(fp);
-       fnidx = (char*)calloc(strlen(fn) + 5, 1);
-       strcpy(fnidx, fn); strcat(fnidx, ".bai");
-       assert(fpidx = fopen(fnidx, "w"));
+       if (_fnidx == 0) {
+               fnidx = (char*)calloc(strlen(fn) + 5, 1);
+               strcpy(fnidx, fn); strcat(fnidx, ".bai");
+       } else fnidx = strdup(_fnidx);
+       fpidx = fopen(fnidx, "w");
+       if (fpidx == 0) {
+               fprintf(stderr, "[bam_index_build2] fail to create the index file.\n");
+               free(fnidx);
+               return 1;
+       }
        bam_index_save(idx, fpidx);
        bam_index_destroy(idx);
        fclose(fpidx);
@@ -346,13 +363,19 @@ int bam_index_build(const char *fn)
        return 0;
 }
 
+int bam_index_build(const char *fn)
+{
+       return bam_index_build2(fn, 0);
+}
+
 int bam_index(int argc, char *argv[])
 {
        if (argc < 2) {
-               fprintf(stderr, "Usage: samtools index <in.bam>\n");
+               fprintf(stderr, "Usage: samtools index <in.bam> [<out.index>]\n");
                return 1;
        }
-       bam_index_build(argv[1]);
+       if (argc >= 3) bam_index_build2(argv[1], argv[2]);
+       else bam_index_build(argv[1]);
        return 0;
 }