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 41 #include <google/protobuf/compiler/csharp/csharp_doc_comment.h> 42 #include <google/protobuf/compiler/csharp/csharp_enum.h> 43 #include <google/protobuf/compiler/csharp/csharp_helpers.h> 44 #include <google/protobuf/compiler/csharp/csharp_options.h> 45 46 using google::protobuf::internal::scoped_ptr; 47 48 namespace google { 49 namespace protobuf { 50 namespace compiler { 51 namespace csharp { 52 53 EnumGenerator::EnumGenerator(const EnumDescriptor* descriptor, const Options* options) : 54 SourceGeneratorBase(descriptor->file(), options), 55 descriptor_(descriptor) { 56 } 57 58 EnumGenerator::~EnumGenerator() { 59 } 60 61 void EnumGenerator::Generate(io::Printer* printer) { 62 WriteEnumDocComment(printer, descriptor_); 63 WriteGeneratedCodeAttributes(printer); 64 printer->Print("$access_level$ enum $name$ {\n", 65 "access_level", class_access_level(), 66 "name", descriptor_->name()); 67 printer->Indent(); 68 std::set<string> used_names; 69 for (int i = 0; i < descriptor_->value_count(); i++) { 70 WriteEnumValueDocComment(printer, descriptor_->value(i)); 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())); 86 } 87 printer->Outdent(); 88 printer->Print("}\n"); 89 printer->Print("\n"); 90 } 91 92 } // namespace csharp 93 } // namespace compiler 94 } // namespace protobuf 95 } // namespace google 96
