forked from google/or-tools
-
Notifications
You must be signed in to change notification settings - Fork 0
/
proto_tools.cc
64 lines (58 loc) · 2.43 KB
/
proto_tools.cc
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
// Copyright 2010-2021 Google LLC
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
#include "ortools/util/proto_tools.h"
#include "absl/strings/str_cat.h"
#include "google/protobuf/descriptor.h"
#include "google/protobuf/message.h"
#include "google/protobuf/text_format.h"
namespace operations_research {
namespace {
using ::google::protobuf::Descriptor;
using ::google::protobuf::FieldDescriptor;
using ::google::protobuf::Reflection;
using ::google::protobuf::TextFormat;
void WriteFullProtocolMessage(const google::protobuf::Message& message,
int indent_level, std::string* out) {
std::string temp_string;
const std::string indent(indent_level * 2, ' ');
const Descriptor* desc = message.GetDescriptor();
const Reflection* refl = message.GetReflection();
for (int i = 0; i < desc->field_count(); ++i) {
const FieldDescriptor* fd = desc->field(i);
const bool repeated = fd->is_repeated();
const int start = repeated ? 0 : -1;
const int limit = repeated ? refl->FieldSize(message, fd) : 0;
for (int j = start; j < limit; ++j) {
absl::StrAppend(out, indent, fd->name());
if (fd->cpp_type() == FieldDescriptor::CPPTYPE_MESSAGE) {
absl::StrAppend(out, " {\n");
const google::protobuf::Message& nested_message =
repeated ? refl->GetRepeatedMessage(message, fd, j)
: refl->GetMessage(message, fd);
WriteFullProtocolMessage(nested_message, indent_level + 1, out);
absl::StrAppend(out, indent, "}\n");
} else {
TextFormat::PrintFieldValueToString(message, fd, j, &temp_string);
absl::StrAppend(out, ": ", temp_string, "\n");
}
}
}
}
} // namespace
std::string FullProtocolMessageAsString(
const google::protobuf::Message& message, int indent_level) {
std::string message_str;
WriteFullProtocolMessage(message, indent_level, &message_str);
return message_str;
}
} // namespace operations_research