chromium/third_party/mediapipe/src/mediapipe/framework/scheduler_queue.cc

// Copyright 2019 The MediaPipe Authors.
//
// 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 "mediapipe/framework/scheduler_queue.h"

#include <memory>
#include <queue>
#include <utility>

#include "absl/log/absl_check.h"
#include "absl/synchronization/mutex.h"
#include "mediapipe/framework/calculator_node.h"
#include "mediapipe/framework/executor.h"
#include "mediapipe/framework/port/canonical_errors.h"
#include "mediapipe/framework/port/logging.h"
#include "mediapipe/framework/port/status.h"

#ifdef __APPLE__
#define AUTORELEASEPOOL …
#else
#define AUTORELEASEPOOL
#endif  // __APPLE__

namespace mediapipe {
namespace internal {

SchedulerQueue::Item::Item(CalculatorNode* node, CalculatorContext* cc)
    : … { … }

SchedulerQueue::Item::Item(CalculatorNode* node)
    : … { … }

// Returning true means "this runs after that".
bool SchedulerQueue::Item::operator<(const SchedulerQueue::Item& that) const { … }

void SchedulerQueue::Reset() { … }

void SchedulerQueue::SetExecutor(Executor* executor) { … }

bool SchedulerQueue::IsIdle() { … }

void SchedulerQueue::SetRunning(bool running) { … }

void SchedulerQueue::AddNode(CalculatorNode* node, CalculatorContext* cc) { … }

void SchedulerQueue::AddNodeForOpen(CalculatorNode* node) { … }

void SchedulerQueue::AddItemToQueue(Item&& item) { … }

int SchedulerQueue::GetTasksToSubmitToExecutor() { … }

void SchedulerQueue::SubmitWaitingTasksToExecutor() { … }

void SchedulerQueue::RunNextTask() { … }

void SchedulerQueue::RunCalculatorNode(CalculatorNode* node,
                                       CalculatorContext* cc) { … }

void SchedulerQueue::OpenCalculatorNode(CalculatorNode* node) { … }

void SchedulerQueue::CleanupAfterRun() { … }

}  // namespace internal
}  // namespace mediapipe