ebe5acb27a
Adds ability to specify desired frame size separate from actual clip resolution, as well as clip and desired fps. This allows e.g. reading an HD clip but running benchmarks in VGA, and to specify e.g. 60fps for the clip but 30for encoding where frame dropping kicks in so that motion is actually correct rather than just plaing the clip slowly. Bug: webrtc:12229 Change-Id: I4ad4fcc335611a449dc2723ffafbec6731e89f55 Reviewed-on: https://webrtc-review.googlesource.com/c/src/+/195324 Commit-Queue: Erik Språng <sprang@webrtc.org> Reviewed-by: Philip Eliasson <philipel@webrtc.org> Cr-Commit-Position: refs/heads/master@{#32839}
120 lines
3.7 KiB
C++
120 lines
3.7 KiB
C++
/*
|
|
* Copyright (c) 2011 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 TEST_TESTSUPPORT_FRAME_READER_H_
|
|
#define TEST_TESTSUPPORT_FRAME_READER_H_
|
|
|
|
#include <stdio.h>
|
|
|
|
#include <string>
|
|
|
|
#include "absl/types/optional.h"
|
|
#include "api/scoped_refptr.h"
|
|
|
|
namespace webrtc {
|
|
class I420Buffer;
|
|
namespace test {
|
|
|
|
// Handles reading of I420 frames from video files.
|
|
class FrameReader {
|
|
public:
|
|
virtual ~FrameReader() {}
|
|
|
|
// Initializes the frame reader, i.e. opens the input file.
|
|
// This must be called before reading of frames has started.
|
|
// Returns false if an error has occurred, in addition to printing to stderr.
|
|
virtual bool Init() = 0;
|
|
|
|
// Reads a frame from the input file. On success, returns the frame.
|
|
// Returns nullptr if encountering end of file or a read error.
|
|
virtual rtc::scoped_refptr<I420Buffer> ReadFrame() = 0;
|
|
|
|
// Closes the input file if open. Essentially makes this class impossible
|
|
// to use anymore. Will also be invoked by the destructor.
|
|
virtual void Close() = 0;
|
|
|
|
// Frame length in bytes of a single frame image.
|
|
virtual size_t FrameLength() = 0;
|
|
// Total number of frames in the input video source.
|
|
virtual int NumberOfFrames() = 0;
|
|
};
|
|
|
|
class YuvFrameReaderImpl : public FrameReader {
|
|
public:
|
|
enum class RepeatMode { kSingle, kRepeat, kPingPong };
|
|
class DropperUtil {
|
|
public:
|
|
DropperUtil(int source_fps, int target_fps);
|
|
|
|
enum class DropDecision { kDropframe, kKeepFrame };
|
|
DropDecision UpdateLevel();
|
|
|
|
private:
|
|
const double frame_size_buckets_;
|
|
double bucket_level_;
|
|
};
|
|
|
|
// Creates a file handler. The input file is assumed to exist and be readable.
|
|
// Parameters:
|
|
// input_filename The file to read from.
|
|
// width, height Size of each frame to read.
|
|
YuvFrameReaderImpl(std::string input_filename, int width, int height);
|
|
YuvFrameReaderImpl(std::string input_filename,
|
|
int input_width,
|
|
int input_height,
|
|
int desired_width,
|
|
int desired_height,
|
|
RepeatMode repeat_mode,
|
|
absl::optional<int> clip_fps,
|
|
int target_fps);
|
|
~YuvFrameReaderImpl() override;
|
|
bool Init() override;
|
|
rtc::scoped_refptr<I420Buffer> ReadFrame() override;
|
|
void Close() override;
|
|
size_t FrameLength() override;
|
|
int NumberOfFrames() override;
|
|
|
|
protected:
|
|
const std::string input_filename_;
|
|
// It is not const, so subclasses will be able to add frame header size.
|
|
size_t frame_length_in_bytes_;
|
|
const int input_width_;
|
|
const int input_height_;
|
|
const int desired_width_;
|
|
const int desired_height_;
|
|
const size_t frame_size_bytes_;
|
|
const RepeatMode repeat_mode_;
|
|
int number_of_frames_;
|
|
int current_frame_index_;
|
|
std::unique_ptr<DropperUtil> dropper_;
|
|
FILE* input_file_;
|
|
};
|
|
|
|
class Y4mFrameReaderImpl : public YuvFrameReaderImpl {
|
|
public:
|
|
// Creates a file handler. The input file is assumed to exist and be readable.
|
|
// Parameters:
|
|
// input_filename The file to read from.
|
|
// width, height Size of each frame to read.
|
|
Y4mFrameReaderImpl(std::string input_filename, int width, int height);
|
|
~Y4mFrameReaderImpl() override;
|
|
bool Init() override;
|
|
rtc::scoped_refptr<I420Buffer> ReadFrame() override;
|
|
|
|
private:
|
|
// Buffer that is used to read file and frame headers.
|
|
uint8_t* buffer_;
|
|
};
|
|
|
|
} // namespace test
|
|
} // namespace webrtc
|
|
|
|
#endif // TEST_TESTSUPPORT_FRAME_READER_H_
|