java_enum_field.cc 36 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005
  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. // Author: kenton@google.com (Kenton Varda)
  31. // Based on original Protocol Buffers design by
  32. // Sanjay Ghemawat, Jeff Dean, and others.
  33. #include <map>
  34. #include <string>
  35. #include <google/protobuf/stubs/logging.h>
  36. #include <google/protobuf/stubs/common.h>
  37. #include <google/protobuf/compiler/java/java_context.h>
  38. #include <google/protobuf/compiler/java/java_doc_comment.h>
  39. #include <google/protobuf/compiler/java/java_enum_field.h>
  40. #include <google/protobuf/compiler/java/java_helpers.h>
  41. #include <google/protobuf/compiler/java/java_name_resolver.h>
  42. #include <google/protobuf/io/printer.h>
  43. #include <google/protobuf/wire_format.h>
  44. #include <google/protobuf/stubs/strutil.h>
  45. namespace google {
  46. namespace protobuf {
  47. namespace compiler {
  48. namespace java {
  49. namespace {
  50. void SetEnumVariables(const FieldDescriptor* descriptor,
  51. int messageBitIndex,
  52. int builderBitIndex,
  53. const FieldGeneratorInfo* info,
  54. ClassNameResolver* name_resolver,
  55. std::map<string, string>* variables) {
  56. SetCommonFieldVariables(descriptor, info, variables);
  57. (*variables)["type"] =
  58. name_resolver->GetImmutableClassName(descriptor->enum_type());
  59. (*variables)["mutable_type"] =
  60. name_resolver->GetMutableClassName(descriptor->enum_type());
  61. (*variables)["default"] = ImmutableDefaultValue(descriptor, name_resolver);
  62. (*variables)["default_number"] = SimpleItoa(
  63. descriptor->default_value_enum()->number());
  64. (*variables)["tag"] =
  65. SimpleItoa(static_cast<int32>(internal::WireFormat::MakeTag(descriptor)));
  66. (*variables)["tag_size"] = SimpleItoa(
  67. internal::WireFormat::TagSize(descriptor->number(), GetType(descriptor)));
  68. // TODO(birdo): Add @deprecated javadoc when generating javadoc is supported
  69. // by the proto compiler
  70. (*variables)["deprecation"] = descriptor->options().deprecated()
  71. ? "@java.lang.Deprecated " : "";
  72. (*variables)["on_changed"] = "onChanged();";
  73. // Use deprecated valueOf() method to be compatible with old generated code
  74. // for v2.5.0/v2.6.1.
  75. // TODO(xiaofeng): Use "forNumber" when we no longer support compatibility
  76. // with v2.5.0/v2.6.1, and remove the @SuppressWarnings annotations.
  77. (*variables)["for_number"] = "valueOf";
  78. if (SupportFieldPresence(descriptor->file())) {
  79. // For singular messages and builders, one bit is used for the hasField bit.
  80. (*variables)["get_has_field_bit_message"] = GenerateGetBit(messageBitIndex);
  81. (*variables)["get_has_field_bit_builder"] = GenerateGetBit(builderBitIndex);
  82. // Note that these have a trailing ";".
  83. (*variables)["set_has_field_bit_message"] =
  84. GenerateSetBit(messageBitIndex) + ";";
  85. (*variables)["set_has_field_bit_builder"] =
  86. GenerateSetBit(builderBitIndex) + ";";
  87. (*variables)["clear_has_field_bit_builder"] =
  88. GenerateClearBit(builderBitIndex) + ";";
  89. (*variables)["is_field_present_message"] = GenerateGetBit(messageBitIndex);
  90. } else {
  91. (*variables)["set_has_field_bit_message"] = "";
  92. (*variables)["set_has_field_bit_builder"] = "";
  93. (*variables)["clear_has_field_bit_builder"] = "";
  94. (*variables)["is_field_present_message"] =
  95. (*variables)["name"] + "_ != " +
  96. (*variables)["default"] + ".getNumber()";
  97. }
  98. // For repated builders, one bit is used for whether the array is immutable.
  99. (*variables)["get_mutable_bit_builder"] = GenerateGetBit(builderBitIndex);
  100. (*variables)["set_mutable_bit_builder"] = GenerateSetBit(builderBitIndex);
  101. (*variables)["clear_mutable_bit_builder"] = GenerateClearBit(builderBitIndex);
  102. // For repeated fields, one bit is used for whether the array is immutable
  103. // in the parsing constructor.
  104. (*variables)["get_mutable_bit_parser"] =
  105. GenerateGetBitMutableLocal(builderBitIndex);
  106. (*variables)["set_mutable_bit_parser"] =
  107. GenerateSetBitMutableLocal(builderBitIndex);
  108. (*variables)["get_has_field_bit_from_local"] =
  109. GenerateGetBitFromLocal(builderBitIndex);
  110. (*variables)["set_has_field_bit_to_local"] =
  111. GenerateSetBitToLocal(messageBitIndex);
  112. if (SupportUnknownEnumValue(descriptor->file())) {
  113. (*variables)["unknown"] = (*variables)["type"] + ".UNRECOGNIZED";
  114. } else {
  115. (*variables)["unknown"] = (*variables)["default"];
  116. }
  117. }
  118. } // namespace
  119. // ===================================================================
  120. ImmutableEnumFieldGenerator::
  121. ImmutableEnumFieldGenerator(const FieldDescriptor* descriptor,
  122. int messageBitIndex,
  123. int builderBitIndex,
  124. Context* context)
  125. : descriptor_(descriptor), messageBitIndex_(messageBitIndex),
  126. builderBitIndex_(builderBitIndex),
  127. name_resolver_(context->GetNameResolver()) {
  128. SetEnumVariables(descriptor, messageBitIndex, builderBitIndex,
  129. context->GetFieldGeneratorInfo(descriptor),
  130. name_resolver_, &variables_);
  131. }
  132. ImmutableEnumFieldGenerator::~ImmutableEnumFieldGenerator() {}
  133. int ImmutableEnumFieldGenerator::GetNumBitsForMessage() const {
  134. return 1;
  135. }
  136. int ImmutableEnumFieldGenerator::GetNumBitsForBuilder() const {
  137. return 1;
  138. }
  139. void ImmutableEnumFieldGenerator::
  140. GenerateInterfaceMembers(io::Printer* printer) const {
  141. if (SupportFieldPresence(descriptor_->file())) {
  142. WriteFieldDocComment(printer, descriptor_);
  143. printer->Print(variables_,
  144. "$deprecation$boolean has$capitalized_name$();\n");
  145. }
  146. if (SupportUnknownEnumValue(descriptor_->file())) {
  147. WriteFieldDocComment(printer, descriptor_);
  148. printer->Print(variables_,
  149. "$deprecation$int get$capitalized_name$Value();\n");
  150. }
  151. WriteFieldDocComment(printer, descriptor_);
  152. printer->Print(variables_,
  153. "$deprecation$$type$ get$capitalized_name$();\n");
  154. }
  155. void ImmutableEnumFieldGenerator::
  156. GenerateMembers(io::Printer* printer) const {
  157. printer->Print(variables_,
  158. "private int $name$_;\n");
  159. PrintExtraFieldInfo(variables_, printer);
  160. if (SupportFieldPresence(descriptor_->file())) {
  161. WriteFieldDocComment(printer, descriptor_);
  162. printer->Print(variables_,
  163. "$deprecation$public boolean ${$has$capitalized_name$$}$() {\n"
  164. " return $get_has_field_bit_message$;\n"
  165. "}\n");
  166. printer->Annotate("{", "}", descriptor_);
  167. }
  168. if (SupportUnknownEnumValue(descriptor_->file())) {
  169. WriteFieldDocComment(printer, descriptor_);
  170. printer->Print(variables_,
  171. "$deprecation$public int ${$get$capitalized_name$Value$}$() {\n"
  172. " return $name$_;\n"
  173. "}\n");
  174. printer->Annotate("{", "}", descriptor_);
  175. }
  176. WriteFieldDocComment(printer, descriptor_);
  177. printer->Print(variables_,
  178. "$deprecation$public $type$ ${$get$capitalized_name$$}$() {\n"
  179. " @SuppressWarnings(\"deprecation\")\n"
  180. " $type$ result = $type$.$for_number$($name$_);\n"
  181. " return result == null ? $unknown$ : result;\n"
  182. "}\n");
  183. printer->Annotate("{", "}", descriptor_);
  184. }
  185. void ImmutableEnumFieldGenerator::
  186. GenerateBuilderMembers(io::Printer* printer) const {
  187. printer->Print(variables_,
  188. "private int $name$_ = $default_number$;\n");
  189. if (SupportFieldPresence(descriptor_->file())) {
  190. WriteFieldDocComment(printer, descriptor_);
  191. printer->Print(variables_,
  192. "$deprecation$public boolean ${$has$capitalized_name$$}$() {\n"
  193. " return $get_has_field_bit_builder$;\n"
  194. "}\n");
  195. printer->Annotate("{", "}", descriptor_);
  196. }
  197. if (SupportUnknownEnumValue(descriptor_->file())) {
  198. WriteFieldDocComment(printer, descriptor_);
  199. printer->Print(variables_,
  200. "$deprecation$public int ${$get$capitalized_name$Value$}$() {\n"
  201. " return $name$_;\n"
  202. "}\n");
  203. printer->Annotate("{", "}", descriptor_);
  204. WriteFieldDocComment(printer, descriptor_);
  205. printer->Print(variables_,
  206. "$deprecation$public Builder "
  207. "${$set$capitalized_name$Value$}$(int value) {\n"
  208. " $name$_ = value;\n"
  209. " $on_changed$\n"
  210. " return this;\n"
  211. "}\n");
  212. printer->Annotate("{", "}", descriptor_);
  213. }
  214. WriteFieldDocComment(printer, descriptor_);
  215. printer->Print(variables_,
  216. "$deprecation$public $type$ ${$get$capitalized_name$$}$() {\n"
  217. " @SuppressWarnings(\"deprecation\")\n"
  218. " $type$ result = $type$.$for_number$($name$_);\n"
  219. " return result == null ? $unknown$ : result;\n"
  220. "}\n");
  221. printer->Annotate("{", "}", descriptor_);
  222. WriteFieldDocComment(printer, descriptor_);
  223. printer->Print(variables_,
  224. "$deprecation$public Builder ${$set$capitalized_name$$}$($type$ value) {\n"
  225. " if (value == null) {\n"
  226. " throw new NullPointerException();\n"
  227. " }\n"
  228. " $set_has_field_bit_builder$\n"
  229. " $name$_ = value.getNumber();\n"
  230. " $on_changed$\n"
  231. " return this;\n"
  232. "}\n");
  233. printer->Annotate("{", "}", descriptor_);
  234. WriteFieldDocComment(printer, descriptor_);
  235. printer->Print(variables_,
  236. "$deprecation$public Builder ${$clear$capitalized_name$$}$() {\n"
  237. " $clear_has_field_bit_builder$\n"
  238. " $name$_ = $default_number$;\n"
  239. " $on_changed$\n"
  240. " return this;\n"
  241. "}\n");
  242. printer->Annotate("{", "}", descriptor_);
  243. }
  244. void ImmutableEnumFieldGenerator::
  245. GenerateFieldBuilderInitializationCode(io::Printer* printer) const {
  246. // noop for enums
  247. }
  248. void ImmutableEnumFieldGenerator::
  249. GenerateInitializationCode(io::Printer* printer) const {
  250. printer->Print(variables_, "$name$_ = $default_number$;\n");
  251. }
  252. void ImmutableEnumFieldGenerator::
  253. GenerateBuilderClearCode(io::Printer* printer) const {
  254. printer->Print(variables_,
  255. "$name$_ = $default_number$;\n"
  256. "$clear_has_field_bit_builder$\n");
  257. }
  258. void ImmutableEnumFieldGenerator::
  259. GenerateMergingCode(io::Printer* printer) const {
  260. if (SupportFieldPresence(descriptor_->file())) {
  261. printer->Print(variables_,
  262. "if (other.has$capitalized_name$()) {\n"
  263. " set$capitalized_name$(other.get$capitalized_name$());\n"
  264. "}\n");
  265. } else if (SupportUnknownEnumValue(descriptor_->file())) {
  266. printer->Print(variables_,
  267. "if (other.$name$_ != $default_number$) {\n"
  268. " set$capitalized_name$Value(other.get$capitalized_name$Value());\n"
  269. "}\n");
  270. } else {
  271. GOOGLE_LOG(FATAL) << "Can't reach here.";
  272. }
  273. }
  274. void ImmutableEnumFieldGenerator::
  275. GenerateBuildingCode(io::Printer* printer) const {
  276. if (SupportFieldPresence(descriptor_->file())) {
  277. printer->Print(variables_,
  278. "if ($get_has_field_bit_from_local$) {\n"
  279. " $set_has_field_bit_to_local$;\n"
  280. "}\n");
  281. }
  282. printer->Print(variables_,
  283. "result.$name$_ = $name$_;\n");
  284. }
  285. void ImmutableEnumFieldGenerator::
  286. GenerateParsingCode(io::Printer* printer) const {
  287. if (SupportUnknownEnumValue(descriptor_->file())) {
  288. printer->Print(variables_,
  289. "int rawValue = input.readEnum();\n"
  290. "$set_has_field_bit_message$\n"
  291. "$name$_ = rawValue;\n");
  292. } else {
  293. printer->Print(variables_,
  294. "int rawValue = input.readEnum();\n"
  295. " @SuppressWarnings(\"deprecation\")\n"
  296. "$type$ value = $type$.$for_number$(rawValue);\n"
  297. "if (value == null) {\n"
  298. " unknownFields.mergeVarintField($number$, rawValue);\n"
  299. "} else {\n"
  300. " $set_has_field_bit_message$\n"
  301. " $name$_ = rawValue;\n"
  302. "}\n");
  303. }
  304. }
  305. void ImmutableEnumFieldGenerator::
  306. GenerateParsingDoneCode(io::Printer* printer) const {
  307. // noop for enums
  308. }
  309. void ImmutableEnumFieldGenerator::
  310. GenerateSerializationCode(io::Printer* printer) const {
  311. printer->Print(variables_,
  312. "if ($is_field_present_message$) {\n"
  313. " output.writeEnum($number$, $name$_);\n"
  314. "}\n");
  315. }
  316. void ImmutableEnumFieldGenerator::
  317. GenerateSerializedSizeCode(io::Printer* printer) const {
  318. printer->Print(variables_,
  319. "if ($is_field_present_message$) {\n"
  320. " size += com.google.protobuf.CodedOutputStream\n"
  321. " .computeEnumSize($number$, $name$_);\n"
  322. "}\n");
  323. }
  324. void ImmutableEnumFieldGenerator::
  325. GenerateEqualsCode(io::Printer* printer) const {
  326. printer->Print(variables_,
  327. "result = result && $name$_ == other.$name$_;\n");
  328. }
  329. void ImmutableEnumFieldGenerator::
  330. GenerateHashCode(io::Printer* printer) const {
  331. printer->Print(variables_,
  332. "hash = (37 * hash) + $constant_name$;\n"
  333. "hash = (53 * hash) + $name$_;\n");
  334. }
  335. string ImmutableEnumFieldGenerator::GetBoxedType() const {
  336. return name_resolver_->GetImmutableClassName(descriptor_->enum_type());
  337. }
  338. // ===================================================================
  339. ImmutableEnumOneofFieldGenerator::
  340. ImmutableEnumOneofFieldGenerator(const FieldDescriptor* descriptor,
  341. int messageBitIndex,
  342. int builderBitIndex,
  343. Context* context)
  344. : ImmutableEnumFieldGenerator(
  345. descriptor, messageBitIndex, builderBitIndex, context) {
  346. const OneofGeneratorInfo* info =
  347. context->GetOneofGeneratorInfo(descriptor->containing_oneof());
  348. SetCommonOneofVariables(descriptor, info, &variables_);
  349. }
  350. ImmutableEnumOneofFieldGenerator::
  351. ~ImmutableEnumOneofFieldGenerator() {}
  352. void ImmutableEnumOneofFieldGenerator::
  353. GenerateMembers(io::Printer* printer) const {
  354. PrintExtraFieldInfo(variables_, printer);
  355. if (SupportFieldPresence(descriptor_->file())) {
  356. WriteFieldDocComment(printer, descriptor_);
  357. printer->Print(variables_,
  358. "$deprecation$public boolean ${$has$capitalized_name$$}$() {\n"
  359. " return $has_oneof_case_message$;\n"
  360. "}\n");
  361. printer->Annotate("{", "}", descriptor_);
  362. }
  363. if (SupportUnknownEnumValue(descriptor_->file())) {
  364. WriteFieldDocComment(printer, descriptor_);
  365. printer->Print(variables_,
  366. "$deprecation$public int ${$get$capitalized_name$Value$}$() {\n"
  367. " if ($has_oneof_case_message$) {\n"
  368. " return (java.lang.Integer) $oneof_name$_;\n"
  369. " }\n"
  370. " return $default_number$;\n"
  371. "}\n");
  372. printer->Annotate("{", "}", descriptor_);
  373. }
  374. WriteFieldDocComment(printer, descriptor_);
  375. printer->Print(variables_,
  376. "$deprecation$public $type$ ${$get$capitalized_name$$}$() {\n"
  377. " if ($has_oneof_case_message$) {\n"
  378. " @SuppressWarnings(\"deprecation\")\n"
  379. " $type$ result = $type$.$for_number$(\n"
  380. " (java.lang.Integer) $oneof_name$_);\n"
  381. " return result == null ? $unknown$ : result;\n"
  382. " }\n"
  383. " return $default$;\n"
  384. "}\n");
  385. printer->Annotate("{", "}", descriptor_);
  386. }
  387. void ImmutableEnumOneofFieldGenerator::
  388. GenerateBuilderMembers(io::Printer* printer) const {
  389. if (SupportFieldPresence(descriptor_->file())) {
  390. WriteFieldDocComment(printer, descriptor_);
  391. printer->Print(variables_,
  392. "$deprecation$public boolean ${$has$capitalized_name$$}$() {\n"
  393. " return $has_oneof_case_message$;\n"
  394. "}\n");
  395. printer->Annotate("{", "}", descriptor_);
  396. }
  397. if (SupportUnknownEnumValue(descriptor_->file())) {
  398. WriteFieldDocComment(printer, descriptor_);
  399. printer->Print(variables_,
  400. "$deprecation$public int ${$get$capitalized_name$Value$}$() {\n"
  401. " if ($has_oneof_case_message$) {\n"
  402. " return ((java.lang.Integer) $oneof_name$_).intValue();\n"
  403. " }\n"
  404. " return $default_number$;\n"
  405. "}\n");
  406. printer->Annotate("{", "}", descriptor_);
  407. WriteFieldDocComment(printer, descriptor_);
  408. printer->Print(variables_,
  409. "$deprecation$public Builder "
  410. "${$set$capitalized_name$Value$}$(int value) {\n"
  411. " $set_oneof_case_message$;\n"
  412. " $oneof_name$_ = value;\n"
  413. " $on_changed$\n"
  414. " return this;\n"
  415. "}\n");
  416. printer->Annotate("{", "}", descriptor_);
  417. }
  418. WriteFieldDocComment(printer, descriptor_);
  419. printer->Print(variables_,
  420. "$deprecation$public $type$ ${$get$capitalized_name$$}$() {\n"
  421. " if ($has_oneof_case_message$) {\n"
  422. " @SuppressWarnings(\"deprecation\")\n"
  423. " $type$ result = $type$.$for_number$(\n"
  424. " (java.lang.Integer) $oneof_name$_);\n"
  425. " return result == null ? $unknown$ : result;\n"
  426. " }\n"
  427. " return $default$;\n"
  428. "}\n");
  429. printer->Annotate("{", "}", descriptor_);
  430. WriteFieldDocComment(printer, descriptor_);
  431. printer->Print(variables_,
  432. "$deprecation$public Builder ${$set$capitalized_name$$}$($type$ value) {\n"
  433. " if (value == null) {\n"
  434. " throw new NullPointerException();\n"
  435. " }\n"
  436. " $set_oneof_case_message$;\n"
  437. " $oneof_name$_ = value.getNumber();\n"
  438. " $on_changed$\n"
  439. " return this;\n"
  440. "}\n");
  441. printer->Annotate("{", "}", descriptor_);
  442. WriteFieldDocComment(printer, descriptor_);
  443. printer->Print(variables_,
  444. "$deprecation$public Builder ${$clear$capitalized_name$$}$() {\n"
  445. " if ($has_oneof_case_message$) {\n"
  446. " $clear_oneof_case_message$;\n"
  447. " $oneof_name$_ = null;\n"
  448. " $on_changed$\n"
  449. " }\n"
  450. " return this;\n"
  451. "}\n");
  452. printer->Annotate("{", "}", descriptor_);
  453. }
  454. void ImmutableEnumOneofFieldGenerator::
  455. GenerateBuildingCode(io::Printer* printer) const {
  456. printer->Print(variables_,
  457. "if ($has_oneof_case_message$) {\n"
  458. " result.$oneof_name$_ = $oneof_name$_;\n"
  459. "}\n");
  460. }
  461. void ImmutableEnumOneofFieldGenerator::
  462. GenerateMergingCode(io::Printer* printer) const {
  463. if (SupportUnknownEnumValue(descriptor_->file())) {
  464. printer->Print(variables_,
  465. "set$capitalized_name$Value(other.get$capitalized_name$Value());\n");
  466. } else {
  467. printer->Print(variables_,
  468. "set$capitalized_name$(other.get$capitalized_name$());\n");
  469. }
  470. }
  471. void ImmutableEnumOneofFieldGenerator::
  472. GenerateParsingCode(io::Printer* printer) const {
  473. if (SupportUnknownEnumValue(descriptor_->file())) {
  474. printer->Print(variables_,
  475. "int rawValue = input.readEnum();\n"
  476. "$set_oneof_case_message$;\n"
  477. "$oneof_name$_ = rawValue;\n");
  478. } else {
  479. printer->Print(variables_,
  480. "int rawValue = input.readEnum();\n"
  481. "@SuppressWarnings(\"deprecation\")\n"
  482. "$type$ value = $type$.$for_number$(rawValue);\n"
  483. "if (value == null) {\n"
  484. " unknownFields.mergeVarintField($number$, rawValue);\n"
  485. "} else {\n"
  486. " $set_oneof_case_message$;\n"
  487. " $oneof_name$_ = rawValue;\n"
  488. "}\n");
  489. }
  490. }
  491. void ImmutableEnumOneofFieldGenerator::
  492. GenerateSerializationCode(io::Printer* printer) const {
  493. printer->Print(variables_,
  494. "if ($has_oneof_case_message$) {\n"
  495. " output.writeEnum($number$, ((java.lang.Integer) $oneof_name$_));\n"
  496. "}\n");
  497. }
  498. void ImmutableEnumOneofFieldGenerator::
  499. GenerateSerializedSizeCode(io::Printer* printer) const {
  500. printer->Print(variables_,
  501. "if ($has_oneof_case_message$) {\n"
  502. " size += com.google.protobuf.CodedOutputStream\n"
  503. " .computeEnumSize($number$, ((java.lang.Integer) $oneof_name$_));\n"
  504. "}\n");
  505. }
  506. void ImmutableEnumOneofFieldGenerator::
  507. GenerateEqualsCode(io::Printer* printer) const {
  508. if (SupportUnknownEnumValue(descriptor_->file())) {
  509. printer->Print(variables_,
  510. "result = result && get$capitalized_name$Value()\n"
  511. " == other.get$capitalized_name$Value();\n");
  512. } else {
  513. printer->Print(variables_,
  514. "result = result && get$capitalized_name$()\n"
  515. " .equals(other.get$capitalized_name$());\n");
  516. }
  517. }
  518. void ImmutableEnumOneofFieldGenerator::
  519. GenerateHashCode(io::Printer* printer) const {
  520. if (SupportUnknownEnumValue(descriptor_->file())) {
  521. printer->Print(variables_,
  522. "hash = (37 * hash) + $constant_name$;\n"
  523. "hash = (53 * hash) + get$capitalized_name$Value();\n");
  524. } else {
  525. printer->Print(variables_,
  526. "hash = (37 * hash) + $constant_name$;\n"
  527. "hash = (53 * hash) + get$capitalized_name$().getNumber();\n");
  528. }
  529. }
  530. // ===================================================================
  531. RepeatedImmutableEnumFieldGenerator::
  532. RepeatedImmutableEnumFieldGenerator(const FieldDescriptor* descriptor,
  533. int messageBitIndex,
  534. int builderBitIndex,
  535. Context* context)
  536. : descriptor_(descriptor), messageBitIndex_(messageBitIndex),
  537. builderBitIndex_(builderBitIndex), context_(context),
  538. name_resolver_(context->GetNameResolver()) {
  539. SetEnumVariables(descriptor, messageBitIndex, builderBitIndex,
  540. context->GetFieldGeneratorInfo(descriptor),
  541. name_resolver_, &variables_);
  542. }
  543. RepeatedImmutableEnumFieldGenerator::~RepeatedImmutableEnumFieldGenerator() {}
  544. int RepeatedImmutableEnumFieldGenerator::GetNumBitsForMessage() const {
  545. return 0;
  546. }
  547. int RepeatedImmutableEnumFieldGenerator::GetNumBitsForBuilder() const {
  548. return 1;
  549. }
  550. void RepeatedImmutableEnumFieldGenerator::
  551. GenerateInterfaceMembers(io::Printer* printer) const {
  552. WriteFieldDocComment(printer, descriptor_);
  553. printer->Print(variables_,
  554. "$deprecation$java.util.List<$type$> get$capitalized_name$List();\n");
  555. WriteFieldDocComment(printer, descriptor_);
  556. printer->Print(variables_,
  557. "$deprecation$int get$capitalized_name$Count();\n");
  558. WriteFieldDocComment(printer, descriptor_);
  559. printer->Print(variables_,
  560. "$deprecation$$type$ get$capitalized_name$(int index);\n");
  561. if (SupportUnknownEnumValue(descriptor_->file())) {
  562. WriteFieldDocComment(printer, descriptor_);
  563. printer->Print(variables_,
  564. "$deprecation$java.util.List<java.lang.Integer>\n"
  565. "get$capitalized_name$ValueList();\n");
  566. WriteFieldDocComment(printer, descriptor_);
  567. printer->Print(variables_,
  568. "$deprecation$int get$capitalized_name$Value(int index);\n");
  569. }
  570. }
  571. void RepeatedImmutableEnumFieldGenerator::
  572. GenerateMembers(io::Printer* printer) const {
  573. printer->Print(variables_,
  574. "private java.util.List<java.lang.Integer> $name$_;\n"
  575. "private static final com.google.protobuf.Internal.ListAdapter.Converter<\n"
  576. " java.lang.Integer, $type$> $name$_converter_ =\n"
  577. " new com.google.protobuf.Internal.ListAdapter.Converter<\n"
  578. " java.lang.Integer, $type$>() {\n"
  579. " public $type$ convert(java.lang.Integer from) {\n"
  580. " @SuppressWarnings(\"deprecation\")\n"
  581. " $type$ result = $type$.$for_number$(from);\n"
  582. " return result == null ? $unknown$ : result;\n"
  583. " }\n"
  584. " };\n");
  585. PrintExtraFieldInfo(variables_, printer);
  586. WriteFieldDocComment(printer, descriptor_);
  587. printer->Print(variables_,
  588. "$deprecation$public java.util.List<$type$> "
  589. "${$get$capitalized_name$List$}$() {\n"
  590. " return new com.google.protobuf.Internal.ListAdapter<\n"
  591. " java.lang.Integer, $type$>($name$_, $name$_converter_);\n"
  592. "}\n");
  593. printer->Annotate("{", "}", descriptor_);
  594. WriteFieldDocComment(printer, descriptor_);
  595. printer->Print(variables_,
  596. "$deprecation$public int ${$get$capitalized_name$Count$}$() {\n"
  597. " return $name$_.size();\n"
  598. "}\n");
  599. printer->Annotate("{", "}", descriptor_);
  600. WriteFieldDocComment(printer, descriptor_);
  601. printer->Print(variables_,
  602. "$deprecation$public $type$ ${$get$capitalized_name$$}$(int index) {\n"
  603. " return $name$_converter_.convert($name$_.get(index));\n"
  604. "}\n");
  605. printer->Annotate("{", "}", descriptor_);
  606. if (SupportUnknownEnumValue(descriptor_->file())) {
  607. WriteFieldDocComment(printer, descriptor_);
  608. printer->Print(variables_,
  609. "$deprecation$public java.util.List<java.lang.Integer>\n"
  610. "${$get$capitalized_name$ValueList$}$() {\n"
  611. " return $name$_;\n"
  612. "}\n");
  613. printer->Annotate("{", "}", descriptor_);
  614. WriteFieldDocComment(printer, descriptor_);
  615. printer->Print(variables_,
  616. "$deprecation$public int ${$get$capitalized_name$Value$}$(int index) {\n"
  617. " return $name$_.get(index);\n"
  618. "}\n");
  619. printer->Annotate("{", "}", descriptor_);
  620. }
  621. if (descriptor_->is_packed()) {
  622. printer->Print(variables_,
  623. "private int $name$MemoizedSerializedSize;\n");
  624. }
  625. }
  626. void RepeatedImmutableEnumFieldGenerator::
  627. GenerateBuilderMembers(io::Printer* printer) const {
  628. printer->Print(variables_,
  629. // One field is the list and the other field keeps track of whether the
  630. // list is immutable. If it's immutable, the invariant is that it must
  631. // either an instance of Collections.emptyList() or it's an ArrayList
  632. // wrapped in a Collections.unmodifiableList() wrapper and nobody else has
  633. // a refererence to the underlying ArrayList. This invariant allows us to
  634. // share instances of lists between protocol buffers avoiding expensive
  635. // memory allocations. Note, immutable is a strong guarantee here -- not
  636. // just that the list cannot be modified via the reference but that the
  637. // list can never be modified.
  638. "private java.util.List<java.lang.Integer> $name$_ =\n"
  639. " java.util.Collections.emptyList();\n"
  640. "private void ensure$capitalized_name$IsMutable() {\n"
  641. " if (!$get_mutable_bit_builder$) {\n"
  642. " $name$_ = new java.util.ArrayList<java.lang.Integer>($name$_);\n"
  643. " $set_mutable_bit_builder$;\n"
  644. " }\n"
  645. "}\n");
  646. WriteFieldDocComment(printer, descriptor_);
  647. printer->Print(variables_,
  648. // Note: We return an unmodifiable list because otherwise the caller
  649. // could hold on to the returned list and modify it after the message
  650. // has been built, thus mutating the message which is supposed to be
  651. // immutable.
  652. "$deprecation$public java.util.List<$type$> "
  653. "${$get$capitalized_name$List$}$() {\n"
  654. " return new com.google.protobuf.Internal.ListAdapter<\n"
  655. " java.lang.Integer, $type$>($name$_, $name$_converter_);\n"
  656. "}\n");
  657. printer->Annotate("{", "}", descriptor_);
  658. WriteFieldDocComment(printer, descriptor_);
  659. printer->Print(variables_,
  660. "$deprecation$public int ${$get$capitalized_name$Count$}$() {\n"
  661. " return $name$_.size();\n"
  662. "}\n");
  663. printer->Annotate("{", "}", descriptor_);
  664. WriteFieldDocComment(printer, descriptor_);
  665. printer->Print(variables_,
  666. "$deprecation$public $type$ ${$get$capitalized_name$$}$(int index) {\n"
  667. " return $name$_converter_.convert($name$_.get(index));\n"
  668. "}\n");
  669. printer->Annotate("{", "}", descriptor_);
  670. WriteFieldDocComment(printer, descriptor_);
  671. printer->Print(variables_,
  672. "$deprecation$public Builder ${$set$capitalized_name$$}$(\n"
  673. " int index, $type$ value) {\n"
  674. " if (value == null) {\n"
  675. " throw new NullPointerException();\n"
  676. " }\n"
  677. " ensure$capitalized_name$IsMutable();\n"
  678. " $name$_.set(index, value.getNumber());\n"
  679. " $on_changed$\n"
  680. " return this;\n"
  681. "}\n");
  682. printer->Annotate("{", "}", descriptor_);
  683. WriteFieldDocComment(printer, descriptor_);
  684. printer->Print(variables_,
  685. "$deprecation$public Builder ${$add$capitalized_name$$}$($type$ value) {\n"
  686. " if (value == null) {\n"
  687. " throw new NullPointerException();\n"
  688. " }\n"
  689. " ensure$capitalized_name$IsMutable();\n"
  690. " $name$_.add(value.getNumber());\n"
  691. " $on_changed$\n"
  692. " return this;\n"
  693. "}\n");
  694. printer->Annotate("{", "}", descriptor_);
  695. WriteFieldDocComment(printer, descriptor_);
  696. printer->Print(variables_,
  697. "$deprecation$public Builder ${$addAll$capitalized_name$$}$(\n"
  698. " java.lang.Iterable<? extends $type$> values) {\n"
  699. " ensure$capitalized_name$IsMutable();\n"
  700. " for ($type$ value : values) {\n"
  701. " $name$_.add(value.getNumber());\n"
  702. " }\n"
  703. " $on_changed$\n"
  704. " return this;\n"
  705. "}\n");
  706. printer->Annotate("{", "}", descriptor_);
  707. WriteFieldDocComment(printer, descriptor_);
  708. printer->Print(variables_,
  709. "$deprecation$public Builder ${$clear$capitalized_name$$}$() {\n"
  710. " $name$_ = java.util.Collections.emptyList();\n"
  711. " $clear_mutable_bit_builder$;\n"
  712. " $on_changed$\n"
  713. " return this;\n"
  714. "}\n");
  715. printer->Annotate("{", "}", descriptor_);
  716. if (SupportUnknownEnumValue(descriptor_->file())) {
  717. WriteFieldDocComment(printer, descriptor_);
  718. printer->Print(variables_,
  719. "$deprecation$public java.util.List<java.lang.Integer>\n"
  720. "${$get$capitalized_name$ValueList$}$() {\n"
  721. " return java.util.Collections.unmodifiableList($name$_);\n"
  722. "}\n");
  723. printer->Annotate("{", "}", descriptor_);
  724. WriteFieldDocComment(printer, descriptor_);
  725. printer->Print(variables_,
  726. "$deprecation$public int ${$get$capitalized_name$Value$}$(int index) {\n"
  727. " return $name$_.get(index);\n"
  728. "}\n");
  729. printer->Annotate("{", "}", descriptor_);
  730. WriteFieldDocComment(printer, descriptor_);
  731. printer->Print(variables_,
  732. "$deprecation$public Builder ${$set$capitalized_name$Value$}$(\n"
  733. " int index, int value) {\n"
  734. " ensure$capitalized_name$IsMutable();\n"
  735. " $name$_.set(index, value);\n"
  736. " $on_changed$\n"
  737. " return this;\n"
  738. "}\n");
  739. printer->Annotate("{", "}", descriptor_);
  740. WriteFieldDocComment(printer, descriptor_);
  741. printer->Print(variables_,
  742. "$deprecation$public Builder "
  743. "${$add$capitalized_name$Value$}$(int value) {\n"
  744. " ensure$capitalized_name$IsMutable();\n"
  745. " $name$_.add(value);\n"
  746. " $on_changed$\n"
  747. " return this;\n"
  748. "}\n");
  749. printer->Annotate("{", "}", descriptor_);
  750. WriteFieldDocComment(printer, descriptor_);
  751. printer->Print(variables_,
  752. "$deprecation$public Builder ${$addAll$capitalized_name$Value$}$(\n"
  753. " java.lang.Iterable<java.lang.Integer> values) {\n"
  754. " ensure$capitalized_name$IsMutable();\n"
  755. " for (int value : values) {\n"
  756. " $name$_.add(value);\n"
  757. " }\n"
  758. " $on_changed$\n"
  759. " return this;\n"
  760. "}\n");
  761. printer->Annotate("{", "}", descriptor_);
  762. }
  763. }
  764. void RepeatedImmutableEnumFieldGenerator::
  765. GenerateFieldBuilderInitializationCode(io::Printer* printer) const {
  766. // noop for enums
  767. }
  768. void RepeatedImmutableEnumFieldGenerator::
  769. GenerateInitializationCode(io::Printer* printer) const {
  770. printer->Print(variables_, "$name$_ = java.util.Collections.emptyList();\n");
  771. }
  772. void RepeatedImmutableEnumFieldGenerator::
  773. GenerateBuilderClearCode(io::Printer* printer) const {
  774. printer->Print(variables_,
  775. "$name$_ = java.util.Collections.emptyList();\n"
  776. "$clear_mutable_bit_builder$;\n");
  777. }
  778. void RepeatedImmutableEnumFieldGenerator::
  779. GenerateMergingCode(io::Printer* printer) const {
  780. // The code below does two optimizations:
  781. // 1. If the other list is empty, there's nothing to do. This ensures we
  782. // don't allocate a new array if we already have an immutable one.
  783. // 2. If the other list is non-empty and our current list is empty, we can
  784. // reuse the other list which is guaranteed to be immutable.
  785. printer->Print(variables_,
  786. "if (!other.$name$_.isEmpty()) {\n"
  787. " if ($name$_.isEmpty()) {\n"
  788. " $name$_ = other.$name$_;\n"
  789. " $clear_mutable_bit_builder$;\n"
  790. " } else {\n"
  791. " ensure$capitalized_name$IsMutable();\n"
  792. " $name$_.addAll(other.$name$_);\n"
  793. " }\n"
  794. " $on_changed$\n"
  795. "}\n");
  796. }
  797. void RepeatedImmutableEnumFieldGenerator::
  798. GenerateBuildingCode(io::Printer* printer) const {
  799. // The code below ensures that the result has an immutable list. If our
  800. // list is immutable, we can just reuse it. If not, we make it immutable.
  801. printer->Print(variables_,
  802. "if ($get_mutable_bit_builder$) {\n"
  803. " $name$_ = java.util.Collections.unmodifiableList($name$_);\n"
  804. " $clear_mutable_bit_builder$;\n"
  805. "}\n"
  806. "result.$name$_ = $name$_;\n");
  807. }
  808. void RepeatedImmutableEnumFieldGenerator::
  809. GenerateParsingCode(io::Printer* printer) const {
  810. // Read and store the enum
  811. if (SupportUnknownEnumValue(descriptor_->file())) {
  812. printer->Print(variables_,
  813. "int rawValue = input.readEnum();\n"
  814. "if (!$get_mutable_bit_parser$) {\n"
  815. " $name$_ = new java.util.ArrayList<java.lang.Integer>();\n"
  816. " $set_mutable_bit_parser$;\n"
  817. "}\n"
  818. "$name$_.add(rawValue);\n");
  819. } else {
  820. printer->Print(variables_,
  821. "int rawValue = input.readEnum();\n"
  822. "@SuppressWarnings(\"deprecation\")\n"
  823. "$type$ value = $type$.$for_number$(rawValue);\n"
  824. "if (value == null) {\n"
  825. " unknownFields.mergeVarintField($number$, rawValue);\n"
  826. "} else {\n"
  827. " if (!$get_mutable_bit_parser$) {\n"
  828. " $name$_ = new java.util.ArrayList<java.lang.Integer>();\n"
  829. " $set_mutable_bit_parser$;\n"
  830. " }\n"
  831. " $name$_.add(rawValue);\n"
  832. "}\n");
  833. }
  834. }
  835. void RepeatedImmutableEnumFieldGenerator::
  836. GenerateParsingCodeFromPacked(io::Printer* printer) const {
  837. // Wrap GenerateParsingCode's contents with a while loop.
  838. printer->Print(variables_,
  839. "int length = input.readRawVarint32();\n"
  840. "int oldLimit = input.pushLimit(length);\n"
  841. "while(input.getBytesUntilLimit() > 0) {\n");
  842. printer->Indent();
  843. GenerateParsingCode(printer);
  844. printer->Outdent();
  845. printer->Print(variables_,
  846. "}\n"
  847. "input.popLimit(oldLimit);\n");
  848. }
  849. void RepeatedImmutableEnumFieldGenerator::
  850. GenerateParsingDoneCode(io::Printer* printer) const {
  851. printer->Print(variables_,
  852. "if ($get_mutable_bit_parser$) {\n"
  853. " $name$_ = java.util.Collections.unmodifiableList($name$_);\n"
  854. "}\n");
  855. }
  856. void RepeatedImmutableEnumFieldGenerator::
  857. GenerateSerializationCode(io::Printer* printer) const {
  858. if (descriptor_->is_packed()) {
  859. printer->Print(variables_,
  860. "if (get$capitalized_name$List().size() > 0) {\n"
  861. " output.writeUInt32NoTag($tag$);\n"
  862. " output.writeUInt32NoTag($name$MemoizedSerializedSize);\n"
  863. "}\n"
  864. "for (int i = 0; i < $name$_.size(); i++) {\n"
  865. " output.writeEnumNoTag($name$_.get(i));\n"
  866. "}\n");
  867. } else {
  868. printer->Print(variables_,
  869. "for (int i = 0; i < $name$_.size(); i++) {\n"
  870. " output.writeEnum($number$, $name$_.get(i));\n"
  871. "}\n");
  872. }
  873. }
  874. void RepeatedImmutableEnumFieldGenerator::
  875. GenerateSerializedSizeCode(io::Printer* printer) const {
  876. printer->Print(variables_,
  877. "{\n"
  878. " int dataSize = 0;\n");
  879. printer->Indent();
  880. printer->Print(variables_,
  881. "for (int i = 0; i < $name$_.size(); i++) {\n"
  882. " dataSize += com.google.protobuf.CodedOutputStream\n"
  883. " .computeEnumSizeNoTag($name$_.get(i));\n"
  884. "}\n");
  885. printer->Print(
  886. "size += dataSize;\n");
  887. if (descriptor_->is_packed()) {
  888. printer->Print(variables_,
  889. "if (!get$capitalized_name$List().isEmpty()) {"
  890. " size += $tag_size$;\n"
  891. " size += com.google.protobuf.CodedOutputStream\n"
  892. " .computeUInt32SizeNoTag(dataSize);\n"
  893. "}");
  894. } else {
  895. printer->Print(variables_,
  896. "size += $tag_size$ * $name$_.size();\n");
  897. }
  898. // cache the data size for packed fields.
  899. if (descriptor_->is_packed()) {
  900. printer->Print(variables_,
  901. "$name$MemoizedSerializedSize = dataSize;\n");
  902. }
  903. printer->Outdent();
  904. printer->Print("}\n");
  905. }
  906. void RepeatedImmutableEnumFieldGenerator::
  907. GenerateEqualsCode(io::Printer* printer) const {
  908. printer->Print(variables_,
  909. "result = result && $name$_.equals(other.$name$_);\n");
  910. }
  911. void RepeatedImmutableEnumFieldGenerator::
  912. GenerateHashCode(io::Printer* printer) const {
  913. printer->Print(variables_,
  914. "if (get$capitalized_name$Count() > 0) {\n"
  915. " hash = (37 * hash) + $constant_name$;\n"
  916. " hash = (53 * hash) + $name$_.hashCode();\n"
  917. "}\n");
  918. }
  919. string RepeatedImmutableEnumFieldGenerator::GetBoxedType() const {
  920. return name_resolver_->GetImmutableClassName(descriptor_->enum_type());
  921. }
  922. } // namespace java
  923. } // namespace compiler
  924. } // namespace protobuf
  925. } // namespace google