OLD | NEW |
| (Empty) |
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 | |
3 // LICENSE file. | |
4 | |
5 #include "media/tools/omx_test/file_sink.h" | |
6 | |
7 #include "base/file_util.h" | |
8 #include "base/logging.h" | |
9 #include "media/tools/omx_test/color_space_util.h" | |
10 | |
11 namespace media { | |
12 | |
13 FileSink::FileSink(const FilePath& output_path, | |
14 bool simulate_copy, | |
15 bool enable_csc) | |
16 : output_path_(output_path), | |
17 simulate_copy_(simulate_copy), | |
18 enable_csc_(enable_csc), | |
19 width_(0), | |
20 height_(0), | |
21 copy_buf_size_(0), | |
22 csc_buf_size_(0) { | |
23 } | |
24 | |
25 FileSink::~FileSink() {} | |
26 | |
27 void FileSink::BufferReady(int size, uint8* buffer) { | |
28 if (size > copy_buf_size_) { | |
29 copy_buf_.reset(new uint8[size]); | |
30 copy_buf_size_ = size; | |
31 } | |
32 if (size > csc_buf_size_) { | |
33 csc_buf_.reset(new uint8[size]); | |
34 csc_buf_size_ = size; | |
35 } | |
36 | |
37 // Copy the output of the decoder to user memory. | |
38 if (simulate_copy_ || output_file_.get()) // Implies a copy. | |
39 memcpy(copy_buf_.get(), buffer, size); | |
40 | |
41 uint8* out_buffer = copy_buf_.get(); | |
42 if (enable_csc_) { | |
43 // Now assume the raw output is NV21. | |
44 media::NV21toIYUV(copy_buf_.get(), csc_buf_.get(), width_, height_); | |
45 out_buffer = csc_buf_.get(); | |
46 } | |
47 | |
48 if (output_file_.get()) | |
49 fwrite(out_buffer, sizeof(uint8), size, output_file_.get()); | |
50 } | |
51 | |
52 bool FileSink::Initialize() { | |
53 // Opens the output file for writing. | |
54 if (!output_path_.empty()) { | |
55 output_file_.Set(file_util::OpenFile(output_path_, "wb")); | |
56 if (!output_file_.get()) { | |
57 LOG(ERROR) << "can't open dump file %s" << output_path_.value(); | |
58 return false; | |
59 } | |
60 } | |
61 return true; | |
62 } | |
63 | |
64 void FileSink::UpdateSize(int width, int height) { | |
65 width_ = width; | |
66 height_ = height; | |
67 } | |
68 | |
69 } // namespace media | |
OLD | NEW |