blob: 3c2fb246f2a23b5a627486e03bee07ad97dfcab5 [file] [log] [blame]
Jon Skeet60c059b2008-10-23 21:17:56 +01001using System;
Jon Skeet68036862008-10-22 13:30:34 +01002using Google.ProtocolBuffers.Descriptors;
3
4namespace Google.ProtocolBuffers.ProtoGen {
5 // TODO(jonskeet): Refactor this. There's loads of common code here.
6 internal class PrimitiveFieldGenerator : FieldGeneratorBase, IFieldSourceGenerator {
7
8 internal PrimitiveFieldGenerator(FieldDescriptor descriptor)
9 : base(descriptor) {
10 }
11
12 public void GenerateMembers(TextGenerator writer) {
Jon Skeet4cf9e3c2008-11-24 11:11:28 +000013 writer.WriteLine("private bool has{0};", PropertyName);
Jon Skeet68036862008-10-22 13:30:34 +010014 writer.WriteLine("private {0} {1}_ = {2};", TypeName, Name, DefaultValue);
Jon Skeet4cf9e3c2008-11-24 11:11:28 +000015 writer.WriteLine("public bool Has{0} {{", PropertyName);
16 writer.WriteLine(" get {{ return has{0}; }}", PropertyName);
Jon Skeet68036862008-10-22 13:30:34 +010017 writer.WriteLine("}");
18 writer.WriteLine("public {0} {1} {{", TypeName, PropertyName);
19 writer.WriteLine(" get {{ return {0}_; }}", Name);
20 writer.WriteLine("}");
21 }
22
23 public void GenerateBuilderMembers(TextGenerator writer) {
Jon Skeet4cf9e3c2008-11-24 11:11:28 +000024 writer.WriteLine("public bool Has{0} {{", PropertyName);
25 writer.WriteLine(" get {{ return result.Has{0}; }}", PropertyName);
Jon Skeet68036862008-10-22 13:30:34 +010026 writer.WriteLine("}");
27 writer.WriteLine("public {0} {1} {{", TypeName, PropertyName);
28 writer.WriteLine(" get {{ return result.{0}; }}", PropertyName);
Jon Skeet4cf9e3c2008-11-24 11:11:28 +000029 writer.WriteLine(" set {{ Set{0}(value); }}", PropertyName);
Jon Skeet68036862008-10-22 13:30:34 +010030 writer.WriteLine("}");
Jon Skeet4cf9e3c2008-11-24 11:11:28 +000031 writer.WriteLine("public Builder Set{0}({1} value) {{", PropertyName, TypeName);
Jon Skeet642a8142009-01-27 12:25:21 +000032 AddNullCheck(writer);
Jon Skeet4cf9e3c2008-11-24 11:11:28 +000033 writer.WriteLine(" result.has{0} = true;", PropertyName);
Jon Skeet68036862008-10-22 13:30:34 +010034 writer.WriteLine(" result.{0}_ = value;", Name);
35 writer.WriteLine(" return this;");
36 writer.WriteLine("}");
Jon Skeet4cf9e3c2008-11-24 11:11:28 +000037 writer.WriteLine("public Builder Clear{0}() {{", PropertyName);
38 writer.WriteLine(" result.has{0} = false;", PropertyName);
Jon Skeet68036862008-10-22 13:30:34 +010039 writer.WriteLine(" result.{0}_ = {1};", Name, DefaultValue);
40 writer.WriteLine(" return this;");
41 writer.WriteLine("}");
42 }
43
44 public void GenerateMergingCode(TextGenerator writer) {
Jon Skeet4cf9e3c2008-11-24 11:11:28 +000045 writer.WriteLine("if (other.Has{0}) {{", PropertyName);
Jon Skeet68036862008-10-22 13:30:34 +010046 writer.WriteLine(" {0} = other.{0};", PropertyName);
47 writer.WriteLine("}");
48 }
49
50 public void GenerateBuildingCode(TextGenerator writer) {
51 // Nothing to do here for primitive types
52 }
53
54 public void GenerateParsingCode(TextGenerator writer) {
55 writer.WriteLine("{0} = input.Read{1}();", PropertyName, CapitalizedTypeName);
56 }
57
58 public void GenerateSerializationCode(TextGenerator writer) {
Jon Skeet4cf9e3c2008-11-24 11:11:28 +000059 writer.WriteLine("if (Has{0}) {{", PropertyName);
Jon Skeet68036862008-10-22 13:30:34 +010060 writer.WriteLine(" output.Write{0}({1}, {2});", CapitalizedTypeName, Number, PropertyName);
61 writer.WriteLine("}");
62 }
63
64 public void GenerateSerializedSizeCode(TextGenerator writer) {
Jon Skeet4cf9e3c2008-11-24 11:11:28 +000065 writer.WriteLine("if (Has{0}) {{", PropertyName);
Jon Skeet68036862008-10-22 13:30:34 +010066 writer.WriteLine(" size += pb::CodedOutputStream.Compute{0}Size({1}, {2});",
67 CapitalizedTypeName, Number, PropertyName);
68 writer.WriteLine("}");
69 }
70 }
71}