path: root/csharp/src/ProtoGen/SourceGeneratorBase.cs

#region Copyright notice and license

// Protocol Buffers - Google's data interchange format
// Copyright 2008 Google Inc.  All rights reserved.
// http://github.com/jskeet/dotnet-protobufs/
// Original C++/Java/Python code:
// http://code.google.com/p/protobuf/
//
// Redistribution and use in source and binary forms, with or without
// modification, are permitted provided that the following conditions are
// met:
//
//     * Redistributions of source code must retain the above copyright
// notice, this list of conditions and the following disclaimer.
//     * Redistributions in binary form must reproduce the above
// copyright notice, this list of conditions and the following disclaimer
// in the documentation and/or other materials provided with the
// distribution.
//     * Neither the name of Google Inc. nor the names of its
// contributors may be used to endorse or promote products derived from
// this software without specific prior written permission.
//
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

#endregion

using System.Collections.Generic;
using Google.ProtocolBuffers.DescriptorProtos;
using Google.ProtocolBuffers.Descriptors;

namespace Google.ProtocolBuffers.ProtoGen
{
    internal abstract class SourceGeneratorBase<T> where T : IDescriptor
    {
        private readonly T descriptor;

        protected readonly bool OptimizeSpeed;
        protected readonly bool OptimizeSize;
        protected readonly bool UseLiteRuntime;
        protected readonly string RuntimeSuffix;

        protected SourceGeneratorBase(T descriptor)
        {
            this.descriptor = descriptor;

            OptimizeSize = descriptor.File.Options.OptimizeFor ==
                           FileOptions.Types.OptimizeMode.CODE_SIZE;
            OptimizeSpeed = descriptor.File.Options.OptimizeFor ==
                            FileOptions.Types.OptimizeMode.SPEED;
            UseLiteRuntime = descriptor.File.Options.OptimizeFor ==
                             FileOptions.Types.OptimizeMode.LITE_RUNTIME;
            //Lite runtime uses OptimizeSpeed code branches
            OptimizeSpeed |= UseLiteRuntime;
            RuntimeSuffix = UseLiteRuntime ? "Lite" : "";
        }

        protected T Descriptor
        {
            get { return descriptor; }
        }

        internal static string GetClassName(IDescriptor descriptor)
        {
            return ToCSharpName(descriptor.FullName, descriptor.File);
        }

        // Groups are hacky:  The name of the field is just the lower-cased name
        // of the group type.  In C#, though, we would like to retain the original
        // capitalization of the type name.
        internal static string GetFieldName(FieldDescriptor descriptor)
        {
            if (descriptor.FieldType == FieldType.Group)
            {
                return descriptor.MessageType.Name;
            }
            else
            {
                return descriptor.Name;
            }
        }

        internal static string GetFieldConstantName(FieldDescriptor field)
        {
            return field.CSharpOptions.PropertyName + "FieldNumber";
        }

        private static string ToCSharpName(string name, FileDescriptor file)
        {
            string result = file.CSharpOptions.Namespace;
            if (file.CSharpOptions.NestClasses)
            {
                if (result != "")
                {
                    result += ".";
                }
                result += file.CSharpOptions.UmbrellaClassname;
            }
            if (result != "")
            {
                result += '.';
            }
            string classname;
            if (file.Package == "")
            {
                classname = name;
            }
            else
            {
                // Strip the proto package from full_name since we've replaced it with
                // the C# namespace.
                classname = name.Substring(file.Package.Length + 1);
            }
            result += classname.Replace(".", ".Types.");
            return "global::" + result;
        }

        protected string ClassAccessLevel
        {
            get { return descriptor.File.CSharpOptions.PublicClasses ? "public" : "internal"; }
        }

        protected void WriteGeneratedCodeAttributes(TextGenerator writer)
        {
            if (descriptor.File.CSharpOptions.GeneratedCodeAttributes)
            {
                writer.WriteLine("[global::System.Runtime.CompilerServices.CompilerGeneratedAttribute()]");
                writer.WriteLine("[global::System.CodeDom.Compiler.GeneratedCodeAttribute(\"{0}\", \"{1}\")]",
                                 GetType().Assembly.GetName().Name, GetType().Assembly.GetName().Version);
            }
        }

        protected void WriteChildren<TChild>(TextGenerator writer, string region, IEnumerable<TChild> children)
            where TChild : IDescriptor
        {
            // Copy the set of children; makes access easier
            List<TChild> copy = new List<TChild>(children);
            if (copy.Count == 0)
            {
                return;
            }

            if (region != null)
            {
                writer.WriteLine("#region {0}", region);
            }
            foreach (TChild child in children)
            {
                SourceGenerators.CreateGenerator(child).Generate(writer);
            }
            if (region != null)
            {
                writer.WriteLine("#endregion");
                writer.WriteLine();
            }
        }
    }
}