Jan Tattermusch | 685ae36 | 2015-03-16 19:07:16 -0700 | [diff] [blame] | 1 | // Protocol Buffers - Google's data interchange format |
| 2 | // Copyright 2008 Google Inc. All rights reserved. |
| 3 | // https://developers.google.com/protocol-buffers/ |
| 4 | // |
| 5 | // Redistribution and use in source and binary forms, with or without |
| 6 | // modification, are permitted provided that the following conditions are |
| 7 | // met: |
| 8 | // |
| 9 | // * Redistributions of source code must retain the above copyright |
| 10 | // notice, this list of conditions and the following disclaimer. |
| 11 | // * Redistributions in binary form must reproduce the above |
| 12 | // copyright notice, this list of conditions and the following disclaimer |
| 13 | // in the documentation and/or other materials provided with the |
| 14 | // distribution. |
| 15 | // * Neither the name of Google Inc. nor the names of its |
| 16 | // contributors may be used to endorse or promote products derived from |
| 17 | // this software without specific prior written permission. |
| 18 | // |
| 19 | // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
| 20 | // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
| 21 | // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
| 22 | // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
| 23 | // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
| 24 | // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
| 25 | // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
| 26 | // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
| 27 | // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
| 28 | // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
| 29 | // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
| 30 | |
| 31 | #include <sstream> |
| 32 | |
| 33 | #include <google/protobuf/compiler/code_generator.h> |
| 34 | #include <google/protobuf/compiler/plugin.h> |
| 35 | #include <google/protobuf/descriptor.h> |
| 36 | #include <google/protobuf/descriptor.pb.h> |
| 37 | #include <google/protobuf/io/printer.h> |
| 38 | #include <google/protobuf/io/zero_copy_stream.h> |
| 39 | #include <google/protobuf/stubs/strutil.h> |
| 40 | |
Jon Skeet | 67dd42c | 2015-10-01 10:36:58 +0100 | [diff] [blame] | 41 | #include <google/protobuf/compiler/csharp/csharp_doc_comment.h> |
Jan Tattermusch | 685ae36 | 2015-03-16 19:07:16 -0700 | [diff] [blame] | 42 | #include <google/protobuf/compiler/csharp/csharp_enum.h> |
| 43 | #include <google/protobuf/compiler/csharp/csharp_helpers.h> |
Jon Skeet | 75626ed | 2016-04-08 12:19:10 +0100 | [diff] [blame] | 44 | #include <google/protobuf/compiler/csharp/csharp_options.h> |
Jan Tattermusch | 685ae36 | 2015-03-16 19:07:16 -0700 | [diff] [blame] | 45 | |
| 46 | using google::protobuf::internal::scoped_ptr; |
| 47 | |
| 48 | namespace google { |
| 49 | namespace protobuf { |
| 50 | namespace compiler { |
| 51 | namespace csharp { |
| 52 | |
Gaurav Vaish | 74d8b0b | 2016-03-24 11:44:56 -0700 | [diff] [blame] | 53 | EnumGenerator::EnumGenerator(const EnumDescriptor* descriptor, const Options* options) : |
| 54 | SourceGeneratorBase(descriptor->file(), options), |
Jan Tattermusch | 685ae36 | 2015-03-16 19:07:16 -0700 | [diff] [blame] | 55 | descriptor_(descriptor) { |
| 56 | } |
| 57 | |
| 58 | EnumGenerator::~EnumGenerator() { |
| 59 | } |
| 60 | |
Jie Luo | 90da351 | 2015-06-03 18:02:17 -0700 | [diff] [blame] | 61 | void EnumGenerator::Generate(io::Printer* printer) { |
Jon Skeet | 67dd42c | 2015-10-01 10:36:58 +0100 | [diff] [blame] | 62 | WriteEnumDocComment(printer, descriptor_); |
Jie Luo | 90da351 | 2015-06-03 18:02:17 -0700 | [diff] [blame] | 63 | WriteGeneratedCodeAttributes(printer); |
| 64 | printer->Print("$access_level$ enum $name$ {\n", |
| 65 | "access_level", class_access_level(), |
| 66 | "name", descriptor_->name()); |
| 67 | printer->Indent(); |
Jon Skeet | 75626ed | 2016-04-08 12:19:10 +0100 | [diff] [blame] | 68 | std::set<string> used_names; |
Jan Tattermusch | 685ae36 | 2015-03-16 19:07:16 -0700 | [diff] [blame] | 69 | for (int i = 0; i < descriptor_->value_count(); i++) { |
Jon Skeet | 67dd42c | 2015-10-01 10:36:58 +0100 | [diff] [blame] | 70 | WriteEnumValueDocComment(printer, descriptor_->value(i)); |
Jon Skeet | 75626ed | 2016-04-08 12:19:10 +0100 | [diff] [blame] | 71 | string original_name = descriptor_->value(i)->name(); |
| 72 | string name = options()->legacy_enum_values |
| 73 | ? descriptor_->value(i)->name() |
| 74 | : GetEnumValueName(descriptor_->name(), descriptor_->value(i)->name()); |
| 75 | // Make sure we don't get any duplicate names due to prefix removal. |
| 76 | while (!used_names.insert(name).second) { |
| 77 | // It's possible we'll end up giving this warning multiple times, but that's better than not at all. |
| 78 | GOOGLE_LOG(WARNING) << "Duplicate enum value " << name << " (originally " << original_name |
| 79 | << ") in " << descriptor_->name() << "; adding underscore to distinguish"; |
| 80 | name += "_"; |
| 81 | } |
| 82 | printer->Print("[pbr::OriginalName(\"$original_name$\")] $name$ = $number$,\n", |
| 83 | "original_name", original_name, |
| 84 | "name", name, |
| 85 | "number", SimpleItoa(descriptor_->value(i)->number())); |
Jan Tattermusch | 685ae36 | 2015-03-16 19:07:16 -0700 | [diff] [blame] | 86 | } |
Jie Luo | 90da351 | 2015-06-03 18:02:17 -0700 | [diff] [blame] | 87 | printer->Outdent(); |
| 88 | printer->Print("}\n"); |
| 89 | printer->Print("\n"); |
Jan Tattermusch | 685ae36 | 2015-03-16 19:07:16 -0700 | [diff] [blame] | 90 | } |
| 91 | |
| 92 | } // namespace csharp |
| 93 | } // namespace compiler |
| 94 | } // namespace protobuf |
| 95 | } // namespace google |