164d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci// Protocol Buffers - Google's data interchange format 264d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci// Copyright 2008 Google Inc. All rights reserved. 364d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci// http://code.google.com/p/protobuf/ 464d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci// 564d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci// Redistribution and use in source and binary forms, with or without 664d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci// modification, are permitted provided that the following conditions are 764d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci// met: 864d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci// 964d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci// * Redistributions of source code must retain the above copyright 1064d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci// notice, this list of conditions and the following disclaimer. 1164d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci// * Redistributions in binary form must reproduce the above 1264d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci// copyright notice, this list of conditions and the following disclaimer 1364d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci// in the documentation and/or other materials provided with the 1464d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci// distribution. 1564d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci// * Neither the name of Google Inc. nor the names of its 1664d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci// contributors may be used to endorse or promote products derived from 1764d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci// this software without specific prior written permission. 1864d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci// 1964d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 2064d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 2164d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 2264d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 2364d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 2464d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 2564d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 2664d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 2764d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 2864d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 2964d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 3064d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci 3164d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci// Author: kenton@google.com (Kenton Varda) 3264d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci// Based on original Protocol Buffers design by 3364d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci// Sanjay Ghemawat, Jeff Dean, and others. 3464d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci 3564d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci#include <map> 3664d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci#include <string> 3764d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci 3864d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci#include <google/protobuf/compiler/javanano/javanano_params.h> 3964d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci#include <google/protobuf/compiler/javanano/javanano_enum.h> 4064d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci#include <google/protobuf/compiler/javanano/javanano_helpers.h> 4164d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci#include <google/protobuf/io/printer.h> 4264d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci#include <google/protobuf/descriptor.pb.h> 4364d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci#include <google/protobuf/stubs/strutil.h> 4464d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci 4564d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazcinamespace google { 4664d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazcinamespace protobuf { 4764d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazcinamespace compiler { 4864d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazcinamespace javanano { 4964d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci 5064d8d8f89050c5ada85341f967af391f4716a7cbUlas KirazciEnumGenerator::EnumGenerator(const EnumDescriptor* descriptor, const Params& params) 5164d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci : params_(params), descriptor_(descriptor) { 5264d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci for (int i = 0; i < descriptor_->value_count(); i++) { 5364d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci const EnumValueDescriptor* value = descriptor_->value(i); 5464d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci const EnumValueDescriptor* canonical_value = 5564d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci descriptor_->FindValueByNumber(value->number()); 5664d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci 5764d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci if (value == canonical_value) { 5864d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci canonical_values_.push_back(value); 5964d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci } else { 6064d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci Alias alias; 6164d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci alias.value = value; 6264d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci alias.canonical_value = canonical_value; 6364d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci aliases_.push_back(alias); 6464d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci } 6564d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci } 6664d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci} 6764d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci 6864d8d8f89050c5ada85341f967af391f4716a7cbUlas KirazciEnumGenerator::~EnumGenerator() {} 6964d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci 7064d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazcivoid EnumGenerator::Generate(io::Printer* printer) { 7126266cd4660ffe1f3d6015b715713ee654c5b936Max Cai printer->Print( 72286271f75aa6ff1f3746379b77d6dc55415baa16Max Cai "\n" 7326266cd4660ffe1f3d6015b715713ee654c5b936Max Cai "// enum $classname$\n", 7426266cd4660ffe1f3d6015b715713ee654c5b936Max Cai "classname", descriptor_->name()); 7526266cd4660ffe1f3d6015b715713ee654c5b936Max Cai 7626266cd4660ffe1f3d6015b715713ee654c5b936Max Cai // Start of container interface 7726266cd4660ffe1f3d6015b715713ee654c5b936Max Cai bool use_shell_class = params_.java_enum_style(); 7826266cd4660ffe1f3d6015b715713ee654c5b936Max Cai if (use_shell_class) { 7926266cd4660ffe1f3d6015b715713ee654c5b936Max Cai printer->Print( 8026266cd4660ffe1f3d6015b715713ee654c5b936Max Cai "public interface $classname$ {\n", 8126266cd4660ffe1f3d6015b715713ee654c5b936Max Cai "classname", RenameJavaKeywords(descriptor_->name())); 8226266cd4660ffe1f3d6015b715713ee654c5b936Max Cai printer->Indent(); 8364d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci } 8464d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci 8526266cd4660ffe1f3d6015b715713ee654c5b936Max Cai // Canonical values 8626266cd4660ffe1f3d6015b715713ee654c5b936Max Cai for (int i = 0; i < canonical_values_.size(); i++) { 8726266cd4660ffe1f3d6015b715713ee654c5b936Max Cai printer->Print( 8826266cd4660ffe1f3d6015b715713ee654c5b936Max Cai "public static final int $name$ = $canonical_value$;\n", 8926266cd4660ffe1f3d6015b715713ee654c5b936Max Cai "name", RenameJavaKeywords(canonical_values_[i]->name()), 9026266cd4660ffe1f3d6015b715713ee654c5b936Max Cai "canonical_value", SimpleItoa(canonical_values_[i]->number())); 9126266cd4660ffe1f3d6015b715713ee654c5b936Max Cai } 9264d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci 9326266cd4660ffe1f3d6015b715713ee654c5b936Max Cai // Aliases 9464d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci for (int i = 0; i < aliases_.size(); i++) { 9526266cd4660ffe1f3d6015b715713ee654c5b936Max Cai printer->Print( 9626266cd4660ffe1f3d6015b715713ee654c5b936Max Cai "public static final int $name$ = $canonical_name$;\n", 9726266cd4660ffe1f3d6015b715713ee654c5b936Max Cai "name", RenameJavaKeywords(aliases_[i].value->name()), 9826266cd4660ffe1f3d6015b715713ee654c5b936Max Cai "canonical_name", RenameJavaKeywords(aliases_[i].canonical_value->name())); 9964d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci } 10064d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci 10126266cd4660ffe1f3d6015b715713ee654c5b936Max Cai // End of container interface 10226266cd4660ffe1f3d6015b715713ee654c5b936Max Cai if (use_shell_class) { 10326266cd4660ffe1f3d6015b715713ee654c5b936Max Cai printer->Outdent(); 10426266cd4660ffe1f3d6015b715713ee654c5b936Max Cai printer->Print("}\n"); 10526266cd4660ffe1f3d6015b715713ee654c5b936Max Cai } 10664d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci} 10764d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci 10864d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci} // namespace javanano 10964d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci} // namespace compiler 11064d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci} // namespace protobuf 11164d8d8f89050c5ada85341f967af391f4716a7cbUlas Kirazci} // namespace google 112