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/descriptor.h>
35 #include <google/protobuf/descriptor.pb.h>
36 #include <google/protobuf/io/printer.h>
37 #include <google/protobuf/io/zero_copy_stream.h>
38 #include <google/protobuf/stubs/strutil.h>
39
40 #include <google/protobuf/compiler/csharp/csharp_doc_comment.h>
41 #include <google/protobuf/compiler/csharp/csharp_enum.h>
42 #include <google/protobuf/compiler/csharp/csharp_helpers.h>
43 #include <google/protobuf/compiler/csharp/csharp_options.h>
44
45 namespace google {
46 namespace protobuf {
47 namespace compiler {
48 namespace csharp {
49
EnumGenerator(const EnumDescriptor * descriptor,const Options * options)50 EnumGenerator::EnumGenerator(const EnumDescriptor* descriptor, const Options* options) :
51 SourceGeneratorBase(descriptor->file(), options),
52 descriptor_(descriptor) {
53 }
54
~EnumGenerator()55 EnumGenerator::~EnumGenerator() {
56 }
57
Generate(io::Printer * printer)58 void EnumGenerator::Generate(io::Printer* printer) {
59 WriteEnumDocComment(printer, descriptor_);
60 printer->Print("$access_level$ enum $name$ {\n",
61 "access_level", class_access_level(),
62 "name", descriptor_->name());
63 printer->Indent();
64 std::set<string> used_names;
65 std::set<int> used_number;
66 for (int i = 0; i < descriptor_->value_count(); i++) {
67 WriteEnumValueDocComment(printer, descriptor_->value(i));
68 string original_name = descriptor_->value(i)->name();
69 string name = GetEnumValueName(descriptor_->name(), descriptor_->value(i)->name());
70 // Make sure we don't get any duplicate names due to prefix removal.
71 while (!used_names.insert(name).second) {
72 // It's possible we'll end up giving this warning multiple times, but that's better than not at all.
73 GOOGLE_LOG(WARNING) << "Duplicate enum value " << name << " (originally " << original_name
74 << ") in " << descriptor_->name() << "; adding underscore to distinguish";
75 name += "_";
76 }
77 int number = descriptor_->value(i)->number();
78 if (!used_number.insert(number).second) {
79 printer->Print("[pbr::OriginalName(\"$original_name$\", PreferredAlias = false)] $name$ = $number$,\n",
80 "original_name", original_name,
81 "name", name,
82 "number", StrCat(number));
83 } else {
84 printer->Print("[pbr::OriginalName(\"$original_name$\")] $name$ = $number$,\n",
85 "original_name", original_name,
86 "name", name,
87 "number", StrCat(number));
88 }
89 }
90 printer->Outdent();
91 printer->Print("}\n");
92 printer->Print("\n");
93 }
94
95 } // namespace csharp
96 } // namespace compiler
97 } // namespace protobuf
98 } // namespace google
99