aboutsummaryrefslogtreecommitdiff
path: root/src/ProtoGen/RepeatedPrimitiveFieldGenerator.cs
diff options
context:
space:
mode:
Diffstat (limited to 'src/ProtoGen/RepeatedPrimitiveFieldGenerator.cs')
-rw-r--r--src/ProtoGen/RepeatedPrimitiveFieldGenerator.cs84
1 files changed, 84 insertions, 0 deletions
diff --git a/src/ProtoGen/RepeatedPrimitiveFieldGenerator.cs b/src/ProtoGen/RepeatedPrimitiveFieldGenerator.cs
new file mode 100644
index 00000000..cdb7cd6f
--- /dev/null
+++ b/src/ProtoGen/RepeatedPrimitiveFieldGenerator.cs
@@ -0,0 +1,84 @@
+using System;
+using System.Collections.Generic;
+using System.Text;
+using Google.ProtocolBuffers.Descriptors;
+
+namespace Google.ProtocolBuffers.ProtoGen {
+ internal class RepeatedPrimitiveFieldGenerator : FieldGeneratorBase, IFieldSourceGenerator {
+
+ internal RepeatedPrimitiveFieldGenerator(FieldDescriptor descriptor)
+ : base(descriptor) {
+ }
+
+ public void GenerateMembers(TextGenerator writer) {
+ writer.WriteLine("private pbc::PopsicleList<{0}> {1}_ = new pbc::PopsicleList<{0}>();", TypeName, Name);
+ writer.WriteLine("public scg::IList<{0}> {1}List {{", TypeName, CapitalizedName);
+ writer.WriteLine(" get {{ return pbc::Lists.AsReadOnly({0}_); }}", Name);
+ writer.WriteLine("}");
+
+ // TODO(jonskeet): Redundant API calls? Possibly - include for portability though. Maybe create an option.
+ writer.WriteLine("public int {0}Count {{", CapitalizedName);
+ writer.WriteLine(" get {{ return {0}_.Count; }}", Name);
+ writer.WriteLine("}");
+
+ writer.WriteLine("public {0} Get{1}(int index) {{", TypeName, CapitalizedName);
+ writer.WriteLine(" return {0}_[index];", Name);
+ writer.WriteLine("}");
+ }
+
+ public void GenerateBuilderMembers(TextGenerator writer) {
+ // Note: We can return the original list here, because we make it unmodifiable when we build
+ writer.WriteLine("public scg::IList<{0}> {1}List {{", TypeName, CapitalizedName);
+ writer.WriteLine(" get {{ return result.{0}_; }}", Name);
+ writer.WriteLine("}");
+ writer.WriteLine("public int {0}Count {{", CapitalizedName);
+ writer.WriteLine(" get {{ return result.{0}Count; }}", CapitalizedName);
+ writer.WriteLine("}");
+ writer.WriteLine("public {0} Get{1}(int index) {{", TypeName, CapitalizedName);
+ writer.WriteLine(" return result.Get{0}(index);", CapitalizedName);
+ writer.WriteLine("}");
+ writer.WriteLine("public Builder Set{0}(int index, {1} value) {{", CapitalizedName, TypeName);
+ writer.WriteLine(" result.{0}_[index] = value;", Name);
+ writer.WriteLine(" return this;");
+ writer.WriteLine("}");
+ writer.WriteLine("public Builder Add{0}({1} value) {{", CapitalizedName, TypeName);
+ writer.WriteLine(" result.{0}_.Add(value);", Name, TypeName);
+ writer.WriteLine(" return this;");
+ writer.WriteLine("}");
+ writer.WriteLine("public Builder AddRange{0}(scg::IEnumerable<{1}> values) {{", CapitalizedName, TypeName);
+ writer.WriteLine(" base.AddRange(values, result.{0}_);", Name);
+ writer.WriteLine(" return this;");
+ writer.WriteLine("}");
+ writer.WriteLine("public Builder Clear{0}() {{", CapitalizedName);
+ writer.WriteLine(" result.{0}_.Clear();", Name);
+ writer.WriteLine(" return this;");
+ writer.WriteLine("}");
+ }
+
+ public void GenerateMergingCode(TextGenerator writer) {
+ writer.WriteLine("if (other.{0}_.Count != 0) {{", Name);
+ writer.WriteLine(" base.AddRange(other.{0}_, result.{0}_);", Name);
+ writer.WriteLine("}");
+ }
+
+ public void GenerateBuildingCode(TextGenerator writer) {
+ writer.WriteLine("result.{0}_.MakeReadOnly();", Name);
+ }
+
+ public void GenerateParsingCode(TextGenerator writer) {
+ writer.WriteLine("Add{0}(input.Read{1}());", CapitalizedName, CapitalizedTypeName);
+ }
+
+ public void GenerateSerializationCode(TextGenerator writer) {
+ writer.WriteLine("foreach ({0} element in {1}List) {{", TypeName, CapitalizedName);
+ writer.WriteLine(" output.Write{0}({1}, element);", CapitalizedTypeName, Number);
+ writer.WriteLine("}");
+ }
+
+ public void GenerateSerializedSizeCode(TextGenerator writer) {
+ writer.WriteLine("foreach ({0} element in {1}List) {{", TypeName, CapitalizedName);
+ writer.WriteLine(" size += pb::CodedOutputStream.Compute{0}Size({1}, element);", CapitalizedTypeName, Number);
+ writer.WriteLine("}");
+ }
+ }
+}