cpp-lazy-stream/include/stream/streams/streams.hpp

#pragma once

#include <algorithm>
#include <memory>
#include <numeric>
#include <vector>

#define DELEGATE_ITERATOR_IMPL_BASE(impl) \
  bool operator==(iterator const&other) const { \
    return impl == other.impl; \
  }

#define DELEGATE_ITERATOR_IMPL(impl) \
  DELEGATE_ITERATOR_IMPL_BASE(impl) \
  iterator& operator++() { \
    ++impl; \
    return *this; \
  }

namespace stream {
#define STREAM_ITERATOR_COPY() \
  copy(other.copy), dereference(other.dereference), compare(other.compare), destroy(other.destroy), advance(other.advance), type_(other.type_)
  
  template <typename T>
  class iterator {
  public:
    using value_type = typename std::remove_reference<T>::type;
    using reference = value_type &;
    using pointer = value_type *;
    using difference_type = std::ptrdiff_t;
    using iterator_category = std::forward_iterator_tag;
  public:
    iterator() = default;
    
    template <typename Iter>
    iterator(Iter impl) {
      copy = [](void * p) { return (void*) new Iter(*(Iter*)(p)); };
      dereference = [](void * p) -> T { return **((Iter*)p); };
      compare = [](void * l, void * r) { return *((Iter*)l) == *((Iter*)r); };
      destroy = [](void * p) { delete (Iter*)(p); };
      advance = [](void * p) { ++(*(Iter*)(p)); };
      type_ = typeid(Iter).name();
      impl_ = copy(&impl);
    }
    
    iterator(iterator const& other)
    : STREAM_ITERATOR_COPY()
    , impl_(copy(other.impl_)) {
    }
    
    iterator(iterator && other)
    : STREAM_ITERATOR_COPY()
    , impl_(other.impl_) {
      other.impl_ = nullptr;
    }
    
    iterator & operator=(iterator const & other) {
      return *this = iterator{other};
    }

    iterator & operator=(iterator && other) {
      swap(*this, other);
      return *this;
    }

    ~iterator() { if (destroy) destroy(impl_); }
    
    T operator*() const { return dereference(impl_); }
    iterator& operator++() { advance(impl_); return *this; }
    bool operator==(iterator const&other) const {
      if (strcmp(type_, other.type_)) { return false; }
      return compare(impl_, other.impl_);
    }
    bool operator!=(iterator const&other) const {
      if (strcmp(type_, other.type_)) { return false; }
      return !compare(impl_, other.impl_);
    }
  private:
    friend void swap(iterator & lhs, iterator & rhs) {
      using std::swap;
      swap(lhs.copy, rhs.copy);
      swap(lhs.dereference, rhs.dereference);
      swap(lhs.compare, rhs.compare);
      swap(lhs.destroy, rhs.destroy);
      swap(lhs.advance, rhs.advance);
      swap(lhs.type_, rhs.type_);
      swap(lhs.impl_, rhs.impl_);
    }
    using delegate = void(*)(void*);
    void* (*copy)(void*){nullptr};
    T (*dereference)(void*){nullptr};
    bool (*compare)(void*, void*){nullptr};
    delegate destroy{nullptr}, advance{nullptr};
    char const * type_{nullptr};
    void * impl_{nullptr};
  };
  
  namespace detail {
    template <typename T, typename = void>
    class stream_base_pointer_impl {};
    
    template <typename T>
    class stream_base_pointer_impl<T, typename std::enable_if<traits::is_dereferencable<T>::value>::type> {
    private:
      using self_t = stream_base<T>;
      using pointer = typename std::remove_reference<T>::type;
      using element_type = typename std::pointer_traits<pointer>::element_type;
    public:
      auto deref() const & -> stream_base<element_type &> {
        return static_cast<self_t const *>(this)->map([](T const & p) -> element_type & { return *p; });
      }
      
      auto deref() && -> stream_base<element_type &> {
        return static_cast<self_t &&>(*this).map([](T const & p) -> element_type & { return *p; });
      }
    };
    
    template <typename T>
    class stream_base : public stream_base_pointer_impl<T> {
    private:
      using value_type = typename std::decay<T>::type;
    public:
      template <typename Stream>
      stream_base(std::shared_ptr<Stream> && impl) {
        do_begin = [](std::shared_ptr<void> p) -> iterator<T> { return std::static_pointer_cast<Stream>(p)->begin(); };
        do_end = [](std::shared_ptr<void> p) -> iterator<T> { return std::static_pointer_cast<Stream>(p)->end(); };
        impl_ = std::static_pointer_cast<void>(impl);
      }
      
      ::stream::iterator<T> begin() const { return do_begin(impl_); }
      ::stream::iterator<T> end  () const { return do_end  (impl_); }
      
      bool empty() const { return begin() == end(); }
      
      std::vector<value_type> collect() const {
        std::vector<value_type> coll;
        collect(coll);
        return coll;
      }
      
      template <typename C, typename = typename std::enable_if<!std::is_void<typename C::value_type>::value, void>::type>
      C & collect(C & coll) const {
        std::copy(begin(), end(), std::inserter(coll, coll.end()));
        return coll;
      }
      
      template <typename F>
      value_type accumulate(F&& fold, value_type const& accum) const {
        return std::accumulate(begin(), end(), accum, fold);
      }
      
      value_type accumulate(value_type const& accum) const {
        return std::accumulate(begin(), end(), accum);
      }
      
      template <typename F>
      void each(F && consumer) const {
        std::for_each(begin(), end(), consumer);
      }
      
      template <typename F>
      stream_base<traits::mapped_t<T, F>> map(F&& func) const &;
      template <typename F>
      stream_base<T> filter(F&& func) const &;
      template <typename F>
      stream_base<traits::fmapped_t<T, F>> flatmap(F&& func) const &;
      
      template <typename F>
      stream_base<traits::mapped_t<T, F>> map(F&& func) &&;
      template <typename F>
      stream_base<T> filter(F&& func) &&;
      template <typename F>
      stream_base<traits::fmapped_t<T, F>> flatmap(F&& func) &&;
      
      template <typename Cast>
      stream_base<Cast const &> cast() const & {
        return map([](T const & p) -> Cast const & { return p; });
      }
      
      template <typename Cast>
      stream_base<Cast const &> cast() && {
        return std::move(*this).map([](T const & p) -> Cast const & { return p; });
      }
      
      template <typename F, typename = traits::is_memvar_t<F>>
      stream_base<traits::memvar_f<F>> map(F && memvar) const & {
        return map(map_member_object<F>{memvar});
      }
      
      template <typename F, typename = traits::is_memfun_t<F>>
      stream_base<traits::memfun_f<F>> map(F && memvar) const & {
        return map(map_member_function<F>{memvar});
      }

      template <typename F, typename = traits::is_memvar_t<F>>
      stream_base<traits::memvar_f<F>> map(F && memvar) && {
        return std::move(*this).map(map_member_object<F>{memvar});
      }

      template <typename F, typename = traits::is_memfun_t<F>>
      stream_base<traits::memfun_f<F>> map(F && memvar) && {
        return std::move(*this).map(map_member_function<F>{memvar});
      }
    private:
      iterator<T> (*do_begin)(std::shared_ptr<void>){nullptr};
      iterator<T> (*do_end)(std::shared_ptr<void>){nullptr};
      std::shared_ptr<void> impl_{nullptr};
    };
  }
}