Refactor frame generation code so it can be used by multiple modules.
R=pbos@webrtc.org, stefan@webrtc.org, pbos, stefan
BUG=
Review URL: https://webrtc-codereview.appspot.com/2240004
git-svn-id: http://webrtc.googlecode.com/svn/trunk@4791 4adac7df-926f-26a2-2b94-8c16560cd09d
diff --git a/webrtc/common_video/test/frame_generator.cc b/webrtc/common_video/test/frame_generator.cc
new file mode 100644
index 0000000..2f93d7a
--- /dev/null
+++ b/webrtc/common_video/test/frame_generator.cc
@@ -0,0 +1,80 @@
+/*
+ * Copyright (c) 2013 The WebRTC project authors. All Rights Reserved.
+ *
+ * Use of this source code is governed by a BSD-style license
+ * that can be found in the LICENSE file in the root of the source
+ * tree. An additional intellectual property rights grant can be found
+ * in the file PATENTS. All contributing project authors may
+ * be found in the AUTHORS file in the root of the source tree.
+ */
+#include "webrtc/common_video/test/frame_generator.h"
+
+#include <stdio.h>
+
+#include "webrtc/common_video/libyuv/include/webrtc_libyuv.h"
+
+namespace webrtc {
+namespace test {
+namespace {
+
+class YuvFileGenerator : public FrameGenerator {
+ public:
+ YuvFileGenerator(FILE* file, size_t width, size_t height)
+ : file_(file), width_(width), height_(height) {
+ assert(file);
+ assert(width > 0);
+ assert(height > 0);
+ frame_size_ = CalcBufferSize(
+ kI420, static_cast<int>(width_), static_cast<int>(height_));
+ frame_buffer_ = new uint8_t[frame_size_];
+ frame_.CreateEmptyFrame(static_cast<int>(width),
+ static_cast<int>(height),
+ static_cast<int>(width),
+ static_cast<int>((width + 1) / 2),
+ static_cast<int>((width + 1) / 2));
+ }
+
+ virtual ~YuvFileGenerator() {
+ fclose(file_);
+ delete[] frame_buffer_;
+ }
+
+ virtual I420VideoFrame& NextFrame() OVERRIDE {
+ size_t count = fread(frame_buffer_, 1, frame_size_, file_);
+ if (count < frame_size_) {
+ rewind(file_);
+ return NextFrame();
+ }
+
+ ConvertToI420(kI420,
+ frame_buffer_,
+ 0,
+ 0,
+ static_cast<int>(width_),
+ static_cast<int>(height_),
+ 0,
+ kRotateNone,
+ &frame_);
+ return frame_;
+ }
+
+ private:
+ FILE* file_;
+ size_t width_;
+ size_t height_;
+ size_t frame_size_;
+ uint8_t* frame_buffer_;
+ I420VideoFrame frame_;
+};
+} // namespace
+
+FrameGenerator* FrameGenerator::CreateFromYuvFile(const char* file,
+ size_t width,
+ size_t height) {
+ FILE* file_handle = fopen(file, "r");
+ assert(file_handle);
+ return new YuvFileGenerator(file_handle, width, height);
+}
+
+} // namespace test
+} // namespace webrtc
