ResInsight/ApplicationCode/UnitTests/RiaPolyArcLineSampler-Test.cpp

#include "gtest/gtest.h"

#include "RiaPolyArcLineSampler.h"
#include <iostream>

//--------------------------------------------------------------------------------------------------
TEST(RiaPolyArcLineSampler, Basic)
{
    std::vector<cvf::Vec3d> points{{0, 0, 0}, {0, 0, -10}, {0, 10, -20}, {0, 20, -20}, {0, 30, -30}};

    RiaPolyArcLineSampler sampler({0, 0, -1}, points);

    std::vector<cvf::Vec3d> sampledPoints;
    std::vector<double>     mds;

    sampler.sampledPointsAndMDs(2, true, &sampledPoints, &mds);
#if 0
    for (size_t pIdx = 0; pIdx < sampledPoints.size(); ++pIdx)
    {
        std::cout <<  sampledPoints[pIdx].x() << " " 
                  <<  sampledPoints[pIdx].y() << " " 
                  <<  sampledPoints[pIdx].z() << " md: " << mds[pIdx] << std::endl;
    }
#endif
    EXPECT_EQ(27, (int)sampledPoints.size());
    EXPECT_NEAR(51.4159, mds.back(), 1e-4);

    EXPECT_NEAR(points[2].y(), sampledPoints[12].y(), 2);
    EXPECT_NEAR(points[2].z(), sampledPoints[12].z(), 2);

    EXPECT_NEAR(points[4].y(), sampledPoints[25].y(), 2);
    EXPECT_NEAR(points[4].z(), sampledPoints[25].z(), 2);
}

//--------------------------------------------------------------------------------------------------
TEST(RiaPolyArcLineSampler, TestInvalidInput)
{
    {
        // Two identical points after each other
        std::vector<cvf::Vec3d> points{{0, 0, -20}, {0, 0, -20}};

        RiaPolyArcLineSampler sampler({0, 0, -1}, points);

        std::vector<cvf::Vec3d> sampledPoints;
        std::vector<double>     mds;

        sampler.sampledPointsAndMDs(2, true, &sampledPoints, &mds);

        EXPECT_EQ(0, (int)sampledPoints.size());
    }

    {
        std::vector<cvf::Vec3d> points;

        RiaPolyArcLineSampler sampler({0, 0, -1}, points);

        std::vector<cvf::Vec3d> sampledPoints;
        std::vector<double>     mds;

        sampler.sampledPointsAndMDs(2, true, &sampledPoints, &mds);

        EXPECT_EQ(0, (int)sampledPoints.size());
    }

    {
        std::vector<cvf::Vec3d> points{{0, 0, 0}};

        RiaPolyArcLineSampler sampler({0, 0, -1}, points);

        std::vector<cvf::Vec3d> sampledPoints;
        std::vector<double>     mds;

        sampler.sampledPointsAndMDs(2, true, &sampledPoints, &mds);

        EXPECT_EQ(0, (int)sampledPoints.size());
    }
}