LUA極速文字列分割split関数実装

言うまでもなく、luaのstringがsplit関数を提供していないことに気づき、書くしかなかった.

//    
//      ，     O(n)，     O(1)
LUA_API int split(lua_State * L)
{
    int n = 1;
    int pos = 0;
    const char * src = lua_tostring(L, 1);
    int len = strlen(src); 
    char c = lua_tostring(L, 2)[0];
    assert(src[0] != c);    //             
    assert(src[len-1] != c); 
    //
    for(int i=0; i<len; ++i){
        if(src[i] == c){ 
            lua_pushlstring(L, src+pos, i-pos);
            pos = i+1;
            ++n;
        }
    }
    lua_pushlstring(L, src+pos, len-pos);
    return n;
}