DayZOfficialSourceCode/Scripts/3_game/gui/effects/radialmenu.c

enum RadialMenuControlType
{
	MOUSE,
	CONTROLLER	
}

class RadialMenu : ScriptedWidgetEventHandler
{
	protected Widget					m_Parent;
	protected Widget 					m_ItemCardsContainer;
	protected Widget					m_RadialSelector;
	protected ImageWidget				m_RadialSelectorImage;
	protected ImageWidget 				m_RadialSelectorPointerImage;
	protected int 						m_RadialSelectorOriginalColor;
	protected int 						m_RadialSelectorDisabledColor;
	protected Widget 					m_SelectedObject;
	protected ref map<Widget, float> 	m_RadialItemCards;
	
	protected float 					m_AngleRadOffset;
	protected ref Timer 				m_UpdateTimer;
	
	//widget
	static const string 				RADIAL_SELECTOR	 			= "RadialSelector";
	static const string 				RADIAL_SELECTOR_IMAGE		= "SelectorImage";
	static const string 				RADIAL_SELECTOR_POINTER		= "SelectorPointer";
	static const string 				RADIAL_DELIMITER_CONTAINER 	= "RadialDelimiterContainer";
	static const string 				RADIAL_ITEM_CARD_CONTAINER 	= "RadialItemCardContainer";

	//controls
	protected RadialMenuControlType 	m_ControlType;
	private UAIDWrapper 				m_SelectInputWrapper;
	private UAIDWrapper 				m_BackInputWrapper;
	protected float 					m_ControllerAngle;
	protected float 					m_ControllerTilt;

	//controller
	protected int 						m_ControllerTimout;
	protected bool						m_IsControllerTimoutEnabled 		= true;			//enables/disables controller deselect timeout reset
	protected const float				CONTROLLER_DESELECT_TIMEOUT 		= 1000;			//timeout [ms] after which selection is automatically deselect when controller is not active
	protected const float 				CONTROLLER_TILT_TRESHOLD_SELECT		= 0.8;			//tilt value (0.0-1.0) for controller sticks after which the selection will be selected
	protected const float 				CONTROLLER_TILT_TRESHOLD_EXECUTE	= 1.0;			//tilt value (0.0-1.0) for controller sticks after which the selection will be executed 
	
	//mouse
	protected bool 						m_WidgetInitialized;
	protected const float 				MOUSE_SAFE_ZONE_RADIUS = 120;		//Radius [px] of safe zone where every previous selection is deselected
	
	//References
	protected float 					m_RadiusOffset;					//Radius [% of the main container size]
	protected float 					m_ExecuteDistanceOffset;		//Distance offset [% of the main container size] after which the selection will be automatically executed
	protected float						m_OffsetFromTop;				//first item in the menu won't be directly on top but offset by a rad angle value (clock-wise)
	protected float						m_ItemCardRadiusOffset;			//Radius [% of the main container size] for item cards
	protected string					m_DelimiterLayout;				//layout file name with path
	
	ref UIScriptedMenu					m_RegisteredClass;	
	ref static RadialMenu				m_Instance;
	
	/*
	
			RADIAL MENU EVENTS
	
	Mouse:
		OnMouseSelect
		OnMouseDeselect
		OnMouseExecute - unused, press events used instead
		OnMousePressLeft
		OnMousePressRight
	
	Controller:
		OnControllerSelect
		OnControllerDeselect
		OnControllerExecute - unused, press events used instead
		OnControllerPressSelect
		OnControllerPressBack
	
	Common:
		OnControlsChanged		- controls has been changed (mouse<->controller)
	
	*/
	
	
	//============================================
	// RadialMenu
	//============================================
	void RadialMenu()
	{
		m_Instance = this;
		
		//set default control type
#ifdef PLATFORM_CONSOLE
		Input inp = GetGame().GetInput();
		if (inp && inp.IsEnabledMouseAndKeyboardEvenOnServer())
		{
			m_ControlType = RadialMenuControlType.MOUSE;
		}
		else
		{
			m_ControlType = RadialMenuControlType.CONTROLLER;
		}
#endif
#ifdef PLATFORM_WINDOWS
		m_ControlType = RadialMenuControlType.MOUSE;
#endif
		
		m_SelectInputWrapper = GetUApi().GetInputByID(UAUISelect).GetPersistentWrapper();
		m_BackInputWrapper= GetUApi().GetInputByID(UAUIBack).GetPersistentWrapper();
		
		//radial cards
		m_RadialItemCards = new map<Widget, float>;
		m_UpdateTimer = new Timer();
		m_UpdateTimer.Run(0.01, this, "Update", NULL, true);
	}

	void ~RadialMenu()
	{
	}
	
	static RadialMenu GetInstance()
	{
		return m_Instance;
	}
	
	//Set handler
	void OnWidgetScriptInit(Widget w)
	{
		m_ItemCardsContainer = w.FindAnyWidget(RADIAL_ITEM_CARD_CONTAINER);
		m_RadialSelector = w.FindAnyWidget(RADIAL_SELECTOR);
		m_RadialSelectorImage = ImageWidget.Cast(m_RadialSelector.FindAnyWidget(RADIAL_SELECTOR_IMAGE));
		m_RadialSelectorPointerImage = ImageWidget.Cast(m_RadialSelector.FindAnyWidget(RADIAL_SELECTOR_POINTER));
		
		m_RadialSelectorOriginalColor = m_RadialSelectorImage.GetColor();
		m_RadialSelectorDisabledColor = ARGB(255,150,150,150);
		
		//parent
		m_Parent = w;
		m_Parent.SetHandler(this);
	}

	//controls
	void SetControlType(RadialMenuControlType type)
	{
		if (m_ControlType != type)
		{
			m_ControlType = type;
			GetGame().GameScript.CallFunction(m_RegisteredClass, "OnControlsChanged", NULL, type);
		}
	}
	
	bool IsUsingMouse()
	{
		if (m_ControlType == RadialMenuControlType.MOUSE)
		{
			return true;
		}
		
		return false;
	}

	bool IsUsingController()
	{
		if (m_ControlType == RadialMenuControlType.CONTROLLER)
		{
			return true;
		}
		
		return false;
	}
	
	void SetWidgetInitialized(bool state)
	{
		m_WidgetInitialized = state;
	}
	
	bool IsWidgetInitialized()
	{
		return m_WidgetInitialized;
	}
			
	//============================================
	// Setup
	//============================================	
	void RegisterClass(UIScriptedMenu class_name)
	{
		m_RegisteredClass = class_name;
		if (m_UpdateTimer && !m_UpdateTimer.IsRunning())
			m_UpdateTimer.Run(0.01, this, "Update", NULL, true);
	}	

	//Set radial menu parameters
	//Radius offset [% of the main container size]
	void SetRadiusOffset(float radius_offset)
	{
		m_RadiusOffset = radius_offset;
	}
	
	//Distance offset [% of the main container size] after which the selection will be automatically executed
	void SetExecuteDistOffset(float execute_dist_offset)
	{
		m_ExecuteDistanceOffset	= execute_dist_offset;
	}
	
	//First item in the menu won't be directly on top but offset by a rad angle value (clock-wise)
	void SetOffsetFromTop(float offset_from_top)
	{
		m_OffsetFromTop	= offset_from_top;
	}
	
	//Radius [% of the main container size] for item cards	
	void SetItemCardRadiusOffset(float item_card_radius_offset)
	{
		m_ItemCardRadiusOffset = item_card_radius_offset;
	}
	
	//Enable/Disable controller timeout
	void ActivateControllerTimeout(bool state)
	{
		m_IsControllerTimoutEnabled = state;
	}
	
	void SetWidgetProperties(string delimiter_layout)
	{
		m_DelimiterLayout = delimiter_layout;
	}
	
	//============================================
	// Visual
	//============================================	
	//hide_selector => shows/hides radial selector when refreshing radial menu
	void Refresh(bool hide_selector = true)
	{
		int item_cards_count = GetItemCardsCount();
		if (item_cards_count > 0)
			m_AngleRadOffset = 2 * Math.PI / item_cards_count;		
		float angle_rad = -Math.PI / 2;
		
		//--PARAM top offset--
		if (m_OffsetFromTop != 0)
		{
			angle_rad = angle_rad + m_OffsetFromTop;
		}
		//--------------------
		
		//delete all delimiters
		Widget delimiters_panel = m_Parent.FindAnyWidget(RADIAL_DELIMITER_CONTAINER);
		if (delimiters_panel)
		{
			Widget del_child = delimiters_panel.GetChildren();
			while (del_child)
			{
				Widget child_to_destroy1 = del_child;
				del_child = del_child.GetSibling();
				
				delete child_to_destroy1;
			}
		}
		
		//Position item cards, crate radial delimiters
		Widget item_cards_panel = m_Parent.FindAnyWidget(RADIAL_ITEM_CARD_CONTAINER);
		Widget item_card = item_cards_panel.GetChildren();
		
		//get radius
		float original_r = GetRadius();
		float item_cards_r = original_r;
		
		//--PARAM top offset--....
		if (m_ItemCardRadiusOffset != 0)
		{
			item_cards_r = item_cards_r * m_ItemCardRadiusOffset;
			if (item_cards_r < 0) item_cards_r = 0;				//min radius is 0
		}
		
		m_RadialItemCards.Clear();
		for (int i = 0; i < item_cards_count; ++i)
		{
			//position item cards
			if (item_card)
			{
				//creates circle from simple widget items
				float pos_x = item_cards_r * Math.Cos(angle_rad);
				float pos_y = item_cards_r * Math.Sin(angle_rad);
				
				pos_x = pos_x / original_r;
				pos_y =	pos_y / original_r;
				
				item_card.SetPos(pos_x, pos_y);
				
				//store item card
				m_RadialItemCards.Insert(item_card, angle_rad);
				
				//get next child
				item_card = item_card.GetSibling();
			}
			//-------------------------
			
			//create delimiter
			if (item_cards_count > 1 && delimiters_panel && m_DelimiterLayout)
			{
				Widget delimiter_widget = GetGame().GetWorkspace().CreateWidgets(m_DelimiterLayout, delimiters_panel);
				float delim_angle_rad = angle_rad + (m_AngleRadOffset / 2);
				delimiter_widget.SetPos(0, 0);
				delimiter_widget.SetRotation(0, 0, GetAngleInDegrees(delim_angle_rad) + 90);
			}
						
			//calculate next angle
			angle_rad += m_AngleRadOffset;
		}
		
		//hide selector on refresh
		if (hide_selector)
		{
			HideRadialSelector();
		}
	}

	//Radial selector
	protected void ShowRadialSelector(Widget selected_item)
	{
		if (m_RadialSelector && selected_item)
		{
			int item_count = m_RadialItemCards.Count();
			if (item_count > 1)
			{
				int angle_deg = GetAngleInDegrees(m_RadialItemCards.Get(selected_item));
				m_RadialSelector.SetRotation(0, 0, angle_deg + 90);			//rotate widget according to its desired rotation
				
				//set radial selector size
				float progress = (1 / item_count) * 2;
				m_RadialSelectorImage.SetMaskProgress(progress);
				
				m_RadialSelector.Show(true);
				
				bool grey_selector = selected_item.GetFlags() & WidgetFlags.DISABLED;
				if (!grey_selector)
				{
					m_RadialSelectorImage.SetColor(m_RadialSelectorDisabledColor);
					m_RadialSelectorPointerImage.SetColor(m_RadialSelectorDisabledColor);
				}
				else
				{
					m_RadialSelectorImage.SetColor(m_RadialSelectorOriginalColor);
					m_RadialSelectorPointerImage.SetColor(m_RadialSelectorOriginalColor);
				}
			}
		}
	}
	
	protected void HideRadialSelector()
	{
		if (m_RadialSelector)
		{
			m_RadialSelector.Show(false);
		}
	}
	
	//============================================
	// Widget size calculations
	//============================================	
	protected int GetItemCardsCount()
	{
		Widget child = m_ItemCardsContainer.GetChildren();
		int count = 0;
		
		while (child)
		{
			++count;
			
			child = child.GetSibling();
		}
		
		return count;
	}
	
	protected float GetRadius()
	{
		float radius = Math.AbsFloat(GetParentMinSize() * 0.5);
		
		//PARAM --radius--
		if (m_RadiusOffset > 0)
		{
			return radius * m_RadiusOffset;
		}
		//----------------
		
		return radius;
	}
	
	protected void GetParentCenter(out float center_x, out float center_y)
	{
		if (m_Parent)
		{
			float wx;
			float wy;
			m_Parent.GetScreenPos(wx, wy);
			
			float ww;
			float wh;
			m_Parent.GetScreenSize(ww, wh);
			
			center_x = wx + ww / 2;	//center
			center_y = wy + wh / 2;
		}
	}
	
	protected float GetParentMinSize()
	{
		if (m_Parent)
		{
			float size_x;
			float size_y;
			m_Parent.GetScreenSize(size_x, size_y);
			
			return Math.Min(size_x, size_y);
		}
		
		return 0;
	}

	//============================================
	// Angle calculations
	//============================================	
	//get object by angle (degrees)
	protected Widget GetObjectByDegAngle(float deg_angle)
	{
		for (int i = 0; i < m_RadialItemCards.Count(); ++i)
		{
			Widget w = m_RadialItemCards.GetKey(i);
			float w_angle = GetAngleInDegrees(m_RadialItemCards.Get(w));
			float offset = GetAngleInDegrees(m_AngleRadOffset) / 2;
			float min_angle = w_angle - offset;
			float max_angle = w_angle + offset;
			
			if (min_angle < 0) min_angle += 360;	//clamp 0-360
			if (max_angle > 360) max_angle -= 360;
			
			if (min_angle > max_angle)		//angle radius is in the cycling point 360->
			{
				if (min_angle <= deg_angle)	//is cursor position also before this point
				{
					if (deg_angle > max_angle)
					{
						return w;
					}
				}
				else							//is cursor position after this point
				{
					if (deg_angle < max_angle)
					{
						return w;
					}
				}
			}
			else
			{
				if (deg_angle >= min_angle && deg_angle < max_angle)	//min, max angles are within 0-360 radius
				{
					return w;
				}				
			}
		}
		
		return NULL;
	}
	
	//returns GUI compatible mouse-to-parent angle
	protected float GetMousePointerAngle()
	{
		int mouse_x;
		int mouse_y;
		GetMousePos(mouse_x, mouse_y);
		
		float center_x;
		float center_y;
		GetParentCenter(center_x, center_y);

		float tan_x = mouse_x - center_x;
		float tan_y = mouse_y - center_y;
		float angle = Math.Atan2(tan_y, tan_x);
		
		return angle;
	}
	
	//returns distance from parent center
	protected float GetMouseDistance()
	{
		int mouse_x;
		int mouse_y;
		GetMousePos(mouse_x, mouse_y);
		
		float center_x;
		float center_y;
		GetParentCenter(center_x, center_y);

		float distance = vector.Distance(Vector(mouse_x, mouse_y, 0), Vector(center_x, center_y, 0));
		
		return distance;
	}
	
	//return angle 0-360 deg
	protected float GetAngleInDegrees(float rad_angle)
	{
		float rad_deg = rad_angle * Math.RAD2DEG;
		
		int angle_mp = rad_deg / 360;
		
		if (rad_deg < 0)
		{
			rad_deg = rad_deg - (360 * angle_mp);
			rad_deg += 360;
		}
		
		return rad_deg;
	}

	//============================================
	// Update
	//============================================
	//mouse
	int last_time = -1;
	protected void Update()
	{
		if (this && !m_RegisteredClass)
		{
			m_UpdateTimer.Stop();
			return;
		}
		
		//get delta time
		if (last_time < 0)
		{
			last_time = GetGame().GetTime();
		}
		int delta_time = GetGame().GetTime() - last_time;
		last_time = GetGame().GetTime();
		
		//controls
		if (this && m_RegisteredClass && m_RegisteredClass.IsVisible())
		{
			//mouse controls
			if (IsUsingMouse() && m_WidgetInitialized)
			{
				float mouse_angle = GetMousePointerAngle();
				float mouse_distance = GetMouseDistance();
				
				//--PARAM --safe zone radius--
				if (mouse_distance <= MOUSE_SAFE_ZONE_RADIUS)
				{
					//Deselect
					GetGame().GameScript.CallFunction(m_RegisteredClass, "OnMouseDeselect", NULL, m_SelectedObject);
					m_SelectedObject = NULL;
					//hide selector
					HideRadialSelector();
				}
				else
				{
					//Deselect
					GetGame().GameScript.CallFunction(m_RegisteredClass, "OnMouseDeselect", NULL, m_SelectedObject);
					
					//Select
					m_SelectedObject = GetObjectByDegAngle(GetAngleInDegrees(mouse_angle));
					GetGame().GameScript.CallFunction(m_RegisteredClass, "OnMouseSelect", NULL, m_SelectedObject);				
					//show selector
					ShowRadialSelector(m_SelectedObject);
				}
			}
			//controller controls
			else if (IsUsingController())
			{
				UpdataControllerInput();
				
				//Controller tilt
				if (m_ControllerAngle > -1 && m_ControllerTilt > -1)
				{
					//Right analogue stick
					Widget w_selected = GetObjectByDegAngle(m_ControllerAngle);
					//Select
					if (w_selected)
					{
						if (w_selected != m_SelectedObject)
						{
							if (m_ControllerTilt >= CONTROLLER_TILT_TRESHOLD_SELECT)
							{
								//Deselect
								GetGame().GameScript.CallFunction(m_RegisteredClass, "OnControllerDeselect", NULL, m_SelectedObject);
							
								//Select new object
								m_SelectedObject = w_selected;
								GetGame().GameScript.CallFunction(m_RegisteredClass, "OnControllerSelect", NULL, m_SelectedObject);
								//show selector
								ShowRadialSelector(m_SelectedObject);
							}
						}
					}
					else
					{
						GetGame().GameScript.CallFunction(m_RegisteredClass, "OnControllerDeselect", NULL, m_SelectedObject);
						m_SelectedObject = NULL;
						//hide selector
						HideRadialSelector();						
					}		
				}
				//if controller is giving no feedback
				else
				{
					if (m_IsControllerTimoutEnabled)
					{
						m_ControllerTimout += delta_time;
						
						if (m_ControllerTimout >= CONTROLLER_DESELECT_TIMEOUT)
						{
							GetGame().GameScript.CallFunction(m_RegisteredClass, "OnControllerDeselect", NULL, m_SelectedObject);
							m_SelectedObject = NULL;
							//hide selector
							HideRadialSelector();						
							
							m_ControllerTimout = 0;		//reset controller timeout
						}
					}
				}
				
				m_ControllerAngle = -1;				//reset angle and tilt
				m_ControllerTilt = -1;
			}
			
			m_WidgetInitialized = true;
		}
	}
	
	float NormalizeInvertAngle(float angle)
	{
		float new_angle = 360 - angle;
		int angle_mp = new_angle / 360;
			
		new_angle = new_angle - (360 * angle_mp);
			
		return new_angle;
	}
	
	//============================================
	// Controls
	//============================================
	void UpdataControllerInput()
	{
		Input input = GetGame().GetInput();
		
		//Controller radial
		float angle;
		float tilt;
		input.GetGamepadThumbDirection(GamepadButton.THUMB_RIGHT, angle, tilt);
		angle = NormalizeInvertAngle(angle * Math.RAD2DEG);
			
		m_ControllerAngle = angle;
		m_ControllerTilt = tilt;
		m_ControllerTimout	= 0;			//reset controller timeout
		
		//Controller buttons
		//Select (A,cross)
		if (m_SelectInputWrapper.InputP().LocalPress())
		{
			GetGame().GameScript.CallFunction(m_RegisteredClass, "OnControllerPressSelect", NULL, m_SelectedObject);
		}
		
		//Back (B,circle)
		if (m_BackInputWrapper.InputP().LocalPress())
		{
			GetGame().GameScript.CallFunction(m_RegisteredClass, "OnControllerPressBack", NULL, m_SelectedObject);
		}
	}
	
	override bool OnMouseButtonUp(Widget w, int x, int y, int button)
	{
		if (button == MouseState.LEFT && m_SelectedObject/* && w == m_SelectedObject*/)
		{
			//Execute
			GetGame().GameScript.CallFunction(m_RegisteredClass, "OnMousePressLeft", NULL, m_SelectedObject);
			
			return true;
		}
		
		if (button == MouseState.RIGHT)
		{
			//Back one level
			GetGame().GameScript.CallFunction(m_RegisteredClass, "OnMousePressRight", NULL, NULL);
			
			return true;
		}
		
		return false;
	}
}