From 74d9e9e7371c80394698fb9805cbf0cbde67a8f3 Mon Sep 17 00:00:00 2001
From: Determinant <ted.sybil@gmail.com>
Date: Tue, 2 Jun 2015 20:28:16 +0800
Subject: add ParamRepo, LayerRepo, DAGLayer

---
 nn/param_repo.lua | 26 ++++++++++++++++++++++++++
 1 file changed, 26 insertions(+)
 create mode 100644 nn/param_repo.lua

(limited to 'nn/param_repo.lua')

diff --git a/nn/param_repo.lua b/nn/param_repo.lua
new file mode 100644
index 0000000..3e37c31
--- /dev/null
+++ b/nn/param_repo.lua
@@ -0,0 +1,26 @@
+local ParamRepo = nerv.class("nerv.ParamRepo")
+
+function ParamRepo:__init(param_files)
+    local param_table = {}
+    if type(param_files) ~= "table" then
+        nerv.error("param file table is need")
+    end
+    for i = 1, #param_files do
+        local pf = nerv.ChunkFile(param_files[i], "r")
+        for cid, cspec in pairs(pf.metadata) do
+            if param_table[cid] ~= nil then
+                nerv.error("conflicting chunk id in param files")
+            end
+            param_table[cid] = pf
+        end
+    end
+    self.param_table = param_table
+end
+
+function ParamRepo:get_param(pid, global_conf)
+    local pf = self.param_table[pid]
+    if pf == nil then
+        nerv.error("param with id %s not found", pid)
+    end
+    return pf:read_chunk(pid, global_conf)
+end
-- 
cgit v1.2.3