// SPDX-License-Identifier: LGPL-2.1-or-later /*************************************************************************** * Copyright (c) 2023 David Carter * * * * This file is part of FreeCAD. * * * * FreeCAD 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 2.1 of the * * License, or (at your option) any later version. * * * * FreeCAD 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 FreeCAD. If not, see * * . * * * **************************************************************************/ #include #include #include #include #include #include #include #include #include #include #include #include #include #include #ifdef _MSC_VER # pragma warning(disable : 4834) #endif // clang-format off class TestMaterial : public ::testing::Test { protected: static void SetUpTestSuite() { if (App::Application::GetARGC() == 0) { tests::initApplication(); } } void SetUp() override { Base::Interpreter().runString("import Part"); _modelManager = &(Materials::ModelManager::getManager()); _materialManager = &(Materials::MaterialManager::getManager()); } // void TearDown() override {} Materials::ModelManager* _modelManager; Materials::MaterialManager* _materialManager; }; TEST_F(TestMaterial, TestInstallation) { ASSERT_NE(_modelManager, nullptr); // We should have loaded at least the system library auto libraries = _materialManager->getLibraries(); ASSERT_GT(libraries->size(), 0); // We should have at least one material auto materials = _materialManager->getLocalMaterials(); ASSERT_GT(materials->size(), 0); } TEST_F(TestMaterial, TestMaterialsWithModel) { ASSERT_NE(_materialManager, nullptr); auto materials = _materialManager->materialsWithModel( QStringLiteral("f6f9e48c-b116-4e82-ad7f-3659a9219c50")); // IsotropicLinearElastic EXPECT_GT(materials->size(), 0); auto materialsComplete = _materialManager->materialsWithModelComplete( QStringLiteral("f6f9e48c-b116-4e82-ad7f-3659a9219c50")); // IsotropicLinearElastic EXPECT_LE(materialsComplete->size(), materials->size()); auto materialsLinearElastic = _materialManager->materialsWithModel( QStringLiteral("7b561d1d-fb9b-44f6-9da9-56a4f74d7536")); // LinearElastic // All LinearElastic models should be in IsotropicLinearElastic since it is inherited EXPECT_LE(materialsLinearElastic->size(), materials->size()); for (auto &itp : *materialsLinearElastic) { auto mat = itp.first; EXPECT_NO_THROW(materials->at(mat)); } } TEST_F(TestMaterial, TestMaterialByPath) { ASSERT_NE(_materialManager, nullptr); auto steel = _materialManager->getMaterialByPath( QStringLiteral("Standard/Metal/Steel/CalculiX-Steel.FCMat"), QStringLiteral("System")); EXPECT_NE(&steel, nullptr); EXPECT_EQ(steel->getName(), QStringLiteral("CalculiX-Steel")); EXPECT_EQ(steel->getUUID(), QStringLiteral("92589471-a6cb-4bbc-b748-d425a17dea7d")); // The same but with a leading '/' auto steel2 = _materialManager->getMaterialByPath( QStringLiteral("/Standard/Metal/Steel/CalculiX-Steel.FCMat"), QStringLiteral("System")); EXPECT_NE(&steel2, nullptr); EXPECT_EQ(steel2->getName(), QStringLiteral("CalculiX-Steel")); EXPECT_EQ(steel2->getUUID(), QStringLiteral("92589471-a6cb-4bbc-b748-d425a17dea7d")); // Same with the library name as a prefix auto steel3 = _materialManager->getMaterialByPath( QStringLiteral("/System/Standard/Metal/Steel/CalculiX-Steel.FCMat"), QStringLiteral("System")); EXPECT_NE(&steel3, nullptr); EXPECT_EQ(steel3->getName(), QStringLiteral("CalculiX-Steel")); EXPECT_EQ(steel3->getUUID(), QStringLiteral("92589471-a6cb-4bbc-b748-d425a17dea7d")); } TEST_F(TestMaterial, TestAddPhysicalModel) { // Start with an empty material Materials::Material material; auto models = material.getPhysicalModels(); EXPECT_NE(&models, nullptr); EXPECT_EQ(models->size(), 0); // Add a model material.addPhysical(Materials::ModelUUIDs::ModelUUID_Electromagnetic_Default); models = material.getPhysicalModels(); EXPECT_EQ(models->size(), 1); // Add a second model material.addPhysical(Materials::ModelUUIDs::ModelUUID_Mechanical_LinearElastic); models = material.getPhysicalModels(); EXPECT_EQ(models->size(), 2); // Add an inherited model material.addPhysical(Materials::ModelUUIDs::ModelUUID_Mechanical_IsotropicLinearElastic); models = material.getPhysicalModels(); EXPECT_EQ(models->size(), 2); // Add a super model material.clearModels(); EXPECT_EQ(models->size(), 0); material.addPhysical(Materials::ModelUUIDs::ModelUUID_Mechanical_IsotropicLinearElastic); models = material.getPhysicalModels(); EXPECT_EQ(models->size(), 1); material.addPhysical(Materials::ModelUUIDs::ModelUUID_Mechanical_LinearElastic); models = material.getPhysicalModels(); EXPECT_EQ(models->size(), 1); // Remove the inherited model material.removePhysical(Materials::ModelUUIDs::ModelUUID_Mechanical_IsotropicLinearElastic); models = material.getPhysicalModels(); EXPECT_EQ(models->size(), 1); // Remove the super model material.removePhysical(Materials::ModelUUIDs::ModelUUID_Mechanical_LinearElastic); models = material.getPhysicalModels(); EXPECT_EQ(models->size(), 0); } TEST_F(TestMaterial, TestAddAppearanceModel) { // Start with an empty material Materials::Material material; auto models = material.getAppearanceModels(); EXPECT_NE(models, nullptr); EXPECT_EQ(models->size(), 0); // Add a model material.addAppearance(Materials::ModelUUIDs::ModelUUID_Rendering_Vector); models = material.getAppearanceModels(); EXPECT_EQ(models->size(), 1); // Add a second model material.addAppearance(Materials::ModelUUIDs::ModelUUID_Rendering_Advanced); models = material.getAppearanceModels(); EXPECT_EQ(models->size(), 2); // Add an inherited model material.addAppearance(Materials::ModelUUIDs::ModelUUID_Rendering_Basic); models = material.getAppearanceModels(); EXPECT_EQ(models->size(), 2); // Add a super model material.clearModels(); EXPECT_EQ(models->size(), 0); material.addAppearance(Materials::ModelUUIDs::ModelUUID_Rendering_Basic); models = material.getAppearanceModels(); EXPECT_EQ(models->size(), 1); material.addAppearance(Materials::ModelUUIDs::ModelUUID_Rendering_Advanced); models = material.getAppearanceModels(); EXPECT_EQ(models->size(), 1); // Remove the inherited model material.removeAppearance(Materials::ModelUUIDs::ModelUUID_Rendering_Basic); models = material.getAppearanceModels(); EXPECT_EQ(models->size(), 1); // Remove the super model material.removeAppearance(Materials::ModelUUIDs::ModelUUID_Rendering_Advanced); models = material.getAppearanceModels(); EXPECT_EQ(models->size(), 0); } QString parseQuantity(const std::string& value) { auto quantity = Base::Quantity::parse(value); quantity.setFormat(Materials::MaterialValue::getQuantityFormat()); return QString::fromStdString(quantity.getUserString()); } TEST_F(TestMaterial, TestCalculiXSteel) { ASSERT_NE(_materialManager, nullptr); auto steel = _materialManager->getMaterial(QStringLiteral("92589471-a6cb-4bbc-b748-d425a17dea7d")); EXPECT_EQ(steel->getName(), QStringLiteral("CalculiX-Steel")); EXPECT_EQ(steel->getUUID(), QStringLiteral("92589471-a6cb-4bbc-b748-d425a17dea7d")); EXPECT_TRUE(steel->hasPhysicalModel(Materials::ModelUUIDs::ModelUUID_Mechanical_Density)); // Density EXPECT_TRUE(steel->hasPhysicalModel(Materials::ModelUUIDs::ModelUUID_Mechanical_IsotropicLinearElastic)); // IsotropicLinearElastic EXPECT_TRUE(steel->hasPhysicalModel(Materials::ModelUUIDs::ModelUUID_Thermal_Default)); // Thermal EXPECT_FALSE(steel->hasPhysicalModel(Materials::ModelUUIDs::ModelUUID_Mechanical_LinearElastic)); // Legacy linear elastic - Not in the model EXPECT_TRUE(steel->hasAppearanceModel(Materials::ModelUUIDs::ModelUUID_Rendering_Basic)); // BasicRendering - inherited from Steel.FCMat EXPECT_TRUE(steel->isPhysicalModelComplete(Materials::ModelUUIDs::ModelUUID_Mechanical_Density)); // Density EXPECT_FALSE(steel->isPhysicalModelComplete(Materials::ModelUUIDs::ModelUUID_Mechanical_IsotropicLinearElastic)); // IsotropicLinearElastic - incomplete EXPECT_FALSE(steel->isPhysicalModelComplete(Materials::ModelUUIDs::ModelUUID_Thermal_Default)); // Thermal EXPECT_FALSE(steel->isPhysicalModelComplete(Materials::ModelUUIDs::ModelUUID_Mechanical_LinearElastic)); // Legacy linear elastic - Not in the model EXPECT_TRUE(steel->isAppearanceModelComplete(Materials::ModelUUIDs::ModelUUID_Rendering_Basic)); // BasicRendering - inherited from Steel.FCMat EXPECT_TRUE(steel->hasPhysicalProperty(QStringLiteral("Density"))); EXPECT_TRUE(steel->hasPhysicalProperty(QStringLiteral("BulkModulus"))); EXPECT_TRUE(steel->hasPhysicalProperty(QStringLiteral("PoissonRatio"))); EXPECT_TRUE(steel->hasPhysicalProperty(QStringLiteral("YoungsModulus"))); EXPECT_TRUE(steel->hasPhysicalProperty(QStringLiteral("ShearModulus"))); EXPECT_TRUE(steel->hasPhysicalProperty(QStringLiteral("SpecificHeat"))); EXPECT_TRUE(steel->hasPhysicalProperty(QStringLiteral("ThermalConductivity"))); EXPECT_TRUE(steel->hasPhysicalProperty(QStringLiteral("ThermalExpansionCoefficient"))); EXPECT_TRUE(steel->hasAppearanceProperty(QStringLiteral("AmbientColor"))); EXPECT_TRUE(steel->hasAppearanceProperty(QStringLiteral("DiffuseColor"))); EXPECT_TRUE(steel->hasAppearanceProperty(QStringLiteral("EmissiveColor"))); EXPECT_TRUE(steel->hasAppearanceProperty(QStringLiteral("Shininess"))); EXPECT_TRUE(steel->hasAppearanceProperty(QStringLiteral("SpecularColor"))); EXPECT_TRUE(steel->hasAppearanceProperty(QStringLiteral("Transparency"))); auto& properties = steel->getPhysicalProperties(); EXPECT_NO_THROW(properties.at(QStringLiteral("Density"))); EXPECT_NO_THROW(properties.at(QStringLiteral("BulkModulus"))); // This is different from the Python behaviour EXPECT_NO_THROW(properties.at(QStringLiteral("PoissonRatio"))); EXPECT_NO_THROW(properties.at(QStringLiteral("YoungsModulus"))); EXPECT_NO_THROW(properties.at(QStringLiteral("ShearModulus"))); EXPECT_NO_THROW(properties.at(QStringLiteral("SpecificHeat"))); EXPECT_NO_THROW(properties.at(QStringLiteral("ThermalConductivity"))); EXPECT_NO_THROW(properties.at(QStringLiteral("ThermalExpansionCoefficient"))); EXPECT_THROW(properties.at(QStringLiteral("AmbientColor")), std::out_of_range); EXPECT_THROW(properties.at(QStringLiteral("DiffuseColor")), std::out_of_range); EXPECT_THROW(properties.at(QStringLiteral("EmissiveColor")), std::out_of_range); EXPECT_THROW(properties.at(QStringLiteral("Shininess")), std::out_of_range); EXPECT_THROW(properties.at(QStringLiteral("SpecularColor")), std::out_of_range); EXPECT_THROW(properties.at(QStringLiteral("Transparency")), std::out_of_range); auto& properties1 = steel->getAppearanceProperties(); EXPECT_THROW(properties1.at(QStringLiteral("Density")), std::out_of_range); EXPECT_THROW(properties1.at(QStringLiteral("BulkModulus")), std::out_of_range); EXPECT_THROW(properties1.at(QStringLiteral("PoissonRatio")), std::out_of_range); EXPECT_THROW(properties1.at(QStringLiteral("YoungsModulus")), std::out_of_range); EXPECT_THROW(properties1.at(QStringLiteral("ShearModulus")), std::out_of_range); EXPECT_THROW(properties1.at(QStringLiteral("SpecificHeat")), std::out_of_range); EXPECT_THROW(properties1.at(QStringLiteral("ThermalConductivity")), std::out_of_range); EXPECT_THROW(properties1.at(QStringLiteral("ThermalExpansionCoefficient")), std::out_of_range); EXPECT_NO_THROW(properties1.at(QStringLiteral("AmbientColor"))); EXPECT_NO_THROW(properties1.at(QStringLiteral("DiffuseColor"))); EXPECT_NO_THROW(properties1.at(QStringLiteral("EmissiveColor"))); EXPECT_NO_THROW(properties1.at(QStringLiteral("Shininess"))); EXPECT_NO_THROW(properties1.at(QStringLiteral("SpecularColor"))); EXPECT_NO_THROW(properties1.at(QStringLiteral("Transparency"))); EXPECT_FALSE(properties[QStringLiteral("Density")]->isNull()); EXPECT_TRUE(properties[QStringLiteral("BulkModulus")]->isNull()); EXPECT_FALSE(properties[QStringLiteral("PoissonRatio")]->isNull()); EXPECT_FALSE(properties[QStringLiteral("YoungsModulus")]->isNull()); EXPECT_TRUE(properties[QStringLiteral("ShearModulus")]->isNull()); EXPECT_FALSE(properties[QStringLiteral("SpecificHeat")]->isNull()); EXPECT_FALSE(properties[QStringLiteral("ThermalConductivity")]->isNull()); EXPECT_FALSE(properties[QStringLiteral("ThermalExpansionCoefficient")]->isNull()); EXPECT_FALSE(properties1[QStringLiteral("AmbientColor")]->isNull()); EXPECT_FALSE(properties1[QStringLiteral("DiffuseColor")]->isNull()); EXPECT_FALSE(properties1[QStringLiteral("EmissiveColor")]->isNull()); EXPECT_FALSE(properties1[QStringLiteral("Shininess")]->isNull()); EXPECT_FALSE(properties1[QStringLiteral("SpecularColor")]->isNull()); EXPECT_FALSE(properties1[QStringLiteral("Transparency")]->isNull()); QLocale locale; EXPECT_EQ(properties[QStringLiteral("Density")]->getString(), parseQuantity("7900.00 kg/m^3")); EXPECT_EQ(properties[QStringLiteral("PoissonRatio")]->getString(), locale.toString(0.3)); EXPECT_EQ(properties[QStringLiteral("YoungsModulus")]->getString(), parseQuantity("210.00 GPa")); EXPECT_EQ(properties[QStringLiteral("SpecificHeat")]->getString(), parseQuantity("590.00 J/kg/K")); EXPECT_EQ(properties[QStringLiteral("ThermalConductivity")]->getString(), parseQuantity("43.00 W/m/K")); EXPECT_EQ(properties[QStringLiteral("ThermalExpansionCoefficient")]->getString(), parseQuantity("12.00 µm/m/K")); EXPECT_EQ(properties1[QStringLiteral("AmbientColor")]->getString(), QStringLiteral("(0.0020, 0.0020, 0.0020, 1.0)")); EXPECT_EQ(properties1[QStringLiteral("DiffuseColor")]->getString(), QStringLiteral("(0.0000, 0.0000, 0.0000, 1.0)")); EXPECT_EQ(properties1[QStringLiteral("EmissiveColor")]->getString(), QStringLiteral("(0.0000, 0.0000, 0.0000, 1.0)")); EXPECT_EQ(properties1[QStringLiteral("Shininess")]->getString(), locale.toString(0.06)); EXPECT_EQ(properties1[QStringLiteral("SpecularColor")]->getString(), QStringLiteral("(0.9800, 0.9800, 0.9800, 1.0)")); EXPECT_EQ(properties1[QStringLiteral("Transparency")]->getString(), QStringLiteral("0")); EXPECT_TRUE(properties[QStringLiteral("BulkModulus")]->getString().isEmpty()); EXPECT_TRUE(properties[QStringLiteral("ShearModulus")]->getString().isEmpty()); // These are the preferred method of access // EXPECT_DOUBLE_EQ(steel->getPhysicalQuantity(QStringLiteral("Density")).getValue(), 7.9e-06); EXPECT_NEAR(steel->getPhysicalValue(QStringLiteral("PoissonRatio")).toDouble(), 0.3, 1e-6); EXPECT_DOUBLE_EQ(steel->getPhysicalQuantity(QStringLiteral("YoungsModulus")).getValue(), 210000000.0); EXPECT_DOUBLE_EQ(steel->getPhysicalQuantity(QStringLiteral("SpecificHeat")).getValue(), 590000000.0); EXPECT_DOUBLE_EQ(steel->getPhysicalQuantity(QStringLiteral("ThermalConductivity")).getValue(), 43000.0); EXPECT_DOUBLE_EQ(steel->getPhysicalQuantity(QStringLiteral("ThermalExpansionCoefficient")).getValue(), 1.2e-05); EXPECT_EQ(steel->getAppearanceValue(QStringLiteral("AmbientColor")), QStringLiteral("(0.0020, 0.0020, 0.0020, 1.0)")); EXPECT_EQ(steel->getAppearanceValue(QStringLiteral("DiffuseColor")), QStringLiteral("(0.0000, 0.0000, 0.0000, 1.0)")); EXPECT_EQ(steel->getAppearanceValue(QStringLiteral("EmissiveColor")), QStringLiteral("(0.0000, 0.0000, 0.0000, 1.0)")); EXPECT_NEAR(steel->getAppearanceValue(QStringLiteral("Shininess")).toDouble(), 0.06, 1e-6); EXPECT_EQ(steel->getAppearanceValue(QStringLiteral("SpecularColor")), QStringLiteral("(0.9800, 0.9800, 0.9800, 1.0)")); EXPECT_DOUBLE_EQ(steel->getAppearanceValue(QStringLiteral("Transparency")).toDouble(), 0.0); EXPECT_EQ(steel->getPhysicalQuantity(QStringLiteral("Density")).getUserString(), parseQuantity("7900.00 kg/m^3").toStdString()); EXPECT_EQ(steel->getPhysicalQuantity(QStringLiteral("YoungsModulus")).getUserString(), parseQuantity("210.00 GPa").toStdString()); EXPECT_EQ(steel->getPhysicalQuantity(QStringLiteral("SpecificHeat")).getUserString(), parseQuantity("590.00 J/kg/K").toStdString()); EXPECT_EQ(steel->getPhysicalQuantity(QStringLiteral("ThermalConductivity")).getUserString(), parseQuantity("43.00 W/m/K").toStdString()); EXPECT_EQ(steel->getPhysicalQuantity(QStringLiteral("ThermalExpansionCoefficient")).getUserString(), parseQuantity("12.00 µm/m/K").toStdString()); } TEST_F(TestMaterial, TestColumns) { // Start with an empty material Materials::Material testMaterial; auto models = testMaterial.getPhysicalModels(); EXPECT_NE(&models, nullptr); EXPECT_EQ(models->size(), 0); // Add a model testMaterial.addPhysical(Materials::ModelUUIDs::ModelUUID_Test_Model); models = testMaterial.getPhysicalModels(); EXPECT_EQ(models->size(), 1); EXPECT_TRUE(testMaterial.hasPhysicalProperty(QStringLiteral("TestArray2D"))); auto array2d = testMaterial.getPhysicalProperty(QStringLiteral("TestArray2D"))->getMaterialValue(); EXPECT_TRUE(array2d); EXPECT_EQ(dynamic_cast(*array2d).columns(), 2); EXPECT_TRUE(testMaterial.hasPhysicalProperty(QStringLiteral("TestArray2D3Column"))); auto array2d3Column = testMaterial.getPhysicalProperty(QStringLiteral("TestArray2D3Column"))->getMaterialValue(); EXPECT_TRUE(array2d3Column); EXPECT_EQ(dynamic_cast(*array2d3Column).columns(), 3); EXPECT_TRUE(testMaterial.hasPhysicalProperty(QStringLiteral("TestArray3D"))); auto array3d = testMaterial.getPhysicalProperty(QStringLiteral("TestArray3D"))->getMaterialValue(); EXPECT_TRUE(array3d); EXPECT_EQ(dynamic_cast(*array3d).columns(), 2); } // clang-format on