[Part][test] Add tests for GeomBSplineCurve::Trim()

This commit is contained in:
Ajinkya Dahale
2024-06-05 18:24:04 +05:30
parent 7c15affce6
commit 16cc6bf56a
2 changed files with 93 additions and 0 deletions

View File

@@ -16,6 +16,7 @@ target_sources(
${CMAKE_CURRENT_SOURCE_DIR}/FeaturePartCut.cpp
${CMAKE_CURRENT_SOURCE_DIR}/FeaturePartFuse.cpp
${CMAKE_CURRENT_SOURCE_DIR}/FeatureRevolution.cpp
${CMAKE_CURRENT_SOURCE_DIR}/Geometry.cpp
${CMAKE_CURRENT_SOURCE_DIR}/PartFeature.cpp
${CMAKE_CURRENT_SOURCE_DIR}/PartFeatures.cpp
${CMAKE_CURRENT_SOURCE_DIR}/PartTestHelpers.cpp

View File

@@ -0,0 +1,92 @@
// SPDX-License-Identifier: LGPL-2.1-or-later
#include <gtest/gtest.h>
#include <boost/core/ignore_unused.hpp>
#include "Mod/Part/App/Geometry.h"
#include <src/App/InitApplication.h>
#include <BRepBuilderAPI_MakeVertex.hxx>
#include "PartTestHelpers.h"
#include "App/MappedElement.h"
// using namespace Part;
// using namespace PartTestHelpers;
class GeometryTest: public ::testing::Test, public PartTestHelpers::PartTestHelperClass
{
protected:
static void SetUpTestSuite()
{
tests::initApplication();
}
void SetUp() override
{
createTestDoc();
}
void TearDown() override
{}
};
TEST_F(GeometryTest, testTrimBSpline)
{
// Arrange
// create arbitrary B-splines periodic and non-periodic, with arbitrary knots
// NOTE: Avoid B-spline with typical knots like those ranging from [0,1] or [-1,1]
int degree = 3;
std::vector<Base::Vector3d> poles;
poles.emplace_back(1, 0, 0);
poles.emplace_back(1, 1, 0);
poles.emplace_back(1, 0.5, 0);
poles.emplace_back(0, 1, 0);
poles.emplace_back(0, 0, 0);
std::vector<double> weights(5, 1.0);
std::vector<double> knotsNonPeriodic = {0.0, 1.0, 2.0};
std::vector<int> multiplicitiesNonPeriodic = {degree + 1, 1, degree + 1};
Part::GeomBSplineCurve nonPeriodicBSpline1(poles,
weights,
knotsNonPeriodic,
multiplicitiesNonPeriodic,
degree,
false);
Part::GeomBSplineCurve nonPeriodicBSpline2(poles,
weights,
knotsNonPeriodic,
multiplicitiesNonPeriodic,
degree,
false);
std::vector<double> knotsPeriodic = {0.0, 0.3, 1.0, 1.5, 1.8, 2.0};
double period = knotsPeriodic.back() - knotsPeriodic.front();
std::vector<int> multiplicitiesPeriodic(6, 1);
Part::GeomBSplineCurve periodicBSpline1(poles,
weights,
knotsPeriodic,
multiplicitiesPeriodic,
degree,
true);
Part::GeomBSplineCurve periodicBSpline2(poles,
weights,
knotsPeriodic,
multiplicitiesPeriodic,
degree,
true);
// NOTE: These should be within the knot range, with param1 < param2
double param1 = 0.5, param2 = 1.4;
// TODO: Decide what to do if params are outside the range
// Act
periodicBSpline1.Trim(param1, param2);
periodicBSpline2.Trim(param2, param1);
nonPeriodicBSpline1.Trim(param1, param2);
// TODO: What happens when a non-periodic B-spline is trimmed this way?
nonPeriodicBSpline2.Trim(param2, param1);
// Assert
EXPECT_DOUBLE_EQ(periodicBSpline1.getFirstParameter(), param1);
EXPECT_DOUBLE_EQ(periodicBSpline1.getLastParameter(), param2);
EXPECT_DOUBLE_EQ(periodicBSpline2.getFirstParameter(), param2);
EXPECT_DOUBLE_EQ(periodicBSpline2.getLastParameter(), param1 + period);
EXPECT_DOUBLE_EQ(nonPeriodicBSpline1.getFirstParameter(), param1);
EXPECT_DOUBLE_EQ(nonPeriodicBSpline1.getLastParameter(), param2);
}