blob: dc733c0b690ef231ad6048e084d54d63045a06d6 [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);
32 writer.WriteLine(" result.has{0} = true;", PropertyName);
Jon Skeet68036862008-10-22 13:30:34 +010033 writer.WriteLine(" result.{0}_ = value;", Name);
34 writer.WriteLine(" return this;");
35 writer.WriteLine("}");
Jon Skeet4cf9e3c2008-11-24 11:11:28 +000036 writer.WriteLine("public Builder Clear{0}() {{", PropertyName);
37 writer.WriteLine(" result.has{0} = false;", PropertyName);
Jon Skeet68036862008-10-22 13:30:34 +010038 writer.WriteLine(" result.{0}_ = {1};", Name, DefaultValue);
39 writer.WriteLine(" return this;");
40 writer.WriteLine("}");
41 }
42
43 public void GenerateMergingCode(TextGenerator writer) {
Jon Skeet4cf9e3c2008-11-24 11:11:28 +000044 writer.WriteLine("if (other.Has{0}) {{", PropertyName);
Jon Skeet68036862008-10-22 13:30:34 +010045 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) {
Jon Skeet4cf9e3c2008-11-24 11:11:28 +000058 writer.WriteLine("if (Has{0}) {{", PropertyName);
Jon Skeet68036862008-10-22 13:30:34 +010059 writer.WriteLine(" output.Write{0}({1}, {2});", CapitalizedTypeName, Number, PropertyName);
60 writer.WriteLine("}");
61 }
62
63 public void GenerateSerializedSizeCode(TextGenerator writer) {
Jon Skeet4cf9e3c2008-11-24 11:11:28 +000064 writer.WriteLine("if (Has{0}) {{", PropertyName);
Jon Skeet68036862008-10-22 13:30:34 +010065 writer.WriteLine(" size += pb::CodedOutputStream.Compute{0}Size({1}, {2});",
66 CapitalizedTypeName, Number, PropertyName);
67 writer.WriteLine("}");
68 }
69 }
70}