Index: third_party/protobuf/src/google/protobuf/compiler/java/java_extension.h |
diff --git a/third_party/protobuf/src/google/protobuf/compiler/java/java_extension.h b/third_party/protobuf/src/google/protobuf/compiler/java/java_extension.h |
index 009ed9ffacaf207d4fb6322dca43697499b32991..bdd4226379b04a81cc8b4cbd5ab0c6b803421de8 100644 |
--- a/third_party/protobuf/src/google/protobuf/compiler/java/java_extension.h |
+++ b/third_party/protobuf/src/google/protobuf/compiler/java/java_extension.h |
@@ -1,6 +1,6 @@ |
// Protocol Buffers - Google's data interchange format |
// Copyright 2008 Google Inc. All rights reserved. |
-// http://code.google.com/p/protobuf/ |
+// https://developers.google.com/protocol-buffers/ |
// |
// Redistribution and use in source and binary forms, with or without |
// modification, are permitted provided that the following conditions are |
@@ -35,6 +35,7 @@ |
#ifndef GOOGLE_PROTOBUF_COMPILER_JAVA_EXTENSION_H__ |
#define GOOGLE_PROTOBUF_COMPILER_JAVA_EXTENSION_H__ |
+#include <map> |
#include <string> |
#include <google/protobuf/stubs/common.h> |
@@ -42,6 +43,12 @@ |
namespace google { |
namespace protobuf { |
class FieldDescriptor; // descriptor.h |
+ namespace compiler { |
+ namespace java { |
+ class Context; // context.h |
+ class ClassNameResolver; // name_resolver.h |
+ } |
+ } |
namespace io { |
class Printer; // printer.h |
} |
@@ -56,17 +63,46 @@ namespace java { |
// since extensions are just simple identifiers with interesting types. |
class ExtensionGenerator { |
public: |
- explicit ExtensionGenerator(const FieldDescriptor* descriptor); |
- ~ExtensionGenerator(); |
+ explicit ExtensionGenerator() {} |
+ virtual ~ExtensionGenerator() {} |
+ |
+ virtual void Generate(io::Printer* printer) = 0; |
+ |
+ // Returns an estimate of the number of bytes the printed code will compile to |
+ virtual int GenerateNonNestedInitializationCode(io::Printer* printer) = 0; |
+ |
+ // Returns an estimate of the number of bytes the printed code will compile to |
+ virtual int GenerateRegistrationCode(io::Printer* printer) = 0; |
- void Generate(io::Printer* printer); |
- void GenerateNonNestedInitializationCode(io::Printer* printer); |
- void GenerateRegistrationCode(io::Printer* printer); |
+ protected: |
+ static void InitTemplateVars(const FieldDescriptor* descriptor, |
+ const string& scope, |
+ bool immutable, |
+ ClassNameResolver* name_resolver, |
+ map<string, string>* vars_pointer); |
private: |
+ GOOGLE_DISALLOW_EVIL_CONSTRUCTORS(ExtensionGenerator); |
+}; |
+ |
+class ImmutableExtensionGenerator : public ExtensionGenerator { |
+ public: |
+ explicit ImmutableExtensionGenerator(const FieldDescriptor* descriptor, |
+ Context* context); |
+ virtual ~ImmutableExtensionGenerator(); |
+ |
+ virtual void Generate(io::Printer* printer); |
+ virtual int GenerateNonNestedInitializationCode(io::Printer* printer); |
+ virtual int GenerateRegistrationCode(io::Printer* printer); |
+ |
+ protected: |
const FieldDescriptor* descriptor_; |
+ Context* context_; |
+ ClassNameResolver* name_resolver_; |
string scope_; |
- GOOGLE_DISALLOW_EVIL_CONSTRUCTORS(ExtensionGenerator); |
+ |
+ private: |
+ GOOGLE_DISALLOW_EVIL_CONSTRUCTORS(ImmutableExtensionGenerator); |
}; |
} // namespace java |
@@ -74,4 +110,4 @@ class ExtensionGenerator { |
} // namespace protobuf |
} // namespace google |
-#endif // GOOGLE_PROTOBUF_COMPILER_JAVA_MESSAGE_H__ |
+#endif // GOOGLE_PROTOBUF_COMPILER_JAVA_EXTENSION_H__ |