add isReferenced function to querry the referenced state

1 files changed, 13 insertions, 0 deletions

diff --git a/File.lua b/File.lua
index c86d106..9e5773d 100644
--- a/File.lua
+++ b/File.lua
@@ -42,12 +42,25 @@ function File:isWritableObject(object)
 end
 
 function File:referenced(ref)
+   -- we use an environment to keep a record of written objects
+   if not torch.getenv(self).writeObjects then
+      torch.setenv(self, {writeObjects={}, writeObjectsRef={}, readObjects={}})
+   end
    local env = torch.getenv(self)
    env.force = not ref
    torch.setenv(self,env)
    return self
 end
 
+function File:isReferenced()
+   -- if no environment, then no forcing setup yet
+   if not torch.getenv(self).writeObjects then
+      return true
+   end
+   local env = torch.getenv(self)
+   return not env.force
+end
+
 function File:writeObject(object)
    -- we use an environment to keep a record of written objects
    if not torch.getenv(self).writeObjects then