openshot/libopenshot
0
0
Fork 0
mirror of https://github.com/OpenShot/libopenshot.git synced 2026-03-02 08:53:52 -08:00
Code Issues Packages Projects Releases Wiki Activity
Files
d24c2e451c5ea6a21ab08495b9532c2ced60352d
libopenshot/tests/KeyFrame_Tests.cpp
Brenno c0c2a82336 Merge branch 'develop' into keyframe-refactor
2021-02-01 14:22:41 -03:00

707 lines
21 KiB
C++
Raw Blame History

/**
* @file
* @brief Unit tests for openshot::Keyframe
* @author Jonathan Thomas <jonathan@openshot.org>
*
* @ref License
*/
/* LICENSE
*
* Copyright (c) 2008-2019 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 Lesser 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 Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with OpenShot Library. If not, see <http://www.gnu.org/licenses/>.
*/
#include <sstream>
#include <memory>
#include "UnitTest++.h"
// Prevent name clashes with juce::UnitTest
#define DONT_SET_USING_JUCE_NAMESPACE 1
#include "KeyFrame.h"
#include "TrackedObjectBBox.h"
#include "Exceptions.h"
#include "Coordinate.h"
#include "Fraction.h"
#include "Clip.h"
#include "Timeline.h"
#include "effects/Tracker.h"
#include "Point.h"
using namespace std;
using namespace openshot;
SUITE(Keyframe) {
TEST(GetPoint_With_No_Points)
{
// Create an empty keyframe
Keyframe k1;
CHECK_THROW(k1.GetPoint(0), OutOfBoundsPoint);
}
TEST(GetPoint_With_1_Points)
{
// Create an empty keyframe
Keyframe k1;
k1.AddPoint(openshot::Point(2,3));
CHECK_THROW(k1.GetPoint(-1), OutOfBoundsPoint);
CHECK_EQUAL(1, k1.GetCount());
CHECK_CLOSE(2.0f, k1.GetPoint(0).co.X, 0.00001);
CHECK_CLOSE(3.0f, k1.GetPoint(0).co.Y, 0.00001);
CHECK_THROW(k1.GetPoint(1), OutOfBoundsPoint);
}
TEST(AddPoint_With_1_Point)
{
// Create an empty keyframe
Keyframe k1;
k1.AddPoint(openshot::Point(2,9));
CHECK_CLOSE(2.0f, k1.GetPoint(0).co.X, 0.00001);
CHECK_THROW(k1.GetPoint(-1), OutOfBoundsPoint);
CHECK_THROW(k1.GetPoint(1), OutOfBoundsPoint);
}
TEST(AddPoint_With_2_Points)
{
// Create an empty keyframe
Keyframe k1;
k1.AddPoint(openshot::Point(2,9));
k1.AddPoint(openshot::Point(5,20));
CHECK_CLOSE(2.0f, k1.GetPoint(0).co.X, 0.00001);
CHECK_CLOSE(5.0f, k1.GetPoint(1).co.X, 0.00001);
CHECK_THROW(k1.GetPoint(-1), OutOfBoundsPoint);
CHECK_THROW(k1.GetPoint(2), OutOfBoundsPoint);
}
TEST(GetValue_For_Bezier_Curve_2_Points)
{
// Create a keyframe curve with 2 points
Keyframe kf;
kf.AddPoint(openshot::Point(Coordinate(1, 1), BEZIER));
kf.AddPoint(openshot::Point(Coordinate(50, 4), BEZIER));
// Spot check values from the curve
CHECK_CLOSE(1.0f, kf.GetValue(-1), 0.0001);
CHECK_CLOSE(1.0f, kf.GetValue(0), 0.0001);
CHECK_CLOSE(1.0f, kf.GetValue(1), 0.0001);
CHECK_CLOSE(1.12414f, kf.GetValue(9), 0.0001);
CHECK_CLOSE(1.86370f, kf.GetValue(20), 0.0001);
CHECK_CLOSE(3.79733f, kf.GetValue(40), 0.0001);
CHECK_CLOSE(4.0f, kf.GetValue(50), 0.0001);
// Check the expected number of values
CHECK_EQUAL(51, kf.GetLength());
}
TEST(GetValue_For_Bezier_Curve_5_Points_40_Percent_Handle)
{
// Create a keyframe curve with 2 points
Keyframe kf;
kf.AddPoint(openshot::Point(Coordinate(1, 1), BEZIER));
kf.AddPoint(openshot::Point(Coordinate(50, 4), BEZIER));
kf.AddPoint(openshot::Point(Coordinate(100, 10), BEZIER));
kf.AddPoint(openshot::Point(Coordinate(150, 0), BEZIER));
kf.AddPoint(openshot::Point(Coordinate(200, 3), BEZIER));
// Spot check values from the curve
CHECK_CLOSE(kf.GetValue(-1), 1.0f, 0.0001);
CHECK_CLOSE(1.0f, kf.GetValue(0), 0.0001);
CHECK_CLOSE(1.0f, kf.GetValue(1), 0.0001);
CHECK_CLOSE(2.68197f, kf.GetValue(27), 0.0001);
CHECK_CLOSE(7.47719f, kf.GetValue(77), 0.0001);
CHECK_CLOSE(4.20468f, kf.GetValue(127), 0.0001);
CHECK_CLOSE(1.73860f, kf.GetValue(177), 0.0001);
CHECK_CLOSE(3.0f, kf.GetValue(200), 0.0001);
// Check the expected number of values
CHECK_EQUAL(201, kf.GetLength());
}
TEST(GetValue_For_Bezier_Curve_5_Points_25_Percent_Handle)
{
// Create a keyframe curve with 2 points
Keyframe kf;
kf.AddPoint(openshot::Point(Coordinate(1, 1), BEZIER));
kf.AddPoint(openshot::Point(Coordinate(50, 4), BEZIER));
kf.AddPoint(openshot::Point(Coordinate(100, 10), BEZIER));
kf.AddPoint(openshot::Point(Coordinate(150, 0), BEZIER));
kf.AddPoint(openshot::Point(Coordinate(200, 3), BEZIER));
// Spot check values from the curve
CHECK_CLOSE(1.0f, kf.GetValue(-1), 0.0001);
CHECK_CLOSE(1.0f, kf.GetValue(0), 0.0001);
CHECK_CLOSE(1.0f, kf.GetValue(1), 0.0001);
CHECK_CLOSE(2.68197f, kf.GetValue(27), 0.0001);
CHECK_CLOSE(7.47719f, kf.GetValue(77), 0.0001);
CHECK_CLOSE(4.20468f, kf.GetValue(127), 0.0001);
CHECK_CLOSE(1.73860f, kf.GetValue(177), 0.0001);
CHECK_CLOSE(3.0f, kf.GetValue(200), 0.0001);
// Check the expected number of values
CHECK_EQUAL(201, kf.GetLength());
}
TEST(GetValue_For_Linear_Curve_3_Points)
{
// Create a keyframe curve with 2 points
Keyframe kf;
kf.AddPoint(openshot::Point(Coordinate(1, 1), LINEAR));
kf.AddPoint(openshot::Point(Coordinate(25, 8), LINEAR));
kf.AddPoint(openshot::Point(Coordinate(50, 2), LINEAR));
// Spot check values from the curve
CHECK_CLOSE(1.0f, kf.GetValue(-1), 0.0001);
CHECK_CLOSE(1.0f, kf.GetValue(0), 0.0001);
CHECK_CLOSE(1.0f, kf.GetValue(1), 0.0001);
CHECK_CLOSE(3.33333f, kf.GetValue(9), 0.0001);
CHECK_CLOSE(6.54167f, kf.GetValue(20), 0.0001);
CHECK_CLOSE(4.4f, kf.GetValue(40), 0.0001);
CHECK_CLOSE(2.0f, kf.GetValue(50), 0.0001);
// Check the expected number of values
CHECK_EQUAL(51, kf.GetLength());
}
TEST(GetValue_For_Constant_Curve_3_Points)
{
// Create a keyframe curve with 2 points
Keyframe kf;
kf.AddPoint(openshot::Point(Coordinate(1, 1), CONSTANT));
kf.AddPoint(openshot::Point(Coordinate(25, 8), CONSTANT));
kf.AddPoint(openshot::Point(Coordinate(50, 2), CONSTANT));
// Spot check values from the curve
CHECK_CLOSE(1.0f, kf.GetValue(-1), 0.0001);
CHECK_CLOSE(1.0f, kf.GetValue(0), 0.0001);
CHECK_CLOSE(1.0f, kf.GetValue(1), 0.0001);
CHECK_CLOSE(1.0f, kf.GetValue(24), 0.0001);
CHECK_CLOSE(8.0f, kf.GetValue(25), 0.0001);
CHECK_CLOSE(8.0f, kf.GetValue(40), 0.0001);
CHECK_CLOSE(8.0f, kf.GetValue(49), 0.0001);
CHECK_CLOSE(2.0f, kf.GetValue(50), 0.0001);
// Check the expected number of values
CHECK_EQUAL(51, kf.GetLength());
}
TEST(Check_Direction_and_Repeat_Fractions)
{
// Create a keyframe curve with 2 points
Keyframe kf;
kf.AddPoint(1, 500);
kf.AddPoint(400, 100);
kf.AddPoint(500, 500);
// Spot check values from the curve
CHECK_EQUAL(500, kf.GetInt(1));
CHECK_EQUAL(false, kf.IsIncreasing(1));
CHECK_EQUAL(1, kf.GetRepeatFraction(1).num);
CHECK_EQUAL(13, kf.GetRepeatFraction(1).den);
CHECK_EQUAL(500, kf.GetDelta(1));
CHECK_EQUAL(498, kf.GetInt(24));
CHECK_EQUAL(false, kf.IsIncreasing(24));
CHECK_EQUAL(3, kf.GetRepeatFraction(24).num);
CHECK_EQUAL(6, kf.GetRepeatFraction(24).den);
CHECK_EQUAL(0, kf.GetDelta(24));
CHECK_EQUAL(100, kf.GetLong(390));
CHECK_EQUAL(true, kf.IsIncreasing(390));
CHECK_EQUAL(3, kf.GetRepeatFraction(390).num);
CHECK_EQUAL(16, kf.GetRepeatFraction(390).den);
CHECK_EQUAL(0, kf.GetDelta(390));
CHECK_EQUAL(100, kf.GetLong(391));
CHECK_EQUAL(true, kf.IsIncreasing(391));
CHECK_EQUAL(4, kf.GetRepeatFraction(391).num);
CHECK_EQUAL(16, kf.GetRepeatFraction(391).den);
CHECK_EQUAL(-1, kf.GetDelta(388));
}
TEST(Get_Closest_Point)
{
// Create a keyframe curve with 2 points
Keyframe kf;
kf.AddPoint(1, 0.0);
kf.AddPoint(1000, 1.0);
kf.AddPoint(2500, 0.0);
// Spot check values from the curve (to the right)
CHECK_EQUAL(1000, kf.GetClosestPoint(openshot::Point(900, 900)).co.X);
CHECK_EQUAL(1, kf.GetClosestPoint(openshot::Point(1, 1)).co.X);
CHECK_EQUAL(1000, kf.GetClosestPoint(openshot::Point(5, 5)).co.X);
CHECK_EQUAL(1000, kf.GetClosestPoint(openshot::Point(1000, 1000)).co.X);
CHECK_EQUAL(2500, kf.GetClosestPoint(openshot::Point(1001, 1001)).co.X);
CHECK_EQUAL(2500, kf.GetClosestPoint(openshot::Point(2500, 2500)).co.X);
CHECK_EQUAL(2500, kf.GetClosestPoint(openshot::Point(3000, 3000)).co.X);
// Spot check values from the curve (to the left)
CHECK_EQUAL(1, kf.GetClosestPoint(openshot::Point(900, 900), true).co.X);
CHECK_EQUAL(1, kf.GetClosestPoint(openshot::Point(1, 1), true).co.X);
CHECK_EQUAL(1, kf.GetClosestPoint(openshot::Point(5, 5), true).co.X);
CHECK_EQUAL(1, kf.GetClosestPoint(openshot::Point(1000, 1000), true).co.X);
CHECK_EQUAL(1000, kf.GetClosestPoint(openshot::Point(1001, 1001), true).co.X);
CHECK_EQUAL(1000, kf.GetClosestPoint(openshot::Point(2500, 2500), true).co.X);
CHECK_EQUAL(2500, kf.GetClosestPoint(openshot::Point(3000, 3000), true).co.X);
}
TEST(Get_Previous_Point)
{
// Create a keyframe curve with 2 points
Keyframe kf;
kf.AddPoint(1, 0.0);
kf.AddPoint(1000, 1.0);
kf.AddPoint(2500, 0.0);
// Spot check values from the curve
CHECK_EQUAL(1, kf.GetPreviousPoint(kf.GetClosestPoint(openshot::Point(900, 900))).co.X);
CHECK_EQUAL(1, kf.GetPreviousPoint(kf.GetClosestPoint(openshot::Point(1, 1))).co.X);
CHECK_EQUAL(1, kf.GetPreviousPoint(kf.GetClosestPoint(openshot::Point(5, 5))).co.X);
CHECK_EQUAL(1, kf.GetPreviousPoint(kf.GetClosestPoint(openshot::Point(1000, 1000))).co.X);
CHECK_EQUAL(1000, kf.GetPreviousPoint(kf.GetClosestPoint(openshot::Point(1001, 1001))).co.X);
CHECK_EQUAL(1000, kf.GetPreviousPoint(kf.GetClosestPoint(openshot::Point(2500, 2500))).co.X);
CHECK_EQUAL(1000, kf.GetPreviousPoint(kf.GetClosestPoint(openshot::Point(3000, 3000))).co.X);
}
TEST(Get_Max_Point)
{
// Create a keyframe curve
Keyframe kf;
kf.AddPoint(1, 1.0);
// Spot check values from the curve
CHECK_EQUAL(1.0, kf.GetMaxPoint().co.Y);
kf.AddPoint(2, 0.0);
// Spot check values from the curve
CHECK_EQUAL(1.0, kf.GetMaxPoint().co.Y);
kf.AddPoint(3, 2.0);
// Spot check values from the curve
CHECK_EQUAL(2.0, kf.GetMaxPoint().co.Y);
kf.AddPoint(4, 1.0);
// Spot check values from the curve
CHECK_EQUAL(2.0, kf.GetMaxPoint().co.Y);
}
TEST(Scale_Keyframe)
{
// Create a keyframe curve with 2 points
Keyframe kf;
kf.AddPoint(openshot::Point(Coordinate(1, 1), BEZIER));
kf.AddPoint(openshot::Point(Coordinate(25, 8), BEZIER));
kf.AddPoint(openshot::Point(Coordinate(50, 2), BEZIER));
// Spot check values from the curve
CHECK_CLOSE(1.0f, kf.GetValue(1), 0.01);
CHECK_CLOSE(7.99f, kf.GetValue(24), 0.01);
CHECK_CLOSE(8.0f, kf.GetValue(25), 0.01);
CHECK_CLOSE(3.85f, kf.GetValue(40), 0.01);
CHECK_CLOSE(2.01f, kf.GetValue(49), 0.01);
CHECK_CLOSE(2.0f, kf.GetValue(50), 0.01);
// Resize / Scale the keyframe
kf.ScalePoints(2.0); // 100% larger
// Spot check values from the curve
CHECK_CLOSE(1.0f, kf.GetValue(1), 0.01);
CHECK_CLOSE(4.08f, kf.GetValue(24), 0.01);
CHECK_CLOSE(4.36f, kf.GetValue(25), 0.01);
CHECK_CLOSE(7.53f, kf.GetValue(40), 0.01);
CHECK_CLOSE(7.99f, kf.GetValue(49), 0.01);
CHECK_CLOSE(8.0f, kf.GetValue(50), 0.01);
CHECK_CLOSE(2.39f, kf.GetValue(90), 0.01);
CHECK_CLOSE(2.0f, kf.GetValue(100), 0.01);
// Resize / Scale the keyframe
kf.ScalePoints(0.5); // 50% smaller, which should match the original size
// Spot check values from the curve
CHECK_CLOSE(1.0f, kf.GetValue(1), 0.01);
CHECK_CLOSE(7.99f, kf.GetValue(24), 0.01);
CHECK_CLOSE(8.0f, kf.GetValue(25), 0.01);
CHECK_CLOSE(3.85f, kf.GetValue(40), 0.01);
CHECK_CLOSE(2.01f, kf.GetValue(49), 0.01);
CHECK_CLOSE(2.0f, kf.GetValue(50), 0.01);
}
TEST(Flip_Keyframe)
{
// Create a keyframe curve with 2 points
Keyframe kf;
kf.AddPoint(openshot::Point(Coordinate(1, 1), LINEAR));
kf.AddPoint(openshot::Point(Coordinate(25, 8), LINEAR));
kf.AddPoint(openshot::Point(Coordinate(50, 2), LINEAR));
kf.AddPoint(openshot::Point(Coordinate(100, 10), LINEAR));
// Spot check values from the curve
CHECK_CLOSE(1.0f, kf.GetValue(1), 0.01);
CHECK_CLOSE(8.0f, kf.GetValue(25), 0.01);
CHECK_CLOSE(2.0f, kf.GetValue(50), 0.01);
CHECK_CLOSE(10.0f, kf.GetValue(100), 0.01);
// Flip the points
kf.FlipPoints();
// Spot check values from the curve
CHECK_CLOSE(10.0f, kf.GetValue(1), 0.01);
CHECK_CLOSE(2.0f, kf.GetValue(25), 0.01);
CHECK_CLOSE(8.0f, kf.GetValue(50), 0.01);
CHECK_CLOSE(1.0f, kf.GetValue(100), 0.01);
// Flip the points again (back to the original)
kf.FlipPoints();
// Spot check values from the curve
CHECK_CLOSE(1.0f, kf.GetValue(1), 0.01);
CHECK_CLOSE(8.0f, kf.GetValue(25), 0.01);
CHECK_CLOSE(2.0f, kf.GetValue(50), 0.01);
CHECK_CLOSE(10.0f, kf.GetValue(100), 0.01);
}
TEST(Remove_Duplicate_Point)
{
// Create a keyframe curve with 2 points
Keyframe kf;
kf.AddPoint(1, 0.0);
kf.AddPoint(1, 1.0);
kf.AddPoint(1, 2.0);
// Spot check values from the curve
CHECK_EQUAL(1, kf.GetLength());
CHECK_CLOSE(2.0, kf.GetPoint(0).co.Y, 0.01);
}
TEST(Large_Number_Values)
{
// Large value
int64_t const large_value = 30 * 60 * 90;
// Create a keyframe curve with 2 points
Keyframe kf;
kf.AddPoint(1, 1.0);
kf.AddPoint(large_value, 100.0); // 90 minutes long
// Spot check values from the curve
CHECK_EQUAL(large_value + 1, kf.GetLength());
CHECK_CLOSE(1.0, kf.GetPoint(0).co.Y, 0.01);
CHECK_CLOSE(100.0, kf.GetPoint(1).co.Y, 0.01);
}
TEST(Remove_Point)
{
Keyframe kf;
kf.AddPoint(openshot::Point(Coordinate(1, 1), CONSTANT));
kf.AddPoint(openshot::Point(Coordinate(3, 100), CONSTANT));
CHECK_EQUAL(1, kf.GetInt(2));
kf.AddPoint(openshot::Point(Coordinate(2, 50), CONSTANT));
CHECK_EQUAL(50, kf.GetInt(2));
kf.RemovePoint(1); // This is the index of point with X == 2
CHECK_EQUAL(1, kf.GetInt(2));
CHECK_THROW(kf.RemovePoint(100), OutOfBoundsPoint);
}
TEST(Constant_Interpolation_First_Segment)
{
Keyframe kf;
kf.AddPoint(Point(Coordinate(1, 1), CONSTANT));
kf.AddPoint(Point(Coordinate(2, 50), CONSTANT));
kf.AddPoint(Point(Coordinate(3, 100), CONSTANT));
CHECK_EQUAL(1, kf.GetInt(0));
CHECK_EQUAL(1, kf.GetInt(1));
CHECK_EQUAL(50, kf.GetInt(2));
CHECK_EQUAL(100, kf.GetInt(3));
CHECK_EQUAL(100, kf.GetInt(4));
}
TEST(isIncreasing)
{
// Which cases need to be tested to keep same behaviour as
// previously?
//
// - "invalid point" => true
// - point where all next values are equal => false
// - point where first non-eq next value is smaller => false
// - point where first non-eq next value is larger => true
Keyframe kf;
kf.AddPoint(1, 1, LINEAR); // testing with linear
kf.AddPoint(3, 5, BEZIER); // testing with bezier
kf.AddPoint(6, 10, CONSTANT); // first non-eq is smaller
kf.AddPoint(8, 8, CONSTANT); // first non-eq is larger
kf.AddPoint(10, 10, CONSTANT); // all next values are equal
kf.AddPoint(15, 10, CONSTANT);
// "invalid points"
CHECK_EQUAL(true, kf.IsIncreasing(0));
CHECK_EQUAL(true, kf.IsIncreasing(15));
// all next equal
CHECK_EQUAL(false, kf.IsIncreasing(12));
// first non-eq is larger
CHECK_EQUAL(true, kf.IsIncreasing(8));
// first non-eq is smaller
CHECK_EQUAL(false, kf.IsIncreasing(6));
// bezier and linear
CHECK_EQUAL(true, kf.IsIncreasing(4));
CHECK_EQUAL(true, kf.IsIncreasing(2));
}
TEST(GetLength)
{
Keyframe f;
CHECK_EQUAL(0, f.GetLength());
f.AddPoint(1, 1);
CHECK_EQUAL(1, f.GetLength());
f.AddPoint(2, 1);
CHECK_EQUAL(3, f.GetLength());
f.AddPoint(200, 1);
CHECK_EQUAL(201, f.GetLength());
Keyframe g;
g.AddPoint(200, 1);
CHECK_EQUAL(1, g.GetLength());
g.AddPoint(1,1);
CHECK_EQUAL(201, g.GetLength());
}
TEST(Use_Interpolation_of_Segment_End_Point)
{
Keyframe f;
f.AddPoint(1,0, CONSTANT);
f.AddPoint(100,155, BEZIER);
CHECK_CLOSE(75.9, f.GetValue(50), 0.1);
}
TEST(Handle_Large_Segment)
{
Keyframe kf;
kf.AddPoint(1, 0, CONSTANT);
kf.AddPoint(1000000, 1, LINEAR);
UNITTEST_TIME_CONSTRAINT(10); // 10 milliseconds would still be relatively slow, but need to think about slower build machines!
CHECK_CLOSE(0.5, kf.GetValue(500000), 0.01);
CHECK_EQUAL(true, kf.IsIncreasing(10));
Fraction fr = kf.GetRepeatFraction(250000);
CHECK_CLOSE(0.5, (double)fr.num / fr.den, 0.01);
}
TEST(Point_Vector_Constructor)
{
std::vector<Point> points{Point(1, 10), Point(5, 20), Point(10, 30)};
Keyframe k1(points);
CHECK_EQUAL(11, k1.GetLength());
CHECK_CLOSE(30.0f, k1.GetValue(10), 0.0001);
}
}; // SUITE
TEST(TrackedObjectBBox_init_test) {
TrackedObjectBBox kfb;
}
TEST(TrackedObjectBBox_addBox_test) {
TrackedObjectBBox kfb;
kfb.AddBox(1, 10.0, 10.0, 100.0, 100.0, 0.0);
CHECK_EQUAL(true, kfb.Contains(1));
CHECK_EQUAL(1, kfb.GetLength());
kfb.RemoveBox(1);
CHECK_EQUAL(false, kfb.Contains(1));
CHECK_EQUAL(0, kfb.GetLength());
}
TEST(TrackedObjectBBox_GetVal_test) {
TrackedObjectBBox kfb;
kfb.AddBox(1, 10.0, 10.0, 100.0, 100.0, 0.0);
BBox val = kfb.GetBox(1);
CHECK_EQUAL(10.0, val.cx);
CHECK_EQUAL(10.0, val.cy);
CHECK_EQUAL(100.0,val.width);
CHECK_EQUAL(100.0,val.height);
CHECK_EQUAL(0.0, val.angle);
}
TEST(TrackedObjectBBox_GetVal_Interpolation) {
TrackedObjectBBox kfb;
kfb.AddBox(1, 10.0, 10.0, 100.0, 100.0, 0.0);
kfb.AddBox(11, 20.0, 20.0, 100.0, 100.0, 0.0);
kfb.AddBox(21, 30.0, 30.0, 100.0, 100.0, 0.0);
kfb.AddBox(31, 40.0, 40.0, 100.0, 100.0, 0.0);
BBox val = kfb.GetBox(5);
CHECK_EQUAL(14.0, val.cx);
CHECK_EQUAL(14.0, val.cy);
CHECK_EQUAL(100.0,val.width);
CHECK_EQUAL(100.0,val.height);
val = kfb.GetBox(15);
CHECK_EQUAL(24.0, val.cx);
CHECK_EQUAL(24.0, val.cy);
CHECK_EQUAL(100.0,val.width);
CHECK_EQUAL(100.0, val.height);
val = kfb.GetBox(25);
CHECK_EQUAL(34.0, val.cx);
CHECK_EQUAL(34.0, val.cy);
CHECK_EQUAL(100.0,val.width);
CHECK_EQUAL(100.0, val.height);
}
TEST(TrackedObjectBBox_Json_set) {
TrackedObjectBBox kfb;
kfb.AddBox(1, 10.0, 10.0, 100.0, 100.0, 0.0);
kfb.AddBox(10, 20.0, 20.0, 100.0, 100.0, 0.0);
kfb.AddBox(20, 30.0, 30.0, 100.0, 100.0, 0.0);
kfb.AddBox(30, 40.0, 40.0, 100.0, 100.0, 0.0);
kfb.scale_x.AddPoint(1, 2.0);
kfb.scale_x.AddPoint(10, 3.0);
kfb.SetBaseFPS(Fraction(24.0, 1.0));
auto dataJSON = kfb.Json();
TrackedObjectBBox fromJSON_kfb;
fromJSON_kfb.SetJson(dataJSON);
CHECK_EQUAL(kfb.GetBaseFPS().num, fromJSON_kfb.GetBaseFPS().num);
double time_kfb = kfb.FrameNToTime(1, 1.0);
double time_fromJSON_kfb = fromJSON_kfb.FrameNToTime(1, 1.0);
CHECK_EQUAL(time_kfb, time_fromJSON_kfb);
BBox kfb_bbox = kfb.BoxVec[time_kfb];
BBox fromJSON_bbox = fromJSON_kfb.BoxVec[time_fromJSON_kfb];
CHECK_EQUAL(kfb_bbox.cx, fromJSON_bbox.cx);
CHECK_EQUAL(kfb_bbox.cy, fromJSON_bbox.cy);
CHECK_EQUAL(kfb_bbox.width, fromJSON_bbox.width);
CHECK_EQUAL(kfb_bbox.height, fromJSON_bbox.height);
CHECK_EQUAL(kfb_bbox.angle, fromJSON_bbox.angle);
}
TEST(TrackedObjectBBox_Scale_test){
TrackedObjectBBox kfb;
kfb.AddBox(1, 10.0, 10.0, 10.0, 10.0, 0.0);
kfb.scale_x.AddPoint(1.0, 2.0);
kfb.scale_y.AddPoint(1.0, 3.0);
BBox bbox = kfb.GetBox(1);
CHECK_EQUAL(20.0, bbox.width);
CHECK_EQUAL(30.0, bbox.height);
}
TEST(Attach_test){
std::stringstream path1, path2;
path1 << TEST_MEDIA_PATH << "test.avi";
path2 << TEST_MEDIA_PATH << "run.mp4";
// Create Timelime
Timeline t(1280, 720, Fraction(25,1), 44100, 2, ChannelLayout::LAYOUT_STEREO);
// Create Clip and add it to the Timeline
Clip clip(new FFmpegReader(path1.str()));
clip.Id("AAAA1234");
// Create a child clip and add it to the Timeline
Clip childClip(new FFmpegReader(path2.str()));
childClip.Id("CHILD123");
// Add clips to timeline
t.AddClip(&childClip);
t.AddClip(&clip);
// Create tracker and add it to clip
Tracker tracker;
clip.AddEffect(&tracker);
// Save a pointer to trackedData
std::shared_ptr<TrackedObjectBBox> trackedData = tracker.trackedData;
// Change trackedData scale
trackedData->scale_x.AddPoint(1, 2.0);
CHECK_EQUAL(2.0, trackedData->scale_x.GetValue(1));
// Tracked Data JSON
auto trackedDataJson = trackedData->JsonValue();
// Get and cast the trakcedObject
auto trackedObject_base = t.GetTrackedObject("None");
std::shared_ptr<TrackedObjectBBox> trackedObject = std::static_pointer_cast<TrackedObjectBBox>(trackedObject_base);
CHECK_EQUAL(trackedData, trackedObject);
// Set trackedObject Json Value
trackedObject->SetJsonValue(trackedDataJson);
// Attach childClip to tracked object
std::string tracked_id = trackedData->Id();
childClip.Open();
childClip.AttachToObject(tracked_id);
std::shared_ptr<TrackedObjectBBox> trackedTest = std::static_pointer_cast<TrackedObjectBBox>(childClip.GetAttachedObject());
CHECK_EQUAL(trackedData->scale_x.GetValue(1), trackedTest->scale_x.GetValue(1));
auto frameTest = childClip.GetFrame(1);
childClip.Close();
}
TEST(GetBoxValues_test){
TrackedObjectBBox trackedDataObject;
trackedDataObject.AddBox(1, 10.0, 10.0, 20.0, 20.0, 30.0);
std::shared_ptr<TrackedObjectBase> trackedData = std::make_shared<TrackedObjectBBox>(trackedDataObject);
auto boxValues = trackedData->GetBoxValues(1);
CHECK_EQUAL(10.0, boxValues["cx"]);
CHECK_EQUAL(10.0, boxValues["cy"]);
CHECK_EQUAL(20.0, boxValues["w"]);
CHECK_EQUAL(20.0, boxValues["h"]);
CHECK_EQUAL(30.0, boxValues["ang"]);
}
Reference in New Issue View Git Blame Copy Permalink