• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
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 
31 // Author: kenton@google.com (Kenton Varda)
32 //  Based on original Protocol Buffers design by
33 //  Sanjay Ghemawat, Jeff Dean, and others.
34 
35 #ifndef GOOGLE_PROTOBUF_COMPILER_JAVA_MESSAGE_H__
36 #define GOOGLE_PROTOBUF_COMPILER_JAVA_MESSAGE_H__
37 
38 #include <string>
39 #include <map>
40 #include <google/protobuf/compiler/java/java_field.h>
41 
42 namespace google {
43 namespace protobuf {
44   namespace compiler {
45     namespace java {
46       class Context;           // context.h
47       class ClassNameResolver; // name_resolver.h
48     }
49   }
50   namespace io {
51     class Printer;             // printer.h
52   }
53 }
54 
55 namespace protobuf {
56 namespace compiler {
57 namespace java {
58 
59 static const int kMaxStaticSize = 1 << 15;  // aka 32k
60 
61 class MessageGenerator {
62  public:
63   explicit MessageGenerator(const Descriptor* descriptor);
64   virtual ~MessageGenerator();
65 
66   // All static variables have to be declared at the top-level of the file
67   // so that we can control initialization order, which is important for
68   // DescriptorProto bootstrapping to work.
69   virtual void GenerateStaticVariables(
70       io::Printer* printer, int* bytecode_estimate) = 0;
71 
72   // Output code which initializes the static variables generated by
73   // GenerateStaticVariables(). Returns an estimate of bytecode size.
74   virtual int GenerateStaticVariableInitializers(io::Printer* printer) = 0;
75 
76   // Generate the class itself.
77   virtual void Generate(io::Printer* printer) = 0;
78 
79   // Generates the base interface that both the class and its builder implement
80   virtual void GenerateInterface(io::Printer* printer) = 0;
81 
82   // Generate code to register all contained extensions with an
83   // ExtensionRegistry.
84   virtual void GenerateExtensionRegistrationCode(io::Printer* printer) = 0;
85 
86  protected:
87   const Descriptor* descriptor_;
88 
89  private:
90   GOOGLE_DISALLOW_EVIL_CONSTRUCTORS(MessageGenerator);
91 };
92 
93 class ImmutableMessageGenerator : public MessageGenerator {
94  public:
95   explicit ImmutableMessageGenerator(const Descriptor* descriptor,
96                                      Context* context);
97   virtual ~ImmutableMessageGenerator();
98 
99   virtual void Generate(io::Printer* printer);
100   virtual void GenerateInterface(io::Printer* printer);
101   virtual void GenerateExtensionRegistrationCode(io::Printer* printer);
102   virtual void GenerateStaticVariables(
103       io::Printer* printer, int* bytecode_estimate);
104 
105   // Returns an estimate of the number of bytes the printed code will compile to
106   virtual int GenerateStaticVariableInitializers(io::Printer* printer);
107 
108  private:
109 
110   void GenerateFieldAccessorTable(io::Printer* printer, int* bytecode_estimate);
111 
112   // Returns an estimate of the number of bytes the printed code will compile to
113   int GenerateFieldAccessorTableInitializer(io::Printer* printer);
114 
115   void GenerateMessageSerializationMethods(io::Printer* printer);
116   void GenerateParseFromMethods(io::Printer* printer);
117   void GenerateSerializeOneField(io::Printer* printer,
118                                  const FieldDescriptor* field);
119   void GenerateSerializeOneExtensionRange(
120       io::Printer* printer, const Descriptor::ExtensionRange* range);
121 
122   void GenerateBuilder(io::Printer* printer);
123   void GenerateIsInitialized(io::Printer* printer);
124   void GenerateDescriptorMethods(io::Printer* printer);
125   void GenerateInitializers(io::Printer* printer);
126   void GenerateEqualsAndHashCode(io::Printer* printer);
127   void GenerateParser(io::Printer* printer);
128   void GenerateParsingConstructor(io::Printer* printer);
129   void GenerateAnyMethods(io::Printer* printer);
130 
131   Context* context_;
132   ClassNameResolver* name_resolver_;
133   FieldGeneratorMap<ImmutableFieldGenerator> field_generators_;
134 
135   GOOGLE_DISALLOW_EVIL_CONSTRUCTORS(ImmutableMessageGenerator);
136 };
137 
138 }  // namespace java
139 }  // namespace compiler
140 }  // namespace protobuf
141 
142 }  // namespace google
143 #endif  // GOOGLE_PROTOBUF_COMPILER_JAVA_MESSAGE_H__
144