video_source.h 6.46 KB
Newer Older
John Koleszar's avatar
John Koleszar committed
1 2 3 4 5 6 7 8 9 10 11
/*
 *  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_
Johann's avatar
Johann committed
12

13
#if defined(_WIN32)
14 15
#define NOMINMAX
#define WIN32_LEAN_AND_MEAN
16 17
#include <windows.h>
#endif
18 19 20
#include <cstdio>
#include <cstdlib>
#include <string>
Johann's avatar
Johann committed
21
#include "test/acm_random.h"
John Koleszar's avatar
John Koleszar committed
22 23 24 25
#include "vpx/vpx_encoder.h"

namespace libvpx_test {

26 27 28 29 30 31
// 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)
32

33 34 35 36 37 38 39 40 41 42 43 44
// A simple function to encapsulate cross platform retrieval of test data path
static 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
45
  }
46 47
  return data_path;
}
48

49 50 51 52
// Undefining stringification macros because they are not used elsewhere
#undef TO_STRING
#undef STRINGIFY

53
inline FILE *OpenTestDataFile(const std::string& file_name) {
54
  const std::string path_to_source = GetDataPath() + "/" + file_name;
55 56 57
  return fopen(path_to_source.c_str(), "rb");
}

58 59
static FILE *GetTempOutFile(std::string *file_name) {
  file_name->clear();
60 61
#if defined(_WIN32)
  char fname[MAX_PATH];
62 63 64 65 66 67 68
  char tmppath[MAX_PATH];
  if (GetTempPathA(MAX_PATH, tmppath)) {
    // Assume for now that the filename generated is unique per process
    if (GetTempFileNameA(tmppath, "lvx", 0, fname)) {
      file_name->assign(fname);
      return fopen(fname, "wb+");
    }
69
  }
70
  return NULL;
71
#else
72
  return tmpfile();
73
#endif
74 75
}

76 77 78
class TempOutFile {
 public:
  TempOutFile() {
79
    file_ = GetTempOutFile(&file_name_);
80 81 82 83
  }
  ~TempOutFile() {
    CloseFile();
    if (!file_name_.empty()) {
84
      EXPECT_EQ(0, remove(file_name_.c_str()));
85 86 87 88 89 90 91 92
    }
  }
  FILE *file() {
    return file_;
  }
  const std::string& file_name() {
    return file_name_;
  }
93 94

 protected:
95 96
  void CloseFile() {
    if (file_) {
97
      fclose(file_);
98 99 100 101 102 103 104
      file_ = NULL;
    }
  }
  FILE *file_;
  std::string file_name_;
};

John Koleszar's avatar
John Koleszar committed
105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130
// 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;

  // 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;
131 132 133

  // Get the current file limit.
  virtual unsigned int limit() const = 0;
John Koleszar's avatar
John Koleszar committed
134 135 136 137 138
};


class DummyVideoSource : public VideoSource {
 public:
139 140 141 142 143 144 145
  DummyVideoSource()
      : img_(NULL),
        limit_(100),
        width_(80),
        height_(64),
        format_(VPX_IMG_FMT_I420) {
    ReallocImage();
John Koleszar's avatar
John Koleszar committed
146
  }
John Koleszar's avatar
John Koleszar committed
147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175

  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_; }

176 177
  virtual unsigned int limit() const { return limit_; }

178 179 180 181
  void set_limit(unsigned int limit) {
    limit_ = limit;
  }

John Koleszar's avatar
John Koleszar committed
182
  void SetSize(unsigned int width, unsigned int height) {
John Koleszar's avatar
John Koleszar committed
183 184 185
    if (width != width_ || height != height_) {
      width_ = width;
      height_ = height;
186 187 188 189 190 191 192 193
      ReallocImage();
    }
  }

  void SetImageFormat(vpx_img_fmt_t format) {
    if (format_ != format) {
      format_ = format;
      ReallocImage();
John Koleszar's avatar
John Koleszar committed
194
    }
John Koleszar's avatar
John Koleszar committed
195 196 197
  }

 protected:
198
  virtual void FillFrame() { if (img_) memset(img_->img_data, 0, raw_sz_); }
John Koleszar's avatar
John Koleszar committed
199

200 201 202 203 204 205
  void ReallocImage() {
    vpx_img_free(img_);
    img_ = vpx_img_alloc(NULL, format_, width_, height_, 32);
    raw_sz_ = ((img_->w + 31) & ~31) * img_->h * img_->bps / 8;
  }

John Koleszar's avatar
John Koleszar committed
206 207 208 209
  vpx_image_t *img_;
  size_t       raw_sz_;
  unsigned int limit_;
  unsigned int frame_;
John Koleszar's avatar
John Koleszar committed
210 211
  unsigned int width_;
  unsigned int height_;
212
  vpx_img_fmt_t format_;
John Koleszar's avatar
John Koleszar committed
213 214 215 216
};


class RandomVideoSource : public DummyVideoSource {
Johann's avatar
Johann committed
217
 public:
Johann's avatar
Johann committed
218 219 220
  RandomVideoSource(int seed = ACMRandom::DeterministicSeed())
      : rnd_(seed),
        seed_(seed) { }
Johann's avatar
Johann committed
221

John Koleszar's avatar
John Koleszar committed
222
 protected:
Johann's avatar
Johann committed
223 224 225
  // Reset the RNG to get a matching stream for the second pass
  virtual void Begin() {
    frame_ = 0;
Johann's avatar
Johann committed
226
    rnd_.Reset(seed_);
Johann's avatar
Johann committed
227 228 229
    FillFrame();
  }

John Koleszar's avatar
John Koleszar committed
230 231 232
  // 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() {
233 234 235 236 237 238 239
    if (img_) {
      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_);
    }
John Koleszar's avatar
John Koleszar committed
240
  }
Johann's avatar
Johann committed
241 242

  ACMRandom rnd_;
Johann's avatar
Johann committed
243
  int seed_;
John Koleszar's avatar
John Koleszar committed
244 245
};

246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261
// 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;

Tom Finegan's avatar
Tom Finegan committed
262
  virtual size_t frame_size() const = 0;
263

Tom Finegan's avatar
Tom Finegan committed
264
  virtual unsigned int frame_number() const = 0;
265 266
};

John Koleszar's avatar
John Koleszar committed
267 268 269
}  // namespace libvpx_test

#endif  // TEST_VIDEO_SOURCE_H_