1 /*
2  *  Copyright (c) 2012 The WebM project authors. All Rights Reserved.
3  *
4  *  Use of this source code is governed by a BSD-style license
5  *  that can be found in the LICENSE file in the root of the source
6  *  tree. An additional intellectual property rights grant can be found
7  *  in the file PATENTS.  All contributing project authors may
8  *  be found in the AUTHORS file in the root of the source tree.
9  */
10 #ifndef TEST_WEBM_VIDEO_SOURCE_H_
11 #define TEST_WEBM_VIDEO_SOURCE_H_
12 #include <cstdarg>
13 #include <cstdio>
14 #include <cstdlib>
15 #include <new>
16 #include <string>
17 #include "../tools_common.h"
18 #include "../webmdec.h"
19 #include "test/video_source.h"
20 
21 namespace libvpx_test {
22 
23 // This class extends VideoSource to allow parsing of WebM files,
24 // so that we can do actual file decodes.
25 class WebMVideoSource : public CompressedVideoSource {
26  public:
WebMVideoSource(const std::string & file_name)27   explicit WebMVideoSource(const std::string &file_name)
28       : file_name_(file_name),
29         vpx_ctx_(new VpxInputContext()),
30         webm_ctx_(new WebmInputContext()),
31         buf_(NULL),
32         buf_sz_(0),
33         frame_(0),
34         end_of_file_(false) {
35   }
36 
~WebMVideoSource()37   virtual ~WebMVideoSource() {
38     if (vpx_ctx_->file != NULL)
39       fclose(vpx_ctx_->file);
40     webm_free(webm_ctx_);
41     delete vpx_ctx_;
42     delete webm_ctx_;
43   }
44 
Init()45   virtual void Init() {
46   }
47 
Begin()48   virtual void Begin() {
49     vpx_ctx_->file = OpenTestDataFile(file_name_);
50     ASSERT_TRUE(vpx_ctx_->file != NULL) << "Input file open failed. Filename: "
51         << file_name_;
52 
53     ASSERT_EQ(file_is_webm(webm_ctx_, vpx_ctx_), 1) << "file is not WebM";
54 
55     FillFrame();
56   }
57 
Next()58   virtual void Next() {
59     ++frame_;
60     FillFrame();
61   }
62 
FillFrame()63   void FillFrame() {
64     ASSERT_TRUE(vpx_ctx_->file != NULL);
65     const int status = webm_read_frame(webm_ctx_, &buf_, &buf_sz_, &buf_sz_);
66     ASSERT_GE(status, 0) << "webm_read_frame failed";
67     if (status == 1) {
68       end_of_file_ = true;
69     }
70   }
71 
SeekToNextKeyFrame()72   void SeekToNextKeyFrame() {
73     ASSERT_TRUE(vpx_ctx_->file != NULL);
74     do {
75       const int status = webm_read_frame(webm_ctx_, &buf_, &buf_sz_, &buf_sz_);
76       ASSERT_GE(status, 0) << "webm_read_frame failed";
77       ++frame_;
78       if (status == 1) {
79         end_of_file_ = true;
80       }
81     } while (!webm_ctx_->is_key_frame && !end_of_file_);
82   }
83 
cxdata()84   virtual const uint8_t *cxdata() const {
85     return end_of_file_ ? NULL : buf_;
86   }
frame_size()87   virtual size_t frame_size() const { return buf_sz_; }
frame_number()88   virtual unsigned int frame_number() const { return frame_; }
89 
90  protected:
91   std::string file_name_;
92   VpxInputContext *vpx_ctx_;
93   WebmInputContext *webm_ctx_;
94   uint8_t *buf_;
95   size_t buf_sz_;
96   unsigned int frame_;
97   bool end_of_file_;
98 };
99 
100 }  // namespace libvpx_test
101 
102 #endif  // TEST_WEBM_VIDEO_SOURCE_H_
103