Index: services/media/framework/stages/packet_transform_stage.cc |
diff --git a/services/media/framework/stages/packet_transform_stage.cc b/services/media/framework/stages/packet_transform_stage.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..f7ef85fce443ac8dcea02365287f2cde93423a16 |
--- /dev/null |
+++ b/services/media/framework/stages/packet_transform_stage.cc |
@@ -0,0 +1,54 @@ |
+// Copyright 2016 The Chromium Authors. All rights reserved. |
+// Use of this source code is governed by a BSD-style license that can be |
+// found in the LICENSE file. |
+ |
+#include "services/media/framework/stages/packet_transform_stage.h" |
+ |
+namespace mojo { |
+namespace media { |
+ |
+PacketTransformStage::PacketTransformStage( |
+ PacketTransformPtr transform) : |
+ transform_(transform), |
+ allocator_(nullptr), |
+ input_packet_is_new_(true) { |
+ DCHECK(transform_); |
+} |
+ |
+PacketTransformStage::~PacketTransformStage() {} |
+ |
+bool PacketTransformStage::Prepare(UpdateCallback update_callback) { |
+ allocator_ = output_.Prepare(true); |
+ if (allocator_ == nullptr) { |
+ allocator_ = Allocator::GetDefault(); |
+ } |
+ input_.Prepare(nullptr, false); |
+ return false; |
+} |
+ |
+void PacketTransformStage::Update(Engine& engine) { |
+ DCHECK(allocator_); |
+ |
+ if (input_.packet_from_upstream() && output_.demand() != Demand::kNegative) { |
+ PacketPtr output_packet; |
+ if (transform_->TransformPacket( |
+ input_.packet_from_upstream(), |
+ input_packet_is_new_, |
+ *allocator_, |
+ &output_packet)) { |
+ input_.packet_from_upstream().reset(); |
+ input_packet_is_new_ = true; |
+ } else { |
+ input_packet_is_new_ = false; |
+ } |
+ |
+ if (output_packet) { |
+ output_.supply_packet(std::move(output_packet), engine); |
+ } |
+ } |
+ |
+ input_.set_demand(output_.demand(), engine); |
+} |
+ |
+} // namespace media |
+} // namespace mojo |