restore kartencore to initial value

README.md

@@ -0,0 +1,2 @@
+# Leaderboard plugin for srb2kart
+https://mb.srb2.org/addons/time-attack-leaderboard.3742/

leaderboard.lua

@@ -11,15 +11,20 @@ local prevLap = 0
 local splits = {}
 local PATCH = nil
 local help = true
+local EncoreInitial = nil
+
+-- Tracks if stats have been written or not
+local StatTrack = false
 
 local UNCLAIMED = "Unclaimed Record"
-local HELP_MESSAGE = "\x89Leaderboard Commands:\nretry exit findmap changelevel spba_clearcheats lb_gui rival scroll"
+local HELP_MESSAGE = "\x89Leaderboard Commands:\nretry exit findmap changelevel spba_clearcheats lb_gui rival scroll encore"
 local FILENAME = "leaderboard.txt"
 
 -- Retry / changelevel map
 local nextMap = nil
 
 local Flags = 0
+local F_ENCORE = 0x80
 
 -- SPB flags with the least significance first
 local F_SPBATK = 0x1
@@ -27,9 +32,12 @@ local F_SPBJUS = 0x2
 local F_SPBBIG = 0x4
 local F_SPBEXP = 0x8
 
+-- Score table separator
+local ST_SEP = F_SPBATK
+
 local clearcheats = false
 
-local START_TIME = 6 * TICRATE + (3 * TICRATE / 4)
+local START_TIME = 6 * TICRATE + (3 * TICRATE / 4) + 1
 local AFK_TIMEOUT = TICRATE * 5
 local AFK_BALANCE = TICRATE * 60
 local PREVENT_JOIN_TIME = START_TIME + TICRATE * 5
@@ -61,6 +69,8 @@ local clamp
 local scroll_to
 
 local ticsToTime
+
+local allowJoin
 ---------------
 
 local cv_gui = CV_RegisterVar({
@@ -84,6 +94,9 @@ local cv_enable = CV_RegisterVar({
 	PossibleValue = CV_OnOff,
 	func = function(v)
 		disable = $ or not v.value
+		if disable then
+			allowJoin(true)
+		end
 	end
 })
 
@@ -106,25 +119,79 @@ local cv_interrupt = CV_RegisterVar({
 	end
 })
 
+local function setST(t, map, flags, scoreTable)
+	local mode = flags & ST_SEP
+	t[mode] = t[mode] or {}
+	t[mode][map] = scoreTable
+end
+
+local function getST(t, map, flags)
+	local mode = flags & ST_SEP
+	return t[mode] and t[mode][map] or nil
+end
+
 local function setScoreTable(map, flags, scoreTable)
-	local mode = flags & F_SPBATK
+	setST(lb, map, flags, scoreTable)
-	lb[mode] = lb[mode] or {}
-	lb[mode][map] = scoreTable
 end
 
 local function getScoreTable(map, flags)
-	--local id = tostring(map)
+	return getST(lb, map, flags)
-	--if flags & F_SPBATK then
-	--	id = id + "S"
-	--end
-
-	--return id
-
-	local mode = flags & F_SPBATK
-	return lb[mode] and lb[mode][map] or nil
 end
 
-local function score_t(map, name, skin, color, time, splits, flags, restat)
+-- True if a is better than b
+local function lbComp(a, b)
+	-- Calculates the difficulty, harder has higher priority
+	-- if s is positive then a is harder
+	-- if s is negative then b is harder
+	-- if s is 0 then compare time
+	local s = (a["flags"] & (F_SPBEXP | F_SPBBIG)) - (b["flags"] & (F_SPBEXP | F_SPBBIG))
+	return s > 0 or not(s < 0 or a["time"] >= b["time"])
+end
+
+local function sortScores()
+	for mode, t in pairs(lb) do
+		for map, scoreTable in pairs(t) do
+			table.sort(scoreTable, lbComp)
+			setScoreTable(map, mode, scoreTable)
+		end
+	end
+end
+
+-- Reinitialize lb based on new ST_SEP value
+local function reinit_lb()
+	local nlb = {}
+
+	for mode, t in pairs(lb) do
+		for map, scoreTable in pairs(t) do
+			for i, score in ipairs(scoreTable) do
+				local st = getST(nlb, map, score["flags"]) or {}
+				table.insert(st, score)
+				setST(nlb, map, score["flags"], st)
+			end
+		end
+	end
+
+	lb = nlb
+	sortScores()
+end
+
+local cv_spb_separate = CV_RegisterVar({
+	name = "lb_spb_separate",
+	defaultvalue = 0,
+	flags = CV_NETVAR | CV_CALL,
+	PossibleValue = CV_YesNo,
+	func = function(v)
+		if v.value then
+			ST_SEP = F_SPBATK | F_SPBBIG | F_SPBEXP
+		else
+			ST_SEP = F_SPBATK
+		end
+
+		reinit_lb()
+	end
+})
+
+local function score_t(map, name, skin, color, time, splits, flags, stat)
 	return {
 		["map"]    = map,
 		["name"]   = name,
@@ -133,11 +200,11 @@ local function score_t(map, name, skin, color, time, splits, flags, restat)
 		["time"]   = time,
 		["splits"] = splits,
 		["flags"]  = flags,
-		["restat"] = restat
+		["stat"] = stat
 	}
 end
 
-local function restat_t(speed, weight)
+local function stat_t(speed, weight)
 	if speed and weight then
 		return {
 			["speed"] = speed,
@@ -147,9 +214,9 @@ local function restat_t(speed, weight)
 	return nil
 end
 
-local function restat_str(restat)
+local function stat_str(stat)
-	if restat then
+	if stat then
-		return string.format("%d%d", restat["speed"], restat["weight"])
+		return string.format("%d%d", stat["speed"], stat["weight"])
 	end
 
 	return "0"
@@ -160,7 +227,7 @@ local f = io.open(FILENAME, "r")
 if f then
 	for l in f:lines() do
 		-- Leaderboard is stored in the following tab separated format
-		-- mapnum, name, skin, color, time, splits, flags, restat
+		-- mapnum, name, skin, color, time, splits, flags, stat
 		local t = {}
 		for word in (l+"\t"):gmatch("(.-)\t") do
 			table.insert(t, word)
@@ -183,9 +250,9 @@ if f then
 		local stats = nil
 		if t[8] != nil then
 			if #t[8] >= 2 then
-				local speed = string.sub(t[8], 1, 1)
+				local speed = tonumber(string.sub(t[8], 1, 1))
-				local weight = string.sub(t[8], 2, 2)
+				local weight = tonumber(string.sub(t[8], 2, 2))
-				stats = restat_t(speed, weight)
+				stats = stat_t(speed, weight)
 			end
 		end
 
@@ -202,15 +269,17 @@ if f then
 				stats
 			)
 		)
+
 		setScoreTable(tonumber(t[1]), flags, scoreTable)
 	end
 
+	sortScores()
 	f:close()
 else
 	print("Failed to open file: ", FILENAME)
 end
 
-local function allowJoin(v)
+function allowJoin(v)
 	if not cv_interrupt.value then
 		local y
 		if v then
@@ -242,6 +311,13 @@ local function initLeaderboard(player)
 		disable = disable or ingame() > 1
 	end
 	disable = $ or not cv_enable.value
+
+	-- Restore encore mode to initial value
+	if disable and EncoreInitial != nil then
+		COM_BufInsertText(server, string.format("kartencore %d", EncoreInitial))
+		EncoreInitial = nil
+	end
+
 	player.afkTime = leveltime
 end
 addHook("PlayerSpawn", initLeaderboard)
@@ -352,6 +428,24 @@ local function changelevel(player, ...)
 end
 COM_AddCommand("changelevel", changelevel)
 
+local function toggleEncore(player)
+	if not doyoudare(player) then
+		return
+	end
+
+	local enc = CV_FindVar("kartencore")
+	if EncoreInitial == nil then
+		EncoreInitial = enc.value
+	end
+
+	if enc.value then
+		COM_BufInsertText(server, "kartencore off")
+	else
+		COM_BufInsertText(server, "kartencore on")
+	end
+end
+COM_AddCommand("encore", toggleEncore)
+
 local function clearcheats(player)
 	if not player.spectator then
 		clearcheats = true
@@ -616,12 +710,25 @@ local FACERANK_DIM = 16
 local FACERANK_SPC = FACERANK_DIM + 4
 local function drawScore(v, player, pos, x, y, gui, faceRank, score, drawPos, textVFlags)
 	textVFlags = textVFlags or V_HUDTRANSHALF
+
 	--draw Patch/chili
 	v.draw(x, y, faceRank, V_HUDTRANS | VFLAGS, v.getColormap("sonic", score["color"]))
 	if player.name == score["name"] then
 		v.draw(x, y, PATCH["CHILI"][(leveltime / 4) % 8], V_HUDTRANS | VFLAGS)
 	end
 
+	-- Encore
+	if score["flags"] & F_ENCORE then
+		local bob = sin((leveltime + i * 5) * (ANG10))
+		v.drawScaled(
+			x * FRACUNIT,
+			bob + (y + FACERANK_DIM / 2) * FRACUNIT,
+			FRACUNIT / 6,
+			PATCH["RUBY"],
+			V_HUDTRANS
+		)
+	end
+
 	-- SPB
 	if score["flags"] & F_SPBATK then
 		local scale = FRACUNIT / 4
@@ -670,15 +777,18 @@ local function drawScore(v, player, pos, x, y, gui, faceRank, score, drawPos, te
 		v.drawNum(x, y + 3, pos, textVFlags | VFLAGS)
 	end
 
-	-- Restats
+	-- Stats
-	local restat = score["restat"]
+	local stat = score["stat"]
-	if restat then
+	local pskin = skins[score["skin"]]
-		v.drawString(x + FACERANK_DIM - 2, y + 4, restat["speed"], V_HUDTRANS | VFLAGS, "small")
+	if stat and not (
-		v.drawString(x + FACERANK_DIM - 2, y + 8, restat["weight"], V_HUDTRANS | VFLAGS, "small")
+			pskin
+			and pskin.kartweight == stat["weight"]
+			and pskin.kartspeed == stat["speed"]
+		) then
+		v.drawString(x + FACERANK_DIM - 2, y + 4, stat["speed"], V_HUDTRANS | VFLAGS, "small")
+		v.drawString(x + FACERANK_DIM - 2, y + 8, stat["weight"], V_HUDTRANS | VFLAGS, "small")
 	end
 
-	
-	
 	if gui == GUI_ON or (gui == GUI_SPLITS and showSplit) then
 		local name = score["name"]
 
@@ -766,10 +876,10 @@ local function drawScroll(v, player, scoreTable, gui)
 		end
 
 		local x = 10
-		if #scoreTable > 10 then
+		if #scoreTable >= 10 then
 			x = x + 8
 		end
-		if #scoreTable > 100 then
+		if #scoreTable >= 100 then
 			x = x + 8
 		end
 
@@ -850,19 +960,10 @@ function cachePatches(v)
 		end
 		PATCH["BIG"] = v.cachePatch("K_ISGROW")
 		PATCH["HYUD"] = v.cachePatch("K_ISHYUD")
+		PATCH["RUBY"] = v.cachePatch("RUBYICON")
 	end
 end
 
--- True if a is better than b
-local function lbComp(a, b)
-	-- Calculates the difficulty, harder has higher priority
-	-- if s is positive then a is harder
-	-- if s is negative then b is harder
-	-- if s is 0 then compare time
-	local s = (a["flags"] & (F_SPBEXP | F_SPBBIG)) - (b["flags"] & (F_SPBEXP | F_SPBBIG))
-	return s > 0 or not(s < 0 or a["time"] >= b["time"])
-end
-
 -- Find location of player and scroll to it
 function scroll_to(player)
 	local m = getScoreTable(gamemap, Flags) or {}
@@ -878,9 +979,25 @@ function scroll_to(player)
 	drawState = DS_SCRLTO
 end
 
+-- Write skin stats to each score where there are none
+local function writeStats()
+	for _, t in pairs(lb) do
+		for _, scoreTable in pairs(t) do
+			for _, score in ipairs(scoreTable) do
+				local skin = skins[score["skin"]]
+				if skin and not score["stat"] then
+					local stats = stat_t(skin.kartspeed, skin.kartweight)
+					score["stat"] = stats
+				end
+			end
+		end
+	end
+end
+
 local function saveTime(player)
 	local scoreTable = getScoreTable(gamemap, Flags) or {}
 
+	local pskin = skins[player.mo.skin]
 	local newscore = score_t(
 		gamemap,
 		player.name,
@@ -889,7 +1006,7 @@ local function saveTime(player)
 		timeFinished,
 		splits,
 		Flags,
-		restat_t(player.HMRs, player.HMRw)
+		stat_t(player.HMRs or pskin.kartspeed, player.HMRw or pskin.kartweight)
 	)
 
 	-- Check if you beat your previous best
@@ -923,6 +1040,11 @@ local function saveTime(player)
 
 	setScoreTable(gamemap, Flags, scoreTable)
 
+	if not StatTrack then
+		writeStats()
+		StatTrack = true
+	end
+
 	local f = assert(io.open(FILENAME, "w"))
 	if f == nil then
 		print("Failed to open file for writing: " + FILENAME)
@@ -940,7 +1062,7 @@ local function saveTime(player)
 					score["time"], "\t",
 					table.concat(score["splits"], " "), "\t",
 					score["flags"], "\t",
-					restat_str(score["restat"]), "\n"
+					stat_str(score["stat"]), "\n"
 				)
 			end
 		end
@@ -1021,7 +1143,13 @@ local function think()
 		end
 
 		-- Autospec
+		-- Encore
 		if leveltime == 1 then
+			Flags = $ & !F_ENCORE
+			if encoremode then
+				Flags = $ | F_ENCORE
+			end
+
 			if p then
 				for s in players.iterate do
 					if s.valid and s.spectator then
@@ -1125,9 +1253,12 @@ function clamp(min_v, v, max_v)
 end
 
 local function netvars(net)
-	lb = net($)
+	Flags = net($)
 	splits = net($)
 	prevLap = net($)
 	drawState = net($)
+	StatTrack = net($)
+	EncoreInitial = net($)
+	lb = net($)
 end
 addHook("NetVars", netvars)