OLD | NEW |
(Empty) | |
| 1 /* |
| 2 * Copyright (c) 2017 The WebRTC project authors. All Rights Reserved. |
| 3 * |
| 4 * Use of this source code is governed by a BSD-style license |
| 5 * that can be found in the LICENSE file in the root of the source |
| 6 * tree. An additional intellectual property rights grant can be found |
| 7 * in the file PATENTS. All contributing project authors may |
| 8 * be found in the AUTHORS file in the root of the source tree. |
| 9 */ |
| 10 |
| 11 #include <stdio.h> |
| 12 |
| 13 |
| 14 #include "webrtc/common_types.h" |
| 15 #include "webrtc/common_video/libyuv/include/webrtc_libyuv.h" |
| 16 #include "webrtc/rtc_base/checks.h" |
| 17 #include "webrtc/rtc_base/logging.h" |
| 18 #include "webrtc/test/testsupport/frame_writer.h" |
| 19 |
| 20 extern "C" { |
| 21 #if defined(USE_SYSTEM_LIBJPEG) |
| 22 #include <jpeglib.h> |
| 23 #elif defined(USE_LIBJPEG_TURBO) |
| 24 #include "third_party/libjpeg_turbo/jpeglib.h" |
| 25 #else |
| 26 #include "third_party/libjpeg/jpeglib.h" |
| 27 #endif |
| 28 } |
| 29 |
| 30 namespace webrtc { |
| 31 namespace test { |
| 32 |
| 33 JpegFrameWriter::JpegFrameWriter(const std::string &output_filename) |
| 34 : frame_written_(false), |
| 35 output_filename_(output_filename), |
| 36 output_file_(nullptr) {} |
| 37 |
| 38 bool JpegFrameWriter::WriteFrame(const VideoFrame& input_frame, int quality) { |
| 39 RTC_CHECK(!frame_written_) << "Only a single frame can be saved to Jpeg."; |
| 40 const int kColorPlanes = 3; // R, G and B. |
| 41 size_t rgb_len = input_frame.height() * input_frame.width() * kColorPlanes; |
| 42 std::unique_ptr<uint8_t[]> rgb_buf(new uint8_t[rgb_len]); |
| 43 |
| 44 // kRGB24 actually corresponds to FourCC 24BG which is 24-bit BGR. |
| 45 if (ConvertFromI420(input_frame, VideoType::kRGB24, 0, rgb_buf.get()) < 0) { |
| 46 LOG(LS_ERROR) << "Could not convert input frame to RGB."; |
| 47 return false; |
| 48 } |
| 49 output_file_ = fopen(output_filename_.c_str(), "wb"); |
| 50 if (!output_file_) { |
| 51 LOG(LS_ERROR) << "Couldn't open file to write jpeg frame to:" << |
| 52 output_filename_; |
| 53 return false; |
| 54 } |
| 55 |
| 56 // Invoking LIBJPEG |
| 57 struct jpeg_compress_struct cinfo; |
| 58 struct jpeg_error_mgr jerr; |
| 59 JSAMPROW row_pointer[1]; |
| 60 cinfo.err = jpeg_std_error(&jerr); |
| 61 jpeg_create_compress(&cinfo); |
| 62 |
| 63 jpeg_stdio_dest(&cinfo, output_file_); |
| 64 |
| 65 cinfo.image_width = input_frame.width(); |
| 66 cinfo.image_height = input_frame.height(); |
| 67 cinfo.input_components = kColorPlanes; |
| 68 cinfo.in_color_space = JCS_EXT_BGR; |
| 69 jpeg_set_defaults(&cinfo); |
| 70 jpeg_set_quality(&cinfo, quality, TRUE); |
| 71 |
| 72 jpeg_start_compress(&cinfo, TRUE); |
| 73 int row_stride = input_frame.width() * kColorPlanes; |
| 74 while (cinfo.next_scanline < cinfo.image_height) { |
| 75 row_pointer[0] = &rgb_buf.get()[cinfo.next_scanline * row_stride]; |
| 76 jpeg_write_scanlines(&cinfo, row_pointer, 1); |
| 77 } |
| 78 |
| 79 jpeg_finish_compress(&cinfo); |
| 80 jpeg_destroy_compress(&cinfo); |
| 81 fclose(output_file_); |
| 82 |
| 83 frame_written_ = true; |
| 84 return true; |
| 85 } |
| 86 |
| 87 } // namespace test |
| 88 } // namespace webrtc |
OLD | NEW |