summaryrefslogtreecommitdiff
path: root/test/ffmpeg_pts_offset.cc
blob: 2090e517bdd7eeea1d6ff8dea84dd89d30c4a55d (plain)
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
/*
    Copyright (C) 2013 Carl Hetherington <cth@carlh.net>

    This program is free software; you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation; either version 2 of the License, or
    (at your option) any later version.

    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with this program; if not, write to the Free Software
    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.

*/

#include <boost/test/unit_test.hpp>
#include "lib/film.h"
#include "lib/ffmpeg_decoder.h"
#include "lib/ffmpeg_content.h"
#include "test.h"

using boost::shared_ptr;

BOOST_AUTO_TEST_CASE (ffmpeg_pts_offset_test)
{
	shared_ptr<Film> film = new_test_film ("ffmpeg_pts_offset_test");
	shared_ptr<FFmpegContent> content (new FFmpegContent (film, "test/data/test.mp4"));
	content->_audio_stream.reset (new FFmpegAudioStream);
	content->_video_frame_rate = 24;

	{
		/* Sound == video so no offset required */
		content->_first_video = 0;
		content->_audio_stream->first_audio = 0;
		FFmpegDecoder decoder (film, content, true, true);
		BOOST_CHECK_EQUAL (decoder._video_pts_offset, 0);
		BOOST_CHECK_EQUAL (decoder._audio_pts_offset, 0);
	}

	{
		/* Common offset should be removed */
		content->_first_video = 600;
		content->_audio_stream->first_audio = 600;
		FFmpegDecoder decoder (film, content, true, true);
		BOOST_CHECK_EQUAL (decoder._video_pts_offset, -600);
		BOOST_CHECK_EQUAL (decoder._audio_pts_offset, -600);
	}

	{
		/* Video is on a frame boundary */
		content->_first_video = 1.0 / 24.0;
		content->_audio_stream->first_audio = 0;
		FFmpegDecoder decoder (film, content, true, true);
		BOOST_CHECK_EQUAL (decoder._video_pts_offset, 0);
		BOOST_CHECK_EQUAL (decoder._audio_pts_offset, 0);
	}

	{
		/* Video is off a frame boundary */
		double const frame = 1.0 / 24.0;
		content->_first_video = frame + 0.0215;
		content->_audio_stream->first_audio = 0;
		FFmpegDecoder decoder (film, content, true, true);
		BOOST_CHECK_CLOSE (decoder._video_pts_offset, (frame - 0.0215), 0.00001);
		BOOST_CHECK_CLOSE (decoder._audio_pts_offset, (frame - 0.0215), 0.00001);
	}

	{
		/* Video is off a frame boundary and both have a common offset */
		double const frame = 1.0 / 24.0;
		content->_first_video = frame + 0.0215 + 4.1;
		content->_audio_stream->first_audio = 4.1;
		FFmpegDecoder decoder (film, content, true, true);
		BOOST_CHECK_EQUAL (decoder._video_pts_offset, (frame - 0.0215) - 4.1);
		BOOST_CHECK_EQUAL (decoder._audio_pts_offset, (frame - 0.0215) - 4.1);
	}
}