javanano_enum.cc revision 286271f75aa6ff1f3746379b77d6dc55415baa16
1// Protocol Buffers - Google's data interchange format 2// Copyright 2008 Google Inc. All rights reserved. 3// http://code.google.com/p/protobuf/ 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// Author: kenton@google.com (Kenton Varda) 32// Based on original Protocol Buffers design by 33// Sanjay Ghemawat, Jeff Dean, and others. 34 35#include <map> 36#include <string> 37 38#include <google/protobuf/compiler/javanano/javanano_params.h> 39#include <google/protobuf/compiler/javanano/javanano_enum.h> 40#include <google/protobuf/compiler/javanano/javanano_helpers.h> 41#include <google/protobuf/io/printer.h> 42#include <google/protobuf/descriptor.pb.h> 43#include <google/protobuf/stubs/strutil.h> 44 45namespace google { 46namespace protobuf { 47namespace compiler { 48namespace javanano { 49 50EnumGenerator::EnumGenerator(const EnumDescriptor* descriptor, const Params& params) 51 : params_(params), descriptor_(descriptor) { 52 for (int i = 0; i < descriptor_->value_count(); i++) { 53 const EnumValueDescriptor* value = descriptor_->value(i); 54 const EnumValueDescriptor* canonical_value = 55 descriptor_->FindValueByNumber(value->number()); 56 57 if (value == canonical_value) { 58 canonical_values_.push_back(value); 59 } else { 60 Alias alias; 61 alias.value = value; 62 alias.canonical_value = canonical_value; 63 aliases_.push_back(alias); 64 } 65 } 66} 67 68EnumGenerator::~EnumGenerator() {} 69 70void EnumGenerator::Generate(io::Printer* printer) { 71 printer->Print( 72 "\n" 73 "// enum $classname$\n", 74 "classname", descriptor_->name()); 75 76 // Start of container interface 77 bool use_shell_class = params_.java_enum_style(); 78 if (use_shell_class) { 79 printer->Print( 80 "public interface $classname$ {\n", 81 "classname", RenameJavaKeywords(descriptor_->name())); 82 printer->Indent(); 83 } 84 85 // Canonical values 86 for (int i = 0; i < canonical_values_.size(); i++) { 87 printer->Print( 88 "public static final int $name$ = $canonical_value$;\n", 89 "name", RenameJavaKeywords(canonical_values_[i]->name()), 90 "canonical_value", SimpleItoa(canonical_values_[i]->number())); 91 } 92 93 // Aliases 94 for (int i = 0; i < aliases_.size(); i++) { 95 printer->Print( 96 "public static final int $name$ = $canonical_name$;\n", 97 "name", RenameJavaKeywords(aliases_[i].value->name()), 98 "canonical_name", RenameJavaKeywords(aliases_[i].canonical_value->name())); 99 } 100 101 // End of container interface 102 if (use_shell_class) { 103 printer->Outdent(); 104 printer->Print("}\n"); 105 } 106} 107 108} // namespace javanano 109} // namespace compiler 110} // namespace protobuf 111} // namespace google 112