diff options
Diffstat (limited to 'csharp/src/ProtoGen/MessageGenerator.cs')
| -rw-r--r-- | csharp/src/ProtoGen/MessageGenerator.cs | 893 |
1 files changed, 0 insertions, 893 deletions
diff --git a/csharp/src/ProtoGen/MessageGenerator.cs b/csharp/src/ProtoGen/MessageGenerator.cs deleted file mode 100644 index e7ed1e86..00000000 --- a/csharp/src/ProtoGen/MessageGenerator.cs +++ /dev/null @@ -1,893 +0,0 @@ -#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;
-using System.Collections.Generic;
-using Google.ProtocolBuffers.DescriptorProtos;
-using Google.ProtocolBuffers.Descriptors;
-
-namespace Google.ProtocolBuffers.ProtoGen
-{
- internal class MessageGenerator : SourceGeneratorBase<MessageDescriptor>, ISourceGenerator
- {
- private string[] _fieldNames;
-
- internal MessageGenerator(MessageDescriptor descriptor) : base(descriptor)
- {
- }
-
- private string ClassName
- {
- get { return Descriptor.Name; }
- }
-
- private string FullClassName
- {
- get { return GetClassName(Descriptor); }
- }
-
- /// <summary>
- /// Get an identifier that uniquely identifies this type within the file.
- /// This is used to declare static variables related to this type at the
- /// outermost file scope.
- /// </summary>
- private static string GetUniqueFileScopeIdentifier(IDescriptor descriptor)
- {
- return "static_" + descriptor.FullName.Replace(".", "_");
- }
-
- internal void GenerateStaticVariables(TextGenerator writer)
- {
- // Because descriptor.proto (Google.ProtocolBuffers.DescriptorProtos) is
- // used in the construction of descriptors, we have a tricky bootstrapping
- // problem. To help control static initialization order, we make sure all
- // descriptors and other static data that depends on them are members of
- // the proto-descriptor class. This way, they will be initialized in
- // a deterministic order.
-
- string identifier = GetUniqueFileScopeIdentifier(Descriptor);
-
- if (!UseLiteRuntime)
- {
- // The descriptor for this type.
- string access = Descriptor.File.CSharpOptions.NestClasses ? "private" : "internal";
- writer.WriteLine("{0} static pbd::MessageDescriptor internal__{1}__Descriptor;", access, identifier);
- writer.WriteLine(
- "{0} static pb::FieldAccess.FieldAccessorTable<{1}, {1}.Builder> internal__{2}__FieldAccessorTable;",
- access, FullClassName, identifier);
- }
- // Generate static members for all nested types.
- foreach (MessageDescriptor nestedMessage in Descriptor.NestedTypes)
- {
- new MessageGenerator(nestedMessage).GenerateStaticVariables(writer);
- }
- }
-
- internal void GenerateStaticVariableInitializers(TextGenerator writer)
- {
- string identifier = GetUniqueFileScopeIdentifier(Descriptor);
-
- if (!UseLiteRuntime)
- {
- writer.Write("internal__{0}__Descriptor = ", identifier);
- if (Descriptor.ContainingType == null)
- {
- writer.WriteLine("Descriptor.MessageTypes[{0}];", Descriptor.Index);
- }
- else
- {
- writer.WriteLine("internal__{0}__Descriptor.NestedTypes[{1}];",
- GetUniqueFileScopeIdentifier(Descriptor.ContainingType), Descriptor.Index);
- }
-
- writer.WriteLine("internal__{0}__FieldAccessorTable = ", identifier);
- writer.WriteLine(
- " new pb::FieldAccess.FieldAccessorTable<{1}, {1}.Builder>(internal__{0}__Descriptor,",
- identifier, FullClassName);
- writer.Print(" new string[] { ");
- foreach (FieldDescriptor field in Descriptor.Fields)
- {
- writer.Write("\"{0}\", ", field.CSharpOptions.PropertyName);
- }
- writer.WriteLine("});");
- }
-
- // Generate static member initializers for all nested types.
- foreach (MessageDescriptor nestedMessage in Descriptor.NestedTypes)
- {
- new MessageGenerator(nestedMessage).GenerateStaticVariableInitializers(writer);
- }
-
- foreach (FieldDescriptor extension in Descriptor.Extensions)
- {
- new ExtensionGenerator(extension).GenerateStaticVariableInitializers(writer);
- }
- }
-
- public string[] FieldNames
- {
- get
- {
- if (_fieldNames == null)
- {
- List<string> names = new List<string>();
- foreach (FieldDescriptor fieldDescriptor in Descriptor.Fields)
- {
- names.Add(fieldDescriptor.Name);
- }
- //if you change this, the search must also change in GenerateBuilderParsingMethods
- names.Sort(StringComparer.Ordinal);
- _fieldNames = names.ToArray();
- }
- return _fieldNames;
- }
- }
-
- internal int FieldOrdinal(FieldDescriptor field)
- {
- return Array.BinarySearch(FieldNames, field.Name, StringComparer.Ordinal);
- }
-
- private IFieldSourceGenerator CreateFieldGenerator(FieldDescriptor fieldDescriptor)
- {
- return SourceGenerators.CreateFieldGenerator(fieldDescriptor, FieldOrdinal(fieldDescriptor));
- }
-
- public void Generate(TextGenerator writer)
- {
- if (Descriptor.File.CSharpOptions.AddSerializable)
- {
- writer.WriteLine("[global::System.SerializableAttribute()]");
- }
- writer.WriteLine("[global::System.Diagnostics.DebuggerNonUserCodeAttribute()]");
- WriteGeneratedCodeAttributes(writer);
- writer.WriteLine("{0} sealed partial class {1} : pb::{2}Message{3}<{1}, {1}.Builder> {{",
- ClassAccessLevel, ClassName,
- Descriptor.Proto.ExtensionRangeCount > 0 ? "Extendable" : "Generated",
- RuntimeSuffix);
- writer.Indent();
- if (Descriptor.File.CSharpOptions.GeneratePrivateCtor)
- {
- writer.WriteLine("private {0}() {{ }}", ClassName);
- }
- // Must call MakeReadOnly() to make sure all lists are made read-only
- writer.WriteLine("private static readonly {0} defaultInstance = new {0}().MakeReadOnly();", ClassName);
-
- if (OptimizeSpeed)
- {
- writer.WriteLine("private static readonly string[] _{0}FieldNames = new string[] {{ {2}{1}{2} }};",
- NameHelpers.UnderscoresToCamelCase(ClassName), String.Join("\", \"", FieldNames),
- FieldNames.Length > 0 ? "\"" : "");
- List<string> tags = new List<string>();
- foreach (string name in FieldNames)
- {
- tags.Add(WireFormat.MakeTag(Descriptor.FindFieldByName(name)).ToString());
- }
-
- writer.WriteLine("private static readonly uint[] _{0}FieldTags = new uint[] {{ {1} }};",
- NameHelpers.UnderscoresToCamelCase(ClassName), String.Join(", ", tags.ToArray()));
- }
- writer.WriteLine("public static {0} DefaultInstance {{", ClassName);
- writer.WriteLine(" get { return defaultInstance; }");
- writer.WriteLine("}");
- writer.WriteLine();
- writer.WriteLine("public override {0} DefaultInstanceForType {{", ClassName);
- writer.WriteLine(" get { return DefaultInstance; }");
- writer.WriteLine("}");
- writer.WriteLine();
- writer.WriteLine("protected override {0} ThisMessage {{", ClassName);
- writer.WriteLine(" get { return this; }");
- writer.WriteLine("}");
- writer.WriteLine();
- if (!UseLiteRuntime)
- {
- writer.WriteLine("public static pbd::MessageDescriptor Descriptor {");
- writer.WriteLine(" get {{ return {0}.internal__{1}__Descriptor; }}",
- DescriptorUtil.GetFullUmbrellaClassName(Descriptor),
- GetUniqueFileScopeIdentifier(Descriptor));
- writer.WriteLine("}");
- writer.WriteLine();
- writer.WriteLine(
- "protected override pb::FieldAccess.FieldAccessorTable<{0}, {0}.Builder> InternalFieldAccessors {{",
- ClassName);
- writer.WriteLine(" get {{ return {0}.internal__{1}__FieldAccessorTable; }}",
- DescriptorUtil.GetFullUmbrellaClassName(Descriptor),
- GetUniqueFileScopeIdentifier(Descriptor));
- writer.WriteLine("}");
- writer.WriteLine();
- }
-
- // Extensions don't need to go in an extra nested type
- WriteChildren(writer, null, Descriptor.Extensions);
-
- if (Descriptor.EnumTypes.Count + Descriptor.NestedTypes.Count > 0)
- {
- writer.WriteLine("#region Nested types");
- writer.WriteLine("[global::System.Diagnostics.DebuggerNonUserCodeAttribute()]");
- WriteGeneratedCodeAttributes(writer);
- writer.WriteLine("public static partial class Types {");
- writer.Indent();
- WriteChildren(writer, null, Descriptor.EnumTypes);
- WriteChildren(writer, null, Descriptor.NestedTypes);
- writer.Outdent();
- writer.WriteLine("}");
- writer.WriteLine("#endregion");
- writer.WriteLine();
- }
-
- foreach (FieldDescriptor fieldDescriptor in Descriptor.Fields)
- {
- if (Descriptor.File.CSharpOptions.ClsCompliance && GetFieldConstantName(fieldDescriptor).StartsWith("_"))
- {
- writer.WriteLine("[global::System.CLSCompliant(false)]");
- }
-
- // Rats: we lose the debug comment here :(
- writer.WriteLine("public const int {0} = {1};", GetFieldConstantName(fieldDescriptor),
- fieldDescriptor.FieldNumber);
- CreateFieldGenerator(fieldDescriptor).GenerateMembers(writer);
- writer.WriteLine();
- }
-
- if (OptimizeSpeed)
- {
- GenerateIsInitialized(writer);
- GenerateMessageSerializationMethods(writer);
- }
- if (UseLiteRuntime)
- {
- GenerateLiteRuntimeMethods(writer);
- }
-
- GenerateParseFromMethods(writer);
- GenerateBuilder(writer);
-
- // Force the static initialization code for the file to run, since it may
- // initialize static variables declared in this class.
- writer.WriteLine("static {0}() {{", ClassName);
- // We call object.ReferenceEquals() just to make it a valid statement on its own.
- // Another option would be GetType(), but that causes problems in DescriptorProtoFile,
- // where the bootstrapping is somewhat recursive - type initializers call
- // each other, effectively. We temporarily see Descriptor as null.
- writer.WriteLine(" object.ReferenceEquals({0}.Descriptor, null);",
- DescriptorUtil.GetFullUmbrellaClassName(Descriptor));
- writer.WriteLine("}");
-
- writer.Outdent();
- writer.WriteLine("}");
- writer.WriteLine();
- }
-
- private void GenerateLiteRuntimeMethods(TextGenerator writer)
- {
- bool callbase = Descriptor.Proto.ExtensionRangeCount > 0;
- writer.WriteLine("#region Lite runtime methods");
- writer.WriteLine("public override int GetHashCode() {");
- writer.Indent();
- writer.WriteLine("int hash = GetType().GetHashCode();");
- foreach (FieldDescriptor fieldDescriptor in Descriptor.Fields)
- {
- CreateFieldGenerator(fieldDescriptor).WriteHash(writer);
- }
- if (callbase)
- {
- writer.WriteLine("hash ^= base.GetHashCode();");
- }
- writer.WriteLine("return hash;");
- writer.Outdent();
- writer.WriteLine("}");
- writer.WriteLine();
-
- writer.WriteLine("public override bool Equals(object obj) {");
- writer.Indent();
- writer.WriteLine("{0} other = obj as {0};", ClassName);
- writer.WriteLine("if (other == null) return false;");
- foreach (FieldDescriptor fieldDescriptor in Descriptor.Fields)
- {
- CreateFieldGenerator(fieldDescriptor).WriteEquals(writer);
- }
- if (callbase)
- {
- writer.WriteLine("if (!base.Equals(other)) return false;");
- }
- writer.WriteLine("return true;");
- writer.Outdent();
- writer.WriteLine("}");
- writer.WriteLine();
-
- writer.WriteLine("public override void PrintTo(global::System.IO.TextWriter writer) {");
- writer.Indent();
- List<FieldDescriptor> sorted = new List<FieldDescriptor>(Descriptor.Fields);
- sorted.Sort(
- new Comparison<FieldDescriptor>(
- delegate(FieldDescriptor a, FieldDescriptor b) { return a.FieldNumber.CompareTo(b.FieldNumber); }));
- foreach (FieldDescriptor fieldDescriptor in sorted)
- {
- CreateFieldGenerator(fieldDescriptor).WriteToString(writer);
- }
- if (callbase)
- {
- writer.WriteLine("base.PrintTo(writer);");
- }
- writer.Outdent();
- writer.WriteLine("}");
- writer.WriteLine("#endregion");
- writer.WriteLine();
- }
-
- private void GenerateMessageSerializationMethods(TextGenerator writer)
- {
- List<FieldDescriptor> sortedFields = new List<FieldDescriptor>(Descriptor.Fields);
- sortedFields.Sort((f1, f2) => f1.FieldNumber.CompareTo(f2.FieldNumber));
-
- List<DescriptorProto.Types.ExtensionRange> sortedExtensions =
- new List<DescriptorProto.Types.ExtensionRange>(Descriptor.Proto.ExtensionRangeList);
- sortedExtensions.Sort((r1, r2) => (r1.Start.CompareTo(r2.Start)));
-
- writer.WriteLine("public override void WriteTo(pb::ICodedOutputStream output) {");
- writer.Indent();
- // Make sure we've computed the serialized length, so that packed fields are generated correctly.
- writer.WriteLine("CalcSerializedSize();");
- writer.WriteLine("string[] field_names = _{0}FieldNames;", NameHelpers.UnderscoresToCamelCase(ClassName));
- if (Descriptor.Proto.ExtensionRangeList.Count > 0)
- {
- writer.WriteLine(
- "pb::ExtendableMessage{1}<{0}, {0}.Builder>.ExtensionWriter extensionWriter = CreateExtensionWriter(this);",
- ClassName, RuntimeSuffix);
- }
-
- // Merge the fields and the extension ranges, both sorted by field number.
- for (int i = 0, j = 0; i < Descriptor.Fields.Count || j < sortedExtensions.Count;)
- {
- if (i == Descriptor.Fields.Count)
- {
- GenerateSerializeOneExtensionRange(writer, sortedExtensions[j++]);
- }
- else if (j == sortedExtensions.Count)
- {
- GenerateSerializeOneField(writer, sortedFields[i++]);
- }
- else if (sortedFields[i].FieldNumber < sortedExtensions[j].Start)
- {
- GenerateSerializeOneField(writer, sortedFields[i++]);
- }
- else
- {
- GenerateSerializeOneExtensionRange(writer, sortedExtensions[j++]);
- }
- }
-
- if (!UseLiteRuntime)
- {
- if (Descriptor.Proto.Options.MessageSetWireFormat)
- {
- writer.WriteLine("UnknownFields.WriteAsMessageSetTo(output);");
- }
- else
- {
- writer.WriteLine("UnknownFields.WriteTo(output);");
- }
- }
-
- writer.Outdent();
- writer.WriteLine("}");
- writer.WriteLine();
- writer.WriteLine("private int memoizedSerializedSize = -1;");
- writer.WriteLine("public override int SerializedSize {");
- writer.Indent();
- writer.WriteLine("get {");
- writer.Indent();
- writer.WriteLine("int size = memoizedSerializedSize;");
- writer.WriteLine("if (size != -1) return size;");
- writer.WriteLine("return CalcSerializedSize();");
- writer.Outdent();
- writer.WriteLine("}");
- writer.Outdent();
- writer.WriteLine("}");
- writer.WriteLine();
-
- writer.WriteLine("private int CalcSerializedSize() {");
- writer.Indent();
- writer.WriteLine("int size = memoizedSerializedSize;");
- writer.WriteLine("if (size != -1) return size;");
- writer.WriteLine();
- writer.WriteLine("size = 0;");
- foreach (FieldDescriptor field in Descriptor.Fields)
- {
- CreateFieldGenerator(field).GenerateSerializedSizeCode(writer);
- }
- if (Descriptor.Proto.ExtensionRangeCount > 0)
- {
- writer.WriteLine("size += ExtensionsSerializedSize;");
- }
-
- if (!UseLiteRuntime)
- {
- if (Descriptor.Options.MessageSetWireFormat)
- {
- writer.WriteLine("size += UnknownFields.SerializedSizeAsMessageSet;");
- }
- else
- {
- writer.WriteLine("size += UnknownFields.SerializedSize;");
- }
- }
- writer.WriteLine("memoizedSerializedSize = size;");
- writer.WriteLine("return size;");
- writer.Outdent();
- writer.WriteLine("}");
- }
-
- private void GenerateSerializeOneField(TextGenerator writer, FieldDescriptor fieldDescriptor)
- {
- CreateFieldGenerator(fieldDescriptor).GenerateSerializationCode(writer);
- }
-
- private static void GenerateSerializeOneExtensionRange(TextGenerator writer,
- DescriptorProto.Types.ExtensionRange extensionRange)
- {
- writer.WriteLine("extensionWriter.WriteUntil({0}, output);", extensionRange.End);
- }
-
- private void GenerateParseFromMethods(TextGenerator writer)
- {
- // Note: These are separate from GenerateMessageSerializationMethods()
- // because they need to be generated even for messages that are optimized
- // for code size.
-
- writer.WriteLine("public static {0} ParseFrom(pb::ByteString data) {{", ClassName);
- writer.WriteLine(" return ((Builder) CreateBuilder().MergeFrom(data)).BuildParsed();");
- writer.WriteLine("}");
- writer.WriteLine(
- "public static {0} ParseFrom(pb::ByteString data, pb::ExtensionRegistry extensionRegistry) {{",
- ClassName);
- writer.WriteLine(" return ((Builder) CreateBuilder().MergeFrom(data, extensionRegistry)).BuildParsed();");
- writer.WriteLine("}");
- writer.WriteLine("public static {0} ParseFrom(byte[] data) {{", ClassName);
- writer.WriteLine(" return ((Builder) CreateBuilder().MergeFrom(data)).BuildParsed();");
- writer.WriteLine("}");
- writer.WriteLine("public static {0} ParseFrom(byte[] data, pb::ExtensionRegistry extensionRegistry) {{",
- ClassName);
- writer.WriteLine(" return ((Builder) CreateBuilder().MergeFrom(data, extensionRegistry)).BuildParsed();");
- writer.WriteLine("}");
- writer.WriteLine("public static {0} ParseFrom(global::System.IO.Stream input) {{", ClassName);
- writer.WriteLine(" return ((Builder) CreateBuilder().MergeFrom(input)).BuildParsed();");
- writer.WriteLine("}");
- writer.WriteLine(
- "public static {0} ParseFrom(global::System.IO.Stream input, pb::ExtensionRegistry extensionRegistry) {{",
- ClassName);
- writer.WriteLine(" return ((Builder) CreateBuilder().MergeFrom(input, extensionRegistry)).BuildParsed();");
- writer.WriteLine("}");
- writer.WriteLine("public static {0} ParseDelimitedFrom(global::System.IO.Stream input) {{", ClassName);
- writer.WriteLine(" return CreateBuilder().MergeDelimitedFrom(input).BuildParsed();");
- writer.WriteLine("}");
- writer.WriteLine(
- "public static {0} ParseDelimitedFrom(global::System.IO.Stream input, pb::ExtensionRegistry extensionRegistry) {{",
- ClassName);
- writer.WriteLine(" return CreateBuilder().MergeDelimitedFrom(input, extensionRegistry).BuildParsed();");
- writer.WriteLine("}");
- writer.WriteLine("public static {0} ParseFrom(pb::ICodedInputStream input) {{", ClassName);
- writer.WriteLine(" return ((Builder) CreateBuilder().MergeFrom(input)).BuildParsed();");
- writer.WriteLine("}");
- writer.WriteLine(
- "public static {0} ParseFrom(pb::ICodedInputStream input, pb::ExtensionRegistry extensionRegistry) {{",
- ClassName);
- writer.WriteLine(" return ((Builder) CreateBuilder().MergeFrom(input, extensionRegistry)).BuildParsed();");
- writer.WriteLine("}");
- }
-
- /// <summary>
- /// Returns whether or not the specified message type has any required fields.
- /// If it doesn't, calls to check for initialization can be optimised.
- /// TODO(jonskeet): Move this into MessageDescriptor?
- /// </summary>
- private static bool HasRequiredFields(MessageDescriptor descriptor,
- Dictionary<MessageDescriptor, object> alreadySeen)
- {
- if (alreadySeen.ContainsKey(descriptor))
- {
- // The type is already in cache. This means that either:
- // a. The type has no required fields.
- // b. We are in the midst of checking if the type has required fields,
- // somewhere up the stack. In this case, we know that if the type
- // has any required fields, they'll be found when we return to it,
- // and the whole call to HasRequiredFields() will return true.
- // Therefore, we don't have to check if this type has required fields
- // here.
- return false;
- }
- alreadySeen[descriptor] = descriptor; // Value is irrelevant
-
- // If the type has extensions, an extension with message type could contain
- // required fields, so we have to be conservative and assume such an
- // extension exists.
- if (descriptor.Extensions.Count > 0)
- {
- return true;
- }
-
- foreach (FieldDescriptor field in descriptor.Fields)
- {
- if (field.IsRequired)
- {
- return true;
- }
- // Message or group
- if (field.MappedType == MappedType.Message)
- {
- if (HasRequiredFields(field.MessageType, alreadySeen))
- {
- return true;
- }
- }
- }
- return false;
- }
-
- private void GenerateBuilder(TextGenerator writer)
- {
- writer.WriteLine("private {0} MakeReadOnly() {{", ClassName);
- writer.Indent();
- foreach (FieldDescriptor field in Descriptor.Fields)
- {
- CreateFieldGenerator(field).GenerateBuildingCode(writer);
- }
- writer.WriteLine("return this;");
- writer.Outdent();
- writer.WriteLine("}");
- writer.WriteLine();
-
- writer.WriteLine("public static Builder CreateBuilder() { return new Builder(); }");
- writer.WriteLine("public override Builder ToBuilder() { return CreateBuilder(this); }");
- writer.WriteLine("public override Builder CreateBuilderForType() { return new Builder(); }");
- writer.WriteLine("public static Builder CreateBuilder({0} prototype) {{", ClassName);
- writer.WriteLine(" return new Builder(prototype);");
- writer.WriteLine("}");
- writer.WriteLine();
- if (Descriptor.File.CSharpOptions.AddSerializable)
- {
- writer.WriteLine("[global::System.SerializableAttribute()]");
- }
- writer.WriteLine("[global::System.Diagnostics.DebuggerNonUserCodeAttribute()]");
- WriteGeneratedCodeAttributes(writer);
- writer.WriteLine("{0} sealed partial class Builder : pb::{2}Builder{3}<{1}, Builder> {{",
- ClassAccessLevel, ClassName,
- Descriptor.Proto.ExtensionRangeCount > 0 ? "Extendable" : "Generated", RuntimeSuffix);
- writer.Indent();
- writer.WriteLine("protected override Builder ThisBuilder {");
- writer.WriteLine(" get { return this; }");
- writer.WriteLine("}");
- GenerateCommonBuilderMethods(writer);
- if (OptimizeSpeed)
- {
- GenerateBuilderParsingMethods(writer);
- }
- foreach (FieldDescriptor field in Descriptor.Fields)
- {
- writer.WriteLine();
- // No field comment :(
- CreateFieldGenerator(field).GenerateBuilderMembers(writer);
- }
- writer.Outdent();
- writer.WriteLine("}");
- }
-
- private void GenerateCommonBuilderMethods(TextGenerator writer)
- {
- //default constructor
- writer.WriteLine("public Builder() {");
- //Durring static initialization of message, DefaultInstance is expected to return null.
- writer.WriteLine(" result = DefaultInstance;");
- writer.WriteLine(" resultIsReadOnly = true;");
- writer.WriteLine("}");
- //clone constructor
- writer.WriteLine("internal Builder({0} cloneFrom) {{", ClassName);
- writer.WriteLine(" result = cloneFrom;");
- writer.WriteLine(" resultIsReadOnly = true;");
- writer.WriteLine("}");
- writer.WriteLine();
- writer.WriteLine("private bool resultIsReadOnly;");
- writer.WriteLine("private {0} result;", ClassName);
- writer.WriteLine();
- writer.WriteLine("private {0} PrepareBuilder() {{", ClassName);
- writer.WriteLine(" if (resultIsReadOnly) {");
- writer.WriteLine(" {0} original = result;", ClassName);
- writer.WriteLine(" result = new {0}();", ClassName);
- writer.WriteLine(" resultIsReadOnly = false;");
- writer.WriteLine(" MergeFrom(original);");
- writer.WriteLine(" }");
- writer.WriteLine(" return result;");
- writer.WriteLine("}");
- writer.WriteLine();
- writer.WriteLine("public override bool IsInitialized {");
- writer.WriteLine(" get { return result.IsInitialized; }");
- writer.WriteLine("}");
- writer.WriteLine();
- writer.WriteLine("protected override {0} MessageBeingBuilt {{", ClassName);
- writer.WriteLine(" get { return PrepareBuilder(); }");
- writer.WriteLine("}");
- writer.WriteLine();
- //Not actually expecting that DefaultInstance would ever be null here; however, we will ensure it does not break
- writer.WriteLine("public override Builder Clear() {");
- writer.WriteLine(" result = DefaultInstance;", ClassName);
- writer.WriteLine(" resultIsReadOnly = true;");
- writer.WriteLine(" return this;");
- writer.WriteLine("}");
- writer.WriteLine();
- writer.WriteLine("public override Builder Clone() {");
- writer.WriteLine(" if (resultIsReadOnly) {");
- writer.WriteLine(" return new Builder(result);");
- writer.WriteLine(" } else {");
- writer.WriteLine(" return new Builder().MergeFrom(result);");
- writer.WriteLine(" }");
- writer.WriteLine("}");
- writer.WriteLine();
- if (!UseLiteRuntime)
- {
- writer.WriteLine("public override pbd::MessageDescriptor DescriptorForType {");
- writer.WriteLine(" get {{ return {0}.Descriptor; }}", FullClassName);
- writer.WriteLine("}");
- writer.WriteLine();
- }
- writer.WriteLine("public override {0} DefaultInstanceForType {{", ClassName);
- writer.WriteLine(" get {{ return {0}.DefaultInstance; }}", FullClassName);
- writer.WriteLine("}");
- writer.WriteLine();
-
- writer.WriteLine("public override {0} BuildPartial() {{", ClassName);
- writer.Indent();
- writer.WriteLine("if (resultIsReadOnly) {");
- writer.WriteLine(" return result;");
- writer.WriteLine("}");
- writer.WriteLine("resultIsReadOnly = true;");
- writer.WriteLine("return result.MakeReadOnly();");
- writer.Outdent();
- writer.WriteLine("}");
- writer.WriteLine();
-
- if (OptimizeSpeed)
- {
- writer.WriteLine("public override Builder MergeFrom(pb::IMessage{0} other) {{", RuntimeSuffix);
- writer.WriteLine(" if (other is {0}) {{", ClassName);
- writer.WriteLine(" return MergeFrom(({0}) other);", ClassName);
- writer.WriteLine(" } else {");
- writer.WriteLine(" base.MergeFrom(other);");
- writer.WriteLine(" return this;");
- writer.WriteLine(" }");
- writer.WriteLine("}");
- writer.WriteLine();
- writer.WriteLine("public override Builder MergeFrom({0} other) {{", ClassName);
- // Optimization: If other is the default instance, we know none of its
- // fields are set so we can skip the merge.
- writer.Indent();
- writer.WriteLine("if (other == {0}.DefaultInstance) return this;", FullClassName);
- writer.WriteLine("PrepareBuilder();");
- foreach (FieldDescriptor field in Descriptor.Fields)
- {
- CreateFieldGenerator(field).GenerateMergingCode(writer);
- }
- // if message type has extensions
- if (Descriptor.Proto.ExtensionRangeCount > 0)
- {
- writer.WriteLine(" this.MergeExtensionFields(other);");
- }
- if (!UseLiteRuntime)
- {
- writer.WriteLine("this.MergeUnknownFields(other.UnknownFields);");
- }
- writer.WriteLine("return this;");
- writer.Outdent();
- writer.WriteLine("}");
- writer.WriteLine();
- }
- }
-
- private void GenerateBuilderParsingMethods(TextGenerator writer)
- {
- List<FieldDescriptor> sortedFields = new List<FieldDescriptor>(Descriptor.Fields);
- sortedFields.Sort((f1, f2) => f1.FieldNumber.CompareTo(f2.FieldNumber));
-
- writer.WriteLine("public override Builder MergeFrom(pb::ICodedInputStream input) {");
- writer.WriteLine(" return MergeFrom(input, pb::ExtensionRegistry.Empty);");
- writer.WriteLine("}");
- writer.WriteLine();
- writer.WriteLine(
- "public override Builder MergeFrom(pb::ICodedInputStream input, pb::ExtensionRegistry extensionRegistry) {");
- writer.Indent();
- writer.WriteLine("PrepareBuilder();");
- if (!UseLiteRuntime)
- {
- writer.WriteLine("pb::UnknownFieldSet.Builder unknownFields = null;");
- }
- writer.WriteLine("uint tag;");
- writer.WriteLine("string field_name;");
- writer.WriteLine("while (input.ReadTag(out tag, out field_name)) {");
- writer.Indent();
- writer.WriteLine("if(tag == 0 && field_name != null) {");
- writer.Indent();
- //if you change from StringComparer.Ordinal, the array sort in FieldNames { get; } must also change
- writer.WriteLine(
- "int field_ordinal = global::System.Array.BinarySearch(_{0}FieldNames, field_name, global::System.StringComparer.Ordinal);",
- NameHelpers.UnderscoresToCamelCase(ClassName));
- writer.WriteLine("if(field_ordinal >= 0)");
- writer.WriteLine(" tag = _{0}FieldTags[field_ordinal];", NameHelpers.UnderscoresToCamelCase(ClassName));
- writer.WriteLine("else {");
- if (!UseLiteRuntime)
- {
- writer.WriteLine(" if (unknownFields == null) {"); // First unknown field - create builder now
- writer.WriteLine(" unknownFields = pb::UnknownFieldSet.CreateBuilder(this.UnknownFields);");
- writer.WriteLine(" }");
- }
- writer.WriteLine(" ParseUnknownField(input, {0}extensionRegistry, tag, field_name);",
- UseLiteRuntime ? "" : "unknownFields, ");
- writer.WriteLine(" continue;");
- writer.WriteLine("}");
- writer.Outdent();
- writer.WriteLine("}");
-
- writer.WriteLine("switch (tag) {");
- writer.Indent();
- writer.WriteLine("case 0: {"); // 0 signals EOF / limit reached
- writer.WriteLine(" throw pb::InvalidProtocolBufferException.InvalidTag();");
- writer.WriteLine("}");
- writer.WriteLine("default: {");
- writer.WriteLine(" if (pb::WireFormat.IsEndGroupTag(tag)) {");
- if (!UseLiteRuntime)
- {
- writer.WriteLine(" if (unknownFields != null) {");
- writer.WriteLine(" this.UnknownFields = unknownFields.Build();");
- writer.WriteLine(" }");
- }
- writer.WriteLine(" return this;"); // it's an endgroup tag
- writer.WriteLine(" }");
- if (!UseLiteRuntime)
- {
- writer.WriteLine(" if (unknownFields == null) {"); // First unknown field - create builder now
- writer.WriteLine(" unknownFields = pb::UnknownFieldSet.CreateBuilder(this.UnknownFields);");
- writer.WriteLine(" }");
- }
- writer.WriteLine(" ParseUnknownField(input, {0}extensionRegistry, tag, field_name);",
- UseLiteRuntime ? "" : "unknownFields, ");
- writer.WriteLine(" break;");
- writer.WriteLine("}");
- foreach (FieldDescriptor field in sortedFields)
- {
- WireFormat.WireType wt = WireFormat.GetWireType(field.FieldType);
- uint tag = WireFormat.MakeTag(field.FieldNumber, wt);
-
- if (field.IsRepeated &&
- (wt == WireFormat.WireType.Varint || wt == WireFormat.WireType.Fixed32 ||
- wt == WireFormat.WireType.Fixed64))
- {
- writer.WriteLine("case {0}:",
- WireFormat.MakeTag(field.FieldNumber, WireFormat.WireType.LengthDelimited));
- }
-
- writer.WriteLine("case {0}: {{", tag);
- writer.Indent();
- CreateFieldGenerator(field).GenerateParsingCode(writer);
- writer.WriteLine("break;");
- writer.Outdent();
- writer.WriteLine("}");
- }
- writer.Outdent();
- writer.WriteLine("}");
- writer.Outdent();
- writer.WriteLine("}");
- writer.WriteLine();
- if (!UseLiteRuntime)
- {
- writer.WriteLine("if (unknownFields != null) {");
- writer.WriteLine(" this.UnknownFields = unknownFields.Build();");
- writer.WriteLine("}");
- }
- writer.WriteLine("return this;");
- writer.Outdent();
- writer.WriteLine("}");
- writer.WriteLine();
- }
-
- private void GenerateIsInitialized(TextGenerator writer)
- {
- writer.WriteLine("public override bool IsInitialized {");
- writer.Indent();
- writer.WriteLine("get {");
- writer.Indent();
-
- // Check that all required fields in this message are set.
- // TODO(kenton): We can optimize this when we switch to putting all the
- // "has" fields into a single bitfield.
- foreach (FieldDescriptor field in Descriptor.Fields)
- {
- if (field.IsRequired)
- {
- writer.WriteLine("if (!has{0}) return false;", field.CSharpOptions.PropertyName);
- }
- }
-
- // Now check that all embedded messages are initialized.
- foreach (FieldDescriptor field in Descriptor.Fields)
- {
- if (field.FieldType != FieldType.Message ||
- !HasRequiredFields(field.MessageType, new Dictionary<MessageDescriptor, object>()))
- {
- continue;
- }
- string propertyName = NameHelpers.UnderscoresToPascalCase(GetFieldName(field));
- if (field.IsRepeated)
- {
- writer.WriteLine("foreach ({0} element in {1}List) {{", GetClassName(field.MessageType),
- propertyName);
- writer.WriteLine(" if (!element.IsInitialized) return false;");
- writer.WriteLine("}");
- }
- else if (field.IsOptional)
- {
- writer.WriteLine("if (Has{0}) {{", propertyName);
- writer.WriteLine(" if (!{0}.IsInitialized) return false;", propertyName);
- writer.WriteLine("}");
- }
- else
- {
- writer.WriteLine("if (!{0}.IsInitialized) return false;", propertyName);
- }
- }
-
- if (Descriptor.Proto.ExtensionRangeCount > 0)
- {
- writer.WriteLine("if (!ExtensionsAreInitialized) return false;");
- }
- writer.WriteLine("return true;");
- writer.Outdent();
- writer.WriteLine("}");
- writer.Outdent();
- writer.WriteLine("}");
- writer.WriteLine();
- }
-
- internal void GenerateExtensionRegistrationCode(TextGenerator writer)
- {
- foreach (FieldDescriptor extension in Descriptor.Extensions)
- {
- new ExtensionGenerator(extension).GenerateExtensionRegistrationCode(writer);
- }
- foreach (MessageDescriptor nestedMessage in Descriptor.NestedTypes)
- {
- new MessageGenerator(nestedMessage).GenerateExtensionRegistrationCode(writer);
- }
- }
- }
-}
\ No newline at end of file |