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
EnumGenerator(const EnumDescriptor * descriptor,const Options * options)53 EnumGenerator::EnumGenerator(const EnumDescriptor* descriptor, const Options* options) :
54 SourceGeneratorBase(descriptor->file(), options),
55 descriptor_(descriptor) {
56 }
57
~EnumGenerator()58 EnumGenerator::~EnumGenerator() {
59 }
60
Generate(io::Printer * printer)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