c20978e581
NOPRESUBMIT=True # cpplint errors that aren't caused by this CL. NOTRY=True NOTREECHECKS=True TBR=kwiberg@webrtc.org, kjellander@webrtc.org Bug: webrtc:7634 Change-Id: I3cca0fbaa807b563c95979cccd6d1bec32055f36 Reviewed-on: https://chromium-review.googlesource.com/562156 Commit-Queue: Edward Lemur <ehmaldonado@webrtc.org> Reviewed-by: Mirko Bonadei <mbonadei@webrtc.org> Cr-Commit-Position: refs/heads/master@{#18919}
61 lines
1.9 KiB
C++
61 lines
1.9 KiB
C++
/*
|
|
* Copyright (c) 2016 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.
|
|
*/
|
|
|
|
#ifndef WEBRTC_MODULES_VIDEO_CODING_UTILITY_IVF_FILE_WRITER_H_
|
|
#define WEBRTC_MODULES_VIDEO_CODING_UTILITY_IVF_FILE_WRITER_H_
|
|
|
|
#include <memory>
|
|
#include <string>
|
|
|
|
#include "webrtc/common_video/include/video_frame.h"
|
|
#include "webrtc/modules/include/module_common_types.h"
|
|
#include "webrtc/rtc_base/constructormagic.h"
|
|
#include "webrtc/rtc_base/file.h"
|
|
#include "webrtc/rtc_base/timeutils.h"
|
|
|
|
namespace webrtc {
|
|
|
|
class IvfFileWriter {
|
|
public:
|
|
// Takes ownership of the file, which will be closed either through
|
|
// Close or ~IvfFileWriter. If writing a frame would take the file above the
|
|
// |byte_limit| the file will be closed, the write (and all future writes)
|
|
// will fail. A |byte_limit| of 0 is equivalent to no limit.
|
|
static std::unique_ptr<IvfFileWriter> Wrap(rtc::File file, size_t byte_limit);
|
|
~IvfFileWriter();
|
|
|
|
bool WriteFrame(const EncodedImage& encoded_image, VideoCodecType codec_type);
|
|
bool Close();
|
|
|
|
private:
|
|
explicit IvfFileWriter(rtc::File file, size_t byte_limit);
|
|
|
|
bool WriteHeader();
|
|
bool InitFromFirstFrame(const EncodedImage& encoded_image,
|
|
VideoCodecType codec_type);
|
|
|
|
VideoCodecType codec_type_;
|
|
size_t bytes_written_;
|
|
size_t byte_limit_;
|
|
size_t num_frames_;
|
|
uint16_t width_;
|
|
uint16_t height_;
|
|
int64_t last_timestamp_;
|
|
bool using_capture_timestamps_;
|
|
rtc::TimestampWrapAroundHandler wrap_handler_;
|
|
rtc::File file_;
|
|
|
|
RTC_DISALLOW_COPY_AND_ASSIGN(IvfFileWriter);
|
|
};
|
|
|
|
} // namespace webrtc
|
|
|
|
#endif // WEBRTC_MODULES_VIDEO_CODING_UTILITY_IVF_FILE_WRITER_H_
|