File size: 4,137 Bytes
985c397 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 | # SPDX-License-Identifier: LGPL-2.1-or-later
# **************************************************************************
# Copyright (c) 2024 Werner Mayer <wmayer[at]users.sourceforge.net> *
# *
# 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 *
# <https://www.gnu.org/licenses/>. *
# *
# **************************************************************************
import os
import tempfile
import unittest
import FreeCAD as App
import ImportGui
from pivy import coin
class ExportImportTest(unittest.TestCase):
def setUp(self):
TempPath = tempfile.gettempdir()
self.fileName = TempPath + os.sep + "ColorPerFaceTest.step"
self.doc = App.newDocument()
def tearDown(self):
App.closeDocument(self.doc.Name)
def testSaveLoadStepFile(self):
"""
Create a STEP file with color per face
"""
part = self.doc.addObject("App::Part", "Part")
box = part.newObject("Part::Box", "Box")
self.doc.recompute()
box.ViewObject.DiffuseColor = [
(1.0, 0.0, 0.0, 1.0),
(1.0, 0.0, 0.0, 1.0),
(1.0, 0.0, 0.0, 1.0),
(1.0, 0.0, 0.0, 1.0),
(1.0, 1.0, 0.0, 1.0),
(1.0, 1.0, 0.0, 1.0),
]
ImportGui.export([part], self.fileName)
self.doc.clearDocument()
ImportGui.insert(name=self.fileName, docName=self.doc.Name, merge=False, useLinkGroup=True)
part_features = list(filter(lambda x: x.isDerivedFrom("Part::Feature"), self.doc.Objects))
self.assertEqual(len(part_features), 1)
feature = part_features[0]
self.assertEqual(len(feature.ViewObject.DiffuseColor), 6)
self.assertEqual(feature.ViewObject.DiffuseColor[0], (1.0, 0.0, 0.0, 1.0))
self.assertEqual(feature.ViewObject.DiffuseColor[1], (1.0, 0.0, 0.0, 1.0))
self.assertEqual(feature.ViewObject.DiffuseColor[2], (1.0, 0.0, 0.0, 1.0))
self.assertEqual(feature.ViewObject.DiffuseColor[3], (1.0, 0.0, 0.0, 1.0))
self.assertEqual(feature.ViewObject.DiffuseColor[4], (1.0, 1.0, 0.0, 1.0))
self.assertEqual(feature.ViewObject.DiffuseColor[5], (1.0, 1.0, 0.0, 1.0))
sa = coin.SoSearchAction()
sa.setType(coin.SoMaterialBinding.getClassTypeId())
# We need an easier way to access nodes of a display mode
sa.setInterest(coin.SoSearchAction.ALL)
sa.apply(feature.ViewObject.RootNode)
paths = sa.getPaths()
bind = paths.get(1).getTail()
self.assertEqual(bind.value.getValue(), bind.PER_PART)
sa = coin.SoSearchAction()
sa.setType(coin.SoMaterial.getClassTypeId())
# We need an easier way to access nodes of a display mode
sa.setInterest(coin.SoSearchAction.ALL)
sa.apply(feature.ViewObject.RootNode)
paths = sa.getPaths()
mat = paths.get(1).getTail()
self.assertEqual(mat.diffuseColor.getNum(), 6)
|