74 lines
2.5 KiB
C
74 lines
2.5 KiB
C
|
// Protocol Buffers - Google's data interchange format
|
||
|
|
// Copyright 2008 Google Inc. All rights reserved.
|
||
|
|
//
|
||
|
|
// Use of this source code is governed by a BSD-style
|
||
|
|
// license that can be found in the LICENSE file or at
|
||
|
|
// https://developers.google.com/open-source/licenses/bsd
|
||
|
|
|
||
|
|
#ifndef GOOGLE_PROTOBUF_COMPILER_OBJECTIVEC_MESSAGE_H__
|
||
|
|
#define GOOGLE_PROTOBUF_COMPILER_OBJECTIVEC_MESSAGE_H__
|
||
|
|
|
||
|
|
#include <cstddef>
|
||
|
|
#include <memory>
|
||
|
|
#include <string>
|
||
|
|
#include <vector>
|
||
|
|
|
||
|
|
#include "absl/container/btree_set.h"
|
||
|
|
#include "absl/container/flat_hash_set.h"
|
||
|
|
#include "google/protobuf/compiler/objectivec/field.h"
|
||
|
|
#include "google/protobuf/compiler/objectivec/oneof.h"
|
||
|
|
#include "google/protobuf/compiler/objectivec/options.h"
|
||
|
|
#include "google/protobuf/descriptor.h"
|
||
|
|
#include "google/protobuf/io/printer.h"
|
||
|
|
|
||
|
|
namespace google {
|
||
|
|
namespace protobuf {
|
||
|
|
namespace compiler {
|
||
|
|
namespace objectivec {
|
||
|
|
|
||
|
|
class ExtensionGenerator;
|
||
|
|
|
||
|
|
class MessageGenerator {
|
||
|
|
public:
|
||
|
|
MessageGenerator(const std::string& file_description_name,
|
||
|
|
const Descriptor* descriptor,
|
||
|
|
const GenerationOptions& generation_options);
|
||
|
|
~MessageGenerator() = default;
|
||
|
|
|
||
|
|
MessageGenerator(const MessageGenerator&) = delete;
|
||
|
|
MessageGenerator& operator=(const MessageGenerator&) = delete;
|
||
|
|
|
||
|
|
void AddExtensionGenerators(
|
||
|
|
std::vector<std::unique_ptr<ExtensionGenerator>>* extension_generators);
|
||
|
|
|
||
|
|
void GenerateMessageHeader(io::Printer* printer) const;
|
||
|
|
void GenerateSource(io::Printer* printer) const;
|
||
|
|
void DetermineObjectiveCClassDefinitions(
|
||
|
|
absl::btree_set<std::string>* fwd_decls) const;
|
||
|
|
void DetermineForwardDeclarations(absl::btree_set<std::string>* fwd_decls,
|
||
|
|
bool include_external_types) const;
|
||
|
|
void DetermineNeededFiles(
|
||
|
|
absl::flat_hash_set<const FileDescriptor*>* deps) const;
|
||
|
|
|
||
|
|
// Checks if the message or a nested message includes a oneof definition.
|
||
|
|
bool IncludesOneOfDefinition() const { return !oneof_generators_.empty(); }
|
||
|
|
|
||
|
|
private:
|
||
|
|
const std::string file_description_name_;
|
||
|
|
const Descriptor* descriptor_;
|
||
|
|
const GenerationOptions& generation_options_;
|
||
|
|
FieldGeneratorMap field_generators_;
|
||
|
|
const std::string class_name_;
|
||
|
|
const std::string deprecated_attribute_;
|
||
|
|
std::vector<const ExtensionGenerator*> extension_generators_;
|
||
|
|
std::vector<std::unique_ptr<OneofGenerator>> oneof_generators_;
|
||
|
|
size_t sizeof_has_storage_;
|
||
|
|
};
|
||
|
|
|
||
|
|
} // namespace objectivec
|
||
|
|
} // namespace compiler
|
||
|
|
} // namespace protobuf
|
||
|
|
} // namespace google
|
||
|
|
|
||
|
|
#endif // GOOGLE_PROTOBUF_COMPILER_OBJECTIVEC_MESSAGE_H__
|