javanano_enum.cc revision 0eca14248fc14a12c5c8a6fc253bef6f58cc60c1
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("// enum $classname$\n", "classname", descriptor_->name()); 72 bool is_own_file = params_.java_multiple_files() || 73 ((descriptor_->containing_type() == NULL) && 74 !params_.has_java_outer_classname(descriptor_->file()->name())); 75 76 if (is_own_file) { 77 printer->Print("public final class $classname$ {\n", "classname", 78 descriptor_->name()); 79 printer->Indent(); 80 printer->Print("private $classname$() {}\n", "classname", 81 descriptor_->name()); 82 } 83 for (int i = 0; i < canonical_values_.size(); i++) { 84 map<string, string> vars; 85 vars["name"] = canonical_values_[i]->name(); 86 vars["canonical_value"] = SimpleItoa(canonical_values_[i]->number()); 87 printer->Print(vars, 88 "public static final int $name$ = $canonical_value$;\n"); 89 } 90 91 // ----------------------------------------------------------------- 92 93 for (int i = 0; i < aliases_.size(); i++) { 94 map<string, string> vars; 95 vars["name"] = aliases_[i].value->name(); 96 vars["canonical_name"] = aliases_[i].canonical_value->name(); 97 printer->Print(vars, 98 "public static final int $name$ = $canonical_name$;\n"); 99 } 100 101 if (is_own_file) { 102 printer->Outdent(); 103 printer->Print("}"); 104 } 105 printer->Print("\n"); 106} 107 108} // namespace javanano 109} // namespace compiler 110} // namespace protobuf 111} // namespace google 112