/** * @file * @brief Unit tests for openshot::Keyframe * @author Jonathan Thomas * * @ref License */ /* LICENSE * * Copyright (c) 2008-2019 OpenShot Studios, LLC * . 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 . * * 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 . */ #include #include #include "UnitTest++.h" // Prevent name clashes with juce::UnitTest #define DONT_SET_USING_JUCE_NAMESPACE 1 #include "KeyFrame.h" #include "TrackedObjectBBox.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; TEST(Keyframe_GetPoint_With_No_Points) { // Create an empty keyframe Keyframe k1; CHECK_THROW(k1.GetPoint(0), OutOfBoundsPoint); } TEST(Keyframe_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(Keyframe_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(Keyframe_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(Keyframe_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(Keyframe_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(Keyframe_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(Keyframe_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(Keyframe_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(Keyframe_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(Keyframe_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(Keyframe_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(Keyframe_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(Keyframe_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(Keyframe_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(Keyframe_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(Keyframe_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(Keyframe_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(Keyframe_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(Keyframe_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(Keyframe_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(Keyframe_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(Keyframe_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(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 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 trackedObject = std::static_pointer_cast(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 trackedTest = std::static_pointer_cast(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 trackedData = std::make_shared(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"]); }