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