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("}"); |
Jon Skeet | d6dd0a4 | 2009-06-05 22:00:05 +0100 | [diff] [blame^] | 18 | AddClsComplianceCheck(writer); |
Jon Skeet | 6803686 | 2008-10-22 13:30:34 +0100 | [diff] [blame] | 19 | 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 Skeet | 4cf9e3c | 2008-11-24 11:11:28 +0000 | [diff] [blame] | 25 | writer.WriteLine("public bool Has{0} {{", PropertyName); |
| 26 | writer.WriteLine(" get {{ return result.Has{0}; }}", PropertyName); |
Jon Skeet | 6803686 | 2008-10-22 13:30:34 +0100 | [diff] [blame] | 27 | writer.WriteLine("}"); |
Jon Skeet | d6dd0a4 | 2009-06-05 22:00:05 +0100 | [diff] [blame^] | 28 | AddClsComplianceCheck(writer); |
Jon Skeet | 6803686 | 2008-10-22 13:30:34 +0100 | [diff] [blame] | 29 | writer.WriteLine("public {0} {1} {{", TypeName, PropertyName); |
| 30 | writer.WriteLine(" get {{ return result.{0}; }}", PropertyName); |
Jon Skeet | 4cf9e3c | 2008-11-24 11:11:28 +0000 | [diff] [blame] | 31 | writer.WriteLine(" set {{ Set{0}(value); }}", PropertyName); |
Jon Skeet | 6803686 | 2008-10-22 13:30:34 +0100 | [diff] [blame] | 32 | writer.WriteLine("}"); |
Jon Skeet | d6dd0a4 | 2009-06-05 22:00:05 +0100 | [diff] [blame^] | 33 | AddClsComplianceCheck(writer); |
Jon Skeet | 4cf9e3c | 2008-11-24 11:11:28 +0000 | [diff] [blame] | 34 | writer.WriteLine("public Builder Set{0}({1} value) {{", PropertyName, TypeName); |
Jon Skeet | 642a814 | 2009-01-27 12:25:21 +0000 | [diff] [blame] | 35 | AddNullCheck(writer); |
Jon Skeet | 4cf9e3c | 2008-11-24 11:11:28 +0000 | [diff] [blame] | 36 | writer.WriteLine(" result.has{0} = true;", PropertyName); |
Jon Skeet | 6803686 | 2008-10-22 13:30:34 +0100 | [diff] [blame] | 37 | writer.WriteLine(" result.{0}_ = value;", Name); |
| 38 | writer.WriteLine(" return this;"); |
| 39 | writer.WriteLine("}"); |
Jon Skeet | 4cf9e3c | 2008-11-24 11:11:28 +0000 | [diff] [blame] | 40 | writer.WriteLine("public Builder Clear{0}() {{", PropertyName); |
| 41 | writer.WriteLine(" result.has{0} = false;", PropertyName); |
Jon Skeet | 6803686 | 2008-10-22 13:30:34 +0100 | [diff] [blame] | 42 | writer.WriteLine(" result.{0}_ = {1};", Name, DefaultValue); |
| 43 | writer.WriteLine(" return this;"); |
| 44 | writer.WriteLine("}"); |
| 45 | } |
| 46 | |
| 47 | public void GenerateMergingCode(TextGenerator writer) { |
Jon Skeet | 4cf9e3c | 2008-11-24 11:11:28 +0000 | [diff] [blame] | 48 | writer.WriteLine("if (other.Has{0}) {{", PropertyName); |
Jon Skeet | 6803686 | 2008-10-22 13:30:34 +0100 | [diff] [blame] | 49 | 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 Skeet | 4cf9e3c | 2008-11-24 11:11:28 +0000 | [diff] [blame] | 62 | writer.WriteLine("if (Has{0}) {{", PropertyName); |
Jon Skeet | 6803686 | 2008-10-22 13:30:34 +0100 | [diff] [blame] | 63 | writer.WriteLine(" output.Write{0}({1}, {2});", CapitalizedTypeName, Number, PropertyName); |
| 64 | writer.WriteLine("}"); |
| 65 | } |
| 66 | |
| 67 | public void GenerateSerializedSizeCode(TextGenerator writer) { |
Jon Skeet | 4cf9e3c | 2008-11-24 11:11:28 +0000 | [diff] [blame] | 68 | writer.WriteLine("if (Has{0}) {{", PropertyName); |
Jon Skeet | 6803686 | 2008-10-22 13:30:34 +0100 | [diff] [blame] | 69 | writer.WriteLine(" size += pb::CodedOutputStream.Compute{0}Size({1}, {2});", |
| 70 | CapitalizedTypeName, Number, PropertyName); |
| 71 | writer.WriteLine("}"); |
| 72 | } |
| 73 | } |
| 74 | } |