CppGenerator.cs

Go to the documentation of this file.

using System;
using System.Linq;

namespace CodeGen.generators
{
    public class CppGenerator : Generator
    {
        private const string ClassFormat = "class {0} {1}\n{{{2}{3}{4}}}";
        private string Indent { get; } = GeneratorConf.GetIndent(UseTabs, 4);

        protected override string GenerateClass(Class @class)
        {
            var parent = "";
            if (@class.Parent?.Length > 0)
            {
                parent = " : " + @class.Parent + " " + @class.Parent;
            }

            var @public = GenerateSection("public", @class);
            var @protected = GenerateSection("protected", @class);
            var @private = GenerateSection("private", @class);
            return string.Format(ClassFormat, @class.Name, parent, @public, @protected, @private);
        }

        public override string GenerateField(Field field)
        {
            if (!new CppValidator().FieldIsValid(field))
            {
                throw new ArgumentNullException();
            }

            var result = Indent;
            if (field.Static)
            {
                result += "static ";
            }

            if (field.Const)
            {
                result += "const ";
            }

            result += field.Type + " " + field.Name;
            if (field.Default != "")
            {
                result += " = " + field.Default;
            }

            result += ";";
            return result;
        }

        public override string GenerateMethod(Method method)
        {
            var result = "";

            if (method.Static)
            {
                result += "static ";
            }

            switch (method.Type)
            {
                case "":
                    result += "void ";
                    break;
                default:
                    result += method.Type + " ";
                    break;
            }

            result += method.Name + "(";
            for (var i = 0; i < method.Parameters?.Length; i++)
            {
                var parameter = method.Parameters[i];
                result += parameter.Type + " " + parameter.Name;
                if (!string.IsNullOrWhiteSpace(parameter.Default))
                {
                    result += " = " + parameter.Default;
                }
                if (i + 1 < method.Parameters?.Length)
                {
                    result += ", ";
                }
            }

            result += ");";
            return result;
        }

        private string GenerateSection(string access, Class @class)
        {
            var result = @class.Fields?.Where(field => access == field.Access)
                .Aggregate("", (current, field) => current + (GenerateField(field) + "\n"));

            result = @class.Methods?.Where(method => access == method.Access).Aggregate(result,
                (current, method) => current + (GeneratorConf.ShiftCode(GenerateMethod(method), 1, Indent) + "\n"));
            result = @class.Classes?.Where(clas => access == clas.Access).Aggregate(result,
                (current, clas) => current + (GeneratorConf.ShiftCode(GenerateClass(clas), 1, Indent) + "\n"));
            if (result != "")
            {
                result = "\n" + access + ":\n" + result;
            }

            return result;
        }
    }

    public class CppValidator : Validator
    {
    }
}