blob: 29d5a7392bda8aaff835fbd4507ddf95789096e9 [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("}");
Jon Skeetd6dd0a42009-06-05 22:00:05 +010018 AddClsComplianceCheck(writer);
Jon Skeet68036862008-10-22 13:30:34 +010019 writer.WriteLine("public {0} {1} {{", TypeName, PropertyName);
20 writer.WriteLine(" get {{ return {0}_; }}", Name);
21 writer.WriteLine("}");
22 }
23
24 public void GenerateBuilderMembers(TextGenerator writer) {
Jon Skeet4cf9e3c2008-11-24 11:11:28 +000025 writer.WriteLine("public bool Has{0} {{", PropertyName);
26 writer.WriteLine(" get {{ return result.Has{0}; }}", PropertyName);
Jon Skeet68036862008-10-22 13:30:34 +010027 writer.WriteLine("}");
Jon Skeetd6dd0a42009-06-05 22:00:05 +010028 AddClsComplianceCheck(writer);
Jon Skeet68036862008-10-22 13:30:34 +010029 writer.WriteLine("public {0} {1} {{", TypeName, PropertyName);
30 writer.WriteLine(" get {{ return result.{0}; }}", PropertyName);
Jon Skeet4cf9e3c2008-11-24 11:11:28 +000031 writer.WriteLine(" set {{ Set{0}(value); }}", PropertyName);
Jon Skeet68036862008-10-22 13:30:34 +010032 writer.WriteLine("}");
Jon Skeetd6dd0a42009-06-05 22:00:05 +010033 AddClsComplianceCheck(writer);
Jon Skeet4cf9e3c2008-11-24 11:11:28 +000034 writer.WriteLine("public Builder Set{0}({1} value) {{", PropertyName, TypeName);
Jon Skeet642a8142009-01-27 12:25:21 +000035 AddNullCheck(writer);
Jon Skeet4cf9e3c2008-11-24 11:11:28 +000036 writer.WriteLine(" result.has{0} = true;", PropertyName);
Jon Skeet68036862008-10-22 13:30:34 +010037 writer.WriteLine(" result.{0}_ = value;", Name);
38 writer.WriteLine(" return this;");
39 writer.WriteLine("}");
Jon Skeet4cf9e3c2008-11-24 11:11:28 +000040 writer.WriteLine("public Builder Clear{0}() {{", PropertyName);
41 writer.WriteLine(" result.has{0} = false;", PropertyName);
Jon Skeet68036862008-10-22 13:30:34 +010042 writer.WriteLine(" result.{0}_ = {1};", Name, DefaultValue);
43 writer.WriteLine(" return this;");
44 writer.WriteLine("}");
45 }
46
47 public void GenerateMergingCode(TextGenerator writer) {
Jon Skeet4cf9e3c2008-11-24 11:11:28 +000048 writer.WriteLine("if (other.Has{0}) {{", PropertyName);
Jon Skeet68036862008-10-22 13:30:34 +010049 writer.WriteLine(" {0} = other.{0};", PropertyName);
50 writer.WriteLine("}");
51 }
52
53 public void GenerateBuildingCode(TextGenerator writer) {
54 // Nothing to do here for primitive types
55 }
56
57 public void GenerateParsingCode(TextGenerator writer) {
58 writer.WriteLine("{0} = input.Read{1}();", PropertyName, CapitalizedTypeName);
59 }
60
61 public void GenerateSerializationCode(TextGenerator writer) {
Jon Skeet4cf9e3c2008-11-24 11:11:28 +000062 writer.WriteLine("if (Has{0}) {{", PropertyName);
Jon Skeet68036862008-10-22 13:30:34 +010063 writer.WriteLine(" output.Write{0}({1}, {2});", CapitalizedTypeName, Number, PropertyName);
64 writer.WriteLine("}");
65 }
66
67 public void GenerateSerializedSizeCode(TextGenerator writer) {
Jon Skeet4cf9e3c2008-11-24 11:11:28 +000068 writer.WriteLine("if (Has{0}) {{", PropertyName);
Jon Skeet68036862008-10-22 13:30:34 +010069 writer.WriteLine(" size += pb::CodedOutputStream.Compute{0}Size({1}, {2});",
70 CapitalizedTypeName, Number, PropertyName);
71 writer.WriteLine("}");
72 }
73 }
74}