Vladimír Vondruš
2 years ago
19 changed files with 423 additions and 14 deletions
@ -0,0 +1,42 @@ |
|||||||
|
.. |
||||||
|
This file is part of Magnum. |
||||||
|
|
||||||
|
Copyright © 2010, 2011, 2012, 2013, 2014, 2015, 2016, 2017, 2018, 2019, |
||||||
|
2020, 2021, 2022, 2023 Vladimír Vondruš <mosra@centrum.cz> |
||||||
|
|
||||||
|
Permission is hereby granted, free of charge, to any person obtaining a |
||||||
|
copy of this software and associated documentation files (the "Software"), |
||||||
|
to deal in the Software without restriction, including without limitation |
||||||
|
the rights to use, copy, modify, merge, publish, distribute, sublicense, |
||||||
|
and/or sell copies of the Software, and to permit persons to whom the |
||||||
|
Software is furnished to do so, subject to the following conditions: |
||||||
|
|
||||||
|
The above copyright notice and this permission notice shall be included |
||||||
|
in all copies or substantial portions of the Software. |
||||||
|
|
||||||
|
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR |
||||||
|
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, |
||||||
|
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL |
||||||
|
THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER |
||||||
|
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING |
||||||
|
FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER |
||||||
|
DEALINGS IN THE SOFTWARE. |
||||||
|
.. |
||||||
|
|
||||||
|
.. py:function:: magnum.materialtools.filter_attributes |
||||||
|
:raise AssertionError: If size of :p:`attributes_to_keep` is different than |
||||||
|
:ref:`trade.MaterialData.attribute_data_offset()` for |
||||||
|
:ref:`trade.MaterialData.layer_count` |
||||||
|
.. py:function:: magnum.materialtools.filter_layers |
||||||
|
:raise AssertionError: If size of :p:`layers_to_keep` is different than |
||||||
|
:ref:`trade.MaterialData.layer_count` |
||||||
|
.. py:function:: magnum.materialtools.filter_attributes_layers |
||||||
|
:raise AssertionError: If size of :p:`attributes_to_keep` is different than |
||||||
|
:ref:`trade.MaterialData.attribute_data_offset()` for |
||||||
|
:ref:`trade.MaterialData.layer_count` |
||||||
|
:raise AssertionError: If size of :p:`layers_to_keep` is different than |
||||||
|
:ref:`trade.MaterialData.layer_count` |
||||||
|
.. py:function:: magnum.materialtools.merge |
||||||
|
:raise RuntimeError: If merge failed due to a conflict |
||||||
|
.. py:function:: magnum.materialtools.phong_to_pbr_metallic_roughness |
||||||
|
:raise RuntimeError: If conversion failed |
||||||
@ -0,0 +1,135 @@ |
|||||||
|
/*
|
||||||
|
This file is part of Magnum. |
||||||
|
|
||||||
|
Copyright © 2010, 2011, 2012, 2013, 2014, 2015, 2016, 2017, 2018, 2019, |
||||||
|
2020, 2021, 2022, 2023 Vladimír Vondruš <mosra@centrum.cz> |
||||||
|
|
||||||
|
Permission is hereby granted, free of charge, to any person obtaining a |
||||||
|
copy of this software and associated documentation files (the "Software"), |
||||||
|
to deal in the Software without restriction, including without limitation |
||||||
|
the rights to use, copy, modify, merge, publish, distribute, sublicense, |
||||||
|
and/or sell copies of the Software, and to permit persons to whom the |
||||||
|
Software is furnished to do so, subject to the following conditions: |
||||||
|
|
||||||
|
The above copyright notice and this permission notice shall be included |
||||||
|
in all copies or substantial portions of the Software. |
||||||
|
|
||||||
|
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR |
||||||
|
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, |
||||||
|
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL |
||||||
|
THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER |
||||||
|
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING |
||||||
|
FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER |
||||||
|
DEALINGS IN THE SOFTWARE. |
||||||
|
*/ |
||||||
|
|
||||||
|
#include <pybind11/pybind11.h> |
||||||
|
#include <pybind11/stl.h> /* for std::vector */ |
||||||
|
#include <Corrade/Containers/BitArrayView.h> |
||||||
|
#include <Corrade/Containers/Iterable.h> |
||||||
|
#include <Corrade/Containers/ArrayViewStl.h> |
||||||
|
#include <Magnum/MaterialTools/Copy.h> |
||||||
|
#include <Magnum/MaterialTools/Filter.h> |
||||||
|
#include <Magnum/MaterialTools/Merge.h> |
||||||
|
#include <Magnum/MaterialTools/PhongToPbrMetallicRoughness.h> |
||||||
|
#include <Magnum/MaterialTools/RemoveDuplicates.h> |
||||||
|
#include <Magnum/Trade/MaterialData.h> |
||||||
|
|
||||||
|
#include "Magnum/Trade/PythonBindings.h" |
||||||
|
#include "corrade/EnumOperators.h" |
||||||
|
#include "magnum/bootstrap.h" |
||||||
|
|
||||||
|
namespace magnum { |
||||||
|
|
||||||
|
void materialtools(py::module_& m) { |
||||||
|
m.doc() = "Material tools"; |
||||||
|
|
||||||
|
#ifndef MAGNUM_BUILD_STATIC |
||||||
|
/* These are a part of the same module in the static build, no need to
|
||||||
|
import (also can't import because there it's _magnum.*) */ |
||||||
|
py::module_::import("magnum.trade"); |
||||||
|
#endif |
||||||
|
|
||||||
|
py::enum_<MaterialTools::MergeConflicts>{m, "MergeConflicts", "Material merge conflict resolution"} |
||||||
|
.value("FAIL", MaterialTools::MergeConflicts::Fail) |
||||||
|
.value("KEEP_FIRST_IF_SAME_TYPE", MaterialTools::MergeConflicts::KeepFirstIfSameType) |
||||||
|
.value("KEEP_FIRST_IGNORE_TYPE", MaterialTools::MergeConflicts::KeepFirstIgnoreType); |
||||||
|
|
||||||
|
py::enum_<MaterialTools::PhongToPbrMetallicRoughnessFlag> phongToPbrMetallicRoughnessFlag{m, "PhongToPbrMetallicRoughnessFlags", "Phong to PBR conversion flags"}; |
||||||
|
phongToPbrMetallicRoughnessFlag |
||||||
|
.value("KEEP_ORIGINAL_ATTRIBUTES", MaterialTools::PhongToPbrMetallicRoughnessFlag::KeepOriginalAttributes) |
||||||
|
.value("DROP_UNCOVERTIBLE_ATTRIBUTES", MaterialTools::PhongToPbrMetallicRoughnessFlag::DropUnconvertibleAttributes) |
||||||
|
.value("FAIL_ON_UNCONVERTIBLE_ATTRIBUTES", MaterialTools::PhongToPbrMetallicRoughnessFlag::FailOnUnconvertibleAttributes) |
||||||
|
.value("NONE", MaterialTools::PhongToPbrMetallicRoughnessFlag{}) |
||||||
|
.value("ALL", MaterialTools::PhongToPbrMetallicRoughnessFlag(Containers::enumCastUnderlyingType(~MaterialTools::PhongToPbrMetallicRoughnessFlags{}))); |
||||||
|
corrade::enumOperators(phongToPbrMetallicRoughnessFlag); |
||||||
|
|
||||||
|
m |
||||||
|
.def("copy", static_cast<Trade::MaterialData(*)(const Trade::MaterialData&)>(MaterialTools::copy), "Make an owned copy of the material", py::arg("material")) |
||||||
|
.def("filter_attributes", [](const Trade::MaterialData& material, const Containers::BitArrayView attributesToKeep, Trade::MaterialType typesToKeep) { |
||||||
|
if(attributesToKeep.size() != material.attributeData().size()) { |
||||||
|
PyErr_Format(PyExc_AssertionError, "expected %u bits but got %zu", material.attributeData().size(), attributesToKeep.size()); |
||||||
|
throw py::error_already_set{}; |
||||||
|
} |
||||||
|
|
||||||
|
return MaterialTools::filterAttributes(material, attributesToKeep, typesToKeep); |
||||||
|
}, "Filter material attributes", py::arg("material"), py::arg("attributes_to_keep"), py::arg("types_to_keep") = Trade::MaterialType(Containers::enumCastUnderlyingType(~Trade::MaterialTypes{}))) |
||||||
|
.def("filter_layers", [](const Trade::MaterialData& material, const Containers::BitArrayView layersToKeep, Trade::MaterialType typesToKeep) { |
||||||
|
if(layersToKeep.size() != material.layerCount()) { |
||||||
|
PyErr_Format(PyExc_AssertionError, "expected %u bits but got %zu", material.layerCount(), layersToKeep.size()); |
||||||
|
throw py::error_already_set{}; |
||||||
|
} |
||||||
|
|
||||||
|
return MaterialTools::filterLayers(material, layersToKeep, typesToKeep); |
||||||
|
}, "Filter material layers", py::arg("material"), py::arg("layers_to_keep"), py::arg("types_to_keep") = Trade::MaterialType(Containers::enumCastUnderlyingType(~Trade::MaterialTypes{}))) |
||||||
|
.def("filter_attributes_layers", [](const Trade::MaterialData& material, const Containers::BitArrayView attributesToKeep, const Containers::BitArrayView layersToKeep, Trade::MaterialType typesToKeep) { |
||||||
|
if(attributesToKeep.size() != material.attributeData().size()) { |
||||||
|
PyErr_Format(PyExc_AssertionError, "expected %u attribute bits but got %zu", material.attributeData().size(), attributesToKeep.size()); |
||||||
|
throw py::error_already_set{}; |
||||||
|
} |
||||||
|
if(layersToKeep.size() != material.layerCount()) { |
||||||
|
PyErr_Format(PyExc_AssertionError, "expected %u layer bits but got %zu", material.layerCount(), layersToKeep.size()); |
||||||
|
throw py::error_already_set{}; |
||||||
|
} |
||||||
|
|
||||||
|
return MaterialTools::filterAttributesLayers(material, attributesToKeep, layersToKeep, typesToKeep); |
||||||
|
}, "Filter material attributes and layers", py::arg("material"), py::arg("attributes_to_keep"), py::arg("layers_to_keep"), py::arg("types_to_keep") = Trade::MaterialType(Containers::enumCastUnderlyingType(~Trade::MaterialTypes{}))) |
||||||
|
.def("merge", [](const Trade::MaterialData& first, const Trade::MaterialData& second, MaterialTools::MergeConflicts conflicts) { |
||||||
|
Containers::Optional<Trade::MaterialData> out = MaterialTools::merge(first, second, conflicts); |
||||||
|
if(!out) { |
||||||
|
PyErr_SetString(PyExc_RuntimeError, "material merge failed"); |
||||||
|
throw py::error_already_set{}; |
||||||
|
} |
||||||
|
|
||||||
|
return *Utility::move(out); |
||||||
|
}, "Merge two materials", py::arg("first"), py::arg("second"), py::arg("conflicts") = MaterialTools::MergeConflicts::Fail) |
||||||
|
.def("phong_to_pbr_metallic_roughness", [](const Trade::MaterialData& material, MaterialTools::PhongToPbrMetallicRoughnessFlag flags) { |
||||||
|
Containers::Optional<Trade::MaterialData> out = MaterialTools::phongToPbrMetallicRoughness(material, flags); |
||||||
|
if(!out) { |
||||||
|
PyErr_SetString(PyExc_RuntimeError, "material conversion failed"); |
||||||
|
throw py::error_already_set{}; |
||||||
|
} |
||||||
|
|
||||||
|
return *Utility::move(out); |
||||||
|
}, "Convert a Phong material to PBR metallic/roughness", py::arg("material"), py::arg("flags") = MaterialTools::PhongToPbrMetallicRoughnessFlag{}) |
||||||
|
.def("remove_duplicates", [](std::vector<std::reference_wrapper<const Trade::MaterialData>> materials) { |
||||||
|
std::vector<UnsignedInt> indices; |
||||||
|
indices.resize(materials.size()); |
||||||
|
const std::size_t count = MaterialTools::removeDuplicatesInto(Containers::Iterable<const Trade::MaterialData>{materials.data(), materials.size(), sizeof(std::reference_wrapper<Trade::MaterialData>), [](const void* data) -> const Trade::MaterialData& { |
||||||
|
return static_cast<const std::reference_wrapper<const Trade::MaterialData>*>(data)->get(); |
||||||
|
}}, Containers::arrayView(indices)); |
||||||
|
|
||||||
|
return std::make_pair(std::move(indices), count); |
||||||
|
}, "Remove duplicate materials from a list", py::arg("materials")); |
||||||
|
} |
||||||
|
|
||||||
|
} |
||||||
|
|
||||||
|
#ifndef MAGNUM_BUILD_STATIC |
||||||
|
/* TODO: remove declaration when https://github.com/pybind/pybind11/pull/1863
|
||||||
|
is released */ |
||||||
|
extern "C" PYBIND11_EXPORT PyObject* PyInit_materialtools(); |
||||||
|
PYBIND11_MODULE(materialtools, m) { |
||||||
|
magnum::materialtools(m); |
||||||
|
} |
||||||
|
#endif |
||||||
@ -0,0 +1,188 @@ |
|||||||
|
# |
||||||
|
# This file is part of Magnum. |
||||||
|
# |
||||||
|
# Copyright © 2010, 2011, 2012, 2013, 2014, 2015, 2016, 2017, 2018, 2019, |
||||||
|
# 2020, 2021, 2022, 2023 Vladimír Vondruš <mosra@centrum.cz> |
||||||
|
# |
||||||
|
# Permission is hereby granted, free of charge, to any person obtaining a |
||||||
|
# copy of this software and associated documentation files (the "Software"), |
||||||
|
# to deal in the Software without restriction, including without limitation |
||||||
|
# the rights to use, copy, modify, merge, publish, distribute, sublicense, |
||||||
|
# and/or sell copies of the Software, and to permit persons to whom the |
||||||
|
# Software is furnished to do so, subject to the following conditions: |
||||||
|
# |
||||||
|
# The above copyright notice and this permission notice shall be included |
||||||
|
# in all copies or substantial portions of the Software. |
||||||
|
# |
||||||
|
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR |
||||||
|
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, |
||||||
|
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL |
||||||
|
# THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER |
||||||
|
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING |
||||||
|
# FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER |
||||||
|
# DEALINGS IN THE SOFTWARE. |
||||||
|
# |
||||||
|
|
||||||
|
import os |
||||||
|
import unittest |
||||||
|
|
||||||
|
from corrade import containers |
||||||
|
from magnum import * |
||||||
|
from magnum import materialtools, trade |
||||||
|
import magnum |
||||||
|
|
||||||
|
class Copy(unittest.TestCase): |
||||||
|
def test(self): |
||||||
|
importer = trade.ImporterManager().load_and_instantiate('GltfImporter') |
||||||
|
importer.open_file(os.path.join(os.path.dirname(__file__), "material.gltf")) |
||||||
|
|
||||||
|
material = importer.material(0) |
||||||
|
self.assertEqual(material.attribute_data_flags, trade.DataFlags.OWNED|trade.DataFlags.MUTABLE) |
||||||
|
|
||||||
|
copy = materialtools.copy(material) |
||||||
|
self.assertEqual(copy.attribute_data_flags, trade.DataFlags.OWNED|trade.DataFlags.MUTABLE) |
||||||
|
|
||||||
|
class Filter(unittest.TestCase): |
||||||
|
def test(self): |
||||||
|
importer = trade.ImporterManager().load_and_instantiate('GltfImporter') |
||||||
|
# This adds extra Diffuse attributes for BaseColor and a Phong type, |
||||||
|
# don't want |
||||||
|
importer.configuration['phongMaterialFallback'] = False |
||||||
|
importer.open_file(os.path.join(os.path.dirname(__file__), "material.gltf")) |
||||||
|
|
||||||
|
material = importer.material(0) |
||||||
|
self.assertEqual(material.types, trade.MaterialTypes.PBR_METALLIC_ROUGHNESS|trade.MaterialTypes.PBR_CLEAR_COAT) |
||||||
|
self.assertEqual(material.layer_count, 2) |
||||||
|
self.assertEqual(material.attribute_count(0), 3) |
||||||
|
self.assertEqual(material.attribute_count(1), 8) |
||||||
|
|
||||||
|
# Unlike MeshData or SceneData the tools always produce a new copy of |
||||||
|
# the attribute list, so we don't need to worry about data ownership |
||||||
|
# and such |
||||||
|
|
||||||
|
attributes_to_keep = containers.BitArray.direct_init(material.attribute_data_offset(material.layer_count), True) |
||||||
|
attributes_to_keep[material.attribute_id(trade.MaterialAttribute.DOUBLE_SIDED)] = False |
||||||
|
attributes_to_keep[material.attribute_data_offset(1) + material.attribute_id(1, trade.MaterialAttribute.LAYER_FACTOR_TEXTURE_MATRIX)] = False |
||||||
|
filtered_attributes = materialtools.filter_attributes(material, attributes_to_keep, ~trade.MaterialTypes.PBR_CLEAR_COAT) |
||||||
|
self.assertEqual(filtered_attributes.types, trade.MaterialTypes.PBR_METALLIC_ROUGHNESS) |
||||||
|
self.assertEqual(filtered_attributes.layer_count, 2) |
||||||
|
self.assertEqual(filtered_attributes.attribute_count(0), 2) |
||||||
|
self.assertEqual(filtered_attributes.attribute_count(1), 7) |
||||||
|
self.assertFalse(filtered_attributes.has_attribute(trade.MaterialAttribute.DOUBLE_SIDED)) |
||||||
|
self.assertFalse(filtered_attributes.has_attribute(1, trade.MaterialAttribute.LAYER_FACTOR_TEXTURE_MATRIX)) |
||||||
|
|
||||||
|
layers_to_keep = containers.BitArray.direct_init(material.layer_count, True) |
||||||
|
layers_to_keep[1] = False |
||||||
|
filtered_layers = materialtools.filter_layers(material, layers_to_keep, ~trade.MaterialTypes.PBR_CLEAR_COAT) |
||||||
|
self.assertEqual(filtered_layers.types, trade.MaterialTypes.PBR_METALLIC_ROUGHNESS) |
||||||
|
self.assertEqual(filtered_layers.layer_count, 1) |
||||||
|
self.assertEqual(filtered_layers.attribute_count(), 3) |
||||||
|
|
||||||
|
filtered_attributes_layers = materialtools.filter_attributes_layers(material, attributes_to_keep, layers_to_keep, ~trade.MaterialTypes.PBR_CLEAR_COAT) |
||||||
|
self.assertEqual(filtered_attributes_layers.types, trade.MaterialTypes.PBR_METALLIC_ROUGHNESS) |
||||||
|
self.assertEqual(filtered_attributes_layers.layer_count, 1) |
||||||
|
self.assertEqual(filtered_attributes_layers.attribute_count(), 2) |
||||||
|
self.assertFalse(filtered_attributes_layers.has_attribute(trade.MaterialAttribute.DOUBLE_SIDED)) |
||||||
|
|
||||||
|
def test_invalid_size(self): |
||||||
|
importer = trade.ImporterManager().load_and_instantiate('GltfImporter') |
||||||
|
# This adds extra Diffuse attributes for BaseColor and a Phong type, |
||||||
|
# don't want |
||||||
|
importer.configuration['phongMaterialFallback'] = False |
||||||
|
importer.open_file(os.path.join(os.path.dirname(__file__), "material.gltf")) |
||||||
|
|
||||||
|
material = importer.material(0) |
||||||
|
self.assertEqual(material.types, trade.MaterialTypes.PBR_METALLIC_ROUGHNESS|trade.MaterialTypes.PBR_CLEAR_COAT) |
||||||
|
self.assertEqual(material.layer_count, 2) |
||||||
|
self.assertEqual(material.attribute_count(0), 3) |
||||||
|
self.assertEqual(material.attribute_count(1), 8) |
||||||
|
|
||||||
|
with self.assertRaisesRegex(AssertionError, "expected 11 bits but got 12"): |
||||||
|
materialtools.filter_attributes(material, containers.BitArray.value_init(12)) |
||||||
|
with self.assertRaisesRegex(AssertionError, "expected 2 bits but got 3"): |
||||||
|
materialtools.filter_layers(material, containers.BitArray.value_init(3)) |
||||||
|
with self.assertRaisesRegex(AssertionError, "expected 11 attribute bits but got 12"): |
||||||
|
materialtools.filter_attributes_layers(material, containers.BitArray.value_init(12), containers.BitArray.value_init(2)) |
||||||
|
with self.assertRaisesRegex(AssertionError, "expected 2 layer bits but got 3"): |
||||||
|
materialtools.filter_attributes_layers(material, containers.BitArray.value_init(11), containers.BitArray.value_init(3)) |
||||||
|
|
||||||
|
class Merge(unittest.TestCase): |
||||||
|
def test(self): |
||||||
|
importer = trade.ImporterManager().load_and_instantiate('GltfImporter') |
||||||
|
# This adds extra Diffuse attributes for BaseColor and a Phong type, |
||||||
|
# don't want |
||||||
|
importer.configuration['phongMaterialFallback'] = False |
||||||
|
importer.open_file(os.path.join(os.path.dirname(__file__), "material.gltf")) |
||||||
|
|
||||||
|
a = importer.material('A material with a layer') |
||||||
|
self.assertEqual(a.types, trade.MaterialTypes.PBR_METALLIC_ROUGHNESS|trade.MaterialTypes.PBR_CLEAR_COAT) |
||||||
|
self.assertEqual(a.layer_count, 2) |
||||||
|
self.assertEqual(a.attribute_count(0), 3) |
||||||
|
self.assertEqual(a.attribute_count(1), 8) |
||||||
|
|
||||||
|
b = importer.material('Material with an empty layer') |
||||||
|
self.assertEqual(b.types, trade.MaterialTypes.PBR_CLEAR_COAT) |
||||||
|
self.assertEqual(b.layer_count, 2) |
||||||
|
self.assertEqual(b.attribute_count(0), 1) |
||||||
|
self.assertEqual(b.attribute_count(1), 3) |
||||||
|
|
||||||
|
merged = materialtools.merge(a, b, materialtools.MergeConflicts.KEEP_FIRST_IF_SAME_TYPE) |
||||||
|
self.assertEqual(merged.types, trade.MaterialTypes.PBR_METALLIC_ROUGHNESS|trade.MaterialTypes.PBR_CLEAR_COAT) |
||||||
|
self.assertEqual(merged.layer_count, 2) |
||||||
|
self.assertEqual(merged.attribute_count(0), 4) |
||||||
|
self.assertEqual(merged.attribute_count(1), 8) |
||||||
|
|
||||||
|
def test_failed(self): |
||||||
|
importer = trade.ImporterManager().load_and_instantiate('GltfImporter') |
||||||
|
importer.open_file(os.path.join(os.path.dirname(__file__), "material.gltf")) |
||||||
|
|
||||||
|
a = importer.material('A material with a layer') |
||||||
|
b = importer.material('Material with an empty layer') |
||||||
|
|
||||||
|
with self.assertRaisesRegex(RuntimeError, "material merge failed"): |
||||||
|
materialtools.merge(a, b) |
||||||
|
|
||||||
|
class PhongToPbrMetallicRoughness(unittest.TestCase): |
||||||
|
def test(self): |
||||||
|
importer = trade.ImporterManager().load_and_instantiate('GltfImporter') |
||||||
|
# We actually *do* want the Phong stuff |
||||||
|
importer.configuration['phongMaterialFallback'] = True |
||||||
|
importer.open_file(os.path.join(os.path.dirname(__file__), "material.gltf")) |
||||||
|
material = importer.material(0) |
||||||
|
|
||||||
|
# Keep just the diffuse color, drop everything else |
||||||
|
attributes_to_keep = containers.BitArray.value_init(material.attribute_data_offset(material.layer_count)) |
||||||
|
attributes_to_keep[material.attribute_id(trade.MaterialAttribute.DIFFUSE_COLOR)] = True |
||||||
|
layers_to_keep = containers.BitArray.value_init(material.layer_count) |
||||||
|
layers_to_keep[0] = True |
||||||
|
filtered = materialtools.filter_attributes_layers(material, attributes_to_keep, layers_to_keep, ~(trade.MaterialTypes.PBR_METALLIC_ROUGHNESS|trade.MaterialTypes.PBR_CLEAR_COAT)) |
||||||
|
self.assertEqual(filtered.types, trade.MaterialTypes.PHONG) |
||||||
|
self.assertEqual(filtered.layer_count, 1) |
||||||
|
self.assertEqual(filtered.attribute_count(), 1) |
||||||
|
self.assertTrue(filtered.has_attribute(trade.MaterialAttribute.DIFFUSE_COLOR)) |
||||||
|
|
||||||
|
pbr = materialtools.phong_to_pbr_metallic_roughness(filtered) |
||||||
|
self.assertEqual(pbr.types, trade.MaterialTypes.PBR_METALLIC_ROUGHNESS) |
||||||
|
self.assertEqual(pbr.layer_count, 1) |
||||||
|
self.assertEqual(pbr.attribute_count(), 1) |
||||||
|
self.assertTrue(pbr.has_attribute(trade.MaterialAttribute.BASE_COLOR)) |
||||||
|
|
||||||
|
def test_failed(self): |
||||||
|
importer = trade.ImporterManager().load_and_instantiate('GltfImporter') |
||||||
|
importer.open_file(os.path.join(os.path.dirname(__file__), "material.gltf")) |
||||||
|
material = importer.material('Specular/glossiness') |
||||||
|
|
||||||
|
with self.assertRaisesRegex(RuntimeError, "material conversion failed"): |
||||||
|
materialtools.phong_to_pbr_metallic_roughness(material, materialtools.PhongToPbrMetallicRoughnessFlags.FAIL_ON_UNCONVERTIBLE_ATTRIBUTES) |
||||||
|
|
||||||
|
class RemoveDuplicates(unittest.TestCase): |
||||||
|
def test(self): |
||||||
|
importer = trade.ImporterManager().load_and_instantiate('GltfImporter') |
||||||
|
importer.open_file(os.path.join(os.path.dirname(__file__), "material.gltf")) |
||||||
|
a = importer.material(0) |
||||||
|
b = importer.material(1) |
||||||
|
c = importer.material(2) |
||||||
|
|
||||||
|
indices, count = materialtools.remove_duplicates([a, b, a, c, c, b]) |
||||||
|
self.assertEqual(indices, [0, 1, 0, 3, 3, 1]) |
||||||
|
self.assertEqual(count, 3) |
||||||
Loading…
Reference in new issue