MLEM/MLEM/Formatting/TokenizedString.cs

using System;
using System.Linq;
using System.Text;
using Microsoft.Xna.Framework;
using Microsoft.Xna.Framework.Graphics;
using MLEM.Font;
using MLEM.Formatting.Codes;
using MLEM.Misc;

namespace MLEM.Formatting {
    public class TokenizedString : GenericDataHolder {

        public readonly string RawString;
        public string String { get; private set; }
        public readonly Token[] Tokens;
        public readonly Code[] AllCodes;

        public TokenizedString(string rawString, string strg, Token[] tokens) {
            this.RawString = rawString;
            this.String = strg;
            this.Tokens = tokens;
            // since a code can be present in multiple tokens, we use Distinct here
            this.AllCodes = tokens.SelectMany(t => t.AppliedCodes).Distinct().ToArray();
        }

        public void Split(GenericFont font, float width, float scale) {
            // a split string has the same character count as the input string
            // but with newline characters added
            this.String = font.SplitString(this.String, width, scale);
            // skip splitting logic for unformatted text
            if (this.Tokens.Length == 1) {
                this.Tokens[0].Substring = this.String;
                return;
            }
            foreach (var token in this.Tokens) {
                var index = 0;
                var length = 0;
                var ret = new StringBuilder();
                // this is basically a substring function that ignores newlines for indexing
                for (var i = 0; i < this.String.Length; i++) {
                    // if we're within the bounds of the token's substring, append to the new substring
                    if (index >= token.Index && length < token.Substring.Length)
                        ret.Append(this.String[i]);
                    // if the current char is not a newline, we simulate length increase
                    if (this.String[i] != '\n') {
                        if (index >= token.Index)
                            length++;
                        index++;
                    }
                }
                token.Substring = ret.ToString();
            }
        }

        public Vector2 Measure(GenericFont font) {
            return font.MeasureString(this.String);
        }

        public void Update(GameTime time) {
            foreach (var code in this.AllCodes)
                code.Update(time);
        }

        public Token GetTokenUnderPos(GenericFont font, Vector2 stringPos, Vector2 target, float scale) {
            var innerOffset = new Vector2();
            foreach (var token in this.Tokens) {
                var split = token.Substring.Split('\n');
                for (var i = 0; i < split.Length; i++) {
                    var size = font.MeasureString(split[i]) * scale;
                    var lineArea = new RectangleF(stringPos + innerOffset, size);
                    if (lineArea.Contains(target))
                        return token;

                    if (i < split.Length - 1) {
                        innerOffset.X = 0;
                        innerOffset.Y += font.LineHeight * scale;
                    } else {
                        innerOffset.X += size.X;
                    }
                }
            }
            return null;
        }

        public void Draw(GameTime time, SpriteBatch batch, Vector2 pos, GenericFont font, Color color, float scale, float depth) {
            var innerOffset = new Vector2();
            foreach (var token in this.Tokens) {
                var drawFont = token.GetFont() ?? font;
                var drawColor = token.GetColor() ?? color;
                for (var i = 0; i < token.Substring.Length; i++) {
                    var c = token.Substring[i];
                    if (c == '\n') {
                        innerOffset.X = 0;
                        innerOffset.Y += font.LineHeight * scale;
                    }
                    if (i == 0)
                        token.DrawSelf(time, batch, pos + innerOffset, font, color, scale, depth);

                    var cString = c.ToString();
                    token.DrawCharacter(time, batch, c, cString, i, pos + innerOffset, drawFont, drawColor, scale, depth);
                    innerOffset.X += font.MeasureString(cString).X * scale;
                }
            }
        }

    }
}