From 1638ed559cac91c7ecd693f292e47d6c1f6916ed Mon Sep 17 00:00:00 2001
From: Luke Hubmayer-Werner <mongoose41@gmail.com>
Date: Tue, 16 Jun 2020 00:39:16 +0930
Subject: [PATCH] Dump unit+weapon definitions for external data analysis

---
 export_UnitDefs.lua | 66 +++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 66 insertions(+)
 create mode 100644 export_UnitDefs.lua

diff --git a/export_UnitDefs.lua b/export_UnitDefs.lua
new file mode 100644
index 0000000..397b038
--- /dev/null
+++ b/export_UnitDefs.lua
@@ -0,0 +1,66 @@
+--------------------------------------------------------------------------------
+--------------------------------------------------------------------------------
+--
+--	file:		export_UnitDefs.lua
+--	brief:	Save the UnitDefs table to json for spreadsheeting
+--	author:	Birdulon
+--
+--	Copyright (C) 2020.
+--	Licensed under MIT or WTFPL, whichever you prefer.
+--
+--------------------------------------------------------------------------------
+--------------------------------------------------------------------------------
+
+--local json = require "json"
+--VFS.Include("LuaUI/Widgets/json.lua")
+
+function widget:GetInfo()
+	return {
+		name = "Export UnitDefs",
+		desc = "Exports UnitDefs table to a json when /exportUnitDefs is written in chat",
+		author = "Birdulon",
+		date = "June 2020",
+		license = "MIT/WTFPL",
+		layer = 0,
+		enabled = true  -- loaded by default?
+	}
+end
+
+
+function widget:Update(dt)
+	local filename = "SPR_" .. Engine.versionFull .. "_ZK_" .. Game.gameVersion .. "_unitdefs"
+	local data = {}
+	for unitID=1,#UnitDefs do
+		data[unitID] = {}
+		for k,v in UnitDefs[unitID]:pairs() do
+			if k == "wDefs" then
+				-- Skip as it's actually empty
+			elseif k == "weapons" then
+				data[unitID][k] = v
+				for i=1,#v do
+					if type(data[unitID].weapons[i].weaponDef) == "number" then
+						local wepDef = WeaponDefs[data[unitID].weapons[i].weaponDef]
+						for name,param in wepDef:pairs() do
+							data[unitID].weapons[i][name] = param  -- This flattens the structure a little bit, but I think it's a useful thing to do.
+						end
+					end
+				end
+			else
+				data[unitID][k] = v
+			end
+		end
+	end
+	table.save(data, filename .. ".lua")
+	Spring.Echo("game_message: UnitDefs table saved to " .. filename .. ".lua")
+-- 	if json then
+-- 		local file = io.open(filename .. ".json", "w")
+-- 		io.output(file)
+-- 		io.write(json.encode(data))
+-- 		io.close(file)
+-- 	Spring.Echo("game_message: UnitDefs table saved to " .. filename .. ".json")
+-- 	end
+	widgetHandler:RemoveWidget()
+end
+
+-- function widget:Initialize()
+-- end