Blame view

3rdparty/opencv-4.5.4/modules/videoio/test/test_container_avi.cpp 3.07 KB
f4334277   Hu Chunming   提交3rdparty
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
  // This file is part of OpenCV project.
  // It is subject to the license terms in the LICENSE file found in the top-level directory
  // of this distribution and at http://opencv.org/license.html.
  
  #include "test_precomp.hpp"
  #include "opencv2/videoio/container_avi.private.hpp"
  #include <cstdio>
  
  using namespace cv;
  
  namespace opencv_test { namespace {
  
  TEST(videoio_builtin, basic_avi)
  {
      String filename = BunnyParameters::getFilename(".mjpg.avi");
      AVIReadContainer in;
      in.initStream(filename);
      frame_list frames;
      ASSERT_TRUE(in.parseRiff(frames));
      EXPECT_EQ(frames.size(), static_cast<unsigned>(BunnyParameters::getCount()));
      EXPECT_EQ(in.getWidth(), static_cast<unsigned>(BunnyParameters::getWidth()));
      EXPECT_EQ(in.getHeight(), static_cast<unsigned>(BunnyParameters::getHeight()));
      EXPECT_EQ(in.getFps(), static_cast<unsigned>(BunnyParameters::getFps()));
  }
  
  TEST(videoio_builtin, invalid_avi)
  {
      String filename = BunnyParameters::getFilename(".avi");
      AVIReadContainer in;
      in.initStream(filename);
      frame_list frames;
      EXPECT_FALSE(in.parseRiff(frames));
      EXPECT_EQ(frames.size(), static_cast<unsigned>(0));
  }
  
  TEST(videoio_builtin, read_write_avi)
  {
      const String filename = cv::tempfile("test.avi");
      const double fps = 100;
      const Size sz(800, 600);
      const size_t count = 10;
      const uchar data[count] = {1, 2, 3, 4, 5, 6, 7, 8, 9, 0xA};
      const Codecs codec = MJPEG;
      {
          AVIWriteContainer out;
          ASSERT_TRUE(out.initContainer(filename, fps, sz, true));
          ASSERT_TRUE(out.isOpenedStream());
          EXPECT_EQ(out.getWidth(), sz.width);
          EXPECT_EQ(out.getHeight(), sz.height);
          EXPECT_EQ(out.getChannels(), 3);
  
          out.startWriteAVI(1);
          {
              out.writeStreamHeader(codec); // starts LIST chunk
              size_t chunkPointer = out.getStreamPos();
              int avi_index = out.getAVIIndex(0, dc);
              {
                  out.startWriteChunk(avi_index);
                  out.putStreamBytes(data, count);
                  size_t tempChunkPointer = out.getStreamPos();
                  size_t moviPointer = out.getMoviPointer();
                  out.pushFrameOffset(chunkPointer - moviPointer);
                  out.pushFrameSize(tempChunkPointer - chunkPointer - 8);
                  out.endWriteChunk();
              }
              out.endWriteChunk(); // ends LIST chunk
          }
          out.writeIndex(0, dc);
          out.finishWriteAVI();
      }
      {
          AVIReadContainer in;
          in.initStream(filename);
          frame_list frames;
          ASSERT_TRUE(in.parseRiff(frames));
          EXPECT_EQ(in.getFps(), fps);
          EXPECT_EQ(in.getWidth(), static_cast<unsigned>(sz.width));
          EXPECT_EQ(in.getHeight(), static_cast<unsigned>(sz.height));
          ASSERT_EQ(frames.size(), static_cast<unsigned>(1));
          std::vector<char> actual = in.readFrame(frames.begin());
          ASSERT_EQ(actual.size(), count);
          for (size_t i = 0; i < count; ++i)
              EXPECT_EQ(actual.at(i), data[i]) << "at index " << i;
      }
      remove(filename.c_str());
  }
  
  }} // opencv_test::<anonymous>::