mirror of
https://github.com/MillironX/XAM.jl.git
synced 2024-11-23 02:09:55 +00:00
Merge pull request #56 from jonathanBieler/bam_index_method
Added method for handling different bam index types.
This commit is contained in:
commit
8272cea71e
2 changed files with 22 additions and 7 deletions
|
@ -8,7 +8,7 @@ Create a data reader of the BAM file format.
|
|||
|
||||
# Arguments
|
||||
* `input`: data source
|
||||
* `index=nothing`: filepath to a random access index (currently *bai* is supported)
|
||||
* `index=nothing`: filepath to a random access index (currently *bai* is supported) or BAI object
|
||||
"""
|
||||
mutable struct Reader{T} <: BioGenerics.IO.AbstractReader
|
||||
stream::BGZFStreams.BGZFStream{T}
|
||||
|
@ -28,13 +28,8 @@ function BioGenerics.IO.stream(reader::Reader)
|
|||
end
|
||||
|
||||
function Reader(input::IO; index=nothing)
|
||||
if isa(index, AbstractString)
|
||||
index = BAI(index)
|
||||
elseif index != nothing
|
||||
error("unrecognizable index argument")
|
||||
end
|
||||
reader = init_bam_reader(input)
|
||||
reader.index = index
|
||||
reader.index = init_bam_index(index)
|
||||
return reader
|
||||
end
|
||||
|
||||
|
@ -125,6 +120,11 @@ function init_bam_reader(input::IO)
|
|||
return init_bam_reader(BGZFStreams.BGZFStream(input))
|
||||
end
|
||||
|
||||
init_bam_index(index::AbstractString) = BAI(index)
|
||||
init_bam_index(index::BAI) = index
|
||||
init_bam_index(index::Nothing) = nothing
|
||||
init_bam_index(index) = error("unrecognizable index argument")
|
||||
|
||||
function _read!(reader::Reader, record)
|
||||
unsafe_read(
|
||||
reader.stream,
|
||||
|
|
|
@ -241,6 +241,21 @@
|
|||
|
||||
end
|
||||
|
||||
@testset "BAI" begin
|
||||
|
||||
filepath = joinpath(bamdir, "GSE25840_GSM424320_GM06985_gencode_spliced.head.bam")
|
||||
|
||||
index = BAM.BAI(filepath * ".bai")
|
||||
reader = open(BAM.Reader, filepath, index=index)
|
||||
|
||||
@test isa(eachoverlap(reader, "chr1", 1:100), BAM.OverlapIterator)
|
||||
|
||||
close(reader)
|
||||
|
||||
@test_throws ErrorException open(BAM.Reader, filepath, index=1234)
|
||||
|
||||
end
|
||||
|
||||
@testset "Random access" begin
|
||||
filepath = joinpath(bamdir, "GSE25840_GSM424320_GM06985_gencode_spliced.head.bam")
|
||||
reader = open(BAM.Reader, filepath, index=filepath * ".bai")
|
||||
|
|
Loading…
Reference in a new issue