mirror of
https://github.com/mollyim/webrtc.git
synced 2025-05-15 06:40:43 +01:00
4dc66c53d0
Bug: webrtc:9378 Change-Id: I8fb3b19cad0ad428abc6c8e6b507180d461882ba Reviewed-on: https://webrtc-review.googlesource.com/c/104002 Reviewed-by: Karl Wiberg <kwiberg@webrtc.org> Reviewed-by: Peter Slatala <psla@webrtc.org> Reviewed-by: Patrik Höglund <phoglund@webrtc.org> Commit-Queue: Niels Moller <nisse@webrtc.org> Cr-Commit-Position: refs/heads/master@{#25033}
60 lines
1.9 KiB
C++
60 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 <memory>
|
|
#include <string>
|
|
|
|
#include "api/video/encoded_image.h"
|
|
#include "modules/include/module_common_types.h"
|
|
#include "rtc_base/constructormagic.h"
|
|
#include "rtc_base/file.h"
|
|
#include "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 // MODULES_VIDEO_CODING_UTILITY_IVF_FILE_WRITER_H_
|