1 files changed, 80 insertions, 0 deletions
diff --git a/nerv/tnn/sutil.lua b/nerv/tnn/sutil.lua
new file mode 100644
index 0000000..6a968b7
--- /dev/null
+++ b/nerv/tnn/sutil.lua
@@ -0,0 +1,80 @@
+local Util = nerv.class("nerv.SUtil") --Scheduler Utility
+
+function Util.simple_split(inputstr, sep)
+    if sep == nil then
+        sep = "%s"
+    end
+    local t={} ; i=1
+    for str in string.gmatch(inputstr, "([^"..sep.."]+)") do
+        t[i] = str
+        i = i + 1
+    end
+    return t
+end
+
+function Util.parse_schedule(str)
+    --parse a string like "1.2*10:1.5" to a list of numbers
+    local sch = {}
+    local s = Util.simple_split(str, ':')
+    for i = 1, #s do
+        local p = Util.simple_split(s[i], "%*")
+        if #p ~= 2 and #p ~= 1 then
+            nerv.error("nerv.SUtil:parse_schedule error, unit(%s) not proper, has %d components.", s[i], #p)
+        end
+        if p[2] == nil then
+            p[2] = "1"
+        end 
+        p[1] = tonumber(p[1])
+        p[2] = tonumber(p[2])
+        for j = 1, p[2] do
+            table.insert(sch, p[1])
+        end 
+    end 
+
+    --for i = 1, #sch do
+    --    print(sch[i])
+    --end
+    return sch
+end
+
+function Util.sche_get(s, it)
+    --get s[it]
+    if s == nil then
+        nerv.info("Util.sche_get: warning, scheule is nil, returning zero...")
+        return 0
+    end
+    if #s >= it then
+        return s[it]
+    else
+        nerv.info("Util.sche_get: warning, it(%d) > #schedule(%d), returning the last one of schedule(%f)...", it, #s, s[#s])
+        return s[#s]
+    end
+end
+
+function Util.parse_commands_set(str)
+    local coms = {}
+    local s = Util.simple_split(str, ':,')
+    for i = 1 ,#s do
+        if coms[s[i]] == 1 then
+            nerv.warning("nerv.SUtil.parse_commands_set command(%s) appered more than once in command_set(%s)", s[i], str)
+        end
+        coms[s[i]] = 1
+    end
+    return coms
+end
+
+function Util.log_redirect(fn)
+    nerv.log_fh = assert(io.open(fn, "w"))
+    nerv.info("CAUTION[LOG_REDIRECT], all nerv.printf/info/warning/error calls will be double-written to %s", fn)
+    nerv.printf = 
+        function (fmt, ...)
+            io.write(nerv.sprintf(fmt, ...))
+            nerv.log_fh:write(nerv.sprintf(fmt, ...))
+            nerv.log_fh:flush()
+        end
+    nerv.error = 
+        function (fmt, ...)
+            nerv.log_fh:write(nerv.sprintf("[nerv] internal error:" .. fmt .. "\n", ...))
+            error(nerv.sprintf("[nerv] internal error: " .. fmt .. "\n", ...))
+        end
+end