talksik/mediapipe
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Project import generated by Copybara.

Browse Source 
GitOrigin-RevId: d8caa66de45839696f5bd0786ad3bfbcb9cff632
This commit is contained in:
MediaPipe Team
2020-12-09 22:43:33 -05:00
committed by chuoling
parent f15da632de
commit 2b58cceec9
750 changed files with 22901 additions and 9478 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mediapipe/framework/calculator_node.cc
+38 -38
View File
@@ -73,7 +73,7 @@ Timestamp CalculatorNode::SourceProcessOrder(
return calculator_->SourceProcessOrder(cc);
}
::mediapipe::Status CalculatorNode::Initialize(
mediapipe::Status CalculatorNode::Initialize(
const ValidatedGraphConfig* validated_graph, int node_id,
InputStreamManager* input_stream_managers,
OutputStreamManager* output_stream_managers,
@@ -159,7 +159,7 @@ Timestamp CalculatorNode::SourceProcessOrder(
return InitializeInputStreams(input_stream_managers, output_stream_managers);
}
::mediapipe::Status CalculatorNode::InitializeOutputSidePackets(
mediapipe::Status CalculatorNode::InitializeOutputSidePackets(
const PacketTypeSet& output_side_packet_types,
OutputSidePacketImpl* output_side_packets) {
output_side_packets_ =
@@ -173,10 +173,10 @@ Timestamp CalculatorNode::SourceProcessOrder(
output_side_packets_->GetPtr(id) =
&output_side_packets[base_index + id.value()];
}
return ::mediapipe::OkStatus();
return mediapipe::OkStatus();
}
::mediapipe::Status CalculatorNode::InitializeInputSidePackets(
mediapipe::Status CalculatorNode::InitializeInputSidePackets(
OutputSidePacketImpl* output_side_packets) {
const NodeTypeInfo& node_type_info =
validated_graph_->CalculatorInfos()[node_id_];
@@ -201,10 +201,10 @@ Timestamp CalculatorNode::SourceProcessOrder(
<< output_side_packet_index;
origin_output_side_packet->AddMirror(&input_side_packet_handler_, id);
}
return ::mediapipe::OkStatus();
return mediapipe::OkStatus();
}
::mediapipe::Status CalculatorNode::InitializeOutputStreams(
mediapipe::Status CalculatorNode::InitializeOutputStreams(
OutputStreamManager* output_stream_managers) {
RET_CHECK(output_stream_managers) << "output_stream_managers is NULL";
const NodeTypeInfo& node_type_info =
@@ -216,7 +216,7 @@ Timestamp CalculatorNode::SourceProcessOrder(
current_output_stream_managers);
}
::mediapipe::Status CalculatorNode::InitializeInputStreams(
mediapipe::Status CalculatorNode::InitializeInputStreams(
InputStreamManager* input_stream_managers,
OutputStreamManager* output_stream_managers) {
RET_CHECK(input_stream_managers) << "input_stream_managers is NULL";
@@ -247,10 +247,10 @@ Timestamp CalculatorNode::SourceProcessOrder(
<< output_stream_index;
origin_output_stream_manager->AddMirror(input_stream_handler_.get(), id);
}
return ::mediapipe::OkStatus();
return mediapipe::OkStatus();
}
::mediapipe::Status CalculatorNode::InitializeInputStreamHandler(
mediapipe::Status CalculatorNode::InitializeInputStreamHandler(
const InputStreamHandlerConfig& handler_config,
const PacketTypeSet& input_stream_types) {
const ProtoString& input_stream_handler_name =
@@ -265,10 +265,10 @@ Timestamp CalculatorNode::SourceProcessOrder(
_ << "\"" << input_stream_handler_name
<< "\" is not a registered input stream handler.");
return ::mediapipe::OkStatus();
return mediapipe::OkStatus();
}
::mediapipe::Status CalculatorNode::InitializeOutputStreamHandler(
mediapipe::Status CalculatorNode::InitializeOutputStreamHandler(
const OutputStreamHandlerConfig& handler_config,
const PacketTypeSet& output_stream_types) {
const ProtoString& output_stream_handler_name =
@@ -282,10 +282,10 @@ Timestamp CalculatorNode::SourceProcessOrder(
/*calculator_run_in_parallel=*/max_in_flight_ > 1),
_ << "\"" << output_stream_handler_name
<< "\" is not a registered output stream handler.");
return ::mediapipe::OkStatus();
return mediapipe::OkStatus();
}
::mediapipe::Status CalculatorNode::ConnectShardsToStreams(
mediapipe::Status CalculatorNode::ConnectShardsToStreams(
CalculatorContext* calculator_context) {
RET_CHECK(calculator_context);
MP_RETURN_IF_ERROR(
@@ -325,13 +325,13 @@ void CalculatorNode::SetMaxInputStreamQueueSize(int max_queue_size) {
input_stream_handler_->SetMaxQueueSize(max_queue_size);
}
::mediapipe::Status CalculatorNode::PrepareForRun(
mediapipe::Status CalculatorNode::PrepareForRun(
const std::map<std::string, Packet>& all_side_packets,
const std::map<std::string, Packet>& service_packets,
std::function<void()> ready_for_open_callback,
std::function<void()> source_node_opened_callback,
std::function<void(CalculatorContext*)> schedule_callback,
std::function<void(::mediapipe::Status)> error_callback,
std::function<void(mediapipe::Status)> error_callback,
CounterFactory* counter_factory) {
RET_CHECK(ready_for_open_callback) << "ready_for_open_callback is NULL";
RET_CHECK(schedule_callback) << "schedule_callback is NULL";
@@ -398,7 +398,7 @@ void CalculatorNode::SetMaxInputStreamQueueSize(int max_queue_size) {
input_side_packets_ready_ =
(input_side_packet_handler_.MissingInputSidePacketCount() == 0);
}
return ::mediapipe::OkStatus();
return mediapipe::OkStatus();
}
namespace {
@@ -410,7 +410,7 @@ const Packet GetPacket(const OutputSidePacket& out) {
}
// Resends the output-side-packets from the previous graph run.
::mediapipe::Status ResendSidePackets(CalculatorContext* cc) {
mediapipe::Status ResendSidePackets(CalculatorContext* cc) {
auto& outs = cc->OutputSidePackets();
for (CollectionItemId id = outs.BeginId(); id < outs.EndId(); ++id) {
Packet packet = GetPacket(outs.Get(id));
@@ -419,7 +419,7 @@ const Packet GetPacket(const OutputSidePacket& out) {
outs.Get(id).Set(packet);
}
}
return ::mediapipe::OkStatus();
return mediapipe::OkStatus();
}
} // namespace
@@ -433,7 +433,7 @@ bool CalculatorNode::OutputsAreConstant(CalculatorContext* cc) {
return true;
}
::mediapipe::Status CalculatorNode::OpenNode() {
mediapipe::Status CalculatorNode::OpenNode() {
VLOG(2) << "CalculatorNode::OpenNode() for " << DebugName();
CalculatorContext* default_context =
@@ -448,7 +448,7 @@ bool CalculatorNode::OutputsAreConstant(CalculatorContext* cc) {
calculator_context_manager_.PushInputTimestampToContext(
default_context, Timestamp::Unstarted());
::mediapipe::Status result;
mediapipe::Status result;
if (OutputsAreConstant(default_context)) {
result = ResendSidePackets(default_context);
} else {
@@ -493,7 +493,7 @@ bool CalculatorNode::OutputsAreConstant(CalculatorContext* cc) {
status_ = kStateOpened;
}
return ::mediapipe::OkStatus();
return mediapipe::OkStatus();
}
void CalculatorNode::ActivateNode() {
@@ -527,8 +527,8 @@ void CalculatorNode::CloseOutputStreams(OutputStreamShardSet* outputs) {
output_stream_handler_->Close(outputs);
}
::mediapipe::Status CalculatorNode::CloseNode(
const ::mediapipe::Status& graph_status, bool graph_run_ended) {
mediapipe::Status CalculatorNode::CloseNode(
const mediapipe::Status& graph_status, bool graph_run_ended) {
{
absl::MutexLock status_lock(&status_mutex_);
RET_CHECK_NE(status_, kStateClosed)
@@ -548,11 +548,11 @@ void CalculatorNode::CloseOutputStreams(OutputStreamShardSet* outputs) {
calculator_context_manager_.SetGraphStatusInContext(default_context,
graph_status);
::mediapipe::Status result;
mediapipe::Status result;
if (OutputsAreConstant(default_context)) {
// Do nothing.
result = ::mediapipe::OkStatus();
result = mediapipe::OkStatus();
} else {
MEDIAPIPE_PROFILING(CLOSE, default_context);
LegacyCalculatorSupport::Scoped<CalculatorContext> s(default_context);
@@ -582,10 +582,10 @@ void CalculatorNode::CloseOutputStreams(OutputStreamShardSet* outputs) {
"Calculator::Close() for node \"$0\" failed: ", DebugName());
VLOG(2) << "Closed node " << DebugName();
return ::mediapipe::OkStatus();
return mediapipe::OkStatus();
}
void CalculatorNode::CleanupAfterRun(const ::mediapipe::Status& graph_status) {
void CalculatorNode::CleanupAfterRun(const mediapipe::Status& graph_status) {
if (needs_to_close_) {
calculator_context_manager_.PushInputTimestampToContext(
calculator_context_manager_.GetDefaultCalculatorContext(),
@@ -754,12 +754,12 @@ std::string CalculatorNode::DebugName() const {
}
// TODO: Split this function.
::mediapipe::Status CalculatorNode::ProcessNode(
mediapipe::Status CalculatorNode::ProcessNode(
CalculatorContext* calculator_context) {
if (IsSource()) {
// This is a source Calculator.
if (Closed()) {
return ::mediapipe::OkStatus();
return mediapipe::OkStatus();
}
const Timestamp input_timestamp = calculator_context->InputTimestamp();
@@ -768,7 +768,7 @@ std::string CalculatorNode::DebugName() const {
output_stream_handler_->PrepareOutputs(input_timestamp, outputs);
VLOG(2) << "Calling Calculator::Process() for node: " << DebugName();
::mediapipe::Status result;
mediapipe::Status result;
{
MEDIAPIPE_PROFILING(PROCESS, calculator_context);
@@ -782,7 +782,7 @@ std::string CalculatorNode::DebugName() const {
// Needs to call CloseNode().
node_stopped = true;
} else {
return ::mediapipe::StatusBuilder(result, MEDIAPIPE_LOC).SetPrepend()
return mediapipe::StatusBuilder(result, MEDIAPIPE_LOC).SetPrepend()
<< absl::Substitute(
"Calculator::Process() for node \"$0\" failed: ",
DebugName());
@@ -791,15 +791,15 @@ std::string CalculatorNode::DebugName() const {
output_stream_handler_->PostProcess(input_timestamp);
if (node_stopped) {
MP_RETURN_IF_ERROR(
CloseNode(::mediapipe::OkStatus(), /*graph_run_ended=*/false));
CloseNode(mediapipe::OkStatus(), /*graph_run_ended=*/false));
}
return ::mediapipe::OkStatus();
return mediapipe::OkStatus();
} else {
// This is not a source Calculator.
InputStreamShardSet* const inputs = &calculator_context->Inputs();
OutputStreamShardSet* const outputs = &calculator_context->Outputs();
::mediapipe::Status result =
::mediapipe::InternalError("Calculator context has no input packets.");
mediapipe::Status result =
mediapipe::InternalError("Calculator context has no input packets.");
int num_invocations = calculator_context_manager_.NumberOfContextTimestamps(
*calculator_context);
@@ -818,7 +818,7 @@ std::string CalculatorNode::DebugName() const {
if (OutputsAreConstant(calculator_context)) {
// Do nothing.
result = ::mediapipe::OkStatus();
result = mediapipe::OkStatus();
} else {
MEDIAPIPE_PROFILING(PROCESS, calculator_context);
LegacyCalculatorSupport::Scoped<CalculatorContext> s(
@@ -838,7 +838,7 @@ std::string CalculatorNode::DebugName() const {
// ensure that all sources will be closed and that packets in input
// streams will be processed before the graph is terminated.
if (!result.ok() && result != tool::StatusStop()) {
return ::mediapipe::StatusBuilder(result, MEDIAPIPE_LOC).SetPrepend()
return mediapipe::StatusBuilder(result, MEDIAPIPE_LOC).SetPrepend()
<< absl::Substitute(
"Calculator::Process() for node \"$0\" failed: ",
DebugName());
@@ -855,7 +855,7 @@ std::string CalculatorNode::DebugName() const {
CHECK_EQ(calculator_context_manager_.NumberOfContextTimestamps(
*calculator_context),
1);
return CloseNode(::mediapipe::OkStatus(), /*graph_run_ended=*/false);
return CloseNode(mediapipe::OkStatus(), /*graph_run_ended=*/false);
} else {
RET_CHECK_FAIL()
<< "Invalid input timestamp in ProcessNode(). timestamp: "