Index: third_party/protobuf/src/google/protobuf/compiler/cpp/cpp_message.h |
diff --git a/third_party/protobuf/src/google/protobuf/compiler/cpp/cpp_message.h b/third_party/protobuf/src/google/protobuf/compiler/cpp/cpp_message.h |
index 8e19a3f0557379a75b16ec7dcde5ece6b04b8793..31223e13ff4d7395940a4fb04cd0c48153f2303b 100644 |
--- a/third_party/protobuf/src/google/protobuf/compiler/cpp/cpp_message.h |
+++ b/third_party/protobuf/src/google/protobuf/compiler/cpp/cpp_message.h |
@@ -67,9 +67,13 @@ class MessageGenerator { |
// Header stuff. |
// Return names for foward declarations of this class and all its nested |
- // types. |
- void FillMessageForwardDeclarations(set<string>* class_names); |
- void FillEnumForwardDeclarations(set<string>* enum_names); |
+ // types. A given key in {class,enum}_names will map from a class name to the |
+ // descriptor that was responsible for its inclusion in the map. This can be |
+ // used to associate the descriptor with the code generated for it. |
+ void FillMessageForwardDeclarations( |
+ map<string, const Descriptor*>* class_names); |
+ void FillEnumForwardDeclarations( |
+ map<string, const EnumDescriptor*>* enum_names); |
// Generate definitions of all nested enums (must come before class |
// definitions because those classes use the enums definitions). |