wiremod-svn-archive/wire/lua/entities/gmod_wire_expression/init.lua

// Written by Syranide, me@syranide.com

// Found some extremely rare bug at line 170 (delta), probably got set to nil for some reason.

AddCSLuaFile( "cl_init.lua" )
AddCSLuaFile( "shared.lua" )
include('shared.lua')
include('parser.lua')

ENT.Delta = 0.001
ENT.OverlayDelay = 0
ENT.WireDebugName = "Expression"

if !WireModPacket then
	WireModPacket = {}
	WireModPacketIndex = 0
end

if !WireModVector then
	WireModVector = {}
	WireModVectorIndex = 0
end

function ENT:Initialize()
	self.Entity:PhysicsInit(SOLID_VPHYSICS)
	self.Entity:SetMoveType(MOVETYPE_VPHYSICS)
	self.Entity:SetSolid(SOLID_VPHYSICS)
	
	self.Xinputs =  {}
	self.Xoutputs = {}
	self.Xlocals =  {}

	self.deltavars = {}
	self.inputvars = {}
	self.triggvars = {}
	self.variables = {}
	
	self.Inputs = Wire_CreateInputs(self.Entity, {})
	self.Outputs = Wire_CreateOutputs(self.Entity, {})
end

function ENT:TriggerInput(key, value)
	if key then
		self.deltavars[key] = self.inputvars[key]
		self.inputvars[key] = value
		self.triggvars[key] = true
		self:Update()
		--self.Entity:NextThink(CurTime()+0.001)
	end
end

function ENT:Think()
	if !self.scheduled then return true end

	self.curtime = self.curtime + self.scheduled / 1000
	if math.abs(self.curtime - CurTime()) > 0.1 then self.curtime = CurTime() end

	self.scheduled = nil
	self.clocked = true
	self:Update()
	self.clocked = nil

	if !self.scheduled then self.curtime = nil end

	return true
end

function ENT:Update()
	for _,key in pairs(self.Xinputs) do
		self.variables[key] = self.inputvars[key]
	end

	self.schedule = nil	
	
	local tbl = self.instructions
	self["_"..tbl[1]](self,tbl)
	self.triggvars = {}

	if self.schedule and math.abs(self.schedule) >= self.Delta then
		if !self.clocked and self.schedule > 0 or !self.curtime then self.curtime = CurTime() end
		if self.schedule < 0  then self.schedule = -self.schedule end
		if self.schedule < 20 then self.schedule = 20 end
		self.scheduled = self.schedule
		self.Entity:NextThink(self.curtime + self.schedule / 1000)
	elseif self.scheduled and self.schedule and math.abs(self.schedule) < self.Delta then
		self.scheduled = nil
	end
	
	for _,key in ipairs(self.Xoutputs) do
		Wire_TriggerOutput(self.Entity, key, self.variables[key]) --major overhead, add lazy updates?
	end
end

function ENT:Reset()
	for _,key in ipairs(self.Xlocals)  do self.variables[key] = 0 self.deltavars[key] = 0 end
	for _,key in ipairs(self.Xoutputs) do self.variables[key] = 0 self.deltavars[key] = 0 end

	self.scheduled = nil
	self.schedule = nil
	self.curtime = nil
	self.initialized = false

	self:Update()

	self.initialized = true
end

function ENT:Setup(name, parser)
	local inputs =  parser:GetInputs()
	local outputs = parser:GetOutputs()
	local locals =  parser:GetLocals()
	
	local inputvars = {}
	local deltavars = {}
	local triggvars = {}
	local variables = {}
	
	for _,key in ipairs(inputs) do
		if !self.inputvars[key] then
			inputvars[key] = 0
			deltavars[key] = 0
		else
			inputvars[key] = self.inputvars[key]
			deltavars[key] = self.deltavars[key]
		end
	end
	
	for _,key in ipairs(outputs) do
		if !self.variables[key] then
			variables[key] = 0
			deltavars[key] = 0
		else
			variables[key] = self.variables[key]
			deltavars[key] = self.deltavars[key]
		end
	end
	
	for _,key in ipairs(locals) do
		if !self.variables[key] then
			variables[key] = 0
			deltavars[key] = 0
		else
			variables[key] = self.variables[key]
			deltavars[key] = self.deltavars[key]
		end
	end
	
	self.inputvars = inputvars
	self.deltavars = deltavars
	self.triggvars = triggvars
	self.variables = variables
	
	self.Xinputs =  inputs
	self.Xoutputs = outputs
	self.Xlocals =  locals
	
	self.instructions = parser:GetInstructions()
	
	Wire_AdjustInputs(self.Entity, inputs)
	Wire_AdjustOutputs(self.Entity, outputs)

	if name == "" then name = "generic" end
	self:SetOverlayText("Expression (" .. name .. ")")

	self.scheduled = nil
	self.schedule = nil
	self.curtime = nil
	self.initialized = false
	
	self:Update()

	self.initialized = true
	return true
end

--function ENT:Compile(tbl)
--	if type(tbl) == "table" then
--		tbl[1] = self["_" .. tbl[1]]
--		for i=2,#tbl do tbl[i] = self:Compile(tbl[i]) end
--	end
--	return tbl
--end

function ENT:_end(tbl)  return false end

function ENT:_num(tbl)  return tbl[2] end
function ENT:_var(tbl)  return self.variables[tbl[2]] end
function ENT:_dlt(tbl)  return self.variables[tbl[2]] - self.deltavars[tbl[2]] end
function ENT:_trg(tbl)  if self.triggvars[tbl[2]] then return 1 else return 0 end end

function ENT:_seq(tbl)  if self["_"..tbl[2][1]](self,tbl[2]) == false then return false elseif self["_"..tbl[3][1]](self,tbl[3]) == false then return false end end

function ENT:_fun(tbl)  local prm = self["_"..tbl[3][1]](self,tbl[3]) local fun = self["_"..tbl[2].."_"..#prm] if !fun and #prm > 0 then fun = self["_"..tbl[2].."_x"] end if fun == nil then return -1 end return fun(self,unpack(prm)) end
function ENT:_prm(tbl)  local prm = self["_"..tbl[2][1]](self,tbl[2]) table.insert(prm, self["_"..tbl[3][1]](self,tbl[3])) return prm end
function ENT:_nil(tbl)  return {} end

function ENT:_con(tbl)
	if self:GetPlayer() then
		local instr = tbl[2]
		local outstr = ""
		
		while true do
			local pos = string.find(instr, "$", 1, true)
			if pos then
				outstr = outstr .. string.sub(instr, 0, pos - 1)
				instr  = string.sub(instr, pos + 1)
				
				local pos = string.find(instr, "$", 1, true)
				if pos then
					local var = string.sub(instr, 0, pos - 1)
					if self.variables[var] then
						outstr = outstr .. tostring(self.variables[var])
					else
						outstr = outstr .. "0"
					end
					
					instr  = string.sub(instr, pos + 1)
				end
			else
				outstr = outstr .. instr
				break
			end
		end
		
		self:GetPlayer():ConCommand(outstr)
	end
end

function ENT:_imp(tbl)  if math.abs(self["_"..tbl[2][1]](self,tbl[2])) >= self.Delta then     if self["_"..tbl[3][1]](self,tbl[3]) == false then return false end end end
function ENT:_cnd(tbl)  if math.abs(self["_"..tbl[2][1]](self,tbl[2])) >= self.Delta then return self["_"..tbl[3][1]](self,tbl[3]) else return self["_"..tbl[4][1]](self,tbl[4]) end end

function ENT:_and(tbl)  if math.abs(self["_"..tbl[2][1]](self,tbl[2])) >= self.Delta and math.abs(self["_"..tbl[3][1]](self,tbl[3])) >= self.Delta then return 1 else return 0 end end
function ENT:_or(tbl)   if math.abs(self["_"..tbl[2][1]](self,tbl[2])) >= self.Delta or  math.abs(self["_"..tbl[3][1]](self,tbl[3])) >= self.Delta then return 1 else return 0 end end

function ENT:_ass(tbl)  self.triggvars[tbl[2][2]] = true self.deltavars[tbl[2][2]] = self.variables[tbl[2][2]] self.variables[tbl[2][2]] =                             self["_"..tbl[3][1]](self,tbl[3]) return self.variables[tbl[2][2]] end
function ENT:_aadd(tbl) self.triggvars[tbl[2][2]] = true self.deltavars[tbl[2][2]] = self.variables[tbl[2][2]] self.variables[tbl[2][2]] = self.variables[tbl[2][2]] + self["_"..tbl[3][1]](self,tbl[3]) return self.variables[tbl[2][2]] end
function ENT:_asub(tbl) self.triggvars[tbl[2][2]] = true self.deltavars[tbl[2][2]] = self.variables[tbl[2][2]] self.variables[tbl[2][2]] = self.variables[tbl[2][2]] - self["_"..tbl[3][1]](self,tbl[3]) return self.variables[tbl[2][2]] end
function ENT:_amul(tbl) self.triggvars[tbl[2][2]] = true self.deltavars[tbl[2][2]] = self.variables[tbl[2][2]] self.variables[tbl[2][2]] = self.variables[tbl[2][2]] * self["_"..tbl[3][1]](self,tbl[3]) return self.variables[tbl[2][2]] end
function ENT:_adiv(tbl) self.triggvars[tbl[2][2]] = true self.deltavars[tbl[2][2]] = self.variables[tbl[2][2]] self.variables[tbl[2][2]] = self.variables[tbl[2][2]] / self["_"..tbl[3][1]](self,tbl[3]) return self.variables[tbl[2][2]] end
function ENT:_amod(tbl) self.triggvars[tbl[2][2]] = true self.deltavars[tbl[2][2]] = self.variables[tbl[2][2]] self.variables[tbl[2][2]] = self.variables[tbl[2][2]] % self["_"..tbl[3][1]](self,tbl[3]) return self.variables[tbl[2][2]] end
function ENT:_aexp(tbl) self.triggvars[tbl[2][2]] = true self.deltavars[tbl[2][2]] = self.variables[tbl[2][2]] self.variables[tbl[2][2]] = self.variables[tbl[2][2]] ^ self["_"..tbl[3][1]](self,tbl[3]) return self.variables[tbl[2][2]] end

function ENT:_eq(tbl)   if math.abs(self["_"..tbl[2][1]](self,tbl[2]) - self["_"..tbl[3][1]](self,tbl[3])) < self.Delta  then return 1 else return 0 end end
function ENT:_neq(tbl)  if math.abs(self["_"..tbl[2][1]](self,tbl[2]) - self["_"..tbl[3][1]](self,tbl[3])) >= self.Delta then return 1 else return 0 end end

function ENT:_gth(tbl)  if self["_"..tbl[2][1]](self,tbl[2]) - self["_"..tbl[3][1]](self,tbl[3]) >= self.Delta  then return 1 else return 0 end end
function ENT:_lth(tbl)  if self["_"..tbl[2][1]](self,tbl[2]) - self["_"..tbl[3][1]](self,tbl[3]) <= -self.Delta then return 1 else return 0 end end
function ENT:_geq(tbl)  if self["_"..tbl[2][1]](self,tbl[2]) - self["_"..tbl[3][1]](self,tbl[3]) > -self.Delta  then return 1 else return 0 end end
function ENT:_leq(tbl)  if self["_"..tbl[2][1]](self,tbl[2]) - self["_"..tbl[3][1]](self,tbl[3]) < self.Delta   then return 1 else return 0 end end

function ENT:_neg(tbl)  return                                   - self["_"..tbl[2][1]](self,tbl[2]) end
function ENT:_add(tbl)  return self["_"..tbl[2][1]](self,tbl[2]) + self["_"..tbl[3][1]](self,tbl[3]) end
function ENT:_sub(tbl)  return self["_"..tbl[2][1]](self,tbl[2]) - self["_"..tbl[3][1]](self,tbl[3]) end
function ENT:_mul(tbl)  return self["_"..tbl[2][1]](self,tbl[2]) * self["_"..tbl[3][1]](self,tbl[3]) end
function ENT:_div(tbl)  return self["_"..tbl[2][1]](self,tbl[2]) / self["_"..tbl[3][1]](self,tbl[3]) end
function ENT:_mod(tbl)  return self["_"..tbl[2][1]](self,tbl[2]) % self["_"..tbl[3][1]](self,tbl[3]) end
function ENT:_exp(tbl)  return self["_"..tbl[2][1]](self,tbl[2]) ^ self["_"..tbl[3][1]](self,tbl[3]) end

function ENT:_not(tbl)  if math.abs(self["_"..tbl[2][1]](self,tbl[2])) < self.Delta then return 1 else return 0 end end

ENT._abs_1 =     function (self, n)    return math.abs(n) end
ENT._ceil_1 =    function (self, n)    return math.ceil(n) end
ENT._ceil_2 =    function (self, n, d) return math.ceil(n * 10 ^ d) / 10 ^ d end
ENT._clamp_3 =   function (self, v, l, u) if v < l then return l elseif v > u then return u else return v end end
ENT._inrange_3 = function (self, v, l, u) if v >= l and v <= u then return 1 else return 0 end end
ENT._exp_1 =     function (self, n)    return math.exp(n) end
ENT._floor_1 =   function (self, n)    return math.floor(n) end
ENT._floor_2 =   function (self, n, d) return math.floor(n * 10 ^ d) / 10 ^ d end
ENT._frac_1 =    function (self, n)    return math.fmod(n, 1) end
ENT._int_1 =     function (self, n)    return math.modf(n, 1) end
ENT._ln_1 =      function (self, n)    return math.log(n) / math.log(math.exp(1)) end
ENT._log_2 =     function (self, n, k) return math.log(n) / math.log(k) end
ENT._log2_1 =    function (self, n)    return math.log(n) end
ENT._log10_1 =   function (self, n)    return math.log10(n) end
ENT._mod_2 =     function (self, x, y) return math.fmod(x, y) end
ENT._sgn_1 =     function (self, n)    if n > 0 then return 1 elseif n < 0 then return -1 else return 0 end end
ENT._sqrt_1 =    function (self, n)    return math.sqrt(n) end
ENT._cbrt_1 =    function (self, n)    return 10 ^ ((1 / 3) * math.log10(n ^ 3)) end
ENT._root_1 =    function (self, n, k) return 10 ^ ((1 / k) * math.log10(n ^ k)) end
ENT._round_1 =   function (self, n)    return math.Round(n) end
ENT._round_2 =   function (self, n, d) return math.Round(n * 10 ^ d) / 10 ^ d end
ENT._e_0 =       function (self)       return math.exp(1) end

ENT._max_x =     function (self, ...)  return math.max(...) end
ENT._min_x =     function (self, ...)  return math.min(...) end
ENT._avg_x =     function (self, ...)  local n = 0 for _,v in ipairs({...}) do n = n + v end return n / #{...} end
ENT._sel_x =     function (self, i, ...) if ({...})[i] == nil then return -1 else return ({...})[i] end end

ENT._random_0 =  function (self)       return math.random() end
ENT._random_2 =  function (self, l, u) return math.random() * (u - l) + l end
ENT._curtime_0 = function (self)       return CurTime() end

ENT._deg_1 =     function (self, r)    return math.deg(r) end
ENT._rad_1 =     function (self, d)    return math.rad(d) end
ENT._pi_0 =      function (self)       return math.pi end

ENT._acosr_1 =   function (self, x)    return math.acos(x) end
ENT._asinr_1 =   function (self, x)    return math.asin(x) end
ENT._atanr_2 =   function (self, x, y) return math.atan2(x, y) end
ENT._atan2r_2 =  function (self, x, y) return math.atan2(x, y) end
ENT._atanr_1 =   function (self, x)    return math.atan(x) end
ENT._coshr_1 =   function (self, r)    return math.cosh(r) end
ENT._cosr_1 =    function (self, r)    return math.cos(r) end
ENT._sinr_1 =    function (self, r)    return math.sin(r) end
ENT._sinhr_1 =   function (self, r)    return math.sinh(r) end
ENT._tanr_1 =    function (self, r)    return math.tan(r) end
ENT._tanhr_1 =   function (self, r)    return math.tanh(r) end

ENT._acos_1 =    function (self, x)    return math.deg(math.acos(x)) end
ENT._asin_1 =    function (self, x)    return math.deg(math.asin(x)) end
ENT._atan_2 =    function (self, x, y) return math.deg(math.atan2(x, y)) end
ENT._atan2_2 =   function (self, x, y) return math.deg(math.atan2(x, y)) end
ENT._atan_1 =    function (self, x)    return math.deg(math.atan(x)) end
ENT._cosh_1 =    function (self, d)    return math.cosh(math.rad(d)) end
ENT._cos_1 =     function (self, d)    return math.cos(math.rad(d)) end
ENT._sin_1 =     function (self, d)    return math.sin(math.rad(d)) end
ENT._sinh_1 =    function (self, d)    return math.sinh(math.rad(d)) end
ENT._tan_1 =     function (self, d)    return math.tan(math.rad(d)) end
ENT._tanh_1 =    function (self, d)    return math.tanh(math.rad(d)) end

ENT._angnorm_1 =  function (self, d)   return (d + 180) % 360 - 180 end
ENT._angnormr_1 = function (self, d)   return (d + math.pi) % (math.pi * 2) - math.pi end

ENT._first_0 =    function (self, n)   if self.initialized then return 0 else return 1 end end

ENT._clk_0 =      function (self, n)   if self.clocked then return 1 else return 0 end end
ENT._schedule_1 = function (self, n)   self.schedule = n return 0 end
ENT._interval_1 = function (self, n)   self.schedule = -n return 0 end

ENT._send_x =    function (self, ...)   WireModPacketIndex = WireModPacketIndex % 90 + 1 WireModPacket[WireModPacketIndex] = {...} return WireModPacketIndex + 9 end
ENT._recv_2 =    function (self, id, p) id = id - 9 if WireModPacket[id] and WireModPacket[id][p] then return WireModPacket[id][p] else return -1 end end

function WireGateExpressionSendPacket(...)
	WireModPacketIndex = WireModPacketIndex % 90 + 1
	WireModPacket[WireModPacketIndex] = {...}
	return WireModPacketIndex + 9
end

function WireGateExpressionRecvPacket(packet, index)
	packet = packet - 9
	if WireModPacket[packet] and WireModPacket[packet][index] then
		return WireModPacket[packet][index]
	else
		return nil
	end
end

-- HIGHLY EXPERIMENTAL FUNCTIONALITY --

function ENT:GetVector(id)
	return WireModVector[id - 9]
end

function ENT:PutVector(v)
	WireModVectorIndex = WireModVectorIndex % 90 + 1
	WireModVector[WireModVectorIndex] = v
	return WireModVectorIndex + 9
end

function ENT:CloneVector(v)
	return Vector(v.x, v.y, v.z)
end

function ENT:IsVector(id1, id2)
	id1 = id1 - 9
	if !WireModVector[id1] then return false end

	if id2 then
		id2 = id2 - 9
		if !WireModVector[id2] then	return false end
	end

	return true
end


ENT._vector_2 =       function (self, x, y)    return self:PutVector(Vector(x, y, 0)) end
ENT._vector_3 =       function (self, x, y, z) return self:PutVector(Vector(x, y, z)) end

ENT._vecx_1 =         function (self, v) if self:IsVector(v) then return self:GetVector(v).x else return -1 end end
ENT._vecy_1 =         function (self, v) if self:IsVector(v) then return self:GetVector(v).y else return -1 end end
ENT._vecz_1 =         function (self, v) if self:IsVector(v) then return self:GetVector(v).z else return -1 end end

ENT._vecpitch_1 =     function (self, v) if self:IsVector(v) then return (self:GetVector(v):Angle().p + 180) % 360 - 180 else return -1 end end
ENT._vecyaw_1 =       function (self, v) if self:IsVector(v) then return (self:GetVector(v):Angle().y + 180) % 360 - 180 else return -1 end end

ENT._veclength_1 =    function (self, v) if self:IsVector(v) then return self:GetVector(v):Length() else return -1 end end
ENT._vecnormalize_1 = function (self, v) if self:IsVector(v) then return self:PutVector(self:GetVector(v):GetNormalized()) else return -1 end end

ENT._vecdot_2 =       function (self, v1, v2) if self:IsVector(v1, v2) then return self:GetVector(v1):Dot(self:GetVector(v2)) else return -1 end end
ENT._veccross_2 =     function (self, v1, v2) if self:IsVector(v1, v2) then return self:PutVector(self:GetVector(v1):Cross(self:GetVector(v2))) else return -1 end end
ENT._vecdistance_2 =  function (self, v1, v2) if self:IsVector(v1, v2) then return self:GetVector(v1):Distance(self:GetVector(v2)) else return -1 end end
ENT._vecadd_2 =       function (self, v1, v2) if self:IsVector(v1, v2) then return self:PutVector(self:GetVector(v1) + self:GetVector(v2)) else return -1 end end
ENT._vecsub_2 =       function (self, v1, v2) if self:IsVector(v1, v2) then return self:PutVector(self:GetVector(v1) - self:GetVector(v2)) else return -1 end end
ENT._vecmul_2 =       function (self, v1, v2) if self:IsVector(v1, v2) then return self:PutVector(self:GetVector(v1) * self:GetVector(v2)) else return -1 end end
ENT._vecsmul_2 =      function (self, v, n)   if self:IsVector(v) then return self:PutVector(self:GetVector(v) * n) else return -1 end end
ENT._vecsdiv_2 =      function (self, v, n)   if self:IsVector(v) then return self:PutVector(self:GetVector(v) / n) else return -1 end end

ENT._vecrotate_4 =    function (self, v, p, y, r) if self:IsVector(v) then local vec = self:CloneVector(self:GetVector(v)) vec:Rotate(Angle(p, y, r)) return self:PutVector(vec) else return -1 end end

-- EXTERNAL INPUT EXTENSION

ENT._extcolor_3 =   function (self, r, g, b)    self:SetColor(math.Clamp(r, 0, 255), math.Clamp(g, 0, 255), math.Clamp(b, 0, 255), 255)                   return 0 end
ENT._extcolor_4 =   function (self, r, g, b, a) self:SetColor(math.Clamp(r, 0, 255), math.Clamp(g, 0, 255), math.Clamp(b, 0, 255), math.Clamp(a, 0, 255)) return 0 end

ENT._extcolorr_0 =  function (self) local tbl = { self:GetColor() } return tbl[0] end
ENT._extcolorg_0 =  function (self) local tbl = { self:GetColor() } return tbl[1] end
ENT._extcolorb_0 =  function (self) local tbl = { self:GetColor() } return tbl[2] end
ENT._extcolora_0 =  function (self) local tbl = { self:GetColor() } return tbl[3] end

ENT._extdirfwx_0 =   function (self) return self:GetForward().x end
ENT._extdirfwy_0 =   function (self) return self:GetForward().y end
ENT._extdirfwz_0 =   function (self) return self:GetForward().z end
ENT._extdirfw_0 =    function (self) return self:PutVector(self:GetForward()) end

ENT._extdirrtx_0 =   function (self) return self:GetRight().x end
ENT._extdirrty_0 =   function (self) return self:GetRight().y end
ENT._extdirrtz_0 =   function (self) return self:GetRight().z end
ENT._extdirrt_0 =    function (self) return self:PutVector(self:GetRight()) end

ENT._extdirupx_0 =   function (self) return self:GetUp().x end
ENT._extdirupy_0 =   function (self) return self:GetUp().y end
ENT._extdirupz_0 =   function (self) return self:GetUp().z end
ENT._extdirup_0 =    function (self) return self:PutVector(self:GetUp()) end

ENT._extposx_0 =   function (self) return self:GetPos().x end
ENT._extposy_0 =   function (self) return self:GetPos().y end
ENT._extposz_0 =   function (self) return self:GetPos().z end
ENT._extpos_0 =    function (self) return self:PutVector(self:GetPos()) end

ENT._extvelx_0 =   function (self) return self:GetVelocity().x end
ENT._extvely_0 =   function (self) return self:GetVelocity().y end
ENT._extvelz_0 =   function (self) return self:GetVelocity().z end
ENT._extvel =      function (self) return self:PutVector(self:GetVelocity()) end

ENT._extangp_0 =   function (self) return self:GetAngles().p end
ENT._extangy_0 =   function (self) return self:GetAngles().y end
ENT._extangr_0 =   function (self) return self:GetAngles().r end

ENT._extangvelp_0 =   function (self) return self:GetPhysicsObject():GetAngleVelocity().x end
ENT._extangvely_0 =   function (self) return self:GetPhysicsObject():GetAngleVelocity().y end
ENT._extangvelr_0 =   function (self) return self:GetPhysicsObject():GetAngleVelocity().z end