chromium/third_party/mediapipe/src/mediapipe/framework/tool/options_registry.cc

#include "mediapipe/framework/tool/options_registry.h"

#include <string>
#include <vector>

#include "absl/synchronization/mutex.h"
#include "mediapipe/framework/port/ret_check.h"
#include "mediapipe/framework/tool/proto_util_lite.h"

namespace mediapipe {
namespace tool {

namespace {

// Returns a canonical message type name, with any leading "." removed.
std::string CanonicalTypeName(const std::string& type_name) { … }

// Returns the values from a protobuf field as typed FieldData.
absl::StatusOr<std::vector<FieldData>> GetFieldValues(
    const FieldData& message_data, std::string field_name) { … }

// Returns a single value from a protobuf string field.
std::string GetFieldString(const FieldData& message_data,
                           std::string field_name) { … }

// Registers the descriptors for the descriptor protobufs.  These four
// descriptors are required to deserialize descriptors for other protobufs.
// This implementation avoids a code size problem introduced by
// proto_ns::DescriptorProto.
void RegisterDescriptorProtos(
    absl::flat_hash_map<std::string, Descriptor>& result) { … }

}  // namespace

RegistrationToken OptionsRegistry::Register(
    const FieldData& file_descriptor_set) { … }

void OptionsRegistry::Register(const FieldData& message_type,
                               const std::string& parent_name) { … }

const Descriptor* OptionsRegistry::GetProtobufDescriptor(
    const std::string& type_name) { … }

void OptionsRegistry::FindAllExtensions(
    absl::string_view extendee, std::vector<const FieldDescriptor*>* result) { … }

absl::flat_hash_map<std::string, Descriptor>& OptionsRegistry::descriptors() { … }

absl::flat_hash_map<std::string, std::vector<FieldDescriptor>>&
OptionsRegistry::extensions() { … }

absl::Mutex& OptionsRegistry::mutex() { … }

Descriptor::Descriptor(const std::string& full_name,
                       const FieldData& descriptor_proto)
    : … { … }

Descriptor::Descriptor(const std::string& full_name,
                       const std::vector<FieldDescriptor>& fields)
    : … { … }

const std::string& Descriptor::full_name() const { … }

const FieldDescriptor* Descriptor::FindFieldByName(
    const std::string& name) const { … }

FieldDescriptor::FieldDescriptor(const FieldData& field_proto) { … }

FieldDescriptor::FieldDescriptor(std::string name, int number, FieldType type,
                                 std::string message_type)
    : … { … }

const std::string& FieldDescriptor::name() const { … }

int FieldDescriptor::number() const { … }

FieldType FieldDescriptor::type() const { … }

const Descriptor* FieldDescriptor::message_type() const { … }

}  // namespace tool
}  // namespace mediapipe