javanano_enum.cc revision 26266cd4660ffe1f3d6015b715713ee654c5b936
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 "// enum $classname$\n", 73 "classname", descriptor_->name()); 74 75 // Start of container interface 76 bool use_shell_class = params_.java_enum_style(); 77 if (use_shell_class) { 78 printer->Print( 79 "public interface $classname$ {\n", 80 "classname", RenameJavaKeywords(descriptor_->name())); 81 printer->Indent(); 82 } 83 84 // Canonical values 85 for (int i = 0; i < canonical_values_.size(); i++) { 86 printer->Print( 87 "public static final int $name$ = $canonical_value$;\n", 88 "name", RenameJavaKeywords(canonical_values_[i]->name()), 89 "canonical_value", SimpleItoa(canonical_values_[i]->number())); 90 } 91 92 // Aliases 93 for (int i = 0; i < aliases_.size(); i++) { 94 printer->Print( 95 "public static final int $name$ = $canonical_name$;\n", 96 "name", RenameJavaKeywords(aliases_[i].value->name()), 97 "canonical_name", RenameJavaKeywords(aliases_[i].canonical_value->name())); 98 } 99 100 // End of container interface 101 if (use_shell_class) { 102 printer->Outdent(); 103 printer->Print("}\n"); 104 } 105 printer->Print("\n"); 106} 107 108} // namespace javanano 109} // namespace compiler 110} // namespace protobuf 111} // namespace google 112