// Copyright (C) 2002-2012 Nikolaus Gebhardt
// This file is part of the "Irrlicht Engine".
// For conditions of distribution and use, see copyright notice in irrlicht.h

#ifndef __C_GUI_BUTTON_H_INCLUDED__
#define __C_GUI_BUTTON_H_INCLUDED__

#include "IrrCompileConfig.h"
#ifdef _IRR_COMPILE_WITH_GUI_

#include "IGUIButton.h"
#include "IGUISpriteBank.h"
#include "ITexture.h"
#include "SColor.h"

namespace irr
{
namespace gui
{

	class CGUIButton : public IGUIButton
	{
	public:

		//! constructor
		CGUIButton(IGUIEnvironment* environment, IGUIElement* parent,
			s32 id, core::rect<s32> rectangle, bool noclip=false);

		//! destructor
		virtual ~CGUIButton();

		//! called if an event happened.
		virtual bool OnEvent(const SEvent& event) _IRR_OVERRIDE_;

		//! draws the element and its children
		virtual void draw() _IRR_OVERRIDE_;

		//! sets another skin independent font. if this is set to zero, the button uses the font of the skin.
		virtual void setOverrideFont(IGUIFont* font=0) _IRR_OVERRIDE_;

		//! Gets the override font (if any)
		virtual IGUIFont* getOverrideFont() const _IRR_OVERRIDE_;

		//! Get the font which is used right now for drawing
		virtual IGUIFont* getActiveFont() const _IRR_OVERRIDE_;

		//! Sets another color for the button text.
		virtual void setOverrideColor(video::SColor color)  _IRR_OVERRIDE_;

		//! Gets the override color
		virtual video::SColor getOverrideColor(void) const  _IRR_OVERRIDE_;

		//! Gets the currently used text color
		virtual video::SColor getActiveColor() const _IRR_OVERRIDE_;

		//! Sets if the button text should use the override color or the color in the gui skin.
		virtual void enableOverrideColor(bool enable)  _IRR_OVERRIDE_;

		//! Checks if an override color is enabled
		virtual bool isOverrideColorEnabled(void) const  _IRR_OVERRIDE_;

		//! Sets an image which should be displayed on the button when it is in the given state.
		virtual void setImage(EGUI_BUTTON_IMAGE_STATE state, video::ITexture* image=0, const core::rect<s32>& sourceRect=core::rect<s32>(0,0,0,0))  _IRR_OVERRIDE_;

		//! Sets an image which should be displayed on the button when it is in normal state.
		virtual void setImage(video::ITexture* image=0) _IRR_OVERRIDE_
		{
			setImage(EGBIS_IMAGE_UP, image);
		}

		//! Sets an image which should be displayed on the button when it is in normal state.
		virtual void setImage(video::ITexture* image, const core::rect<s32>& pos) _IRR_OVERRIDE_
		{
			setImage(EGBIS_IMAGE_UP, image, pos);
		}

		//! Sets an image which should be displayed on the button when it is in pressed state.
		virtual void setPressedImage(video::ITexture* image=0) _IRR_OVERRIDE_
		{
			setImage(EGBIS_IMAGE_DOWN, image);
		}

		//! Sets an image which should be displayed on the button when it is in pressed state.
		virtual void setPressedImage(video::ITexture* image, const core::rect<s32>& pos) _IRR_OVERRIDE_
		{
			setImage(EGBIS_IMAGE_DOWN, image, pos);
		}

		//! Sets the sprite bank used by the button
		virtual void setSpriteBank(IGUISpriteBank* bank=0) _IRR_OVERRIDE_;

		//! Sets the animated sprite for a specific button state
		/** \param index: Number of the sprite within the sprite bank, use -1 for no sprite
		\param state: State of the button to set the sprite for
		\param index: The sprite number from the current sprite bank
		\param color: The color of the sprite
		*/
		virtual void setSprite(EGUI_BUTTON_STATE state, s32 index,
				video::SColor color=video::SColor(255,255,255,255),
				bool loop=false, bool scale=false) _IRR_OVERRIDE_;

		//! Get the sprite-index for the given state or -1 when no sprite is set
		virtual s32 getSpriteIndex(EGUI_BUTTON_STATE state) const _IRR_OVERRIDE_;

		//! Get the sprite color for the given state. Color is only used when a sprite is set.
		virtual video::SColor getSpriteColor(EGUI_BUTTON_STATE state) const _IRR_OVERRIDE_;

		//! Returns if the sprite in the given state does loop
		virtual bool getSpriteLoop(EGUI_BUTTON_STATE state) const _IRR_OVERRIDE_;

		//! Returns if the sprite in the given state is scaled
		virtual bool getSpriteScale(EGUI_BUTTON_STATE state) const _IRR_OVERRIDE_;

		//! Sets if the button should behave like a push button. Which means it
		//! can be in two states: Normal or Pressed. With a click on the button,
		//! the user can change the state of the button.
		virtual void setIsPushButton(bool isPushButton=true) _IRR_OVERRIDE_;

		//! Checks whether the button is a push button
		virtual bool isPushButton() const _IRR_OVERRIDE_;

		//! Sets the pressed state of the button if this is a pushbutton
		virtual void setPressed(bool pressed=true) _IRR_OVERRIDE_;

		//! Returns if the button is currently pressed
		virtual bool isPressed() const _IRR_OVERRIDE_;

		//! Sets if the button should use the skin to draw its border
		virtual void setDrawBorder(bool border=true) _IRR_OVERRIDE_;

		//! Checks if the button face and border are being drawn
		virtual bool isDrawingBorder() const _IRR_OVERRIDE_;

		//! Sets if the alpha channel should be used for drawing images on the button (default is false)
		virtual void setUseAlphaChannel(bool useAlphaChannel=true) _IRR_OVERRIDE_;

		//! Checks if the alpha channel should be used for drawing images on the button
		virtual bool isAlphaChannelUsed() const _IRR_OVERRIDE_;

		//! Sets if the button should scale the button images to fit
		virtual void setScaleImage(bool scaleImage=true) _IRR_OVERRIDE_;

		//! Checks whether the button scales the used images
		virtual bool isScalingImage() const _IRR_OVERRIDE_;

		//! Get if the shift key was pressed in last EGET_BUTTON_CLICKED event
		virtual bool getClickShiftState() const	_IRR_OVERRIDE_
		{
			return ClickShiftState;
		}

		//! Get if the control key was pressed in last EGET_BUTTON_CLICKED event
		virtual bool getClickControlState() const _IRR_OVERRIDE_
		{
			return ClickControlState;
		}

	protected:
		void drawSprite(EGUI_BUTTON_STATE state, u32 startTime, const core::position2di& center);
		EGUI_BUTTON_IMAGE_STATE getImageState(bool pressed) const;

	private:

		struct ButtonSprite
		{
			ButtonSprite() : Index(-1), Loop(false), Scale(false)
			{
			}

			bool operator==(const ButtonSprite& other) const
			{
				return Index == other.Index && Color == other.Color && Loop == other.Loop && Scale == other.Scale;
			}

			s32 Index;
			video::SColor Color;
			bool Loop;
			bool Scale;
		};

		ButtonSprite ButtonSprites[EGBS_COUNT];
		IGUISpriteBank* SpriteBank;

		struct ButtonImage
		{
			ButtonImage() : Texture(0), SourceRect(core::rect<s32>(0,0,0,0))
			{
			}

			ButtonImage(const ButtonImage& other) : Texture(0), SourceRect(core::rect<s32>(0,0,0,0))
			{
				*this = other;
			}

			~ButtonImage()
			{
				if ( Texture )
					Texture->drop();
			}

			ButtonImage& operator=(const ButtonImage& other)
			{
				if ( this == &other )
					return *this;

				if (other.Texture)
					other.Texture->grab();
				if ( Texture )
					Texture->drop();
				Texture = other.Texture;
				SourceRect = other.SourceRect;
				return *this;
			}

			bool operator==(const ButtonImage& other) const
			{
				return Texture == other.Texture && SourceRect == other.SourceRect;
			}


			video::ITexture* Texture;
			core::rect<s32> SourceRect;
		};

		ButtonImage ButtonImages[EGBIS_COUNT];

		IGUIFont* OverrideFont;

		bool OverrideColorEnabled;
		video::SColor OverrideColor;

		u32 ClickTime, HoverTime, FocusTime;

		bool ClickShiftState;
		bool ClickControlState;

		bool IsPushButton;
		bool Pressed;
		bool UseAlphaChannel;
		bool DrawBorder;
		bool ScaleImage;
	};

} // end namespace gui
} // end namespace irr

#endif // _IRR_COMPILE_WITH_GUI_

#endif // __C_GUI_BUTTON_H_INCLUDED__