* samtools-0.1.6-18 (r493)

[samtools.git] / bam.c

diff --git a/bam.c b/bam.c
index 8b517565026f276d860f91634789ec8fa2f36195..ee7642b3434248c9b44ef918435f150f101cae00 100644 (file)
--- a/bam.c
+++ b/bam.c
@@ -292,9 +292,12 @@ void bam_view1(const bam_header_t *header, const bam1_t *b)
        free(s);
 }
 
-const char *bam_get_library(const bam_header_t *header, const bam1_t *b)
+// FIXME: we should also check the LB tag associated with each alignment
+const char *bam_get_library(bam_header_t *h, const bam1_t *b)
 {
        const uint8_t *rg;
+       if (h->dict == 0) h->dict = sam_header_parse2(h->text);
+       if (h->rg2lib == 0) h->rg2lib = sam_header2tbl(h->dict, "RG", "ID", "LB");
        rg = bam_aux_get(b, "RG");
-       return (rg == 0)? 0 : sam_tbl_get(header->rg2lib, (const char*)(rg + 1));
+       return (rg == 0)? 0 : sam_tbl_get(h->rg2lib, (const char*)(rg + 1));
 }