blob: 9107def97bc4f0ba5b550c4dfa4cb45582dda315 [file] [log] [blame]
Jon Skeet68036862008-10-22 13:30:34 +01001using System;
2using 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) {
13 writer.WriteLine("private bool has{0};", CapitalizedName);
14 writer.WriteLine("private {0} {1}_ = {2};", TypeName, Name, DefaultValue);
15 writer.WriteLine("public bool Has{0} {{", CapitalizedName);
16 writer.WriteLine(" get {{ return has{0}; }}", CapitalizedName);
17 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) {
24 writer.WriteLine("public bool Has{0} {{", CapitalizedName);
25 writer.WriteLine(" get {{ return result.Has{0}; }}", CapitalizedName);
26 writer.WriteLine("}");
27 writer.WriteLine("public {0} {1} {{", TypeName, PropertyName);
28 writer.WriteLine(" get {{ return result.{0}; }}", PropertyName);
29 writer.WriteLine(" set {{ Set{0}(value); }}", CapitalizedName);
30 writer.WriteLine("}");
31 writer.WriteLine("public Builder Set{0}({1} value) {{", CapitalizedName, TypeName);
32 writer.WriteLine(" result.has{0} = true;", CapitalizedName);
33 writer.WriteLine(" result.{0}_ = value;", Name);
34 writer.WriteLine(" return this;");
35 writer.WriteLine("}");
36 writer.WriteLine("public Builder Clear{0}() {{", CapitalizedName);
37 writer.WriteLine(" result.has{0} = false;", CapitalizedName);
38 writer.WriteLine(" result.{0}_ = {1};", Name, DefaultValue);
39 writer.WriteLine(" return this;");
40 writer.WriteLine("}");
41 }
42
43 public void GenerateMergingCode(TextGenerator writer) {
44 writer.WriteLine("if (other.Has{0}) {{", CapitalizedName);
45 writer.WriteLine(" {0} = other.{0};", PropertyName);
46 writer.WriteLine("}");
47 }
48
49 public void GenerateBuildingCode(TextGenerator writer) {
50 // Nothing to do here for primitive types
51 }
52
53 public void GenerateParsingCode(TextGenerator writer) {
54 writer.WriteLine("{0} = input.Read{1}();", PropertyName, CapitalizedTypeName);
55 }
56
57 public void GenerateSerializationCode(TextGenerator writer) {
58 writer.WriteLine("if (Has{0}) {{", CapitalizedName);
59 writer.WriteLine(" output.Write{0}({1}, {2});", CapitalizedTypeName, Number, PropertyName);
60 writer.WriteLine("}");
61 }
62
63 public void GenerateSerializedSizeCode(TextGenerator writer) {
64 writer.WriteLine("if (Has{0}) {{", CapitalizedName);
65 writer.WriteLine(" size += pb::CodedOutputStream.Compute{0}Size({1}, {2});",
66 CapitalizedTypeName, Number, PropertyName);
67 writer.WriteLine("}");
68 }
69 }
70}