| OLD | NEW |
| 1 // Copyright (c) 2010 The Chromium Authors. All rights reserved. Use of this | 1 // Copyright (c) 2010 The Chromium Authors. All rights reserved. Use of this |
| 2 // source code is governed by a BSD-style license that can be found in the | 2 // source code is governed by a BSD-style license that can be found in the |
| 3 // LICENSE file. | 3 // LICENSE file. |
| 4 | 4 |
| 5 #include "media/tools/omx_test/file_sink.h" | 5 #include "media/tools/omx_test/file_sink.h" |
| 6 | 6 |
| 7 #include "base/file_util.h" | 7 #include "base/file_util.h" |
| 8 #include "base/logging.h" | 8 #include "base/logging.h" |
| 9 #include "media/tools/omx_test/color_space_util.h" | 9 #include "media/tools/omx_test/color_space_util.h" |
| 10 | 10 |
| 11 namespace media { | 11 namespace media { |
| 12 | 12 |
| 13 bool FileSink::AllocateEGLImages(int width, int height, | |
| 14 std::vector<EGLImageKHR>* images) { | |
| 15 NOTREACHED() << "This method is never used"; | |
| 16 return false; | |
| 17 } | |
| 18 | 13 |
| 19 void FileSink::ReleaseEGLImages(const std::vector<EGLImageKHR>& images) { | 14 void FileSink::BufferReady(int size, uint8* buffer) { |
| 20 NOTREACHED() << "This method is never used"; | |
| 21 } | |
| 22 | |
| 23 void FileSink::UseThisBuffer(int buffer_id, OMX_BUFFERHEADERTYPE* buffer) { | |
| 24 CHECK(omx_buffers_.find(buffer_id) == omx_buffers_.end()); | |
| 25 omx_buffers_[buffer_id] = buffer; | |
| 26 } | |
| 27 | |
| 28 void FileSink::StopUsingThisBuffer(int id) { | |
| 29 omx_buffers_.erase(id); | |
| 30 } | |
| 31 | |
| 32 void FileSink::BufferReady(int buffer_id, BufferUsedCallback* callback) { | |
| 33 CHECK(omx_buffers_.find(buffer_id) != omx_buffers_.end()); | |
| 34 CHECK(callback); | |
| 35 | |
| 36 OMX_BUFFERHEADERTYPE* omx_buffer = omx_buffers_[buffer_id]; | |
| 37 uint8* buffer = omx_buffer->pBuffer; | |
| 38 int size = omx_buffer->nFilledLen; | |
| 39 | |
| 40 // We never receive an end-of-stream buffer here. | |
| 41 CHECK(!(omx_buffer->nFlags & OMX_BUFFERFLAG_EOS)); | |
| 42 | |
| 43 if (size > copy_buf_size_) { | 15 if (size > copy_buf_size_) { |
| 44 copy_buf_.reset(new uint8[size]); | 16 copy_buf_.reset(new uint8[size]); |
| 45 copy_buf_size_ = size; | 17 copy_buf_size_ = size; |
| 46 } | 18 } |
| 47 if (size > csc_buf_size_) { | 19 if (size > csc_buf_size_) { |
| 48 csc_buf_.reset(new uint8[size]); | 20 csc_buf_.reset(new uint8[size]); |
| 49 csc_buf_size_ = size; | 21 csc_buf_size_ = size; |
| 50 } | 22 } |
| 51 | 23 |
| 52 // Copy the output of the decoder to user memory. | 24 // Copy the output of the decoder to user memory. |
| 53 if (simulate_copy_ || output_file_.get()) // Implies a copy. | 25 if (simulate_copy_ || output_file_.get()) // Implies a copy. |
| 54 memcpy(copy_buf_.get(), buffer, size); | 26 memcpy(copy_buf_.get(), buffer, size); |
| 55 | 27 |
| 56 uint8* out_buffer = copy_buf_.get(); | 28 uint8* out_buffer = copy_buf_.get(); |
| 57 if (enable_csc_) { | 29 if (enable_csc_) { |
| 58 // Now assume the raw output is NV21. | 30 // Now assume the raw output is NV21. |
| 59 media::NV21toIYUV(copy_buf_.get(), csc_buf_.get(), width_, height_); | 31 media::NV21toIYUV(copy_buf_.get(), csc_buf_.get(), width_, height_); |
| 60 out_buffer = csc_buf_.get(); | 32 out_buffer = csc_buf_.get(); |
| 61 } | 33 } |
| 62 | 34 |
| 63 if (output_file_.get()) | 35 if (output_file_.get()) |
| 64 fwrite(out_buffer, sizeof(uint8), size, output_file_.get()); | 36 fwrite(out_buffer, sizeof(uint8), size, output_file_.get()); |
| 65 | |
| 66 // Always make the callback. | |
| 67 callback->Run(buffer_id); | |
| 68 delete callback; | |
| 69 } | 37 } |
| 70 | 38 |
| 71 bool FileSink::Initialize() { | 39 bool FileSink::Initialize() { |
| 72 // Opens the output file for writing. | 40 // Opens the output file for writing. |
| 73 if (!output_filename_.empty()) { | 41 if (!output_filename_.empty()) { |
| 74 output_file_.Set(file_util::OpenFile(output_filename_, "wb")); | 42 output_file_.Set(file_util::OpenFile(output_filename_, "wb")); |
| 75 if (!output_file_.get()) { | 43 if (!output_file_.get()) { |
| 76 LOG(ERROR) << "can't open dump file %s" << output_filename_; | 44 LOG(ERROR) << "can't open dump file %s" << output_filename_; |
| 77 return false; | 45 return false; |
| 78 } | 46 } |
| 79 } | 47 } |
| 80 return true; | 48 return true; |
| 81 } | 49 } |
| 82 | 50 |
| 83 void FileSink::UpdateSize(int width, int height) { | 51 void FileSink::UpdateSize(int width, int height) { |
| 84 width_ = width; | 52 width_ = width; |
| 85 height_ = height; | 53 height_ = height; |
| 86 } | 54 } |
| 87 | 55 |
| 88 } // namespace media | 56 } // namespace media |
| OLD | NEW |