htslib 0.1.0 → 0.2.0

data/lib/hts/bcf.rb

@@ -9,6 +9,7 @@ require_relative "bcf/format"
 require_relative "bcf/record"
 
 module HTS
+  # A class for working with VCF, BCF files.
   class Bcf < Hts
     include Enumerable
 
@@ -26,7 +27,7 @@ module HTS
       file
     end
 
-    def initialize(file_name, mode = "r", index: nil, fai: nil, threads: nil,
+    def initialize(file_name, mode = "r", index: nil, threads: nil,
                    create_index: false)
       if block_given?
         message = "HTS::Bcf.new() dose not take block; Please use HTS::Bcf.open() instead"
@@ -42,32 +43,31 @@ module HTS
 
       raise Errno::ENOENT, "Failed to open #{@file_name}" if @hts_file.null?
 
-      if threads&.> 0
-        r = LibHTS.hts_set_threads(@hts_file, threads)
-        raise "Failed to set number of threads: #{threads}" if r < 0
-      end
+      set_threads(threads) if threads
 
       return if @mode[0] == "w"
 
       @header = Bcf::Header.new(@hts_file)
-
       create_index(index) if create_index
-
       @idx = load_index(index)
-
       @start_position = tell
+      super # do nothing
     end
 
     def create_index(index_name = nil)
+      check_closed
+
       warn "Create index for #{@file_name} to #{index_name}"
-      if index
-        LibHTS.bcf_index_build2(@hts_file, index_name, -1)
+      if index_name
+        LibHTS.bcf_index_build2(@file_name, index_name, -1)
       else
-        LibHTS.bcf_index_build(@hts_file, -1)
+        LibHTS.bcf_index_build(@file_name, -1)
       end
     end
 
     def load_index(index_name = nil)
+      check_closed
+
       if index_name
         LibHTS.bcf_index_load2(@file_name, index_name)
       else
@@ -76,39 +76,50 @@ module HTS
     end
 
     def index_loaded?
+      check_closed
+
       !@idx.null?
     end
 
     def write_header
-      raise IOError, "closed stream" if closed?
+      check_closed
 
       @header = header.dup
       LibHTS.hts_set_fai_filename(header, @file_name)
-      LibHTS.bcf_hdr_write(@hts_file, header.struct)
+      LibHTS.bcf_hdr_write(@hts_file, header)
     end
 
     def write(var)
-      raise IOError, "closed stream" if closed?
+      check_closed
 
-      var_dup = var.dup = var.dup
+      var_dup = var.dup
       LibHTS.bcf_write(@hts_file, header, var_dup) > 0 || raise
     end
 
     # Close the current file.
 
     def nsamples
+      check_closed
+
       header.nsamples
     end
 
     def samples
+      check_closed
+
       header.samples
     end
 
-    # Iterate over each record.
-    # Generate a new Record object each time.
-    # Slower than each.
-    def each_copy
-      raise IOError, "closed stream" if closed?
+    def each(copy: false, &block)
+      if copy
+        each_record_copy(&block)
+      else
+        each_record_reuse(&block)
+      end
+    end
+
+    private def each_record_copy
+      check_closed
 
       return to_enum(__method__) unless block_given?
 
@@ -119,12 +130,10 @@ module HTS
       self
     end
 
-    # Iterate over each record.
-    # Record object is reused.
-    # Faster than each_copy.
-    def each
-      raise IOError, "closed stream" if closed?
-
+    private def each_record_reuse
+      check_closed
+      # Each does not always start at the beginning of the file.
+      # This is the common behavior of IO objects in Ruby.
       return to_enum(__method__) unless block_given?
 
       bcf1 = LibHTS.bcf_init
@@ -132,5 +141,76 @@ module HTS
       yield record while LibHTS.bcf_read(@hts_file, header, bcf1) != -1
       self
     end
+
+    # @!macro [attach] define_getter
+    #   @method $1
+    #   Get $1 array
+    #   @return [Array] the $1 array
+    define_getter :chrom
+    define_getter :pos
+    define_getter :endpos
+    define_getter :id
+    define_getter :ref
+    define_getter :alt
+    define_getter :qual
+    define_getter :filter
+
+    def info(key = nil)
+      check_closed
+      position = tell
+      if key
+        ary = map { |r| r.info(key) }
+      else
+        raise NotImplementedError
+        # ary = each_copy.map { |r| r.info }
+        # ary = map { |r| r.info.clone }
+      end
+      seek(position)
+      ary
+    end
+
+    def format(key = nil)
+      check_closed
+      position = tell
+      if key
+        ary = map { |r| r.format(key) }
+      else
+        raise NotImplementedError
+        # ary = each_copy.map { |r| r.format }
+        # ary = map { |r| r.format.clone }
+      end
+      seek(position)
+      ary
+    end
+
+    # @!macro [attach] define_iterator
+    #   @method each_$1
+    #   Get $1 iterator
+    define_iterator :chrom
+    define_iterator :pos
+    define_iterator :endpos
+    define_iterator :id
+    define_iterator :ref
+    define_iterator :alt
+    define_iterator :qual
+    define_iterator :filter
+
+    def each_info(key)
+      check_closed
+      return to_enum(__method__) unless block
+
+      each do |r|
+        yield r.info(key)
+      end
+    end
+
+    def each_format(key)
+      check_closed
+      return to_enum(__method__) unless block
+
+      each do |r|
+        yield r.format(key)
+      end
+    end
   end
 end

data/lib/hts/faidx.rb

@@ -6,22 +6,32 @@ module HTS
   class Faidx
     attr_reader :file_name
 
-    class << self
-      alias open new
+    def self.open(*args, **kw)
+      file = new(*args, **kw) # do not yield
+      return file unless block_given?
+
+      begin
+        yield file
+      ensure
+        file.close
+      end
+      file
     end
 
     def initialize(file_name)
+      if block_given?
+        message = "HTS::Faidx.new() dose not take block; Please use HTS::Faidx.open() instead"
+        raise message
+      end
+
       @file_name = file_name
       @fai = LibHTS.fai_load(@file_name)
 
-      # IO like API
-      if block_given?
-        begin
-          yield self
-        ensure
-          close
-        end
-      end
+      raise Errno::ENOENT, "Failed to open #{@file_name}" if @fai.null?
+    end
+
+    def struct
+      @fai
     end
 
     def close
@@ -29,10 +39,10 @@ module HTS
     end
 
     # the number of sequences in the index.
-    def size
+    def length
       LibHTS.faidx_nseq(@fai)
     end
-    alias length size
+    alias size length
 
     # return the length of the requested chromosome.
     def chrom_size(chrom)
@@ -48,10 +58,10 @@ module HTS
     alias chrom_length chrom_size
 
     # FIXME: naming and syntax
-    def cget; end
+    # def cget; end
 
     # FIXME: naming and syntax
-    def get; end
+    # def get; end
 
     # __iter__
   end

data/lib/hts/hts.rb

@@ -3,7 +3,38 @@
 require_relative "../htslib"
 
 module HTS
+  # A base class for hts files.
   class Hts
+    class << self
+      private
+
+      def define_getter(name)
+        define_method(name) do
+          check_closed
+          position = tell
+          ary = map(&name)
+          seek(position)
+          ary
+        end
+      end
+
+      def define_iterator(name)
+        define_method("each_#{name}") do |&block|
+          check_closed
+          return to_enum(__method__) unless block
+
+          each do |record|
+            block.call(record.public_send(name))
+          end
+          self
+        end
+      end
+    end
+
+    def initialize(*args)
+      # do nothing
+    end
+
     def struct
       @hts_file
     end
@@ -12,11 +43,11 @@ module HTS
       @hts_file.to_ptr
     end
 
-    def format
+    def file_format
       LibHTS.hts_get_format(@hts_file)[:format].to_s
     end
 
-    def format_version
+    def file_format_version
       v = LibHTS.hts_get_format(@hts_file)[:version]
       major = v[:major]
       minor = v[:minor]
@@ -38,6 +69,16 @@ module HTS
       @hts_file.nil? || @hts_file.null?
     end
 
+    def set_threads(n)
+      raise TypeError unless n.is_a(Integer)
+
+      if n > 0
+        r = LibHTS.hts_set_threads(@hts_file, n)
+        raise "Failed to set number of threads: #{threads}" if r < 0
+      end
+      self
+    end
+
     def seek(offset)
       if @hts_file[:is_cram] == 1
         LibHTS.cram_seek(@hts_file[:fp][:cram], offset, IO::SEEK_SET)
@@ -70,5 +111,11 @@ module HTS
         raise "Cannot rewind: no start position"
       end
     end
+
+    private
+
+    def check_closed
+      raise IOError, "closed stream" if closed?
+    end
   end
 end

data/lib/hts/libhts/bgzf.rb

@@ -18,19 +18,19 @@ module HTS
     # Open an existing hFILE stream for reading or writing.
     attach_function \
       :bgzf_hopen,
-      [HFILE, :string],
+      [HFile, :string],
       BGZF.by_ref
 
     # Close the BGZF and free all associated resources.
     attach_function \
       :bgzf_close,
-      [HFILE],
+      [HFile],
       :int
 
     # Read up to _length_ bytes from the file storing into _data_.
     attach_function \
       :bgzf_read,
-      [HFILE, :pointer, :size_t],
+      [HFile, :pointer, :size_t],
       :ssize_t
 
     # Write _length_ bytes from _data_ to the file. If no I/O errors occur,
@@ -181,7 +181,7 @@ module HTS
     # Load BGZF index from an hFILE
     attach_function \
       :bgzf_index_load_hfile,
-      [BGZF, HFILE, :string],
+      [BGZF, HFile, :string],
       :int
 
     # Save BGZF index
@@ -193,7 +193,7 @@ module HTS
     # Write a BGZF index to an hFILE
     attach_function \
       :bgzf_index_dump_hfile,
-      [BGZF, HFILE, :string],
+      [BGZF, HFile, :string],
       :int
   end
 end

data/lib/hts/libhts/constants.rb

@@ -1,6 +1,7 @@
 # frozen_string_literal: true
 
 module HTS
+  # Module for working with C HTSlib.
   module LibHTS
     typedef :int64, :hts_pos_t
     typedef :pointer, :bam_plp_auto_f
@@ -24,9 +25,9 @@ module HTS
         :f,              :pointer # kstream_t
     end
 
-    # HFILE
+    # HFile
 
-    class HFILE < FFI::BitStruct
+    class HFile < FFI::BitStruct
       layout \
         :buffer,        :string,
         :begin,         :string,
@@ -56,7 +57,7 @@ module HTS
         :uncompressed_block,     :pointer,
         :compressed_block,       :pointer,
         :cache,                  :pointer,
-        :fp,                     HFILE.ptr,
+        :fp,                     HFile.ptr,
         :mt,                     :pointer,
         :idx,                    :pointer,
         :idx_build_otf,          :int,
@@ -189,6 +190,16 @@ module HTS
         )
     end
 
+    class HtsReglist < FFI::Struct
+      layout \
+        :reg,            :string,
+        :intervals,      :pointer, # hts_pair_pos_t
+        :tid,            :int,
+        :count,          :uint32_t,
+        :min_beg,        :hts_pos_t,
+        :max_end,        :hts_pos_t
+    end
+
     # HtsFile
     class SamHdr < FFI::Struct
       layout \
@@ -217,7 +228,7 @@ module HTS
         union_layout(
           :bgzf,         BGZF.ptr,
           :cram,         :pointer, # cram_fd
-          :hfile,        HFILE.ptr
+          :hfile,        HFile.ptr
         ),
         :state,          :pointer,
         :format,         HtsFormat,
@@ -263,7 +274,7 @@ module HTS
         :n_reg,          :int,
         :beg,            :int64,
         :end,            :int64,
-        :reg_list,       :pointer,
+        :reg_list,       :pointer, # HtsReglist.ptr,
         :curr_tid,       :int,
         :curr_reg,       :int,
         :curr_intv,      :int,
@@ -516,5 +527,15 @@ module HTS
         LibHTS.bcf_destroy(ptr) unless ptr.null?
       end
     end
+
+    CramContentType = enum(
+      :ct_error,           -1,
+      :file_header,        0,
+      :compression_header, 1,
+      :mapped_slice,       2,
+      :unmapped_slice,     3, # cram v1.0 only
+      :external,           4,
+      :core,               5
+    )
   end
 end