| 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 | 
|---|