libopenshot/tests/CVStabilizer.cpp

/**
 * @file
 * @brief Unit tests for openshot::Frame
 * @author Jonathan Thomas <jonathan@openshot.org>
 * @author FeRD (Frank Dana) <ferdnyc@gmail.com>
 *
 * @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 "Clip.h"
#include "CVStabilization.h"  // for TransformParam, CamTrajectory, CVStabilization
#include "ProcessingController.h"

using namespace openshot;

SUITE(CVStabilizer_Tests)
{

    // Just for the stabilizer constructor, it won't be used
    ProcessingController processingController;

    TEST(Stabilize_Video)
    {
        // Create a video clip
        std::stringstream path;
        path << TEST_MEDIA_PATH << "test.avi";

        // Open clip
        openshot::Clip c1(path.str());
        c1.Open();

        std::string json_data = R"proto(
        {
            "protobuf_data_path": "stabilizer.data",
            "smoothing-window": 30
        } )proto";

        // Create stabilizer
        CVStabilization stabilizer(json_data, processingController);

        // Stabilize clip for frames 0-21
        stabilizer.stabilizeClip(c1, 0, 21, true);

        // Get stabilized data
        TransformParam tp = stabilizer.GetTransformParamData(20);
        CamTrajectory ct = stabilizer.GetCamTrajectoryTrackedData(20);

        // // Compare if stabilized data is equal to pre-tested ones
        int dx = tp.dx*1000;
        int dy = tp.dy*1000;
        int da = tp.da*1000;
        int x = ct.x*1000;
        int y = ct.y*1000;
        int a = ct.a*1000;

        CHECK_EQUAL((int) (58), dx);
        CHECK_EQUAL((int) (-88), dy);
        CHECK_EQUAL((int) (7), da);
        CHECK_EQUAL((int) (0), x);
        CHECK_EQUAL((int) (-1), y);
        CHECK_EQUAL((int) (0), a);
    }


    TEST(SaveLoad_Protobuf)
    {

        // Create a video clip
        std::stringstream path;
        path << TEST_MEDIA_PATH << "test.avi";

        // Open clip
        openshot::Clip c1(path.str());
        c1.Open();

        std::string json_data = R"proto(
        {
            "protobuf_data_path": "stabilizer.data",
            "smoothing-window": 30
        } )proto";

        // Create first stabilizer
        CVStabilization stabilizer_1(json_data, processingController);

        // Stabilize clip for frames 0-20
        stabilizer_1.stabilizeClip(c1, 0, 20+1, true);

        // Get stabilized data
        TransformParam tp_1 = stabilizer_1.GetTransformParamData(20);
        CamTrajectory ct_1 = stabilizer_1.GetCamTrajectoryTrackedData(20);

        // Save stabilized data
        stabilizer_1.SaveStabilizedData();

        // Create second stabilizer
        CVStabilization stabilizer_2(json_data, processingController);

        // Load stabilized data from first stabilizer protobuf data
        stabilizer_2._LoadStabilizedData();

        // Get stabilized data
        TransformParam tp_2 = stabilizer_2.GetTransformParamData(20);
        CamTrajectory ct_2 = stabilizer_2.GetCamTrajectoryTrackedData(20);

        // Compare first stabilizer data with second stabilizer data
        CHECK_EQUAL((int) (tp_1.dx * 10000), (int) (tp_2.dx *10000));
        CHECK_EQUAL((int) (tp_1.dy * 10000), (int) (tp_2.dy * 10000));
        CHECK_EQUAL((int) (tp_1.da * 10000), (int) (tp_2.da * 10000));
        CHECK_EQUAL((int) (ct_1.x * 10000), (int) (ct_2.x * 10000));
        CHECK_EQUAL((int) (ct_1.y * 10000), (int) (ct_2.y * 10000));
        CHECK_EQUAL((int) (ct_1.a * 10000), (int) (ct_2.a * 10000));
    }

} // SUITE(Frame_Tests)