local Widget		= require 'Widget'
local Arc			= require 'Arc'
local Dial 			= require 'Dial'
local CriticalText	= require 'CriticalText'
local Text			= require 'Text'
local TextColumn	= require 'TextColumn'
local Line			= require 'Line'
local LabelPlot		= require 'LabelPlot'
local Table			= require 'Table'
local util			= require 'util'
local schema		= require 'default_patterns'

local __string_match		= string.match
local __cairo_path_destroy 	= cairo_path_destroy

local _MODULE_Y_ = 712
local _DIAL_THICKNESS_ = 8
local _DIAL_SPACING_ = 1
local _TEXT_Y_OFFSET_ = 7
local _TEXT_LEFT_X_OFFSET_ = 30
local _TEXT_SPACING_ = 20
local _PLOT_SECTION_BREAK_ = 30
local _PLOT_HEIGHT_ = 56
local _TABLE_SECTION_BREAK_ = 20
local _TABLE_HEIGHT_ = 114

local MEM_TOTAL_KB = tonumber(util.read_file('/proc/meminfo', 'MemTotal:%s+(%d+)'))

local TABLE_CONKY = {}
for c = 1, 3 do TABLE_CONKY[c] = {} end
for r = 1, 5 do TABLE_CONKY[1][r] = '${top_mem name '..r..'}' end
for r = 1, 5 do TABLE_CONKY[2][r] = '${top_mem pid '..r..'}' end
for r = 1, 5 do TABLE_CONKY[3][r] = '${top_mem mem '..r..'}' end

local header = Widget.Header{
	x = _G_INIT_DATA_.RIGHT_X,
	y = _MODULE_Y_,
	width = _G_INIT_DATA_.SECTION_WIDTH,
	header = 'MEMORY'
}

local DIAL_RADIUS = 32
local DIAL_THETA_0 = math.rad(90)
local DIAL_THETA_1 = math.rad(360)
local DIAL_X = _G_INIT_DATA_.RIGHT_X + DIAL_RADIUS + _DIAL_THICKNESS_ / 2
local DIAL_Y = header.bottom_y + DIAL_RADIUS + _DIAL_THICKNESS_ / 2

local dial = Widget.Dial{
	x 				= DIAL_X,
	y 				= DIAL_Y,			
	radius 			= DIAL_RADIUS,
	thickness 		= _DIAL_THICKNESS_,
	critical_limit 	= '>0.8'
}
local cache_arc = Widget.Arc{
	x 			= DIAL_X,
	y 			= DIAL_Y,			
	radius 		= DIAL_RADIUS,
	thickness 	= _DIAL_THICKNESS_,
	arc_pattern	= schema.purple_rounded
}

local total_used = Widget.CriticalText{
	x 			= DIAL_X,
	y 			= DIAL_Y,
	x_align 	= 'center',
	y_align 	= 'center',
	append_end 	= '%'
}

local inner_ring = Widget.Arc{
	x 		= DIAL_X,
	y 		= DIAL_Y,
	radius 	= DIAL_RADIUS - _DIAL_THICKNESS_ / 2 - 2,
	theta0	= 0,
	theta1	= 360
}

local _LINE_1_Y_ = header.bottom_y + _TEXT_Y_OFFSET_
local _TEXT_LEFT_X_ = _G_INIT_DATA_.RIGHT_X + DIAL_RADIUS * 2 + _TEXT_LEFT_X_OFFSET_
local _RIGHT_X_ = _G_INIT_DATA_.RIGHT_X + _G_INIT_DATA_.SECTION_WIDTH

local swap= {
	label = Widget.Text{
		x 		= _TEXT_LEFT_X_,
		y 		= _LINE_1_Y_,
		spacing = _TEXT_SPACING_,
		text	= 'Swap Usage'
	},
	percent = Widget.CriticalText{
		x 			= _RIGHT_X_,
		y 			= _LINE_1_Y_,
		x_align 	= 'right',
		append_end 	= ' %',
	},
}

local cache = {
	labels = Widget.TextColumn{
		x 		= _TEXT_LEFT_X_,
		y 		= _LINE_1_Y_ + _TEXT_SPACING_,
		spacing = _TEXT_SPACING_,
		'Page Cache',
		'Buffers',
		'Kernel Slab'
	},
	percents = Widget.TextColumn{
		x 			= _RIGHT_X_,
		y 			= _LINE_1_Y_ + _TEXT_SPACING_,
		x_align 	= 'right',
		append_end 	= ' %',
		text_color	= schema.purple,
		'<page_cache>',
		'<buffers>',
		'<kernel_slab>'
	},
}

local _PLOT_Y_ = _PLOT_SECTION_BREAK_ + header.bottom_y + DIAL_RADIUS * 2

local plot = Widget.LabelPlot{
	x = _G_INIT_DATA_.RIGHT_X,
	y = _PLOT_Y_,
	width = _G_INIT_DATA_.SECTION_WIDTH,
	height = _PLOT_HEIGHT_
}

local tbl = Widget.Table{
	x = _G_INIT_DATA_.RIGHT_X,
	y = _PLOT_Y_ + _PLOT_HEIGHT_ + _TABLE_SECTION_BREAK_,
	width = _G_INIT_DATA_.SECTION_WIDTH,
	height = _TABLE_HEIGHT_,
	'Name',
	'PID',
	'Mem (%)'
}

local __update = function(cr)
	local MEM_TOTAL_KB = MEM_TOTAL_KB

	local round = util.round
	local precision_round_to_string = util.precision_round_to_string
	local glob = util.read_file('/proc/meminfo')	--kB

	--see source for "free" for formulas and stuff ;)

	local swap_free		= __string_match(glob, 'SwapFree:%s+(%d+)' )
	local swap_total 	= __string_match(glob, 'SwapTotal:%s+(%d+)')
	local page_cache 	= __string_match(glob, 'Cached:%s+(%d+)'   )
	local slab 			= __string_match(glob, 'Slab:%s+(%d+)'     )
	local buffers 		= __string_match(glob, 'Buffers:%s+(%d+)'  )
	local free 			= __string_match(glob, 'MemFree:%s+(%d+)'  )

	local used_percent = util.round((MEM_TOTAL_KB - free - page_cache - buffers - slab) / MEM_TOTAL_KB, 2)

	Dial.set(dial, used_percent)
	CriticalText.set(total_used, cr, used_percent * 100)

	local cache_theta = (DIAL_THETA_0 - DIAL_THETA_1) / MEM_TOTAL_KB * free + DIAL_THETA_1
	__cairo_path_destroy(cache_arc.path)
	cache_arc.path = Arc.create_path(DIAL_X, DIAL_Y, DIAL_RADIUS, dial.dial_angle, cache_theta)
	
	CriticalText.set(swap.percent, cr, precision_round_to_string((swap_total - swap_free) /	swap_total * 100))

	local percents = cache.percents
	TextColumn.set(percents, cr, 1, precision_round_to_string(page_cache / MEM_TOTAL_KB * 100))
	TextColumn.set(percents, cr, 2, precision_round_to_string(buffers / MEM_TOTAL_KB * 100))
	TextColumn.set(percents, cr, 3, precision_round_to_string(slab / MEM_TOTAL_KB * 100))

	LabelPlot.update(plot, used_percent)

	for c = 1, 3 do
		local column = TABLE_CONKY[c]
		for r = 1, 5 do
			Table.set(tbl, cr, c, r, util.conky(column[r], '(%S+)'))
		end
	end
end

Widget = nil
schema = nil
_MODULE_Y_ = nil
_DIAL_THICKNESS_ = nil
_DIAL_SPACING_ = nil
_TEXT_Y_OFFSET_ = nil
_TEXT_LEFT_X_OFFSET_ = nil
_TEXT_SPACING_ = nil
_PLOT_SECTION_BREAK_ = nil
_PLOT_HEIGHT_ = nil
_TABLE_SECTION_BREAK_ = nil
_TABLE_HEIGHT_ = nil
_LINE_1_Y_ = nil
_TEXT_LEFT_X_ = nil
_RIGHT_X_ = nil
_PLOT_Y_ = nil

local draw = function(cr, current_interface)
	__update(cr)

	if current_interface == 0 then
		Text.draw(header.text, cr)
		Line.draw(header.underline, cr)
		Dial.draw(dial, cr)
		Arc.draw(cache_arc, cr)
		Arc.draw(inner_ring, cr)
		CriticalText.draw(total_used, cr)

		Text.draw(swap.label, cr)
		CriticalText.draw(swap.percent, cr)
		TextColumn.draw(cache.labels, cr)
		TextColumn.draw(cache.percents, cr)

		LabelPlot.draw(plot, cr)

		Table.draw(tbl, cr)
	end
end

return draw