/* Algorithms for OctaSTD.
 *
 * This file is part of OctaSTD. See COPYING.md for futher information.
 */

#ifndef OCTA_ALGORITHM_HH
#define OCTA_ALGORITHM_HH

#include <math.h>

#include "octa/functional.hh"
#include "octa/range.hh"
#include "octa/utility.hh"
#include "octa/initializer_list.hh"

namespace octa {

/* partitioning */

template<typename R, typename U>
R partition(R range, U pred) {
    R ret = range;
    for (; !range.empty(); range.pop_front()) {
        if (pred(range.front())) {
            octa::swap(range.front(), ret.front());
            ret.pop_front();
        }
    }
    return ret;
}

template<typename R, typename P>
bool is_partitioned(R range, P pred) {
    for (; !range.empty() && pred(range.front()); range.pop_front());
    for (; !range.empty(); range.pop_front())
        if (pred(range.front())) return false;
    return true;
}

/* sorting */

namespace detail {
    template<typename R, typename C>
    void insort(R range, C compare) {
        octa::RangeSize<R> rlen = range.size();
        for (octa::RangeSize<R> i = 1; i < rlen; ++i) {
            octa::RangeSize<R> j = i;
            octa::RangeValue<R> v(octa::move(range[i]));
            while (j > 0 && !compare(range[j - 1], v)) {
                range[j] = range[j - 1];
                --j;
            }
            range[j] = octa::move(v);
        }
    }

    template<typename R, typename C>
    void hs_sift_down(R range, octa::RangeSize<R> s,
    octa::RangeSize<R> e, C compare) {
        octa::RangeSize<R> r = s;
        while ((r * 2 + 1) <= e) {
            octa::RangeSize<R> ch = r * 2 + 1;
            octa::RangeSize<R> sw = r;
            if (compare(range[sw], range[ch]))
                sw = ch;
            if (((ch + 1) <= e) && compare(range[sw], range[ch + 1]))
                sw = ch + 1;
            if (sw != r) {
                octa::swap(range[r], range[sw]);
                r = sw;
            } else return;
        }
    }

    template<typename R, typename C>
    void heapsort(R range, C compare) {
        octa::RangeSize<R> len = range.size();
        octa::RangeSize<R> st = (len - 2) / 2;
        for (;;) {
            octa::detail::hs_sift_down(range, st, len - 1, compare);
            if (st-- == 0) break;
        }
        octa::RangeSize<R> e = len - 1;
        while (e > 0) {
            octa::swap(range[e], range[0]);
            --e;
            octa::detail::hs_sift_down(range, 0, e, compare);
        }
    }

    template<typename R, typename C>
    void introloop(R range, C compare, RangeSize<R> depth) {
        if (range.size() <= 10) {
            octa::detail::insort(range, compare);
            return;
        }
        if (depth == 0) {
            octa::detail::heapsort(range, compare);
            return;
        }
        octa::swap(range[range.size() / 2], range.back());
        octa::RangeSize<R> pi = 0;
        R pr = range;
        pr.pop_back();
        for (; !pr.empty(); pr.pop_front()) {
            if (compare(pr.front(), range.back()))
                octa::swap(pr.front(), range[pi++]);
        }
        octa::swap(range[pi], range.back());
        octa::detail::introloop(range.slice(0, pi), compare, depth - 1);
        octa::detail::introloop(range.slice(pi + 1, range.size()), compare,
            depth - 1);
    }

    template<typename R, typename C>
    void introsort(R range, C compare) {
        octa::detail::introloop(range, compare, octa::RangeSize<R>(2
            * (log(range.size()) / log(2))));
    }
} /* namespace detail */

template<typename R, typename C>
void sort(R range, C compare) {
    octa::detail::introsort(range, compare);
}

template<typename R>
void sort(R range) {
    sort(range, octa::Less<RangeValue<R>>());
}

/* min/max(_element) */

template<typename T>
inline const T &min(const T &a, const T &b) {
    return (a < b) ? a : b;
}
template<typename T, typename C>
inline const T &min(const T &a, const T &b, C compare) {
    return compare(a, b) ? a : b;
}

template<typename T>
inline const T &max(const T &a, const T &b) {
    return (a < b) ? b : a;
}
template<typename T, typename C>
inline const T &max(const T &a, const T &b, C compare) {
    return compare(a, b) ? b : a;
}

template<typename R>
inline R min_element(R range) {
    R r = range;
    for (; !range.empty(); range.pop_front())
        if (octa::min(r.front(), range.front()) == range.front())
            r = range;
    return r;
}
template<typename R, typename C>
inline R min_element(R range, C compare) {
    R r = range;
    for (; !range.empty(); range.pop_front())
        if (octa::min(r.front(), range.front(), compare) == range.front())
            r = range;
    return r;
}

template<typename R>
inline R max_element(R range) {
    R r = range;
    for (; !range.empty(); range.pop_front())
        if (octa::max(r.front(), range.front()) == range.front())
            r = range;
    return r;
}
template<typename R, typename C>
inline R max_element(R range, C compare) {
    R r = range;
    for (; !range.empty(); range.pop_front())
        if (octa::max(r.front(), range.front(), compare) == range.front())
            r = range;
    return r;
}

template<typename T>
inline T min(std::initializer_list<T> il) {
    return octa::min_element(octa::iter(il)).front();
}
template<typename T, typename C>
inline T min(std::initializer_list<T> il, C compare) {
    return octa::min_element(octa::iter(il), compare).front();
}

template<typename T>
inline T max(std::initializer_list<T> il) {
    return octa::max_element(octa::iter(il)).front();
}

template<typename T, typename C>
inline T max(std::initializer_list<T> il, C compare) {
    return octa::max_element(octa::iter(il), compare).front();
}

/* clamp */

template<typename T, typename U>
inline T clamp(const T &v, const U &lo, const U &hi) {
    return octa::max(T(lo), octa::min(v, T(hi)));
}

template<typename T, typename U, typename C>
inline T clamp(const T &v, const U &lo, const U &hi, C compare) {
    return octa::max(T(lo), octa::min(v, T(hi), compare), compare);
}

/* algos that don't change the range */

template<typename R, typename F>
F for_each(R range, F func) {
    for (; !range.empty(); range.pop_front())
        func(range.front());
    return octa::move(func);
}

template<typename R, typename P>
bool all_of(R range, P pred) {
    for (; !range.empty(); range.pop_front())
        if (!pred(range.front())) return false;
    return true;
}

template<typename R, typename P>
bool any_of(R range, P pred) {
    for (; !range.empty(); range.pop_front())
        if (pred(range.front())) return true;
    return false;
}

template<typename R, typename P>
bool none_of(R range, P pred) {
    for (; !range.empty(); range.pop_front())
        if (pred(range.front())) return false;
    return true;
}

template<typename R, typename T>
R find(R range, const T &v) {
    for (; !range.empty(); range.pop_front())
        if (range.front() == v)
            break;
    return range;
}

template<typename R, typename P>
R find_if(R range, P pred) {
    for (; !range.empty(); range.pop_front())
        if (pred(range.front()))
            break;
    return range;
}

template<typename R, typename P>
R find_if_not(R range, P pred) {
    for (; !range.empty(); range.pop_front())
        if (!pred(range.front()))
            break;
    return range;
}

template<typename R, typename T>
RangeSize<R> count(R range, const T &v) {
    RangeSize<R> ret = 0;
    for (; !range.empty(); range.pop_front())
        if (range.front() == v)
            ++ret;
    return ret;
}

template<typename R, typename P>
RangeSize<R> count_if(R range, P pred) {
    RangeSize<R> ret = 0;
    for (; !range.empty(); range.pop_front())
        if (pred(range.front()))
            ++ret;
    return ret;
}

template<typename R, typename P>
RangeSize<R> count_if_not(R range, P pred) {
    RangeSize<R> ret = 0;
    for (; !range.empty(); range.pop_front())
        if (!pred(range.front()))
            ++ret;
    return ret;
}

template<typename R>
bool equal(R range1, R range2) {
    for (; !range1.empty(); range1.pop_front()) {
        if (range2.empty() || (range1.front() != range2.front()))
            return false;
        range2.pop_front();
    }
    return range2.empty();
}

/* algos that modify ranges or work with output ranges */

template<typename R1, typename R2>
R2 copy(R1 irange, R2 orange) {
    for (; !irange.empty(); irange.pop_front())
        orange.put(irange.front());
    return orange;
}

template<typename R1, typename R2, typename P>
R2 copy_if(R1 irange, R2 orange, P pred) {
    for (; !irange.empty(); irange.pop_front())
        if (pred(irange.front()))
            orange.put(irange.front());
    return orange;
}

template<typename R1, typename R2, typename P>
R2 copy_if_not(R1 irange, R2 orange, P pred) {
    for (; !irange.empty(); irange.pop_front())
        if (!pred(irange.front()))
            orange.put(irange.front());
    return orange;
}

template<typename R1, typename R2>
R2 move(R1 irange, R2 orange) {
    for (; !irange.empty(); irange.pop_front())
        orange.put(octa::move(irange.front()));
    return orange;
}

template<typename R>
void reverse(R range) {
    while (!range.empty()) {
        octa::swap(range.front(), range.back());
        range.pop_front();
        range.pop_back();
    }
}

template<typename R1, typename R2>
R2 reverse_copy(R1 irange, R2 orange) {
    for (; !irange.empty(); irange.pop_back())
        orange.put(irange.back());
    return orange;
}

template<typename R, typename T>
void fill(R range, const T &v) {
    for (; !range.empty(); range.pop_front())
        range.front() = v;
}

template<typename R, typename F>
void generate(R range, F gen) {
    for (; !range.empty(); range.pop_front())
        range.front() = gen();
}

template<typename R1, typename R2>
octa::Pair<R1, R2> swap_ranges(R1 range1, R2 range2) {
    while (!range1.empty() && !range2.empty()) {
        octa::swap(range1.front(), range2.front());
        range1.pop_front();
        range2.pop_front();
    }
    return octa::make_pair(range1, range2);
}

template<typename R, typename T>
void iota(R range, T value) {
    for (; !range.empty(); range.pop_front())
        range.front() = value++;
}

template<typename R, typename T>
T foldl(R range, T init) {
    for (; !range.empty(); range.pop_front())
        init = init + range.front();
    return init;
}

template<typename R, typename T, typename F>
T foldl(R range, T init, F func) {
    for (; !range.empty(); range.pop_front())
        init = func(init, range.front());
    return init;
}

template<typename R, typename T>
T foldr(R range, T init) {
    for (; !range.empty(); range.pop_back())
        init = init + range.back();
    return init;
}

template<typename R, typename T, typename F>
T foldr(R range, T init, F func) {
    for (; !range.empty(); range.pop_back())
        init = func(init, range.back());
    return init;
}

template<typename T, typename F, typename R>
struct MapRange: InputRange<
    MapRange<T, F, R>, octa::RangeCategory<T>, R, R, octa::RangeSize<T>
> {
private:
    T p_range;
    octa::FunctionMakeDefaultConstructible<F> p_func;

public:
    MapRange() = delete;
    MapRange(const T &range, const F &func):
        p_range(range), p_func(func) {}
    MapRange(const MapRange &it):
        p_range(it.p_range), p_func(it.p_func) {}
    MapRange(MapRange &&it):
        p_range(move(it.p_range)), p_func(move(it.p_func)) {}

    MapRange &operator=(const MapRange &v) {
        p_range = v.p_range;
        p_func  = v.p_func;
        return *this;
    }
    MapRange &operator=(MapRange &&v) {
        p_range = move(v.p_range);
        p_func  = move(v.p_func);
        return *this;
    }

    bool empty() const { return p_range.empty(); }
    octa::RangeSize<T> size() const { return p_range.size(); }

    bool equals_front(const MapRange &r) const {
        return p_range.equals_front(r.p_range);
    }
    bool equals_back(const MapRange &r) const {
        return p_range.equals_front(r.p_range);
    }

    octa::RangeDifference<T> distance_front(const MapRange &r) const {
        return p_range.distance_front(r.p_range);
    }
    octa::RangeDifference<T> distance_back(const MapRange &r) const {
        return p_range.distance_back(r.p_range);
    }

    bool pop_front() { return p_range.pop_front(); }
    bool pop_back() { return p_range.pop_back(); }

    bool push_front() { return p_range.pop_front(); }
    bool push_back() { return p_range.push_back(); }

    octa::RangeSize<T> pop_front_n(octa::RangeSize<T> n) {
        p_range.pop_front_n(n);
    }
    octa::RangeSize<T> pop_back_n(octa::RangeSize<T> n) {
        p_range.pop_back_n(n);
    }

    octa::RangeSize<T> push_front_n(octa::RangeSize<T> n) {
        return p_range.push_front_n(n);
    }
    octa::RangeSize<T> push_back_n(octa::RangeSize<T> n) {
        return p_range.push_back_n(n);
    }

    R front() const { return p_func(p_range.front()); }
    R back() const { return p_func(p_range.back()); }

    R operator[](octa::RangeSize<T> idx) const {
        return p_func(p_range[idx]);
    }

    MapRange slice(octa::RangeSize<T> start,
                           octa::RangeSize<T> end) {
        return MapRange(p_range.slice(start, end), p_func);
    }
};

namespace detail {
    template<typename R, typename F> using MapReturnType
        = decltype(declval<F>()(octa::declval<octa::RangeReference<R>>()));
}

template<typename R, typename F>
MapRange<R, F, octa::detail::MapReturnType<R, F>> map(R range,
                                                      F func) {
    return octa::MapRange<R, F, octa::detail::MapReturnType<R, F>>(range,
        func);
}

template<typename T, typename F>
struct FilterRange: InputRange<
    FilterRange<T, F>, octa::CommonType<octa::RangeCategory<T>,
                                     octa::ForwardRangeTag>,
    octa::RangeValue<T>, octa::RangeReference<T>, octa::RangeSize<T>
> {
private:
    T p_range;
    octa::FunctionMakeDefaultConstructible<F> p_pred;

    void advance_valid() {
        while (!p_range.empty() && !p_pred(front())) p_range.pop_front();
    }

public:
    FilterRange() = delete;
    template<typename P>
    FilterRange(const T &range, const P &pred): p_range(range),
    p_pred(pred) {
        advance_valid();
    }
    FilterRange(const FilterRange &it): p_range(it.p_range),
    p_pred(it.p_pred) {
        advance_valid();
    }
    FilterRange(FilterRange &&it): p_range(move(it.p_range)),
    p_pred(move(it.p_pred)) {
        advance_valid();
    }

    FilterRange &operator=(const FilterRange &v) {
        p_range = v.p_range;
        p_pred  = v.p_pred;
        advance_valid();
        return *this;
    }
    FilterRange &operator=(FilterRange &&v) {
        p_range = move(v.p_range);
        p_pred  = move(v.p_pred);
        advance_valid();
        return *this;
    }

    bool empty() const { return p_range.empty(); }

    bool equals_front(const FilterRange &r) const {
        return p_range.equals_front(r.p_range);
    }

    bool pop_front() {
        bool ret = p_range.pop_front();
        advance_valid();
        return ret;
    }

    octa::RangeReference<T> front() const { return p_range.front(); }
};

namespace detail {
    template<typename R, typename P> using FilterPred
        = octa::EnableIf<IsSame<
            decltype(octa::declval<P>()(octa::declval<
                octa::RangeReference<R>
            >())),
            bool
        >::value, P>;
}

template<typename R, typename P>
FilterRange<R, octa::detail::FilterPred<R, P>> filter(R range, P pred) {
    return octa::FilterRange<R, P>(range, pred);
}

} /* namespace octa */

#endif