C++使用htslib庫讀入和寫出bam文件

  有時候咱們須要使用C++處理bam文件,好比取出read1或者read2等符合特定條件的序列,根據cigar值對序列指定位置的鹼基進行統計或者對序列進行處理並輸出等,這時咱們可使用htslib庫。htslib能夠用來處理SAM, BAM,CRAM 和VCF文件,是samtools、bcftools的核心庫。c++

#include <stdio.h>
#include <stdlib.h>
#include <htslib/sam.h>

using namespace std; 

#define bam_is_read1(b) (((b)->core.flag&BAM_FREAD1) != 0)

uint8_t Base[16] = {0,65,67,0,71,0,0,0,84,0,0,0,0,0,0,78};

int main(int argc, char **argv)
{
    bam_hdr_t *header;
    bam1_t *aln = bam_init1();

    samFile *in = sam_open(argv[1], "r");
    htsFile *outR1 = hts_open(argv[2], "wb");
    header = sam_hdr_read(in);
    if (sam_hdr_write(outR1, header) < 0) {
        fprintf(stderr, "Error writing output.\n");
        exit(-1);
    }
    uint8_t *seq;
    int32_t lseq;
    uint32_t *cigar;
    char* qname;
    while (sam_read1(in, header, aln) >= 0) {
        if (bam_is_read1(aln)){
            sam_write1(outR1, header, aln);
        }
        else {
            seq = bam_get_seq(aln);
            lseq = aln->core.l_qseq;
            qname  = bam_get_qname(aln);
            printf("%s\n",qname);
            cigar = bam_get_cigar(aln);
            for(int i=0; i < aln->core.n_cigar;++i){
                int icigar = cigar[i];
                printf("%d%d\n",bam_cigar_op(icigar),bam_cigar_oplen(icigar));
            }
            for(int i=0; i < lseq;++i){
               printf("%c", Base[bam_seqi(seq, i)]);
            }
            printf("\n");

        }
    }
    sam_close(in);
    sam_close(outR1);
}

cigar值存儲形式

32位int,經過bam_get_cigar得到地址,aln->core.n_cigar返回cigar operation的個數git

  • 低 4位存儲cigar operation;經過函數bam_cigar_op()得到operation
  • 高28位存儲cigar值的長度;經過函數,bam_cigar_oplen得到

seq存儲形式

8位int,4位存儲一個鹼基,1,2,4,8,15分別表明A、C、G、T、N,高四位存儲座標數較小的鹼基,可經過bam_seqi(seq,i)遍歷。



github

參考資料

htslib sam.h文件:https://github.com/samtools/htslib/blob/develop/htslib/sam.h
htslib sam文件格式說明:https://github.com/samtools/hts-specs/blob/master/SAMv1.pdf函數

相關文章
相關標籤/搜索