##
# FILTER UC FILE BY MIN SAMPLES
#
require 'set'

module Lederhosen
  class CLI

    desc "uc_filter",
         "filter UCLUST output to remove small, infrequent clusters"

    method_option :input,    :type => :string,  :required => true
    method_option :output,   :type => :string,  :required => true
    method_option :reads,    :type => :numeric, :required => true
    method_option :samples,  :type => :numeric, :required => true

    def uc_filter
      input   = options[:input]
      output  = options[:output]
      reads   = options[:reads].to_i
      samples = options[:samples].to_i

      ohai "filtering #{input} to #{output}, reads = #{reads} & samples = #{samples}"

      # load UC file
      ohai "loading uc file"
      clstr_info   = Helpers.load_uc_file input
      clstr_counts = clstr_info[:clstr_counts] # clstr_counts[:clstr][sample.to_i] = reads

      # filter
      ohai "filtering"
      survivors = clstr_counts.reject do |a, b|
        b.reject{ |i, j| j < reads }.length < samples
      end

      surviving_clusters = survivors.keys.to_set

      # print filtered uc file
      ohai "saving filtered table"
      out = File.open(output, 'w')

      lines = `wc -l #{input}`.split.first.to_i

      pbar = ProgressBar.new 'saving', lines
      kept, total = 1, 0

      File.open(input) do |handle|
        pbar = ProgressBar.new 'saving', File.size(input)
        handle.each do |line|
          # output lederhosen filtering information because I often
          # forget to write this down :)
          out.puts "# filtered: #{input}"
          out.puts "# #{reads} reads in at least #{samples} samples"

          pbar.inc handle.pos
          if line =~ /^#/
            out.print line
            next
          end
          total += 1

          # check if cluster is in surviving clusters
          if surviving_clusters.include? line.split[1].to_i
            out.print line
            kept += 1
          end

        end
        pbar.finish
      end

      out.close

      ohai "clusters: #{surviving_clusters.length}/#{clstr_counts.keys.length} = #{100*surviving_clusters.length/clstr_counts.keys.length.to_f}%"
      ohai "reads:    #{kept}/#{total} = #{100*kept/total.to_f}%"
    end
  end

end