Jon Skeet | 60c059b | 2008-10-23 21:17:56 +0100 | [diff] [blame] | 1 | using System; |
Jon Skeet | 6803686 | 2008-10-22 13:30:34 +0100 | [diff] [blame] | 2 | using Google.ProtocolBuffers.Descriptors; |
| 3 | |
| 4 | namespace 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 Skeet | 4cf9e3c | 2008-11-24 11:11:28 +0000 | [diff] [blame^] | 13 | writer.WriteLine("private bool has{0};", PropertyName); |
Jon Skeet | 6803686 | 2008-10-22 13:30:34 +0100 | [diff] [blame] | 14 | writer.WriteLine("private {0} {1}_ = {2};", TypeName, Name, DefaultValue); |
Jon Skeet | 4cf9e3c | 2008-11-24 11:11:28 +0000 | [diff] [blame^] | 15 | writer.WriteLine("public bool Has{0} {{", PropertyName); |
| 16 | writer.WriteLine(" get {{ return has{0}; }}", PropertyName); |
Jon Skeet | 6803686 | 2008-10-22 13:30:34 +0100 | [diff] [blame] | 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) { |
Jon Skeet | 4cf9e3c | 2008-11-24 11:11:28 +0000 | [diff] [blame^] | 24 | writer.WriteLine("public bool Has{0} {{", PropertyName); |
| 25 | writer.WriteLine(" get {{ return result.Has{0}; }}", PropertyName); |
Jon Skeet | 6803686 | 2008-10-22 13:30:34 +0100 | [diff] [blame] | 26 | writer.WriteLine("}"); |
| 27 | writer.WriteLine("public {0} {1} {{", TypeName, PropertyName); |
| 28 | writer.WriteLine(" get {{ return result.{0}; }}", PropertyName); |
Jon Skeet | 4cf9e3c | 2008-11-24 11:11:28 +0000 | [diff] [blame^] | 29 | writer.WriteLine(" set {{ Set{0}(value); }}", PropertyName); |
Jon Skeet | 6803686 | 2008-10-22 13:30:34 +0100 | [diff] [blame] | 30 | writer.WriteLine("}"); |
Jon Skeet | 4cf9e3c | 2008-11-24 11:11:28 +0000 | [diff] [blame^] | 31 | writer.WriteLine("public Builder Set{0}({1} value) {{", PropertyName, TypeName); |
| 32 | writer.WriteLine(" result.has{0} = true;", PropertyName); |
Jon Skeet | 6803686 | 2008-10-22 13:30:34 +0100 | [diff] [blame] | 33 | writer.WriteLine(" result.{0}_ = value;", Name); |
| 34 | writer.WriteLine(" return this;"); |
| 35 | writer.WriteLine("}"); |
Jon Skeet | 4cf9e3c | 2008-11-24 11:11:28 +0000 | [diff] [blame^] | 36 | writer.WriteLine("public Builder Clear{0}() {{", PropertyName); |
| 37 | writer.WriteLine(" result.has{0} = false;", PropertyName); |
Jon Skeet | 6803686 | 2008-10-22 13:30:34 +0100 | [diff] [blame] | 38 | writer.WriteLine(" result.{0}_ = {1};", Name, DefaultValue); |
| 39 | writer.WriteLine(" return this;"); |
| 40 | writer.WriteLine("}"); |
| 41 | } |
| 42 | |
| 43 | public void GenerateMergingCode(TextGenerator writer) { |
Jon Skeet | 4cf9e3c | 2008-11-24 11:11:28 +0000 | [diff] [blame^] | 44 | writer.WriteLine("if (other.Has{0}) {{", PropertyName); |
Jon Skeet | 6803686 | 2008-10-22 13:30:34 +0100 | [diff] [blame] | 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) { |
Jon Skeet | 4cf9e3c | 2008-11-24 11:11:28 +0000 | [diff] [blame^] | 58 | writer.WriteLine("if (Has{0}) {{", PropertyName); |
Jon Skeet | 6803686 | 2008-10-22 13:30:34 +0100 | [diff] [blame] | 59 | writer.WriteLine(" output.Write{0}({1}, {2});", CapitalizedTypeName, Number, PropertyName); |
| 60 | writer.WriteLine("}"); |
| 61 | } |
| 62 | |
| 63 | public void GenerateSerializedSizeCode(TextGenerator writer) { |
Jon Skeet | 4cf9e3c | 2008-11-24 11:11:28 +0000 | [diff] [blame^] | 64 | writer.WriteLine("if (Has{0}) {{", PropertyName); |
Jon Skeet | 6803686 | 2008-10-22 13:30:34 +0100 | [diff] [blame] | 65 | writer.WriteLine(" size += pb::CodedOutputStream.Compute{0}Size({1}, {2});", |
| 66 | CapitalizedTypeName, Number, PropertyName); |
| 67 | writer.WriteLine("}"); |
| 68 | } |
| 69 | } |
| 70 | } |