argcheck.lua

local env = require 'argcheck.env' -- retrieve argcheck environement

-- From http://lua-users.org/wiki/SplitJoin
function string:split(sep)
	local sep, fields = sep or ":", {}
	local pattern = string.format("([^%s]+)", sep)
	self:gsub(pattern, function(c) fields[#fields+1] = c end)
	return fields
end

env.istype = function(obj, typename)
	if (typename == "*") then
		return true
	end

	-- From the original argcheck env
	local thname = torch.typename(obj) -- empty if non-torch class
	local thtype = torch.type(obj)
	if (typename == "!table" and thtype ~= "table") then
		return true
	end

	if (typename:match("|")) then
		if (thname) then
			-- Do a recursive search thrhough all the patterns for torch class objects
			for _,subtype in ipairs(typename:split("|")) do
				local ret = env.istype(obj, subtype)
				if (ret) then
					return true
				end
			end

			return false
		else
			-- We only need to find basic variable match + nan values
			for _,subtype in ipairs(typename:split("|")) do
				if ((thtype == subtype) or
					 (thtype == "nan" and isnan(obj)))
				then
					return true
				end
			end

			return false
		end
	end

	if thname then
		-- __typename (see below) might be absent
		local match = thname:match(typename)
		if match and (match ~= typename or match == thname) then
			return true
		end
		local mt = torch.getmetatable(thname)
		while mt do
			if mt.__typename then
				match = mt.__typename:match(typename)
				if match and (match ~= typename or match == mt.__typename) then
					return true
				end
			end
			mt = getmetatable(mt)
		end
		return false
	end

	return type(obj) == typename
end