farmbuyer@112: --[[-----------------------------------------------------------------------------
farmbuyer@112: Checkbox Widget
farmbuyer@112: -------------------------------------------------------------------------------]]
farmbuyer@112: local Type, Version = "CheckBox", 22
farmbuyer@112: local AceGUI = LibStub and LibStub("AceGUI-3.0", true)
farmbuyer@112: if not AceGUI or (AceGUI:GetWidgetVersion(Type) or 0) >= Version then return end
farmbuyer@112: 
farmbuyer@112: -- Lua APIs
farmbuyer@112: local select, pairs = select, pairs
farmbuyer@112: 
farmbuyer@112: -- WoW APIs
farmbuyer@112: local PlaySound = PlaySound
farmbuyer@112: local CreateFrame, UIParent = CreateFrame, UIParent
farmbuyer@112: 
farmbuyer@112: -- Global vars/functions that we don't upvalue since they might get hooked, or upgraded
farmbuyer@112: -- List them here for Mikk's FindGlobals script
farmbuyer@112: -- GLOBALS: SetDesaturation, GameFontHighlight
farmbuyer@112: 
farmbuyer@112: --[[-----------------------------------------------------------------------------
farmbuyer@112: Support functions
farmbuyer@112: -------------------------------------------------------------------------------]]
farmbuyer@112: local function AlignImage(self)
farmbuyer@112: 	local img = self.image:GetTexture()
farmbuyer@112: 	self.text:ClearAllPoints()
farmbuyer@112: 	if not img then
farmbuyer@112: 		self.text:SetPoint("LEFT", self.checkbg, "RIGHT")
farmbuyer@112: 		self.text:SetPoint("RIGHT")
farmbuyer@112: 	else
farmbuyer@112: 		self.text:SetPoint("LEFT", self.image,"RIGHT", 1, 0)
farmbuyer@112: 		self.text:SetPoint("RIGHT")
farmbuyer@112: 	end
farmbuyer@112: end
farmbuyer@112: 
farmbuyer@112: --[[-----------------------------------------------------------------------------
farmbuyer@112: Scripts
farmbuyer@112: -------------------------------------------------------------------------------]]
farmbuyer@112: local function Control_OnEnter(frame)
farmbuyer@112: 	frame.obj:Fire("OnEnter")
farmbuyer@112: end
farmbuyer@112: 
farmbuyer@112: local function Control_OnLeave(frame)
farmbuyer@112: 	frame.obj:Fire("OnLeave")
farmbuyer@112: end
farmbuyer@112: 
farmbuyer@112: local function CheckBox_OnMouseDown(frame)
farmbuyer@112: 	local self = frame.obj
farmbuyer@112: 	if not self.disabled then
farmbuyer@112: 		if self.image:GetTexture() then
farmbuyer@112: 			self.text:SetPoint("LEFT", self.image,"RIGHT", 2, -1)
farmbuyer@112: 		else
farmbuyer@112: 			self.text:SetPoint("LEFT", self.checkbg, "RIGHT", 1, -1)
farmbuyer@112: 		end
farmbuyer@112: 	end
farmbuyer@112: 	AceGUI:ClearFocus()
farmbuyer@112: end
farmbuyer@112: 
farmbuyer@112: local function CheckBox_OnMouseUp(frame)
farmbuyer@112: 	local self = frame.obj
farmbuyer@112: 	if not self.disabled then
farmbuyer@112: 		self:ToggleChecked()
farmbuyer@112: 
farmbuyer@112: 		if self.checked then
farmbuyer@112: 			PlaySound("igMainMenuOptionCheckBoxOn")
farmbuyer@112: 		else -- for both nil and false (tristate)
farmbuyer@112: 			PlaySound("igMainMenuOptionCheckBoxOff")
farmbuyer@112: 		end
farmbuyer@112: 
farmbuyer@112: 		self:Fire("OnValueChanged", self.checked)
farmbuyer@112: 		AlignImage(self)
farmbuyer@112: 	end
farmbuyer@112: end
farmbuyer@112: 
farmbuyer@112: --[[-----------------------------------------------------------------------------
farmbuyer@112: Methods
farmbuyer@112: -------------------------------------------------------------------------------]]
farmbuyer@112: local methods = {
farmbuyer@112: 	["OnAcquire"] = function(self)
farmbuyer@112: 		self:SetType()
farmbuyer@112: 		self:SetValue(false)
farmbuyer@112: 		self:SetTriState(nil)
farmbuyer@112: 		-- height is calculated from the width and required space for the description
farmbuyer@112: 		self:SetWidth(200)
farmbuyer@112: 		self:SetImage()
farmbuyer@112: 		self:SetDisabled(nil)
farmbuyer@112: 		self:SetDescription(nil)
farmbuyer@112: 	end,
farmbuyer@112: 
farmbuyer@112: 	-- ["OnRelease"] = nil,
farmbuyer@112: 
farmbuyer@112: 	["OnWidthSet"] = function(self, width)
farmbuyer@112: 		if self.desc then
farmbuyer@112: 			self.desc:SetWidth(width - 30)
farmbuyer@112: 			if self.desc:GetText() and self.desc:GetText() ~= "" then
farmbuyer@112: 				self:SetHeight(28 + self.desc:GetHeight())
farmbuyer@112: 			end
farmbuyer@112: 		end
farmbuyer@112: 	end,
farmbuyer@112: 
farmbuyer@112: 	["SetDisabled"] = function(self, disabled)
farmbuyer@112: 		self.disabled = disabled
farmbuyer@112: 		if disabled then
farmbuyer@112: 			self.frame:Disable()
farmbuyer@112: 			self.text:SetTextColor(0.5, 0.5, 0.5)
farmbuyer@112: 			SetDesaturation(self.check, true)
farmbuyer@112: 			if self.desc then
farmbuyer@112: 				self.desc:SetTextColor(0.5, 0.5, 0.5)
farmbuyer@112: 			end
farmbuyer@112: 		else
farmbuyer@112: 			self.frame:Enable()
farmbuyer@112: 			self.text:SetTextColor(1, 1, 1)
farmbuyer@112: 			if self.tristate and self.checked == nil then
farmbuyer@112: 				SetDesaturation(self.check, true)
farmbuyer@112: 			else
farmbuyer@112: 				SetDesaturation(self.check, false)
farmbuyer@112: 			end
farmbuyer@112: 			if self.desc then
farmbuyer@112: 				self.desc:SetTextColor(1, 1, 1)
farmbuyer@112: 			end
farmbuyer@112: 		end
farmbuyer@112: 	end,
farmbuyer@112: 
farmbuyer@112: 	["SetValue"] = function(self,value)
farmbuyer@112: 		local check = self.check
farmbuyer@112: 		self.checked = value
farmbuyer@112: 		if value then
farmbuyer@112: 			SetDesaturation(self.check, false)
farmbuyer@112: 			self.check:Show()
farmbuyer@112: 		else
farmbuyer@112: 			--Nil is the unknown tristate value
farmbuyer@112: 			if self.tristate and value == nil then
farmbuyer@112: 				SetDesaturation(self.check, true)
farmbuyer@112: 				self.check:Show()
farmbuyer@112: 			else
farmbuyer@112: 				SetDesaturation(self.check, false)
farmbuyer@112: 				self.check:Hide()
farmbuyer@112: 			end
farmbuyer@112: 		end
farmbuyer@112: 		self:SetDisabled(self.disabled)
farmbuyer@112: 	end,
farmbuyer@112: 
farmbuyer@112: 	["GetValue"] = function(self)
farmbuyer@112: 		return self.checked
farmbuyer@112: 	end,
farmbuyer@112: 
farmbuyer@112: 	["SetTriState"] = function(self, enabled)
farmbuyer@112: 		self.tristate = enabled
farmbuyer@112: 		self:SetValue(self:GetValue())
farmbuyer@112: 	end,
farmbuyer@112: 
farmbuyer@112: 	["SetType"] = function(self, type)
farmbuyer@112: 		local checkbg = self.checkbg
farmbuyer@112: 		local check = self.check
farmbuyer@112: 		local highlight = self.highlight
farmbuyer@112: 
farmbuyer@112: 		local size
farmbuyer@112: 		if type == "radio" then
farmbuyer@112: 			size = 16
farmbuyer@112: 			checkbg:SetTexture("Interface\\Buttons\\UI-RadioButton")
farmbuyer@112: 			checkbg:SetTexCoord(0, 0.25, 0, 1)
farmbuyer@112: 			check:SetTexture("Interface\\Buttons\\UI-RadioButton")
farmbuyer@112: 			check:SetTexCoord(0.25, 0.5, 0, 1)
farmbuyer@112: 			check:SetBlendMode("ADD")
farmbuyer@112: 			highlight:SetTexture("Interface\\Buttons\\UI-RadioButton")
farmbuyer@112: 			highlight:SetTexCoord(0.5, 0.75, 0, 1)
farmbuyer@112: 		else
farmbuyer@112: 			size = 24
farmbuyer@112: 			checkbg:SetTexture("Interface\\Buttons\\UI-CheckBox-Up")
farmbuyer@112: 			checkbg:SetTexCoord(0, 1, 0, 1)
farmbuyer@112: 			check:SetTexture("Interface\\Buttons\\UI-CheckBox-Check")
farmbuyer@112: 			check:SetTexCoord(0, 1, 0, 1)
farmbuyer@112: 			check:SetBlendMode("BLEND")
farmbuyer@112: 			highlight:SetTexture("Interface\\Buttons\\UI-CheckBox-Highlight")
farmbuyer@112: 			highlight:SetTexCoord(0, 1, 0, 1)
farmbuyer@112: 		end
farmbuyer@112: 		checkbg:SetHeight(size)
farmbuyer@112: 		checkbg:SetWidth(size)
farmbuyer@112: 	end,
farmbuyer@112: 
farmbuyer@112: 	["ToggleChecked"] = function(self)
farmbuyer@112: 		local value = self:GetValue()
farmbuyer@112: 		if self.tristate then
farmbuyer@112: 			--cycle in true, nil, false order
farmbuyer@112: 			if value then
farmbuyer@112: 				self:SetValue(nil)
farmbuyer@112: 			elseif value == nil then
farmbuyer@112: 				self:SetValue(false)
farmbuyer@112: 			else
farmbuyer@112: 				self:SetValue(true)
farmbuyer@112: 			end
farmbuyer@112: 		else
farmbuyer@112: 			self:SetValue(not self:GetValue())
farmbuyer@112: 		end
farmbuyer@112: 	end,
farmbuyer@112: 
farmbuyer@112: 	["SetLabel"] = function(self, label)
farmbuyer@112: 		self.text:SetText(label)
farmbuyer@112: 	end,
farmbuyer@112: 
farmbuyer@112: 	["SetDescription"] = function(self, desc)
farmbuyer@112: 		if desc then
farmbuyer@112: 			if not self.desc then
farmbuyer@112: 				local desc = self.frame:CreateFontString(nil, "OVERLAY", "GameFontHighlightSmall")
farmbuyer@112: 				desc:ClearAllPoints()
farmbuyer@112: 				desc:SetPoint("TOPLEFT", self.checkbg, "TOPRIGHT", 5, -21)
farmbuyer@112: 				desc:SetWidth(self.frame.width - 30)
farmbuyer@112: 				desc:SetJustifyH("LEFT")
farmbuyer@112: 				desc:SetJustifyV("TOP")
farmbuyer@112: 				self.desc = desc
farmbuyer@112: 			end
farmbuyer@112: 			self.desc:Show()
farmbuyer@112: 			--self.text:SetFontObject(GameFontNormal)
farmbuyer@112: 			self.desc:SetText(desc)
farmbuyer@112: 			self:SetHeight(28 + self.desc:GetHeight())
farmbuyer@112: 		else
farmbuyer@112: 			if self.desc then
farmbuyer@112: 				self.desc:SetText("")
farmbuyer@112: 				self.desc:Hide()
farmbuyer@112: 			end
farmbuyer@112: 			--self.text:SetFontObject(GameFontHighlight)
farmbuyer@112: 			self:SetHeight(24)
farmbuyer@112: 		end
farmbuyer@112: 	end,
farmbuyer@112: 	
farmbuyer@112: 	["SetImage"] = function(self, path, ...)
farmbuyer@112: 		local image = self.image
farmbuyer@112: 		image:SetTexture(path)
farmbuyer@112: 		
farmbuyer@112: 		if image:GetTexture() then
farmbuyer@112: 			local n = select("#", ...)
farmbuyer@112: 			if n == 4 or n == 8 then
farmbuyer@112: 				image:SetTexCoord(...)
farmbuyer@112: 			else
farmbuyer@112: 				image:SetTexCoord(0, 1, 0, 1)
farmbuyer@112: 			end
farmbuyer@112: 		end
farmbuyer@112: 		AlignImage(self)
farmbuyer@112: 	end
farmbuyer@112: }
farmbuyer@112: 
farmbuyer@112: --[[-----------------------------------------------------------------------------
farmbuyer@112: Constructor
farmbuyer@112: -------------------------------------------------------------------------------]]
farmbuyer@112: local function Constructor()
farmbuyer@112: 	local frame = CreateFrame("Button", nil, UIParent)
farmbuyer@112: 	frame:Hide()
farmbuyer@112: 
farmbuyer@112: 	frame:EnableMouse(true)
farmbuyer@112: 	frame:SetScript("OnEnter", Control_OnEnter)
farmbuyer@112: 	frame:SetScript("OnLeave", Control_OnLeave)
farmbuyer@112: 	frame:SetScript("OnMouseDown", CheckBox_OnMouseDown)
farmbuyer@112: 	frame:SetScript("OnMouseUp", CheckBox_OnMouseUp)
farmbuyer@112: 
farmbuyer@112: 	local checkbg = frame:CreateTexture(nil, "ARTWORK")
farmbuyer@112: 	checkbg:SetWidth(24)
farmbuyer@112: 	checkbg:SetHeight(24)
farmbuyer@112: 	checkbg:SetPoint("TOPLEFT")
farmbuyer@112: 	checkbg:SetTexture("Interface\\Buttons\\UI-CheckBox-Up")
farmbuyer@112: 
farmbuyer@112: 	local check = frame:CreateTexture(nil, "OVERLAY")
farmbuyer@112: 	check:SetAllPoints(checkbg)
farmbuyer@112: 	check:SetTexture("Interface\\Buttons\\UI-CheckBox-Check")
farmbuyer@112: 
farmbuyer@112: 	local text = frame:CreateFontString(nil, "OVERLAY", "GameFontHighlight")
farmbuyer@112: 	text:SetJustifyH("LEFT")
farmbuyer@112: 	text:SetHeight(18)
farmbuyer@112: 	text:SetPoint("LEFT", checkbg, "RIGHT")
farmbuyer@112: 	text:SetPoint("RIGHT")
farmbuyer@112: 
farmbuyer@112: 	local highlight = frame:CreateTexture(nil, "HIGHLIGHT")
farmbuyer@112: 	highlight:SetTexture("Interface\\Buttons\\UI-CheckBox-Highlight")
farmbuyer@112: 	highlight:SetBlendMode("ADD")
farmbuyer@112: 	highlight:SetAllPoints(checkbg)
farmbuyer@112: 
farmbuyer@112: 	local image = frame:CreateTexture(nil, "OVERLAY")
farmbuyer@112: 	image:SetHeight(16)
farmbuyer@112: 	image:SetWidth(16)
farmbuyer@112: 	image:SetPoint("LEFT", checkbg, "RIGHT", 1, 0)
farmbuyer@112: 
farmbuyer@112: 	local widget = {
farmbuyer@112: 		checkbg   = checkbg,
farmbuyer@112: 		check     = check,
farmbuyer@112: 		text      = text,
farmbuyer@112: 		highlight = highlight,
farmbuyer@112: 		image     = image,
farmbuyer@112: 		frame     = frame,
farmbuyer@112: 		type      = Type
farmbuyer@112: 	}
farmbuyer@112: 	for method, func in pairs(methods) do
farmbuyer@112: 		widget[method] = func
farmbuyer@112: 	end
farmbuyer@112: 
farmbuyer@112: 	return AceGUI:RegisterAsWidget(widget)
farmbuyer@112: end
farmbuyer@112: 
farmbuyer@112: AceGUI:RegisterWidgetType(Type, Constructor, Version)