cassdlcm/mediapipe
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Internal change

Browse Source 
PiperOrigin-RevId: 513854481
This commit is contained in:
Hadon Nash 2023-03-03 09:57:34 -08:00 committed by Copybara-Service
parent 3837c92fd5
commit c9c1bf21ae
1 changed files with 8 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
mediapipe/framework/tool/template_parser.cc
View File

@ -1676,15 +1676,18 @@ class TemplateParser::Parser::MediaPipeParserImpl
} }
} }
// Inserts one value into the specified field. // Appends one value to the specified field.
static void InsertFieldValue( static void AppendFieldValue(
Message* message, const FieldDescriptor* field, Message* message, const FieldDescriptor* field,
const std::vector<ProtoUtilLite::FieldValue>& args) { const std::vector<ProtoUtilLite::FieldValue>& args) {
auto field_type = static_cast<ProtoUtilLite::FieldType>(field->type()); auto field_type = static_cast<ProtoUtilLite::FieldType>(field->type());
ProtoUtilLite::FieldValue message_bytes; ProtoUtilLite::FieldValue message_bytes;
CHECK(message->SerializePartialToString(&message_bytes)); CHECK(message->SerializePartialToString(&message_bytes));
int count;
MEDIAPIPE_CHECK_OK(ProtoUtilLite::GetFieldCount(
message_bytes, {{field->number(), 0}}, field_type, &count));
MEDIAPIPE_CHECK_OK(ProtoUtilLite::ReplaceFieldRange( MEDIAPIPE_CHECK_OK(ProtoUtilLite::ReplaceFieldRange(
&message_bytes, {{field->number(), 0}}, 0, field_type, args)); &message_bytes, {{field->number(), count}}, 0, field_type, args));
CHECK(message->ParsePartialFromString(message_bytes)); CHECK(message->ParsePartialFromString(message_bytes));
} }
@ -1701,7 +1704,7 @@ class TemplateParser::Parser::MediaPipeParserImpl
// Leave a dummy value in place of the consumed field. // Leave a dummy value in place of the consumed field.
std::vector<ProtoUtilLite::FieldValue> args; std::vector<ProtoUtilLite::FieldValue> args;
GetEmptyFieldValue(field, &args); GetEmptyFieldValue(field, &args);
InsertFieldValue(message, field, args); AppendFieldValue(message, field, args);
return true; return true;
} }
@ -1718,7 +1721,7 @@ class TemplateParser::Parser::MediaPipeParserImpl
// Leave a dummy value in place of the consumed field. // Leave a dummy value in place of the consumed field.
std::vector<ProtoUtilLite::FieldValue> args; std::vector<ProtoUtilLite::FieldValue> args;
GetEmptyFieldValue(field, &args); GetEmptyFieldValue(field, &args);
InsertFieldValue(message, field, args); AppendFieldValue(message, field, args);
return true; return true;
} }