variant_filters_iterator.h

Go to the documentation of this file.

#ifndef gamgee__variant_filters_iterator__guard
#define gamgee__variant_filters_iterator__guard

#include "variant_filters.h"

#include "../utils/hts_memory.h"

#include "htslib/vcf.h"

#include <string>
#include <memory>

namespace gamgee {

class VariantFiltersIterator : public std::iterator<std::random_access_iterator_tag, std::string> {
 public:

  VariantFiltersIterator(const std::shared_ptr<bcf_hdr_t>& header, const std::shared_ptr<bcf1_t>& body, const uint32_t position) : 
    m_header {header},
    m_body {body},
    m_position {position} 
  {}

  std::string operator[](int index) const { return utils::htslib_filter_name(m_header.get(), m_body.get(), index); }                                      
  std::string operator*() const { return utils::htslib_filter_name(m_header.get(), m_body.get(), m_position); }                                           
  VariantFiltersIterator operator++() { m_position++; return *this; }                                                                                     
  VariantFiltersIterator operator--() { m_position--; return *this; }                                                                                     
  VariantFiltersIterator operator+=(const int n) { m_position += n; return *this; }                                                                       
  VariantFiltersIterator operator-=(const int n) { m_position -= n; return *this; }                                                                       
  bool operator!=(const VariantFiltersIterator& other) { return m_position != other.m_position || m_body != other.m_body || m_header != other.m_header; } 
  bool operator==(const VariantFiltersIterator& other) { return m_position == other.m_position && m_body == other.m_body && m_header == other.m_header; } 
  bool operator<=(const VariantFiltersIterator& other) { return m_position <= other.m_position && m_body == other.m_body && m_header == other.m_header; } 
  bool operator>=(const VariantFiltersIterator& other) { return m_position >= other.m_position && m_body == other.m_body && m_header == other.m_header; } 
  bool operator< (const VariantFiltersIterator& other) { return m_position < other.m_position && m_body == other.m_body && m_header == other.m_header;  } 
  bool operator> (const VariantFiltersIterator& other) { return m_position > other.m_position && m_body == other.m_body && m_header == other.m_header;  } 

  uint32_t size() const { return uint32_t(m_body->d.n_flt); } 

 private:
  std::shared_ptr<bcf_hdr_t> m_header; 
  std::shared_ptr<bcf1_t> m_body;      
  uint32_t m_position;                 
};

}

#endif // gamgee__variant_filters_iterator__guard