Пример #1
0
        public string GenerateCode(BinaryFileSchema schema)
        {
            StringBuilder b = new StringBuilder();
            List<CodeClass> codeClasses = new List<CodeClass>();
            FormatName = schema.FormatBlock.Name;

            b.AppendLine("using System;");
            b.AppendLine("using System.Collections.Generic;");
            b.AppendLine("using System.IO;");
            b.AppendLine("using System.Text;");
            b.AppendLine("namespace " + schema.FormatBlock.Name);
            b.AppendLine("{");

            CodeClass mainClass = new CodeClass(schema.FormatBlock.Name + "Parser");
            codeClasses.Add(mainClass);
            rootParser = mainClass;

            CodeMethod mainMethod = new CodeMethod("void Main(string[] args)");
            mainMethod.CodeLines.Add("try{");
            mainMethod.CodeLines.Add("\t" + schema.FormatBlock.Name + " ps = " + schema.FormatBlock.Name + "Parser.ReadFile(args[0]);");

            mainMethod.CodeLines.Add("\tConsole.WriteLine(ps.ToString());");

            mainMethod.CodeLines.Add("}");
            mainMethod.CodeLines.Add("catch(Exception e){");
            mainMethod.CodeLines.Add("\tConsole.WriteLine(e.Message);");
            mainMethod.CodeLines.Add("\tConsole.WriteLine(e.StackTrace);");
            mainMethod.CodeLines.Add("}");
            mainClass.CodeMethods.Add(mainMethod);

            CodeMethod readFile = new CodeMethod( schema.FormatBlock.Name + " ReadFile(string filename)");
            mainClass.CodeMethods.Add(readFile);

            readFile.CodeLines.Add("FileStream fstream = new FileStream(filename, FileMode.Open);");
            readFile.CodeLines.Add("BinaryReader reader = new BinaryReader(fstream);");
            string endianness = ((schema.ByteOrder.ByteOrder == BfsByteOrderEnum.BigEndian) ? BfsByteOrderEnum.BigEndian : BfsByteOrderEnum.LittleEndian).ToString();
            readFile.CodeLines.Add("BfsBinaryReader bfsReader = new BfsBinaryReader(reader, BfsBinaryReader.Endianness." + endianness + ");");
            readFile.CodeLines.Add("return " + schema.FormatBlock.Name + ".Read(bfsReader);");

            foreach (IBfsDataBlock dataBlock in schema.DatablockList)
            {
                CodeClass newClass = MakeClass(dataBlock);
                if(newClass != null)
                    codeClasses.Add(newClass);
            }

            //Output the classes
            foreach (CodeClass cc in codeClasses)
                b.Append(cc.ToString());

            b.AppendLine("}");
            return b.ToString();
        }
Пример #2
0
        public void MakeBitfield(CodeClass codeClass, BfsBitfield dataField)
        {
            codeClass.ClassType = CodeClass.ClassTypeEnum.Struct;

            //Adds the implicit 'value' variable for storing the read primitive type.
            codeClass.CodeFields.Add(CSHelper.TypeMap(dataField.PrimitiveType).ToLower() + " value;");

            string objectType = dataField.Name;
            string varName = dataField.Name.ToLower();

            StringBuilder b = new StringBuilder("bool ");
            for ( int i = 0; i< dataField.BitFieldFields.Count; i++)
            {
                if (i != 0)
                    b.Append(", ");

                BfsBitfieldField field = dataField.BitFieldFields[i];
                if (field.Name == null)
                    b.Append("bit" + field.BitNumber);
                else
                    b.Append(field.Name);
            }
            b.Append(";");
            codeClass.CodeFields.Add(b.ToString());

            CodeMethod codeMethod = new CodeMethod(objectType + " Read(BfsBinaryReader file)");
            codeClass.CodeMethods.Add(codeMethod);

            codeMethod.CodeLines.Add(objectType + " " + varName + " = new " + objectType + "();");
            codeMethod.CodeLines.Add(varName + ".value = file.Read" + dataField.PrimitiveType.PrimitiveType.ToString() + "();");

            foreach( BfsBitfieldField field in dataField.BitFieldFields )
            {
                int size = Environments.GetSizeOfPrimitiveType(dataField.PrimitiveType);
                string mask = MakeMask(field.BitNumber, size);
                string bitName = ((field.Name != null) ? field.Name : ("bit" + field.BitNumber));

                if (field.Actions.Count > 0)
                {
                    IfBlock block = new IfBlock("(" + varName + ".value & " + mask + ") == 1");
                    block.CodeLines.Add(varName + "." + bitName + " = true;");
                    MakeActionList(dataField, block.CodeLines, field.Actions);
                    codeMethod.AddSplitLines(block.ToString());
                }
                else
                    codeMethod.CodeLines.Add(varName + "." + bitName + " = " + "(" + varName + ".value & " + mask + ") == 1;");
            }

            codeMethod.CodeLines.Add("return " + varName + ";");
        }
Пример #3
0
        public void MakeStruct(CodeClass codeClass, BfsStruct dataStruct)
        {
            foreach (BfsStructField structField in dataStruct.StructFieldList)
            {
                string type = "";
                string name = structField.Name;

                if (structField.FieldType is BfsNamedType)
                {
                    BfsNamedType namedType = structField.FieldType as BfsNamedType;
                    type = namedType.DataBlock.Name;
                }
                else if (structField.FieldType is BfsPrimitiveType)
                {
                    BfsPrimitiveType primitiveType = structField.FieldType as BfsPrimitiveType;
                    type = CSHelper.TypeMap(primitiveType).ToLower();
                }
                else if (structField.FieldType is BfsFunctionType)
                {
                    BfsFunctionType functionType = structField.FieldType as BfsFunctionType;
                    if (functionType.FunctionName == "ascii")
                        type = "string";
                    else
                        BfsCompiler.ReportError(functionType.SourceRange,
                            "Only ASCII strings are supported so far..!");
                }

                if (structField.FieldType.ArrayExtension == null)
                    codeClass.CodeFields.Add(type + " " + name + ";");
                else
                {
                    if (structField.FieldType.ArrayExtension is BfsKnownArray)
                        codeClass.CodeFields.Add(type + " [] " + name + ";");
                    else
                        if (structField.FieldType.ArrayExtension is BfsUnknownArray)
                            codeClass.CodeFields.Add("List<" + type + "> " + name + ";");
                        else
                            BfsCompiler.ReportError(structField.FieldType.ArrayExtension.SourceRange,
                                "Unknown array extended type. Known or unknown array type expected!");
                }

            }

            CodeMethod read = new CodeMethod(dataStruct.Name + " Read(BfsBinaryReader file)");
            codeClass.CodeMethods.Add(read);
            MakeReadStruct(dataStruct, read);

            //ToString() method
            CodeMethod toString = new CodeMethod("override string ToString()");
            codeClass.CodeMethods.Add(toString);

            toString.CodeLines.Add("StringBuilder sb = new StringBuilder();");
            toString.CodeLines.Add("sb.AppendLine(\"== Struct: "+ dataStruct.Name +" ==\");");
            foreach (BfsStructField field in dataStruct.StructFieldList)
                toString.CodeLines.Add("sb.AppendLine(\"" + field.Name + " : \" + " + field.ToString() + ".ToString());");
            toString.CodeLines.Add("return sb.ToString();");
        }
Пример #4
0
        public void MakeEnum(CodeClass codeClass, BfsEnum dataEnum)
        {
            codeClass.ClassType = CodeClass.ClassTypeEnum.Struct;

            //Only add nessecary internal enum aliases if there are any.
            if (dataEnum.EnumAliases.Count > 0)
            {
                StringBuilder b = new StringBuilder();
                b.Append("enum " + dataEnum.Name + "Enum {");

                List<string> foundEnumAliases = new List<string>();
                for (int i = 0; i < dataEnum.EnumFields.Count; i++)
                {
                    BfsEnumField enumField = dataEnum.EnumFields[i];
                    if (enumField.Alias != null && !foundEnumAliases.Contains(enumField.Alias))
                    {
                        if (i != 0)
                            b.Append(", ");
                        b.Append(enumField.Alias);
                        foundEnumAliases.Add(enumField.Alias);
                    }
                }
                b.Append("}");
                codeClass.CodeFields.Add(b.ToString());
                codeClass.CodeFields.Add(dataEnum.Name + "Enum " + dataEnum.Name.ToLower() + ";");
            }

            //Adds the implicit 'value' variable for storing the read primitive type.
            codeClass.CodeFields.Add(CSHelper.TypeMap(dataEnum.PrimitiveType).ToLower() + " value;");

            CodeMethod codeMethod = new CodeMethod(dataEnum.Name + " Read(BfsBinaryReader file)");
            codeClass.CodeMethods.Add(codeMethod);

            string objectType = dataEnum.Name;
            string varName = dataEnum.Name.ToLower();
            codeMethod.CodeLines.Add(objectType + " " + varName + " = new " + objectType + "();");
            codeMethod.CodeLines.Add(varName + ".value = file.Read" + dataEnum.PrimitiveType.PrimitiveType.ToString() + "();");

            //Sorts the enum fields into categories to make a pretty switch-case/if-then-else structure.
            List<BfsEnumField> singleValues = new List<BfsEnumField>();
            List<BfsEnumField> rangeValues = new List<BfsEnumField>();
            BfsEnumField elseValue = null;

            foreach (BfsEnumField enumField in dataEnum.EnumFields)
                if (enumField.EnumMatch is BfsEnumValue)
                    singleValues.Add(enumField);
                else if (enumField.EnumMatch is BfsEnumRange)
                    rangeValues.Add(enumField);
                else if (enumField.EnumMatch is BfsEnumElse)
                    elseValue = enumField;

            string tab = string.Empty;

            //If there are 2 or above single enum matches a switch/case statement is made with the values as cases.
            //In case the value isn't any of the values it checks the ranges one by one in an if/else/elseif/elseif/... kind of way.

            if (singleValues.Count > 1)
            {
                codeMethod.CodeLines.Add("switch(" + varName + ".value)");
                codeMethod.CodeLines.Add("{");
                tab = "\t\t";

                foreach (BfsEnumField enumField in singleValues)
                {
                    BfsEnumValue enumVal = enumField.EnumMatch as BfsEnumValue;
                    codeMethod.CodeLines.Add("\tcase " + enumVal.Value + ":");
                    if (enumField.Alias != null)
                        codeMethod.CodeLines.Add("\t\t" + varName + "." + varName + " = " + objectType + "Enum." + enumField.Alias + ";");

                    List<string> actionList = new List<string>();
                    bool addBreak = MakeActionList(dataEnum, actionList, enumField.Actions);

                    foreach (string str in actionList)
                        codeMethod.CodeLines.Add("\t\t" + str);

                    if (addBreak)
                        codeMethod.CodeLines.Add("\t\tbreak;");
                }

                codeMethod.CodeLines.Add("\tdefault:");
            }

            string defaultTab = "\t";
            if (singleValues.Count <= 1)
                defaultTab = string.Empty;

            //If there only are one single value case, then include it in the if-then-else sequence
            IfThenElseSequence ifthenelse = new IfThenElseSequence(defaultTab, varName + ": '" + "\" + " + varName + ".value + \"");

            if (elseValue != null)
            {
                ifthenelse.ElseBlock = new IfBlock();
                MakeActionList(dataEnum, ifthenelse.ElseBlock.CodeLines, elseValue.Actions);
            }

            int countIfElse = 0;
            if (singleValues.Count == 1)
            {
                BfsEnumField enumField = singleValues[0];
                BfsEnumValue enumVal = enumField.EnumMatch as BfsEnumValue;

                IfBlock ifblock = new IfBlock(varName + ".value == " + enumVal.Value);
                ifthenelse.IfBlocks.Add(ifblock);

                if (enumField.Alias != null)
                    ifblock.CodeLines.Add(varName + "." + varName + " = " + objectType + "Enum." + enumField.Alias + ";");

                MakeActionList(dataEnum, ifblock.CodeLines, enumField.Actions);
                countIfElse++;
            }

            //Enum ranges

            foreach (BfsEnumField enumField in rangeValues)
            {
                BfsEnumRange enumVal = enumField.EnumMatch as BfsEnumRange;
                string firstEquality = ((enumVal.StartInclusion == BfsInclusionEnum.Included) ? ">= " : "> ") + enumVal.StartValue;
                string secondEquality = ((enumVal.EndInclusion == BfsInclusionEnum.Included) ? "<= " : "< ") + enumVal.EndValue;

                IfBlock ifblock = new IfBlock(varName + ".value " + firstEquality + " && " + varName + ".value " + secondEquality);
                ifthenelse.IfBlocks.Add(ifblock);
                if (enumField.Alias != null)
                    ifblock.CodeLines.Add(varName + "." + varName + " = " + objectType + "Enum." + enumField.Alias + ";");

                MakeActionList(dataEnum, ifblock.CodeLines, enumField.Actions);
            }

            codeMethod.AddSplitLines(ifthenelse.ToString());

            //End off the switch
            if (singleValues.Count > 1)
            {
                //Only add the final 'break' if the default case doesn't throw any exception
                //to prevent the code not reachable error

                if (ifthenelse.IfBlocks.Count != 0 || !ifthenelse.ElseBlock.ToString().Contains("throw new "))
                    codeMethod.CodeLines.Add("\t\tbreak;");

                codeMethod.CodeLines.Add("}");
            }

            codeMethod.CodeLines.Add("return " + varName + ";");

            //ToString() method
            CodeMethod toString = new CodeMethod("override string ToString()");
            codeClass.CodeMethods.Add(toString);

            if(dataEnum.EnumAliases.Count > 0)
                toString.CodeLines.Add("return " + dataEnum.Name.ToLower() + ".ToString();");
            else
                toString.CodeLines.Add("return value.ToString();");
        }
Пример #5
0
        public CodeClass MakeClass(IBfsDataBlock block)
        {
            //Append all local fields no matter the type
            CodeClass codeClass = new CodeClass(block.Name);
            foreach (BfsLocalField localField in block.LocalFieldList)
                codeClass.CodeFields.Add( localField.PrimitiveType.PrimitiveType.ToString().ToLower() + " " + localField.Name + ";");

            if (block is BfsStruct)
            {
                BfsStruct dataStruct = block as BfsStruct;
                MakeStruct(codeClass, dataStruct);
                return codeClass;
            }
            else if (block is BfsEnum)
            {
                BfsEnum dataEnum = block as BfsEnum;
                MakeEnum(codeClass, dataEnum);
                return codeClass;
            }
            else if (block is BfsBitfield)
            {
                BfsBitfield dataField = block as BfsBitfield;
                MakeBitfield(codeClass, dataField);
                return codeClass;
            }
            else if (block is BfsAbsOffset || block is BfsRelOffset)
            {
                BfsCompiler.ReportError(block.BlockTypeSourceRange,"Not yet implemented: Support for abs_offset or rel_offset!");
            }

            return null;
        }