/****************************************************************************
 Copyright (c) 2020 Simdsoft Limited.

 https://simdsoft.com

 Apache License 2.0

 Permission is hereby granted, free of charge, to any person obtaining a copy
 of this software and associated documentation files (the "Software"), to deal
 in the Software without restriction, including without limitation the rights
 to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 copies of the Software, and to permit persons to whom the Software is
 furnished to do so, subject to the following conditions:

 The above copyright notice and this permission notice shall be included in
 all copies or substantial portions of the Software.

 THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 THE SOFTWARE.
 ****************************************************************************/
#ifndef SIMDSOFT_FAST_SPLIT_HPP
#define SIMDSOFT_FAST_SPLIT_HPP
#include <string.h>
#include <string>

namespace fastl {
namespace internal {
#define _REDIRECT_DELIM_INTRI(_From,_To,_CStr,_Delim) \
    inline _CStr _To(_CStr s, _Delim d) \
    { \
        return _From(s, d); \
    }

    _REDIRECT_DELIM_INTRI(strchr, xstrchr, const char*, int)
    _REDIRECT_DELIM_INTRI(strchr, xstrchr, char*, int)
    _REDIRECT_DELIM_INTRI(wcschr, xstrchr, const wchar_t*, wchar_t)
    _REDIRECT_DELIM_INTRI(wcschr, xstrchr, wchar_t*, wchar_t)

    _REDIRECT_DELIM_INTRI(strstr, xstrstr, const char*, const char*)
    _REDIRECT_DELIM_INTRI(strstr, xstrstr, char*, const char*)
    _REDIRECT_DELIM_INTRI(wcsstr, xstrstr, const wchar_t*, const wchar_t*)
    _REDIRECT_DELIM_INTRI(wcsstr, xstrstr, wchar_t*, const wchar_t*)

    _REDIRECT_DELIM_INTRI(strpbrk, xstrpbrk, const char*, const char*)
    _REDIRECT_DELIM_INTRI(strpbrk, xstrpbrk, char*, const char*)
    _REDIRECT_DELIM_INTRI(wcspbrk, xstrpbrk, const wchar_t*, const wchar_t*)
    _REDIRECT_DELIM_INTRI(wcspbrk, xstrpbrk, wchar_t*, const wchar_t*)
}

/*
* nsc::fast_split API (override+3)
* op prototype: [](const char* start, const char* end){}
*/
template<typename _CStr, typename _Fn> inline
    void fast_split(_CStr s, size_t slen, typename std::remove_pointer<_CStr>::type delim, _Fn func)
{
    auto _Start = s; // the start of every string
    auto _Ptr = s; // source string iterator
    auto _End = s + slen;
    while ((_Ptr = internal::xstrchr(_Ptr, delim)))
    {
        if (_Start <= _Ptr)
            func(_Start, _Ptr);
        _Start = _Ptr + 1;
        ++_Ptr;
    }
    if (_Start <= _End) {
        func(_Start, _End);
    }
}

template<typename _CStr, typename _Fn> inline
    void fast_split(_CStr s, size_t slen, const typename std::remove_pointer<_CStr>::type* delims, size_t dlen, _Fn func)
{
    auto _Start = s; // the start of every string
    auto _Ptr = s; // source string iterator
    auto _End = s + slen;
    while ((_Ptr = internal::xstrstr(_Ptr, delims)))
    {
        if (_Start <= _Ptr)
        {
            func(_Start, _Ptr);
        }
        _Start = _Ptr + dlen;
        _Ptr += dlen;
    }
    if (_Start <= _End) {
        func(_Start, _End);
    }
}

template<typename _CStr, typename _Fn> inline
    void fast_split_of(_CStr s, size_t slen, const typename std::remove_pointer<_CStr>::type* delims, _Fn func)
{
    auto _Start = s; // the start of every string
    auto _Ptr = s; // source string iterator
    auto _End = s + slen;
    auto _Delim = delims[0];
    while ((_Ptr = internal::xstrpbrk(_Ptr, delims)))
    {
        if (_Start <= _Ptr)
        {
            func(_Start, _Ptr, _Delim);
            _Delim = *_Ptr;
        }
        _Start = _Ptr + 1;
        ++_Ptr;
    }
    if (_Start <= _End) {
        func(_Start, _End, _Delim);
    }
}


/*
* nsc::fast_split_brkif API (override+3)
* op prototype: [](const char* start, const char* end){}
*/
template<typename _CStr, typename _Fn> inline
    void fast_split_brkif(_CStr s, size_t slen, typename std::remove_pointer<_CStr>::type delim, _Fn func)
{
    auto _Start = s; // the start of every string
    auto _Ptr = s; // source string iterator
    auto _End = s + slen;
    while ((_Ptr = internal::xstrchr(_Ptr, delim)))
    {
        if (_Start <= _Ptr)
            if (func(_Start, _Ptr))
                return;
        _Start = _Ptr + 1;
        ++_Ptr;
    }
    if (_Start <= _End) {
        func(_Start, _End);
    }
}

template<typename _CStr, typename _Fn> inline
    void fast_split_brkif(_CStr s, size_t slen, const typename std::remove_pointer<_CStr>::type* delims, size_t dlen, _Fn func)
{
    auto _Start = s; // the start of every string
    auto _Ptr = s; // source string iterator
    auto _End = s + slen;
    while ((_Ptr = internal::xstrstr(_Ptr, delims)))
    {
        if (_Start <= _Ptr)
        {
            if (func(_Start, _Ptr))
                return;
        }
        _Start = _Ptr + dlen;
        _Ptr += dlen;
    }
    if (_Start <= _End) {
        func(_Start, _End);
    }
}

template<typename _CStr, typename _Fn> inline
    void fast_split_of_brkif(_CStr s, size_t slen, const typename std::remove_pointer<_CStr>::type* delims, _Fn func)
{
    auto _Start = s; // the start of every string
    auto _Ptr = s; // source string iterator
    auto _End = s + slen;
    auto _Delim = delims[0];
    while ((_Ptr = internal::xstrpbrk(_Ptr, delims)))
    {
        if (_Start <= _Ptr)
        {
            if (func(_Start, _Ptr, _Delim))
                return;
            _Delim = *_Ptr;
        }
        _Start = _Ptr + 1;
        ++_Ptr;
    }
    if (_Start <= _End) {
        func(_Start, _End, _Delim);
    }
}

/// C++ basic_string wrappers
template<typename _StringCont, typename _Fn> inline
    void fast_split(_StringCont& s, typename _StringCont::value_type delim, _Fn func)
{
    typename _StringCont::value_type dummy[1] = { 0 };
    return fast_split(!s.empty() ? &s.front() : &dummy[0], s.length(), delim, func);
}

#if _HAS_CXX17
/// C++17 basic_string_view wrappers
template<typename _Elem, typename _Fn> inline
    void fast_split(std::basic_string_view<_Elem> s, _Elem delim, _Fn func)
{
    return fast_split(s.data(), s.length(), delim, func);
}
template<typename _Elem, typename _Fn> inline
    void fast_split(std::basic_string_view<_Elem> s, std::basic_string_view<_Elem> delims, _Fn func)
{
    return fast_split(s.data(), s.length(), delims.data(), delims.length(), func);
}
template<typename _Elem, typename _Fn> inline
    void fast_split_of(std::basic_string_view<_Elem> s, const _Elem* delims, _Fn func)
{
    return fast_split_of(s.data(), s.length(), delims, func);
}
#endif

/// simple retun a vector<basic_string> as string array
#if _HAS_CXX17
template<typename _Elem> inline
    std::vector<std::basic_string<_Elem>> split(std::basic_string_view<_Elem> s, _Elem delim)
{
    std::vector<std::basic_string<_Elem>> results;
    fast_split(s, delim, [&](const _Elem* s, const _Elem* e) {
        results.push_back(std::basic_string<_Elem>(s, e));
    });
    return results;
}
template<typename _Elem> inline
    std::vector<std::basic_string<_Elem>> split(const std::basic_string<_Elem>& s, _Elem delim)
{
    return split(std::basic_string_view<_Elem>(s.c_str(), s.length()), delim);
}

template<typename _Elem> inline
    std::vector<std::basic_string<_Elem>> split(std::basic_string_view<_Elem> s, std::basic_string_view<_Elem> delims)
{
    std::vector<std::basic_string<_Elem>> results;
    fast_split(s, delims, [&](const _Elem* s, const _Elem* e) {
        results.push_back(std::basic_string<_Elem>(s, e));
    });
    return results;
}
template<typename _Elem> inline
    std::vector<std::basic_string<_Elem>> split(const std::basic_string<_Elem>& s, std::basic_string_view<_Elem> delims)
{
    return split(std::basic_string_view<_Elem>(s.c_str(), s.length()), delims);
}

template<typename _Elem> inline
    std::vector<std::basic_string<_Elem>> split_of(std::basic_string_view<_Elem> s, const _Elem* delims)
{
    std::vector<std::basic_string<_Elem>> results;
    fast_split_of(s, delims, [&](const _Elem* s, const _Elem* e, _Elem) {
        results.push_back(std::basic_string<_Elem>(s, e));
    });
    return results;
}
template<typename _Elem> inline
    std::vector<std::basic_string<_Elem>> split_of(const std::basic_string<_Elem>& s, const _Elem* delims)
{
    return split_of(std::basic_string_view<_Elem>(s.c_str(), s.length()), delims);
}
#endif

namespace nzls {
    /*
    * nsc::fast_split_nzls API (override+3) ignore empty string
    * op prototype: [](const char* start, const char* end){}
    */
    template<typename _CStr, typename _Fn> inline
        void fast_split(_CStr s, size_t slen, typename std::remove_pointer<_CStr>::type delim, _Fn func)
    {
        auto _Start = s; // the start of every string
        auto _Ptr = s; // source string iterator
        auto _End = s + slen;
        while ((_Ptr = internal::xstrchr(_Ptr, delim)))
        {
            if (_Start < _Ptr)
                func(_Start, _Ptr);
            _Start = _Ptr + 1;
            ++_Ptr;
        }
        if (_Start < _End) {
            func(_Start, _End);
        }
    }

    template<typename _CStr, typename _Fn> inline
        void fast_split(_CStr s, size_t slen, const typename std::remove_pointer<_CStr>::type* delims, size_t dlen, _Fn func)
    {
        auto _Start = s; // the start of every string
        auto _Ptr = s; // source string iterator
        auto _End = s + slen;
        while ((_Ptr = internal::xstrstr(_Ptr, delims)))
        {
            if (_Start < _Ptr)
            {
                func(_Start, _Ptr);
            }
            _Start = _Ptr + dlen;
            _Ptr += dlen;
        }
        if (_Start < _End) {
            func(_Start, _End);
        }
    }

    template<typename _CStr, typename _Fn> inline
        void fast_split_of(_CStr s, size_t slen, const typename std::remove_pointer<_CStr>::type* delims, _Fn func)
    {
        auto _Start = s; // the start of every string
        auto _Ptr = s; // source string iterator
        auto _End = s + slen;
        auto _Delim = delims[0];
        while ((_Ptr = internal::xstrpbrk(_Ptr, delims)))
        {
            if (_Start < _Ptr)
            {
                func(_Start, _Ptr, _Delim);
                _Delim = *_Ptr;
            }
            _Start = _Ptr + 1;
            ++_Ptr;
        }
        if (_Start < _End) {
            func(_Start, _End, _Delim);
        }
    }

    /*
    * nsc::fast_split_nzls_brkif API (override+3) ignore empty string
    * op prototype: [](const char* start, const char* end){}
    */
    template<typename _CStr, typename _Fn> inline
        void fast_split_brkif(_CStr s, size_t slen, typename std::remove_pointer<_CStr>::type delim, _Fn func)
    {
        auto _Start = s; // the start of every string
        auto _Ptr = s; // source string iterator
        auto _End = s + slen;
        while ((_Ptr = internal::xstrchr(_Ptr, delim)))
        {
            if (_Start < _Ptr)
                if (func(_Start, _Ptr))
                    return;
            _Start = _Ptr + 1;
            ++_Ptr;
        }
        if (_Start < _End) {
            func(_Start, _End);
        }
    }

    template<typename _CStr, typename _Fn> inline
        void fast_split_brkif(_CStr s, size_t slen, const typename std::remove_pointer<_CStr>::type* delims, size_t dlen, _Fn func)
    {
        auto _Start = s; // the start of every string
        auto _Ptr = s; // source string iterator
        auto _End = s + slen;
        while ((_Ptr = internal::xstrstr(_Ptr, delims)))
        {
            if (_Start < _Ptr)
            {
                if (func(_Start, _Ptr))
                    return;
            }
            _Start = _Ptr + dlen;
            _Ptr += dlen;
        }
        if (_Start < _End) {
            func(_Start, _End);
        }
    }

    template<typename _CStr, typename _Fn> inline
        void fast_split_of_brkif(_CStr s, size_t slen, const typename std::remove_pointer<_CStr>::type* delims, _Fn func)
    {
        auto _Start = s; // the start of every string
        auto _Ptr = s; // source string iterator
        auto _End = s + slen;
        auto _Delim = delims[0];
        while ((_Ptr = internal::xstrpbrk(_Ptr, delims)))
        {
            if (_Start < _Ptr)
            {
                if (func(_Start, _Ptr, _Delim))
                    return;
                _Delim = *_Ptr;
            }
            _Start = _Ptr + 1;
            ++_Ptr;
        }
        if (_Start < _End) {
            func(_Start, _End, _Delim);
        }
    }

    /// C++ basic_string wrappers
    template<typename _StringCont, typename _Fn> inline
        void fast_split(_StringCont& s, typename _StringCont::value_type delim, _Fn func)
    {
        typename _StringCont::value_type dummy[1] = { 0 };
        return fast_split(!s.empty() ? &s.front() : &dummy[0], s.length(), delim, func);
    }

#if _HAS_CXX17
    /// C++17 basic_string_view wrappers
    template<typename _Elem, typename _Fn> inline
        void fast_split(std::basic_string_view<_Elem> s, _Elem delim, _Fn func)
    {
        return fast_split(s.data(), s.length(), delim, func);
    }
    template<typename _Elem, typename _Fn> inline
        void fast_split(std::basic_string_view<_Elem> s, std::basic_string_view<_Elem> delims, _Fn func)
    {
        return fast_split(s.data(), s.length(), delims.data(), delims.length(), func);
    }
    template<typename _Elem, typename _Fn> inline
        void fast_split_of(std::basic_string_view<_Elem> s, const _Elem* delims, _Fn func)
    {
        return fast_split_of(s.data(), s.length(), delims, func);
    }
#endif

    /// simple retun a vector<basic_string> as string array
#if _HAS_CXX17
    template<typename _Elem> inline
        std::vector<std::basic_string<_Elem>> split(std::basic_string_view<_Elem> s, _Elem delim)
    {
        std::vector<std::basic_string<_Elem>> results;
        fast_split(s, delim, [&](const _Elem* s, const _Elem* e) {
            results.push_back(std::basic_string<_Elem>(s, e));
        });
        return results;
    }
    template<typename _Elem> inline
        std::vector<std::basic_string<_Elem>> split(const std::basic_string<_Elem>& s, _Elem delim)
    {
        return split(std::basic_string_view<_Elem>(s.c_str(), s.length()), delim);
    }

    template<typename _Elem> inline
        std::vector<std::basic_string<_Elem>> split(std::basic_string_view<_Elem> s, std::basic_string_view<_Elem> delims)
    {
        std::vector<std::basic_string<_Elem>> results;
        fast_split(s, delims, [&](const _Elem* s, const _Elem* e) {
            results.push_back(std::basic_string<_Elem>(s, e));
        });
        return results;
    }
    template<typename _Elem> inline
        std::vector<std::basic_string<_Elem>> split(const std::basic_string<_Elem>& s, std::basic_string_view<_Elem> delims)
    {
        return split(std::basic_string_view<_Elem>(s.c_str(), s.length()), delims);
    }

    template<typename _Elem> inline
        std::vector<std::basic_string<_Elem>> split_of(std::basic_string_view<_Elem> s, const _Elem* delims)
    {
        std::vector<std::basic_string<_Elem>> results;
        fast_split_of(s, delims, [&](const _Elem* s, const _Elem* e, _Elem) {
            results.push_back(std::basic_string<_Elem>(s, e));
        });
        return results;
    }
    template<typename _Elem> inline
        std::vector<std::basic_string<_Elem>> split_of(const std::basic_string<_Elem>& s, const _Elem* delims)
    {
        return split_of(std::basic_string_view<_Elem>(s.c_str(), s.length()), delims);
    }
#endif
}

template<typename _Elem, typename _Pr, typename _Fn> inline
    void _splitpath(_Elem* s, _Pr _Pred, _Fn func) // will convert '\\' to '/'
{
    _Elem* _Start = s; // the start of every string
    _Elem* _Ptr = s;   // source string iterator
    while (_Pred(_Ptr))
    {
        if ('\\' == *_Ptr || '/' == *_Ptr)
        {
            if (_Ptr != _Start) {
                auto _Ch = *_Ptr;
                *_Ptr = '\0';
                bool should_brk = func(s);
#if defined(_WIN32)
                *_Ptr = '\\';
#else // For unix linux like system.
                *_Ptr = '/';
#endif
                if (should_brk)
                    return;
            }
            _Start = _Ptr + 1;
        }
        ++_Ptr;
    }
    if (_Start < _Ptr) {
        func(s);
    }
}

template<typename _Elem, typename _Fn> inline
    void splitpath(_Elem* s, size_t size, _Fn func) // will convert '\\' to '/'
{
    auto _Endptr = s + size;
    _splitpath(s, [=](_Elem* _Ptr) {return _Ptr < _Endptr; }, func);
}

template<typename _Elem, typename _Fn> inline
    void splitpath(_Elem* s, _Fn func) // will convert '\\' to '/'
{
    _splitpath(s, [=](_Elem* _Ptr) {return *_Ptr != '\0'; }, func);
}

template<typename _Elem, typename _Fn> inline
void splitpath(std::basic_string<_Elem>& s, _Fn func) // will convert '\\' to '/'
{
    _Elem dummy[1] = { 0 };
    splitpath(!s.empty() ? &s.front() : &dummy[0], func);
}
}