Exemple #1
0
        public void GenerateClass(NodeInfo nodeInfo)
        {
            //Checking if object is not null
            Assert.IsNotNull(nodeInfo);

            string directoryPath = Environment.CurrentDirectory + "/Assets";

            using (StreamWriter sw = new StreamWriter(Path.Combine(directoryPath, nodeInfo.ClassName + ".cs")))
            {
                //adding library's
                sw.WriteLine("using UnityEngine;");
                sw.WriteLine("using System.Collections;");
                sw.WriteLine("using System.Collections.Generic;");

                sw.WriteLine(""); //white space

                sw.WriteLine("/// <Information>");
                sw.WriteLine("/// This class is generated with the UML tool from Geoffrey Hendrikx.");
                sw.WriteLine("/// </Information>");

                if (nodeInfo.Parent != string.Empty)
                {
                    sw.WriteLine(nodeInfo.ClassAccessModifiers.ToString().ToLower() + " class " +
                                 nodeInfo.ClassName + " : " + nodeInfo.Parent);
                }
                else
                {
                    sw.WriteLine(nodeInfo.ClassAccessModifiers.ToString().ToLower() + " class " + nodeInfo.ClassName);
                }

                sw.WriteLine("{");

                //adding variables
                for (int i = 0; i < nodeInfo.VariableInfo.Count; i++)
                {
                    ClassContent variable = nodeInfo.VariableInfo[i];
                    sw.WriteLine("\t" + variable.AccessModifiers.ToString().ToLower() + " " +
                                 variable.Type + " " + variable.Name + ";");
                }

                sw.WriteLine(" "); // whitespace

                for (int i = 0; i < nodeInfo.MethodeInfo.Count; i++)
                {
                    ClassContent methode = nodeInfo.MethodeInfo[i];
                    if (methode.Name.EndsWith(")"))
                    {
                        sw.WriteLine("\t" + methode.AccessModifiers.ToString().ToLower() + " " +
                                     methode.Type + " " + methode.Name + "{ }");
                    }
                    else
                    {
                        sw.WriteLine("\t" + methode.AccessModifiers.ToString().ToLower() + " " +
                                     methode.Type + " " + methode.Name + "() { }");
                    }
                }

                sw.WriteLine("}");

                sw.Flush();

                sw.Close();
            }
            AssetDatabase.Refresh();
        }
Exemple #2
0
        /// <summary>
        /// Drawing the node
        /// </summary>
        public void Draw()
        {
            GUILayout.BeginVertical();
            GUILayout.Label("Class name");
            GUILayout.BeginHorizontal();
            ClassAccessModifiers =
                (AccessModifiers)EditorGUILayout.EnumPopup((AccessModifiers)ClassAccessModifiers, GUILayout.MaxWidth(70));
            ClassName = GUILayout.TextField(ClassName);
            GUILayout.EndHorizontal();

            GUILayout.Label("Inheritance");
            Parent = GUILayout.TextField(Parent);

            GUILayout.Space(15);
            //variables
            GUILayout.Label("Variables \t Type \t Name");

            //Variable
            for (int i = 0; i < variableInfo.Count; i++)
            {
                ClassContent variable = VariableInfo[i];
                GUILayout.BeginHorizontal();
                //AccessModifier Type Variable
                variable.AccessModifiers = (AccessModifiers)EditorGUILayout.EnumPopup((AccessModifiers)variable.AccessModifiers, GUILayout.MaxWidth(70));
                variable.Type            = GUILayout.TextField(variable.Type);
                variable.Name            = GUILayout.TextField(variable.Name);

                if (GUILayout.Button("X", GUILayout.MaxWidth(20)))
                {
                    if (variableInfo.Count == 1)
                    {
                        VariableInfo = new List <ClassContent>();
                    }
                    VariableInfo.Remove(variableInfo[i]);
                }
                GUILayout.EndHorizontal();
            }

            //adding variable to the list.
            if (GUILayout.Button("Add variable", GUILayout.MaxWidth(100)))
            {
                variableInfo.Add(new ClassContent());
            }


            GUILayout.Space(15);

            //Methodes
            GUILayout.Label("Methods \t Type \t Name");
            for (int i = 0; i < methodeInfo.Count; i++)
            {
                ClassContent methode = methodeInfo[i];
                GUILayout.BeginHorizontal();
                //AccessModifier Type Methode
                methode.AccessModifiers = (AccessModifiers)EditorGUILayout.EnumPopup((AccessModifiers)methode.AccessModifiers, GUILayout.MaxWidth(70));
                methode.Type            = GUILayout.TextField(methode.Type);
                methode.Name            = GUILayout.TextField(methode.Name);

                //methode.SetStructVariables(accesModifierTemp, typeTemp, nameTemp);

                if (GUILayout.Button("X", GUILayout.MaxWidth(20)))
                {
                    if (methodeInfo.Count == 1)
                    {
                        methodeInfo = new List <ClassContent>();
                    }
                    methodeInfo.RemoveAt(i);
                }
                GUILayout.EndHorizontal();
            }

            //adding methode to the list.
            if (GUILayout.Button("Add methods", GUILayout.MaxWidth(100)))
            {
                methodeInfo.Add(new ClassContent());
            }

            GUILayout.Space(15);

            GUILayout.EndVertical();
        }