mirror of
https://github.com/mollyim/webrtc.git
synced 2025-05-15 14:50:39 +01:00
b7edf69e9a
Bug: webrtc:6463 Change-Id: Ia0767a2e6bbacc43e63c30ed3bd3edb10ff6e645 Reviewed-on: https://webrtc-review.googlesource.com/c/121943 Commit-Queue: Niels Moller <nisse@webrtc.org> Reviewed-by: Karl Wiberg <kwiberg@webrtc.org> Cr-Commit-Position: refs/heads/master@{#26613}
62 lines
1.9 KiB
C++
62 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 MODULES_VIDEO_CODING_UTILITY_IVF_FILE_WRITER_H_
|
|
#define MODULES_VIDEO_CODING_UTILITY_IVF_FILE_WRITER_H_
|
|
|
|
#include <stddef.h>
|
|
#include <stdint.h>
|
|
#include <memory>
|
|
|
|
#include "api/video/encoded_image.h"
|
|
#include "common_types.h" // NOLINT(build/include)
|
|
#include "rtc_base/constructor_magic.h"
|
|
#include "rtc_base/system/file_wrapper.h"
|
|
#include "rtc_base/time_utils.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(FileWrapper file,
|
|
size_t byte_limit);
|
|
~IvfFileWriter();
|
|
|
|
bool WriteFrame(const EncodedImage& encoded_image, VideoCodecType codec_type);
|
|
bool Close();
|
|
|
|
private:
|
|
explicit IvfFileWriter(FileWrapper 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_;
|
|
FileWrapper file_;
|
|
|
|
RTC_DISALLOW_COPY_AND_ASSIGN(IvfFileWriter);
|
|
};
|
|
|
|
} // namespace webrtc
|
|
|
|
#endif // MODULES_VIDEO_CODING_UTILITY_IVF_FILE_WRITER_H_
|