FreeCAD / tests /src /Mod /Part /App /TopoShapeMakeShape.cpp

Upload folder using huggingface_hub

985c397 verified 9 months ago

5.45 kB

// SPDX-License-Identifier: LGPL-2.1-or-later

// Tests for the makeShape methods, extracted from the main set of tests for TopoShape
// due to length and complexity.

#include <gtest/gtest.h>
#include "src/App/InitApplication.h"
#include "PartTestHelpers.h"
#include <Mod/Part/App/TopoShape.h>
#include <Mod/Part/App/TopoShapeOpCode.h>

using namespace Data;
using namespace Part;
using namespace PartTestHelpers;

class TopoShapeMakeShapeTests: public ::testing::Test
{
protected:
    static void SetUpTestSuite()
    {
        tests::initApplication();
    }

    void SetUp() override
    {
        _docName = App::GetApplication().getUniqueDocumentName("test");
        App::GetApplication().newDocument(_docName.c_str(), "testUser");
        _sids = &_sid;
    }

    void TearDown() override
    {
        App::GetApplication().closeDocument(_docName.c_str());
    }

    Part::TopoShape* Shape()
    {
        return &_shape;
    }

    Part::TopoShape::Mapper* Mapper()
    {
        return &_mapper;
    }

private:
    std::string _docName;
    Data::ElementIDRefs _sid;
    QVector<App::StringIDRef>* _sids = nullptr;
    Part::TopoShape _shape;
    Part::TopoShape::Mapper _mapper;
};

TEST_F(TopoShapeMakeShapeTests, nullShapeThrows)
{
    // Arrange
    auto [cube1, cube2] = CreateTwoCubes();
    std::vector<Part::TopoShape> sources {cube1, cube2};
    TopoDS_Vertex nullShape;

    // Act and assert
    EXPECT_THROW(
        Shape()->makeShapeWithElementMap(nullShape, *Mapper(), sources),
        Part::NullShapeException
    );
}

TEST_F(TopoShapeMakeShapeTests, shapeVertex)
{
    // Arrange
    BRepBuilderAPI_MakeVertex vertexMaker = BRepBuilderAPI_MakeVertex(gp_Pnt(10, 10, 10));
    TopoShape topoShape(vertexMaker.Vertex(), 1L);
    // Act
    TopoShape& result = topoShape.makeElementShape(vertexMaker, topoShape);
    // Assert
    EXPECT_EQ(result.getElementMap().size(), 1);  // Changed with PR#12471. Probably will change
                                                  // again after importing other TopoNaming logics
    EXPECT_EQ(result.countSubElements("Vertex"), 1);
    EXPECT_EQ(result.countSubShapes("Vertex"), 1);
}

TEST_F(TopoShapeMakeShapeTests, thruSections)
{
    // Arrange
    auto [face1, wire1, edge1, edge2, edge3, edge4] = CreateRectFace();
    TopoDS_Wire wire2 = wire1;
    auto transform {gp_Trsf()};
    transform.SetTranslation(gp_Pnt(0.0, 0.0, 0.0), gp_Pnt(0.0, 0.5, 1.0));
    wire2.Move(TopLoc_Location(transform));
    TopoShape wire1ts {wire1, 1L};
    TopoShape wire2ts {wire2, 2L};
    BRepOffsetAPI_ThruSections thruMaker;
    thruMaker.AddWire(wire1);
    thruMaker.AddWire(wire2);
    TopoShape topoShape {};
    // Act
    TopoShape& result
        = topoShape.makeElementShape(thruMaker, {wire1ts, wire2ts}, OpCodes::ThruSections);
    auto elements = elementMap(result);
    // Assert
    EXPECT_EQ(elements.size(), 24);
    EXPECT_EQ(elements.count(IndexedName("Vertex", 1)), 1);
    EXPECT_EQ(getVolume(result.getShape()), 4);
    EXPECT_TRUE(allElementsMatch(
        topoShape,
        {
            "Edge1;:G(Edge1;K-1;:H2:4,E);TRU;:H1:1a,F",     "Edge1;:H1,E",   "Edge1;:H2,E",
            "Edge2;:G(Edge2;K-1;:H2:4,E);TRU;:H1:1a,F",     "Edge2;:H1,E",   "Edge2;:H2,E",
            "Edge3;:G(Edge3;K-1;:H2:4,E);TRU;:H1:1a,F",     "Edge3;:H1,E",   "Edge3;:H2,E",
            "Edge4;:G(Edge4;K-1;:H2:4,E);TRU;:H1:1a,F",     "Edge4;:H1,E",   "Edge4;:H2,E",
            "Vertex1;:G(Vertex1;K-1;:H2:4,V);TRU;:H1:1c,E", "Vertex1;:H1,V", "Vertex1;:H2,V",
            "Vertex2;:G(Vertex2;K-1;:H2:4,V);TRU;:H1:1c,E", "Vertex2;:H1,V", "Vertex2;:H2,V",
            "Vertex3;:G(Vertex3;K-1;:H2:4,V);TRU;:H1:1c,E", "Vertex3;:H1,V", "Vertex3;:H2,V",
            "Vertex4;:G(Vertex4;K-1;:H2:4,V);TRU;:H1:1c,E", "Vertex4;:H1,V", "Vertex4;:H2,V",
        }
    ));
}

TEST_F(TopoShapeMakeShapeTests, sewing)
{
    // Arrange
    auto [face1, wire1, edge1, edge2, edge3, edge4] = CreateRectFace();
    auto face2 = face1;
    auto transform {gp_Trsf()};
    transform.SetTranslation(gp_Pnt(0.0, 0.0, 0.0), gp_Pnt(0.5, 0.5, 0.0));
    face2.Move(TopLoc_Location(transform));
    BRepBuilderAPI_Sewing sewer;
    sewer.Add(face1);
    sewer.Add(face2);
    sewer.Perform();
    std::vector<TopoShape> sources {{face1, 1L}, {face2, 2L}};
    TopoShape topoShape {};
    // Act
    TopoShape& result = topoShape.makeShapeWithElementMap(
        sewer.SewedShape(),
        MapperSewing(sewer),
        sources,
        OpCodes::Sewing
    );

    auto elements = elementMap(result);
    // Assert
    EXPECT_EQ(&result, &topoShape);
    EXPECT_EQ(elements.size(), 18);  // Now a single cube
    EXPECT_EQ(elements.count(IndexedName("Vertex", 1)), 1);
    EXPECT_EQ(getArea(result.getShape()), 12);
    // TODO:  This element map is suspiciously devoid of anything OpCodes::Sewing (SEW).  Is that
    // right?
    EXPECT_TRUE(allElementsMatch(
        topoShape,
        {
            "Face1;:H1,F",
            "Face1;:H2,F",
            "Edge1;:H2,E",
            "Edge2;:H2,E",
            "Edge3;:H2,E",
            "Edge4;:H2,E",
            "Edge1;:H1,E",
            "Edge2;:H1,E",
            "Edge3;:H1,E",
            "Edge4;:H1,E",
            "Vertex1;:H2,V",
            "Vertex2;:H2,V",
            "Vertex3;:H2,V",
            "Vertex4;:H2,V",
            "Vertex1;:H1,V",
            "Vertex2;:H1,V",
            "Vertex3;:H1,V",
            "Vertex4;:H1,V",
        }
    ));
}

	// SPDX-License-Identifier: LGPL-2.1-or-later

	// Tests for the makeShape methods, extracted from the main set of tests for TopoShape
	// due to length and complexity.

	#include <gtest/gtest.h>
	#include "src/App/InitApplication.h"
	#include "PartTestHelpers.h"
	#include <Mod/Part/App/TopoShape.h>
	#include <Mod/Part/App/TopoShapeOpCode.h>

	using namespace Data;
	using namespace Part;
	using namespace PartTestHelpers;

	class TopoShapeMakeShapeTests: public ::testing::Test
	{
	protected:
	static void SetUpTestSuite()
	{
	tests::initApplication();
	}

	void SetUp() override
	{
	_docName = App::GetApplication().getUniqueDocumentName("test");
	App::GetApplication().newDocument(_docName.c_str(), "testUser");
	_sids = &_sid;
	}

	void TearDown() override
	{
	App::GetApplication().closeDocument(_docName.c_str());
	}

	Part::TopoShape* Shape()
	{
	return &_shape;
	}

	Part::TopoShape::Mapper* Mapper()
	{
	return &_mapper;
	}

	private:
	std::string _docName;
	Data::ElementIDRefs _sid;
	QVector<App::StringIDRef>* _sids = nullptr;
	Part::TopoShape _shape;
	Part::TopoShape::Mapper _mapper;
	};

	TEST_F(TopoShapeMakeShapeTests, nullShapeThrows)
	{
	// Arrange
	auto [cube1, cube2] = CreateTwoCubes();
	std::vector<Part::TopoShape> sources {cube1, cube2};
	TopoDS_Vertex nullShape;

	// Act and assert
	EXPECT_THROW(
	Shape()->makeShapeWithElementMap(nullShape, *Mapper(), sources),
	Part::NullShapeException
	);
	}

	TEST_F(TopoShapeMakeShapeTests, shapeVertex)
	{
	// Arrange
	BRepBuilderAPI_MakeVertex vertexMaker = BRepBuilderAPI_MakeVertex(gp_Pnt(10, 10, 10));
	TopoShape topoShape(vertexMaker.Vertex(), 1L);
	// Act
	TopoShape& result = topoShape.makeElementShape(vertexMaker, topoShape);
	// Assert
	EXPECT_EQ(result.getElementMap().size(), 1); // Changed with PR#12471. Probably will change
	// again after importing other TopoNaming logics
	EXPECT_EQ(result.countSubElements("Vertex"), 1);
	EXPECT_EQ(result.countSubShapes("Vertex"), 1);
	}

	TEST_F(TopoShapeMakeShapeTests, thruSections)
	{
	// Arrange
	auto [face1, wire1, edge1, edge2, edge3, edge4] = CreateRectFace();
	TopoDS_Wire wire2 = wire1;
	auto transform {gp_Trsf()};
	transform.SetTranslation(gp_Pnt(0.0, 0.0, 0.0), gp_Pnt(0.0, 0.5, 1.0));
	wire2.Move(TopLoc_Location(transform));
	TopoShape wire1ts {wire1, 1L};
	TopoShape wire2ts {wire2, 2L};
	BRepOffsetAPI_ThruSections thruMaker;
	thruMaker.AddWire(wire1);
	thruMaker.AddWire(wire2);
	TopoShape topoShape {};
	// Act
	TopoShape& result
	= topoShape.makeElementShape(thruMaker, {wire1ts, wire2ts}, OpCodes::ThruSections);
	auto elements = elementMap(result);
	// Assert
	EXPECT_EQ(elements.size(), 24);
	EXPECT_EQ(elements.count(IndexedName("Vertex", 1)), 1);
	EXPECT_EQ(getVolume(result.getShape()), 4);
	EXPECT_TRUE(allElementsMatch(
	topoShape,
	{
	"Edge1;:G(Edge1;K-1;:H2:4,E);TRU;:H1:1a,F", "Edge1;:H1,E", "Edge1;:H2,E",
	"Edge2;:G(Edge2;K-1;:H2:4,E);TRU;:H1:1a,F", "Edge2;:H1,E", "Edge2;:H2,E",
	"Edge3;:G(Edge3;K-1;:H2:4,E);TRU;:H1:1a,F", "Edge3;:H1,E", "Edge3;:H2,E",
	"Edge4;:G(Edge4;K-1;:H2:4,E);TRU;:H1:1a,F", "Edge4;:H1,E", "Edge4;:H2,E",
	"Vertex1;:G(Vertex1;K-1;:H2:4,V);TRU;:H1:1c,E", "Vertex1;:H1,V", "Vertex1;:H2,V",
	"Vertex2;:G(Vertex2;K-1;:H2:4,V);TRU;:H1:1c,E", "Vertex2;:H1,V", "Vertex2;:H2,V",
	"Vertex3;:G(Vertex3;K-1;:H2:4,V);TRU;:H1:1c,E", "Vertex3;:H1,V", "Vertex3;:H2,V",
	"Vertex4;:G(Vertex4;K-1;:H2:4,V);TRU;:H1:1c,E", "Vertex4;:H1,V", "Vertex4;:H2,V",
	}
	));
	}

	TEST_F(TopoShapeMakeShapeTests, sewing)
	{
	// Arrange
	auto [face1, wire1, edge1, edge2, edge3, edge4] = CreateRectFace();
	auto face2 = face1;
	auto transform {gp_Trsf()};
	transform.SetTranslation(gp_Pnt(0.0, 0.0, 0.0), gp_Pnt(0.5, 0.5, 0.0));
	face2.Move(TopLoc_Location(transform));
	BRepBuilderAPI_Sewing sewer;
	sewer.Add(face1);
	sewer.Add(face2);
	sewer.Perform();
	std::vector<TopoShape> sources {{face1, 1L}, {face2, 2L}};
	TopoShape topoShape {};
	// Act
	TopoShape& result = topoShape.makeShapeWithElementMap(
	sewer.SewedShape(),
	MapperSewing(sewer),
	sources,
	OpCodes::Sewing
	);

	auto elements = elementMap(result);
	// Assert
	EXPECT_EQ(&result, &topoShape);
	EXPECT_EQ(elements.size(), 18); // Now a single cube
	EXPECT_EQ(elements.count(IndexedName("Vertex", 1)), 1);
	EXPECT_EQ(getArea(result.getShape()), 12);
	// TODO: This element map is suspiciously devoid of anything OpCodes::Sewing (SEW). Is that
	// right?
	EXPECT_TRUE(allElementsMatch(
	topoShape,
	{
	"Face1;:H1,F",
	"Face1;:H2,F",
	"Edge1;:H2,E",
	"Edge2;:H2,E",
	"Edge3;:H2,E",
	"Edge4;:H2,E",
	"Edge1;:H1,E",
	"Edge2;:H1,E",
	"Edge3;:H1,E",
	"Edge4;:H1,E",
	"Vertex1;:H2,V",
	"Vertex2;:H2,V",
	"Vertex3;:H2,V",
	"Vertex4;:H2,V",
	"Vertex1;:H1,V",
	"Vertex2;:H1,V",
	"Vertex3;:H1,V",
	"Vertex4;:H1,V",
	}
	));
	}