aux-addon-ClassicAPI/history.lua

local private, public = {}, {}
Aux.history = public

private.PUSH_INTERVAL = 57600
private.NEW_RECORD = '0#0#0#0#'

function private.load_data()
	local dataset = Aux.persistence.load_dataset()
	dataset.history = dataset.history or { next_push = time() + private.PUSH_INTERVAL, item_data = {} }
	return dataset.history
end

function public.read_record(item_key)
	local data = private.load_data()
	local record = Aux.persistence.deserialize(data.item_data[item_key] or private.NEW_RECORD, '#')
	return {
		auction_count = tonumber(record[1]),
		day_count = tonumber(record[2]),
		EMA5 = tonumber(record[3]),
		EMA30 = tonumber(record[4]),
		histogram = Aux.util.map(Aux.persistence.deserialize(record[5], ';'), function(value)
			return tonumber(value)
		end),
	}
end

function private.write_record(item_key, record)
	local data = private.load_data()
	data.item_data[item_key] = Aux.persistence.serialize({
		record.auction_count,
		record.day_count,
		record.EMA5,
		record.EMA30,
		Aux.persistence.serialize(record.histogram, ';'),
	},'#')
end

function public.process_auction(auction_info)

	if auction_info.buyout_price == 0 then
		return
	end

	local data = private.load_data()

	if data.next_push < time() then
		private.push_data()
	end

	local buyout = auction_info.buyout_price / auction_info.aux_quantity

	local item_record = public.read_record(auction_info.item_key)

	item_record.auction_count = item_record.auction_count + 1

	for i=1,225 do
		item_record.histogram[i] = item_record.histogram[i] or 0
		if buyout < 1.1 ^ i then
			item_record.histogram[i] = item_record.histogram[i] + 1
			break
		end
	end

	private.write_record(auction_info.item_key, item_record)
end

function public.get_price_data(item_key)
	local item_record = public.read_record(item_key)
	return item_record.auction_count, item_record.day_count, private.daily_market_value(item_record.histogram), item_record.EMA5, item_record.EMA30
end

function public.get_market_value(item_key)
	local auction_count, day_count, daily_market_value, EMA5, EMA30 = public.get_price_data(item_key)

	if day_count == 0 then
		return daily_market_value
	else
		return EMA5
	end
end

function private.daily_market_value(histogram)

	local daily_auction_count = 0
	for _, frequency in ipairs(histogram) do
		daily_auction_count = daily_auction_count + frequency
	end

	if daily_auction_count == 0 then
		return 0
	end

	-- average of lowest 25%
	local sum, count = 0, 0
	local limit = daily_auction_count * 0.25
	for i, frequency in ipairs(histogram) do
		local limited_frequency = min(frequency, limit - count)
		sum = sum + 1.1 ^ (i - 1) * 1.05 * limited_frequency
		count = count + frequency
		if count >= limit then
			break
		end
	end
	return sum / limit
end

function private.push_data()
	local data = private.load_data()
	local item_data = data.item_data

	for item_key, _ in pairs(item_data) do

		local item_record = public.read_record(item_key)

		if getn(item_record.histogram) ~= 0 then

			local daily_market_value = private.daily_market_value(item_record.histogram)

			if item_record.day_count == 0 then
				item_record.EMA5 = daily_market_value
				item_record.EMA30 = daily_market_value
			else
				item_record.EMA5 = 2/3 * item_record.EMA5 + 1/3 * daily_market_value
				item_record.EMA30 = 13/14 * item_record.EMA30 + 1/14 * daily_market_value
			end

			item_record.day_count = item_record.day_count + 1
			item_record.histogram = {}

			private.write_record(item_key, item_record)
		end
	end

	data.next_push = time() + private.PUSH_INTERVAL
end

--function private.max_heap(array)
--	local self = {}
--
--	local ROOT = 1
--
--	local function parent(i)
--		return floor((i - 2) / 2) + 1
--	end
--
--	local function left_child(i)
--		return 2 * (i - 1) + 2
--	end
--
--	local function right_child(i)
--		return 2 * (i - 1) + 3
--	end
--
--	function self.insert(value)
--		local index = getn(array) + 1
--
--		while index > ROOT and array[index - 1] > 0 do
--			index = index - 1
--		end
--	end
--
--	function self.extract(signature)
--		return data[signature] ~= nil and data[signature] >= time()
--	end
--
--	return self
--end