ClassiCube/ClassicalSharp/2D/Widgets/InputWidget.cs

// ClassicalSharp copyright 2014-2016 UnknownShadow200 | Licensed under MIT
using System;
using System.Drawing;
using OpenTK.Input;
#if ANDROID
using Android.Graphics;
#endif

namespace ClassicalSharp.Gui.Widgets {
	public abstract class InputWidget : Widget {
		
		public InputWidget( Game game, Font font ) : base( game ) {
			Text = new WrappableStringBuffer( Utils.StringLength * MaxLines );
			lines = new string[MaxLines];
			lineSizes = new Size[MaxLines];
			
			DrawTextArgs args = new DrawTextArgs( "_", font, true );
			caretTex = game.Drawer2D.MakeChatTextTexture( ref args, 0, 0 );
			caretTex.Width = (short)((caretTex.Width * 3) / 4);
			caretWidth = caretTex.Width; caretHeight = caretTex.Height;
			this.font = font;
			
			if( Prefix == null ) return;
			args = new DrawTextArgs( Prefix, font, true );
			Size size = game.Drawer2D.MeasureChatSize( ref args );
			prefixWidth = Width = size.Width;
			prefixHeight = Height = size.Height;
		}
		
		public InputWidget SetLocation( Anchor horAnchor, Anchor verAnchor, int xOffset, int yOffset ) {
			HorizontalAnchor = horAnchor; VerticalAnchor = verAnchor;
			XOffset = xOffset; YOffset = yOffset;
			CalculatePosition();
			return this;
		}
		
		protected int caret = -1;
		protected Texture inputTex, caretTex, prefixTex;
		protected readonly Font font;
		protected int caretWidth, caretHeight, prefixWidth, prefixHeight;
		protected FastColour caretColour;
		
		/// <summary> The raw text entered. </summary>
		/// <remarks> You should Append() to add more text, as that also updates the caret position and texture. </remarks>
		public WrappableStringBuffer Text;
		
		/// <summary> The maximum number of lines that may be entered. </summary>
		public abstract int MaxLines { get; }
		
		/// <summary> The maximum number of characters that can fit on one line. </summary>
		public abstract int MaxCharsPerLine { get; }
		
		/// <summary> The prefix string that is always shown before the input text. Can be null. </summary>
		public abstract string Prefix { get; }
		
		/// <summary> The horizontal offset (in pixels) from the start of the box background
		/// to the beginning of the input texture. </summary>
		public abstract int Padding { get; }
		
		/// <summary> Whether a caret should be drawn at the position characters 
		/// are inserted/deleted from the input text. </summary>
		public bool ShowCaret;
		
		protected string[] lines; // raw text of each line
		protected Size[] lineSizes; // size of each line in pixels
		protected int caretCol, caretRow; // coordinates of caret
		protected double caretAccumulator;
		
		public override void Init() {
			if( lines.Length > 1 ) {
				Text.WordWrap( game.Drawer2D, lines, MaxCharsPerLine );
			} else {
				lines[0] = Text.ToString();
			}
			
			CalculateLineSizes();
			RemakeTexture();
			UpdateCaret();
		}
		
		public override void Dispose() {
			gfx.DeleteTexture( ref inputTex );
		}
		
		public void DisposeFully() {
			Dispose();
			gfx.DeleteTexture( ref caretTex );
			gfx.DeleteTexture( ref prefixTex );
		}

		public override void CalculatePosition() {
			int oldX = X, oldY = Y;
			base.CalculatePosition();
			
			caretTex.X1 += X - oldX; caretTex.Y1 += Y - oldY;
			inputTex.X1 += X - oldX; inputTex.Y1 += Y - oldY;
		}
		
		
		/// <summary> Calculates the sizes of each line in the text buffer. </summary>
		public void CalculateLineSizes() {
			for( int y = 0; y < lineSizes.Length; y++ )
				lineSizes[y] = Size.Empty;
			lineSizes[0].Width = prefixWidth;
			
			DrawTextArgs args = new DrawTextArgs( null, font, true );
			for( int y = 0; y < MaxLines; y++ ) {
				args.Text = lines[y];
				lineSizes[y] += game.Drawer2D.MeasureChatSize( ref args );
			}
			if( lineSizes[0].Height == 0 ) lineSizes[0].Height = prefixHeight;
		}
		
		/// <summary> Calculates the location and size of the caret character </summary>
		public void UpdateCaret() {
			if( caret >= Text.Length ) caret = -1;
			Text.GetCoords( caret, lines, out caretCol, out caretRow );
			DrawTextArgs args = new DrawTextArgs( null, font, true );
			IDrawer2D drawer = game.Drawer2D;
			caretAccumulator = 0;

			if( caretCol == MaxCharsPerLine ) {
				caretTex.X1 = X + Padding + lineSizes[caretRow].Width;
				caretColour = FastColour.Yellow;
				caretTex.Width = (short)caretWidth;
			} else {
				args.Text = lines[caretRow].Substring( 0, caretCol );
				Size trimmedSize = drawer.MeasureChatSize( ref args );
				if( caretRow == 0 ) trimmedSize.Width += prefixWidth;

				caretTex.X1 = X + Padding + trimmedSize.Width;
				caretColour = FastColour.Scale( FastColour.White, 0.8f );
				
				string line = lines[caretRow];
				if( caretCol < line.Length ) {
					args.Text = new String( line[caretCol], 1 );
					caretTex.Width = (short)drawer.MeasureChatSize( ref args ).Width;
				} else {
					caretTex.Width = (short)caretWidth;
				}
			}
			caretTex.Y1 = lineSizes[0].Height * caretRow + inputTex.Y1 + 2;
			
			// Update the colour of the caret
			char code = GetLastColour( caretCol, caretRow );
			if( code != '\0' ) caretColour = drawer.Colours[code];
		}
		
		protected void RenderCaret( double delta ) {
			if( !ShowCaret ) return;
			
			caretAccumulator += delta;
			if( (caretAccumulator % 1) < 0.5 )
				caretTex.Render( gfx, caretColour );
		}
		
		/// <summary> Remakes the raw texture containg all the chat lines. </summary>
		/// <remarks> Also updates the dimensions of the widget. </remarks>
		public virtual void RemakeTexture() {
			int totalHeight = 0, maxWidth = 0;
			for( int i = 0; i < MaxLines; i++ ) {
				totalHeight += lineSizes[i].Height;
				maxWidth = Math.Max( maxWidth, lineSizes[i].Width );
			}
			Size size = new Size( maxWidth, totalHeight );
			caretAccumulator = 0;
			
			int realHeight = 0;
			using( Bitmap bmp = IDrawer2D.CreatePow2Bitmap( size ) )
				using( IDrawer2D drawer = game.Drawer2D )
			{
				drawer.SetBitmap( bmp );
				DrawTextArgs args = new DrawTextArgs( null, font, true );
				if( Prefix != null ) {
					args.Text = Prefix;
					drawer.DrawChatText( ref args, 0, 0 );
				}
				
				for( int i = 0; i < lines.Length; i++ ) {
					if( lines[i] == null ) break;
					args.Text = lines[i];
					char lastCol = GetLastColour( 0, i );
					if( !IDrawer2D.IsWhiteColour( lastCol ) )
						args.Text = "&" + lastCol + args.Text;
					
					int offset = i == 0 ? prefixWidth : 0;
					drawer.DrawChatText( ref args, offset, realHeight );
					realHeight += lineSizes[i].Height;
				}
				inputTex = drawer.Make2DTexture( bmp, size, 0, 0 );
			}
			
			Width = size.Width;
			Height = realHeight == 0 ? prefixHeight : realHeight;
			CalculatePosition();
			inputTex.X1 = X + Padding; inputTex.Y1 = Y;
		}
		
		protected char GetLastColour( int indexX, int indexY ) {
			int x = indexX;
			IDrawer2D drawer = game.Drawer2D;
			for( int y = indexY; y >= 0; y-- ) {
				string part = lines[y];
				char code = drawer.LastColour( part, x );
				if( code != '\0' ) return code;
				if( y > 0 ) x = lines[y - 1].Length;
			}
			return '\0';
		}

		
		/// <summary> Invoked when the user presses enter. </summary>
		public virtual void EnterInput() {
			Clear();
			Height = prefixHeight;
		}
		
		
		/// <summary> Clears all the characters from the text buffer </summary>
		/// <remarks> Deletes the native texture. </remarks>
		public void Clear() {
			Text.Clear();
			for( int i = 0; i < lines.Length; i++ )
				lines[i] = null;
			
			caret = -1;
			Dispose();
		}

		/// <summary> Appends a sequence of characters to current text buffer. </summary>
		/// <remarks> Potentially recreates the native texture. </remarks>
		public void Append( string text ) {
			int appended = 0;
			foreach( char c in text ) {
				if( TryAppendChar( c ) ) appended++;
			}
			
			if( appended == 0 ) return;
			Recreate();
		}
		
		/// <summary> Appends a single character to current text buffer. </summary>
		/// <remarks> Potentially recreates the native texture. </remarks>
		public void Append( char c ) {
			if( !TryAppendChar( c ) ) return;
			Recreate();
		}
		
		
		bool TryAppendChar( char c ) {
			int totalChars = MaxCharsPerLine * lines.Length;
			if( Text.Length == totalChars ) return false;
			if( !AllowedChar( c ) ) return false;
			
			AppendChar( c );
			return true;
		}
		
		protected virtual bool AllowedChar( char c ) {
			return Utils.IsValidInputChar( c, game );
		}
		
		protected void AppendChar( char c ) {
			if( caret == -1 ) {
				Text.InsertAt( Text.Length, c );
			} else {
				Text.InsertAt( caret, c );
				caret++;
				if( caret >= Text.Length ) caret = -1;
			}
		}
		
		protected void DeleteChar() {
			if( Text.Length == 0 ) return;
			
			if( caret == -1 ) {
				Text.DeleteAt( Text.Length - 1 );
			} else if( caret > 0 ) {
				caret--;
				Text.DeleteAt( caret );
			}
		}
		
		
		#region Input handling
		
		protected bool ControlDown() {
			return OpenTK.Configuration.RunningOnMacOS ?
				(game.IsKeyDown( Key.WinLeft ) || game.IsKeyDown( Key.WinRight ))
				: (game.IsKeyDown( Key.ControlLeft ) || game.IsKeyDown( Key.ControlRight ));
		}
		
		public override bool HandlesKeyPress( char key ) {
			if( !game.HideGui ) Append( key );
			return true;
		}
		
		public override bool HandlesKeyDown( Key key ) {
			if( game.HideGui ) return key < Key.F1 || key > Key.F35;
			bool clipboardDown = ControlDown();
			
			if( key == Key.Left ) LeftKey( clipboardDown );
			else if( key == Key.Right ) RightKey( clipboardDown );
			else if( key == Key.BackSpace ) BackspaceKey( clipboardDown );
			else if( key == Key.Delete ) DeleteKey();
			else if( key == Key.Home ) HomeKey();
			else if( key == Key.End ) EndKey();
			else if( clipboardDown && !OtherKey( key ) ) return false;
			
			return true;
		}
		
		public override bool HandlesKeyUp( Key key ) { return true; }
		
		public override bool HandlesMouseClick( int mouseX, int mouseY, MouseButton button ) {
			if( button == MouseButton.Left )
				SetCaretToCursor( mouseX, mouseY );
			return true;
		}
		
		
		void BackspaceKey( bool controlDown ) {
			if( controlDown ) {
				if( caret == -1 ) caret = Text.Length - 1;
				int len = Text.GetBackLength( caret );
				if( len == 0 ) return;
				
				caret -= len;
				if( caret < 0 ) caret = 0;
				for( int i = 0; i <= len; i++ )
					Text.DeleteAt( caret );
				
				if( caret >= Text.Length ) caret = -1;
				if( caret == -1 &&  Text.Length > 0 ) {
					Text.value[Text.Length] = ' ';
				} else if( caret >= 0 && Text.value[caret] != ' ' ) {
					Text.InsertAt( caret, ' ' );
				}
				Recreate();
			} else if( !Text.Empty && caret != 0 ) {
				int index = caret == -1 ? Text.Length - 1 : caret;
				if( CheckColour( index - 1 ) ) {
					DeleteChar(); // backspace XYZ%e to XYZ
				} else if( CheckColour( index - 2 ) ) {
					DeleteChar(); DeleteChar(); // backspace XYZ%eH to XYZ
				}
				
				DeleteChar();
				Recreate();
			}
		}

		bool CheckColour( int index ) {
			if( index < 0 ) return false;
			char code = Text.value[index], col = Text.value[index + 1];
			return (code == '%' || code == '&') && game.Drawer2D.ValidColour( col );
		}
		
		void DeleteKey() {
			if( !Text.Empty && caret != -1 ) {
				Text.DeleteAt( caret );
				if( caret >= Text.Length ) caret = -1;
				Recreate();
			}
		}
		
		void LeftKey( bool controlDown ) {
			if( controlDown ) {
				if( caret == -1 )
					caret = Text.Length - 1;
				caret -= Text.GetBackLength( caret );
				UpdateCaret();
				return;
			}
			
			if( !Text.Empty ) {
				if( caret == -1 ) caret = Text.Length;
				caret--;
				if( caret < 0 ) caret = 0;
				UpdateCaret();
			}
		}
		
		void RightKey( bool controlDown ) {
			if( controlDown ) {
				caret += Text.GetForwardLength( caret );
				if( caret >= Text.Length ) caret = -1;
				UpdateCaret();
				return;
			}
			
			if( !Text.Empty && caret != -1 ) {
				caret++;
				if( caret >= Text.Length ) caret = -1;
				UpdateCaret();
			}
		}
		
		void HomeKey() {
			if( Text.Empty ) return;
			caret = 0;
			UpdateCaret();
		}
		
		void EndKey() {
			caret = -1;
			UpdateCaret();
		}
		
		static char[] trimChars = {'\r', '\n', '\v', '\f', ' ', '\t', '\0'};
		bool OtherKey( Key key ) {
			int totalChars = MaxCharsPerLine * lines.Length;
			if( key == Key.V && Text.Length < totalChars ) {
				string text = null;
				try {
					text = game.window.ClipboardText.Trim( trimChars );
				} catch( Exception ex ) {
					ErrorHandler.LogError( "Paste from clipboard", ex );
					const string warning = "&cError while trying to paste from clipboard.";
					game.Chat.Add( warning, MessageType.ClientStatus4 );
					return true;
				}

				if( String.IsNullOrEmpty( text ) ) return true;
				Append( text );
				return true;
			} else if( key == Key.C ) {
				if( Text.Empty ) return true;
				try {
					game.window.ClipboardText = Text.ToString();
				} catch( Exception ex ) {
					ErrorHandler.LogError( "Copy to clipboard", ex );
					const string warning = "&cError while trying to copy to clipboard.";
					game.Chat.Add( warning, MessageType.ClientStatus4 );
				}
				return true;
			}
			return false;
		}
		
		
		protected unsafe void SetCaretToCursor( int mouseX, int mouseY ) {
			mouseX -= inputTex.X1; mouseY -= inputTex.Y1;
			DrawTextArgs args = new DrawTextArgs( null, font, true );
			IDrawer2D drawer = game.Drawer2D;
			int offset = 0, elemHeight = caretHeight;
			string oneChar = new String( 'A', 1 );
			
			for( int y = 0; y < lines.Length; y++ ) {
				string line = lines[y];
				int xOffset = y == 0 ? prefixWidth : 0;
				if( line == null ) continue;
				
				for( int x = 0; x < line.Length; x++ ) {
					args.Text = line.Substring( 0, x );
					int trimmedWidth = drawer.MeasureChatSize( ref args ).Width + xOffset;
					// avoid allocating an unnecessary string
					fixed( char* ptr = oneChar )
						ptr[0] = line[x];
					
					args.Text = oneChar;
					int elemWidth = drawer.MeasureChatSize( ref args ).Width;
					
					if( GuiElement.Contains( trimmedWidth, y * elemHeight, elemWidth, elemHeight, mouseX, mouseY ) ) {
						caret = offset + x;
						UpdateCaret(); return;
					}
				}
				offset += line.Length;
			}
			caret = -1;
			UpdateCaret();
		}
		
		#endregion
	}
}