LuaSnippets

UserPreferences

1. 모듈 패스 등록하기
2. 비트 연산
3. CSV 파서
4. 암묵적(implicit)인 전역 변수 선언 막기
1. 첫번째 방법
2. 두번째 방법

1 모듈 패스 등록하기

5.0 버전까지는 LUA_PATH 라는 이름의 전역 변수에다 설정하면 됐는데, 5.1 버전부터 package 테이블 안의 path 변수(package.path)를 이용해야한다.


/// \brief 루아 모듈 패스를 등록한다.
/// \param L 루아 상태 객체
/// \param path 패스 문자열. "D:\data\;D:\data\skill\"과 같은 형식, 즉 각각의 
/// 디렉토리를 ';'로 구분지은 문자열로 들어가 있어야 한다.
/// \param bAppend 기존 경로에 추가하는가의 여부
void SetModulePath(lua_State* L, const char* path, bool bAppend)
{
    size_t before = 0;
    size_t current = 0;

    std::string original(path);
    std::string modified;

    // 012345678901234567890123456789
    // D:\data\;D:\data\skill\
    // b       c
    while (current < original.size())
    {
        current = original.find_first_of(';', before);
        if (current == std::string::npos) current = original.size();

        std::string token = original.substr(before, current - before);
        if (!token.empty())
        {
            if (token[token.size()-1] != '\\')
                token += '\\';

#ifdef _DEBUG
            assert(::PathFileExists(token.c_str()) && "path not found");
#endif

            modified += token + "?.lua;";
            modified += token + "?.dll;";
        }

        before = current + 1;
    }

    if (!modified.empty())
    {
        if (bAppend)
        {
            lua_getfield(L, LUA_GLOBALSINDEX, "package");
            assert(lua_istable(L, -1));

            lua_getfield(L, -1, "path");
            std::string old_path = lua_tostring(L, -1);
            if (old_path[old_path.size()-1] != ';')
                old_path += ';';
            modified = old_path + modified;

            lua_pop(L, 2);
        }

        lua_getfield(L, LUA_GLOBALSINDEX, "package");
        assert(lua_istable(L, -1));

        lua_pushstring(L, modified.c_str());
        lua_setfield(L, -2, "path");

        lua_pop(L, 1);
    }
}

2 비트 연산

function extract(value, index)
    local v = value
    local m = 0
    for i=1,index do
        m = math.mod(v, 2)
        v = v / 2
    end
    if m >= 1 then return 1 end
    return 0
end

function bitwise_and(n1, n2)
    local rvalue = 0
    for i=1,32 do
        if extract(n1, i) == 1 and extract(n2, i) == 1 then
            rvalue = rvalue + math.pow(2, i-1)
        end
    end
    return rvalue
end

function bitwise_or(n1, n2)
    local rvalue = 0
    for i=1,32 do
        if extract(n1, i) == 1 or extract(n2, i) == 1 then
            rvalue = rvalue + math.pow(2, i-1)
        end
    end
    return rvalue
end

assert(bitwise_or(4, 1) == 5)
assert(bitwise_or(4, 2) == 6)
assert(bitwise_or(4, 4) == 4)
assert(bitwise_and(4, 1) == 0)
assert(bitwise_and(4, 2) == 0)
assert(bitwise_and(4, 4) == 4)

크아악... 너무 길다. 루아를 잘 알지 못해서 그런 것일수도 있겠으나, 개인적으로는 mathlib.c 파일을 수정해서 C 함수 하나 추가하는 편이 나을 거라고 생각한다.


...
static int math_bwand(lua_State* L) {
  int lhs = (int)luaL_checknumber(L, 1);
  int rhs = (int)luaL_checknumber(L, 2);
  lua_pushnumber(L, lhs & rhs);
  return 1;
}

static int math_bwor(lua_State* L) {
  int lhs = (int)luaL_checknumber(L, 1);
  int rhs = (int)luaL_checknumber(L, 2);
  lua_pushnumber(L, lhs | rhs);
  return 1;
}
...

static const luaL_reg mathlib[] = {
  {"abs",        math_abs},
  ...생략...
  {"randomseed", math_randomseed},
+  {"bwand",      math_bwand},
+  {"bwor",       math_bwor},
  {NULL,         NULL}
};

3 CSV 파서

"Programming in Lua"란 책에 나오는 CSV 파서를 약간 수정한 파서


CSVRow = 
{
    Columns = {}, -- 컬럼 문자열들
    
    new = function (self) 
        o = {} 
        setmetatable(o, self)
        self.__index = self
        return o
    end,
    
    parse = function(self, line)
        self.Columns = {}
    
        s = line .. ','        -- ending comma
        local fieldstart = 1
        repeat
            -- next field is quoted? (start with `"'?)
            if string.find(s, '^"', fieldstart) then
                local a, c
                local i  = fieldstart
    
                repeat
                    -- find closing quote
                    a, i, c = string.find(s, '"("?)', i+1)
                until c ~= '"'    -- quote not followed by quote?
    
                if not i then 
                    error('unmatched "') 
                end
                    
                local f = string.sub(s, fieldstart+1, i-1)
                table.insert(self.Columns, (string.gsub(f, '""', '"')))
                fieldstart = string.find(s, ',', i) + 1
            else                -- unquoted; find next comma
                local nexti = string.find(s, ',', fieldstart)
                table.insert(self.Columns, string.sub(s, fieldstart, nexti-1))
                fieldstart = nexti + 1
            end
        until fieldstart > string.len(s)
    end,
    
    to_string = function(self)
        local line = ""
        for _, token in ipairs(self.Columns) do
            line = line .. ","
            if string.find(token, '[,"]') then
                line = line .. '"' .. string.gsub(token, '"', '""') .. '"'
            else
                line = line .. token
            end
        end
        return string.sub(line, 2)
    end
}

CSVFile = 
{
    Rows = {},

    new = function (self)
        o = {} 
        setmetatable(o, self)
        self.__index = self
        return o
    end,
    
    load = function(self, filename)
        local file = assert(io.open(filename, "r"))
        
        while true do
            local line = file:read("*line")
            if line == nil then break end
            local row = CSVRow:new()
            row:parse(line)
            table.insert(self.Rows, row)
        end
        
        file:close()
    end,
    
    save = function(self, filename)
        local file = assert(io.open(filename, "w"))
        for _, row in pairs(self.Rows) do
            file:write(row:to_string() .. "\n")
        end
    end
}

-- 여기서부터는 샘플 코드
f = CSVFile:new()
f:load("test.csv")
print(f.Rows[1].Columns[1])
f:save("output.csv")

4 암묵적(implicit)인 전역 변수 선언 막기

루아에서는 local 구문을 사용하지 않고 변수를 선언하면 기본적으로 전역 변수가 되어버린다. LuaGarbageCollection 항목에서도 언급했지만, 전역 테이블에 있는 오브젝트들은 기본적으로 가비지 컬렉션의 대상이 되지 않는다. 조그만 프로그램이라면 모르되, 오랜 시간 동안 실행해야하는 프로그램 같은 경우에는 무시할 수 없는 일이라고 할 수 있다.

전역 오브젝트들이 들어있는 전역 테이블도 일반적인 테이블과 똑같은 테이블이기 때문에, 메타테이블과 메타메서드를 잘 설정해주면, 암묵적인 전역 변수 읽기/쓰기를 막을 수 있다.



4.1 첫번째 방법

function declare(name, initval)
   rawset(_G, name, initval or false)
end

setmetatable(_G, {
    __newindex = 
        function(_, n)
            error("attempt to write to undeclared variable " .. n, 2)
        end,
    __index = 
        function(_, n)
            error("attempt to read undeclared variable " .. n, 2)
        end,
})

위 코드를 실행하고 나서는, 변수 선언은 declare 함수를 이용해야만 한다. 아니면 에러가 발생한다.


> a = 1 
stdin:1: attempt to write to undeclared variable a 
> declare("a") 
> a = 1 

이 방법에는 약간 귀찮은 문제가 하나 있는데, 어떤 변수가 nil 값을 가지고 있는지 검사하는 경우에도 에러가 발생한다는 점이다. 이 문제는 rawset 함수를 이용하면 된다.


-- 이런 코드를...
if var == nil then
   ...
end

-- 이렇게 변경한다...
if rawget(var) == nil then
   ...
end




4.2 두번째 방법

local declared_names = {}

function declare(name, initval)
   rawset(_G, name, initval)
   declared_names[name] = true
end

setmetatable(_G, {
    __newindex = 
        function(t, n, v)
            if not declared_names[n] then
                error("attempt to write to undeclared variable "..n, 2)
            else
                rawset(t, n, v)
            end
        end,
    __index = 
        function(_, n)
            if not declared_names[n] then
                error("attempt to read undeclared variable "..n, 2)
            else
                return nil
            end
        end
})

첫번째 방법에서 rawget 문제를 해결한 방법이다. 사용하기 편한 대신, 메타메서드를 호출할 경우가 있으므로 약간 느리다. 하지만 어떤 변수의 값이 nil인지 검사할 때만이므로 무시할만 하다.

• see also Lua
  

• CategoryLanguage