libopenshot/tests/FFmpegReader_Tests.cpp

/**
 * @file
 * @brief Unit tests for openshot::FFmpegReader
 * @author Jonathan Thomas <jonathan@openshot.org>
 *
 * @section LICENSE
 *
 * Copyright (c) 2008-2014 OpenShot Studios, LLC
 * <http://www.openshotstudios.com/>. This file is part of
 * OpenShot Library (libopenshot), an open-source project dedicated to
 * delivering high quality video editing and animation solutions to the
 * world. For more information visit <http://www.openshot.org/>.
 *
 * OpenShot Library (libopenshot) is free software: you can redistribute it
 * and/or modify it under the terms of the GNU Affero General Public License
 * as published by the Free Software Foundation, either version 3 of the
 * License, or (at your option) any later version.
 *
 * OpenShot Library (libopenshot) 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 Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with OpenShot Library. If not, see <http://www.gnu.org/licenses/>.
 *
 * Also, if your software can interact with users remotely through a computer
 * network, you should also make sure that it provides a way for users to
 * get its source. For example, if your program is a web application, its
 * interface could display a "Source" link that leads users to an archive
 * of the code. There are many ways you could offer source, and different
 * solutions will be better for different programs; see section 13 for the
 * specific requirements.
 *
 * You should also get your employer (if you work as a programmer) or school,
 * if any, to sign a "copyright disclaimer" for the program, if necessary.
 * For more information on this, and how to apply and follow the GNU AGPL, see
 * <http://www.gnu.org/licenses/>.
 */

#include "UnitTest++.h"
#include "../include/OpenShot.h"

using namespace std;
using namespace openshot;

TEST(FFmpegReader_Invalid_Path)
{
	// Check invalid path
	CHECK_THROW(FFmpegReader(""), InvalidFile);
}

TEST(FFmpegReader_GetFrame_Before_Opening)
{
	// Create a reader
	FFmpegReader r("../../src/examples/piano.wav");

	// Check invalid path
	CHECK_THROW(r.GetFrame(1), ReaderClosed);
}

TEST(FFmpegReader_Check_Audio_File)
{
	// Create a reader
	FFmpegReader r("../../src/examples/piano.wav");
	r.Open();

	// Get frame 1
	tr1::shared_ptr<Frame> f = r.GetFrame(1);

	// Get the number of channels and samples
	float *samples = f->GetAudioSamples(0);

	// Check audio properties
	CHECK_EQUAL(2, f->GetAudioChannelsCount());
	CHECK_EQUAL(333, f->GetAudioSamplesCount());

	// Check actual sample values (to be sure the waveform is correct)
	CHECK_CLOSE(0.0f, samples[0], 0.00001);
	CHECK_CLOSE(0.0f, samples[50], 0.00001);
	CHECK_CLOSE(0.0f, samples[100], 0.00001);
	CHECK_CLOSE(0.0f, samples[200], 0.00001);
	CHECK_CLOSE(0.160781, samples[230], 0.00001);
	CHECK_CLOSE(-0.06125f, samples[300], 0.00001);

	// Close reader
	r.Close();
}

TEST(FFmpegReader_Check_Video_File)
{
	// Create a reader
	FFmpegReader r("../../src/examples/test.mp4");
	r.Open();

	// Get frame 1
	tr1::shared_ptr<Frame> f = r.GetFrame(1);

	// Get the image data
	const Magick::PixelPacket* pixels = f->GetPixels(10);

	// Check image properties on scanline 10, pixel 112
	CHECK_EQUAL(5397, pixels[112].red);
	CHECK_EQUAL(0, pixels[112].blue);
	CHECK_EQUAL(49087, pixels[112].green);
	CHECK_EQUAL(0, pixels[112].opacity);

	// Get frame 1
	f = r.GetFrame(2);

	// Get the next frame
	pixels = f->GetPixels(10);

	// Check image properties on scanline 10, pixel 112
	CHECK_EQUAL(0, pixels[112].red);
	CHECK_EQUAL(48316, pixels[112].blue);
	CHECK_EQUAL(24672, pixels[112].green);
	CHECK_EQUAL(0, pixels[112].opacity);

	// Close reader
	r.Close();
}

TEST(FFmpegReader_Seek)
{
	// Create a reader
	FFmpegReader r("../../src/examples/sintel_trailer-720p.mp4");
	r.Open();

	// Get frame
	tr1::shared_ptr<Frame> f = r.GetFrame(1);
	CHECK_EQUAL(1, f->number);
	//f->Display();
	//f->DisplayWaveform();

	// Get frame
	f = r.GetFrame(300);
	CHECK_EQUAL(300, f->number);
	//f->Display();
	//f->DisplayWaveform();

	// Get frame
	f = r.GetFrame(301);
	CHECK_EQUAL(301, f->number);
	//f->Display();
	//f->DisplayWaveform();

	// Get frame
	f = r.GetFrame(315);
	CHECK_EQUAL(315, f->number);
	//f->Display();
	//f->DisplayWaveform();

	// Get frame
	f = r.GetFrame(275);
	CHECK_EQUAL(275, f->number);
	//f->Display();
	//f->DisplayWaveform();

	// Get frame
	f = r.GetFrame(270);
	CHECK_EQUAL(270, f->number);
	//f->Display();
	//f->DisplayWaveform();

	// Get frame
	f = r.GetFrame(500);
	CHECK_EQUAL(500, f->number);
	//f->Display();
	//f->DisplayWaveform();

	// Get frame
	f = r.GetFrame(100);
	CHECK_EQUAL(100, f->number);
	//f->Display();
	//f->DisplayWaveform();

	// Get frame
	f = r.GetFrame(600);
	CHECK_EQUAL(600, f->number);
	//f->Display();
	//f->DisplayWaveform();

	// Get frame
	f = r.GetFrame(1);
	CHECK_EQUAL(1, f->number);
	//f->Display();
	//f->DisplayWaveform();

	// Get frame
	f = r.GetFrame(700);
	CHECK_EQUAL(700, f->number);
	//f->Display();
	//f->DisplayWaveform();

	// Close reader
	r.Close();

}