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

AddTarget -> ConnectTo and documentation

Browse Source 
PiperOrigin-RevId: 488407930
This commit is contained in:
MediaPipe Team 2022-11-14 10:38:38 -08:00 committed by Copybara-Service
parent 05cb40ff79
commit c7d531ebb2
1 changed files with 23 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
mediapipe/framework/api2/builder.h
View File

@ -176,22 +176,40 @@ class SourceImpl {
: SourceImpl(&GetWithAutoGrow(vec, 0)) {} : SourceImpl(&GetWithAutoGrow(vec, 0)) {}
explicit SourceImpl(SourceBase* base) : base_(base) {} explicit SourceImpl(SourceBase* base) : base_(base) {}
// Connects MediaPipe stream or side packet to a destination:
// - node input (input stream) / side input (input side packet)
// - graph output (output stream) / side output (output side packet).
//
// MediaPipe streams and side packets can be connected to multiple
// destinations. Side packets and packets added to streams are sent to all
// connected destinations.
template <typename U, template <typename U,
typename std::enable_if<AllowConnection<U>{}, int>::type = 0> typename std::enable_if<AllowConnection<U>{}, int>::type = 0>
Src& AddTarget(const Dst<U>& dest) { Src& ConnectTo(const Dst<U>& dest) {
CHECK(dest.base_.source == nullptr); CHECK(dest.base_.source == nullptr);
dest.base_.source = base_; dest.base_.source = base_;
base_->dests_.emplace_back(&dest.base_); base_->dests_.emplace_back(&dest.base_);
return *this; return *this;
} }
// Shortcut for `ConnectTo`.
//
// Connects MediaPipe stream or side packet to a destination:
// - node input (input stream) / side input (input side packet)
// - graph output (output stream) / side output (output side packet).
//
// MediaPipe streams and side packets can be connected to multiple
// destinations. Side packets and packets added to streams are sent to all
// connected destinations.
template <typename U>
Src& operator>>(const Dst<U>& dest) {
return ConnectTo(dest);
}
Src& SetName(std::string name) { Src& SetName(std::string name) {
base_->name_ = std::move(name); base_->name_ = std::move(name);
return *this; return *this;
} }
template <typename U>
Src& operator>>(const Dst<U>& dest) {
return AddTarget(dest);
}
template <typename U, template <typename U,
std::enable_if_t<internal_builder::AllowCast<T, U>{}, int> = 0> std::enable_if_t<internal_builder::AllowCast<T, U>{}, int> = 0>