[3857] | 1 | #region Copyright notice and license
|
---|
| 2 | // Protocol Buffers - Google's data interchange format
|
---|
| 3 | // Copyright 2008 Google Inc. All rights reserved.
|
---|
| 4 | // http://github.com/jskeet/dotnet-protobufs/
|
---|
| 5 | // Original C++/Java/Python code:
|
---|
| 6 | // http://code.google.com/p/protobuf/
|
---|
| 7 | //
|
---|
| 8 | // Redistribution and use in source and binary forms, with or without
|
---|
| 9 | // modification, are permitted provided that the following conditions are
|
---|
| 10 | // met:
|
---|
| 11 | //
|
---|
| 12 | // * Redistributions of source code must retain the above copyright
|
---|
| 13 | // notice, this list of conditions and the following disclaimer.
|
---|
| 14 | // * Redistributions in binary form must reproduce the above
|
---|
| 15 | // copyright notice, this list of conditions and the following disclaimer
|
---|
| 16 | // in the documentation and/or other materials provided with the
|
---|
| 17 | // distribution.
|
---|
| 18 | // * Neither the name of Google Inc. nor the names of its
|
---|
| 19 | // contributors may be used to endorse or promote products derived from
|
---|
| 20 | // this software without specific prior written permission.
|
---|
| 21 | //
|
---|
| 22 | // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
---|
| 23 | // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
---|
| 24 | // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
---|
| 25 | // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
---|
| 26 | // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
---|
| 27 | // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
---|
| 28 | // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
---|
| 29 | // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
---|
| 30 | // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
---|
| 31 | // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
---|
| 32 | // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
---|
| 33 | #endregion
|
---|
| 34 |
|
---|
| 35 | using System;
|
---|
| 36 | using System.Collections.Generic;
|
---|
| 37 | using Google.ProtocolBuffers.Descriptors;
|
---|
| 38 |
|
---|
| 39 | namespace Google.ProtocolBuffers.ProtoGen {
|
---|
| 40 | internal static class SourceGenerators {
|
---|
| 41 |
|
---|
| 42 | private static readonly Dictionary<Type, Func<IDescriptor, ISourceGenerator>> GeneratorFactories = new Dictionary<Type, Func<IDescriptor, ISourceGenerator>> {
|
---|
| 43 | { typeof(FileDescriptor), descriptor => new UmbrellaClassGenerator((FileDescriptor) descriptor) },
|
---|
| 44 | { typeof(EnumDescriptor), descriptor => new EnumGenerator((EnumDescriptor) descriptor) },
|
---|
| 45 | { typeof(ServiceDescriptor), descriptor => new ServiceGenerator((ServiceDescriptor) descriptor) },
|
---|
| 46 | { typeof(MessageDescriptor), descriptor => new MessageGenerator((MessageDescriptor) descriptor) },
|
---|
| 47 | // For other fields, we have IFieldSourceGenerators.
|
---|
| 48 | { typeof(FieldDescriptor), descriptor => new ExtensionGenerator((FieldDescriptor) descriptor) }
|
---|
| 49 | };
|
---|
| 50 |
|
---|
| 51 | public static IFieldSourceGenerator CreateFieldGenerator(FieldDescriptor field) {
|
---|
| 52 | switch (field.MappedType) {
|
---|
| 53 | case MappedType.Message :
|
---|
| 54 | return field.IsRepeated
|
---|
| 55 | ? (IFieldSourceGenerator) new RepeatedMessageFieldGenerator(field)
|
---|
| 56 | : new MessageFieldGenerator(field);
|
---|
| 57 | case MappedType.Enum:
|
---|
| 58 | return field.IsRepeated
|
---|
| 59 | ? (IFieldSourceGenerator)new RepeatedEnumFieldGenerator(field)
|
---|
| 60 | : new EnumFieldGenerator(field);
|
---|
| 61 | default:
|
---|
| 62 | return field.IsRepeated
|
---|
| 63 | ? (IFieldSourceGenerator)new RepeatedPrimitiveFieldGenerator(field)
|
---|
| 64 | : new PrimitiveFieldGenerator(field);
|
---|
| 65 | }
|
---|
| 66 | }
|
---|
| 67 |
|
---|
| 68 | public static ISourceGenerator CreateGenerator<T>(T descriptor) where T : IDescriptor {
|
---|
| 69 | Func<IDescriptor, ISourceGenerator> factory;
|
---|
| 70 | if (!GeneratorFactories.TryGetValue(typeof(T), out factory)) {
|
---|
| 71 | throw new ArgumentException("No generator registered for " + typeof(T).Name);
|
---|
| 72 | }
|
---|
| 73 | return factory(descriptor);
|
---|
| 74 | }
|
---|
| 75 | }
|
---|
| 76 | }
|
---|