-
Tom Finegan authored
- Use size_t for frame_size() (fixes warnings elsewhere) - Remove unnecessary use of const. Change-Id: Id2cba182ac3fd2f67d25fd1db3a0bc3e1eda040b
eb2325ee
/* * Copyright (c) 2012 The WebM 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_VIDEO_SOURCE_H_#define TEST_VIDEO_SOURCE_H_#include <cstdio>#include <cstdlib>#include <string>#include "test/acm_random.h"#include "vpx/vpx_encoder.h"namespace libvpx_test {// Helper macros to ensure LIBVPX_TEST_DATA_PATH is a quoted string.// These are undefined right below GetDataPath// NOTE: LIBVPX_TEST_DATA_PATH MUST NOT be a quoted string before// Stringification or the GetDataPath will fail at runtime#define TO_STRING(S) #S#define STRINGIFY(S) TO_STRING(S)// A simple function to encapsulate cross platform retrieval of test data pathstatic std::string GetDataPath() { const char *const data_path = getenv("LIBVPX_TEST_DATA_PATH"); if (data_path == NULL) {#ifdef LIBVPX_TEST_DATA_PATH // In some environments, we cannot set environment variables // Instead, we set the data path by using a preprocessor symbol // which can be set from make files return STRINGIFY(LIBVPX_TEST_DATA_PATH);#else return ".";#endif } return data_path;}// Undefining stringification macros because they are not used elsewhere#undef TO_STRING#undef STRINGIFYstatic FILE *OpenTestDataFile(const std::string& file_name) { const std::string path_to_source = GetDataPath() + "/" + file_name; return fopen(path_to_source.c_str(), "rb");}// Abstract base class for test video sources, which provide a stream of// vpx_image_t images with associated timestamps and duration.class VideoSource { public: virtual ~VideoSource() {} // Prepare the stream for reading, rewind/open as necessary. virtual void Begin() = 0; // Advance the cursor to the next frame virtual void Next() = 0; // Get the current video frame, or NULL on End-Of-Stream. virtual vpx_image_t *img() const = 0; // Get the presentation timestamp of the current frame. virtual vpx_codec_pts_t pts() const = 0;7172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140
// Get the current frame's duration
virtual unsigned long duration() const = 0;
// Get the timebase for the stream
virtual vpx_rational_t timebase() const = 0;
// Get the current frame counter, starting at 0.
virtual unsigned int frame() const = 0;
// Get the current file limit.
virtual unsigned int limit() const = 0;
};
class DummyVideoSource : public VideoSource {
public:
DummyVideoSource() : img_(NULL), limit_(100), width_(0), height_(0) {
SetSize(80, 64);
}
virtual ~DummyVideoSource() { vpx_img_free(img_); }
virtual void Begin() {
frame_ = 0;
FillFrame();
}
virtual void Next() {
++frame_;
FillFrame();
}
virtual vpx_image_t *img() const {
return (frame_ < limit_) ? img_ : NULL;
}
// Models a stream where Timebase = 1/FPS, so pts == frame.
virtual vpx_codec_pts_t pts() const { return frame_; }
virtual unsigned long duration() const { return 1; }
virtual vpx_rational_t timebase() const {
const vpx_rational_t t = {1, 30};
return t;
}
virtual unsigned int frame() const { return frame_; }
virtual unsigned int limit() const { return limit_; }
void SetSize(unsigned int width, unsigned int height) {
if (width != width_ || height != height_) {
vpx_img_free(img_);
raw_sz_ = ((width + 31)&~31) * height * 3 / 2;
img_ = vpx_img_alloc(NULL, VPX_IMG_FMT_I420, width, height, 32);
width_ = width;
height_ = height;
}
}
protected:
virtual void FillFrame() { memset(img_->img_data, 0, raw_sz_); }
vpx_image_t *img_;
size_t raw_sz_;
unsigned int limit_;
unsigned int frame_;
unsigned int width_;
unsigned int height_;
};
141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195
class RandomVideoSource : public DummyVideoSource {
public:
RandomVideoSource(int seed = ACMRandom::DeterministicSeed())
: rnd_(seed),
seed_(seed) { }
protected:
// Reset the RNG to get a matching stream for the second pass
virtual void Begin() {
frame_ = 0;
rnd_.Reset(seed_);
FillFrame();
}
// 15 frames of noise, followed by 15 static frames. Reset to 0 rather
// than holding previous frames to encourage keyframes to be thrown.
virtual void FillFrame() {
if (frame_ % 30 < 15)
for (size_t i = 0; i < raw_sz_; ++i)
img_->img_data[i] = rnd_.Rand8();
else
memset(img_->img_data, 0, raw_sz_);
}
ACMRandom rnd_;
int seed_;
};
// Abstract base class for test video sources, which provide a stream of
// decompressed images to the decoder.
class CompressedVideoSource {
public:
virtual ~CompressedVideoSource() {}
virtual void Init() = 0;
// Prepare the stream for reading, rewind/open as necessary.
virtual void Begin() = 0;
// Advance the cursor to the next frame
virtual void Next() = 0;
virtual const uint8_t *cxdata() const = 0;
virtual size_t frame_size() const = 0;
virtual unsigned int frame_number() const = 0;
};
} // namespace libvpx_test
#endif // TEST_VIDEO_SOURCE_H_