src/Mod/PartDesign/PartDesignTests/TestMaterial.py

# ***************************************************************************
# *   Copyright (c) 2024 David Carter <dcarter@davidcarter.ca>              *
# *                                                                         *
# *   This program is free software; you can redistribute it and/or modify  *
# *   it under the terms of the GNU Lesser General Public License (LGPL)    *
# *   as published by the Free Software Foundation; either version 2 of     *
# *   the License, or (at your option) any later version.                   *
# *   for detail see the LICENCE text file.                                 *
# *                                                                         *
# *   This program 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 Library General Public License for more details.                  *
# *                                                                         *
# *   You should have received a copy of the GNU Library General Public     *
# *   License along with this program; if not, write to the Free Software   *
# *   Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  *
# *   USA                                                                   *
# *                                                                         *
# ***************************************************************************

import unittest

import FreeCAD


class TestMaterial(unittest.TestCase):
    def setUp(self):
        self.Doc = FreeCAD.newDocument("PartDesignTestMaterial")

    def compareColors(self, c1, c2):
        # Colors are in the form of tuples
        self.assertEqual(len(c1), 4)
        self.assertEqual(len(c2), 4)
        self.assertAlmostEqual(c1[0], c2[0])
        self.assertAlmostEqual(c1[1], c2[1])
        self.assertAlmostEqual(c1[2], c2[2])
        self.assertAlmostEqual(c1[3], c2[3])

    def testDiffuseColor(self):
        print("PartDesignTestMaterial.testDiffuseColor()")
        self.Body = self.Doc.addObject("PartDesign::Body", "Body")
        self.Box = self.Doc.addObject("PartDesign::AdditiveBox", "Box")
        self.Body.addObject(self.Box)
        self.Box.Length = 10.00
        self.Box.Width = 10.00
        self.Box.Height = 10.00
        self.Doc.recompute()

        self.Body.ViewObject.DiffuseColor = [
            (1.0, 0.0, 0.0, 0.0),
            (0.9, 0.0, 0.0, 0.0),
            (0.8, 0.0, 0.0, 0.0),
            (0.7, 0.0, 0.0, 0.0),
            (0.6, 1.0, 0.0, 0.0),
            (0.5, 1.0, 0.0, 0.0),
        ]
        self.Doc.recompute()
        self.assertEqual(len(self.Body.ViewObject.DiffuseColor), 6)
        colors = self.Body.ViewObject.DiffuseColor
        self.compareColors(colors[0], (1.0, 0.0, 0.0, 0.0))
        self.compareColors(colors[1], (0.9, 0.0, 0.0, 0.0))
        self.compareColors(colors[2], (0.8, 0.0, 0.0, 0.0))
        self.compareColors(colors[3], (0.7, 0.0, 0.0, 0.0))
        self.compareColors(colors[4], (0.6, 1.0, 0.0, 0.0))
        self.compareColors(colors[5], (0.5, 1.0, 0.0, 0.0))

        self.Body.ViewObject.DiffuseColor = [(1.0, 0.0, 0.0, 0.0)]
        self.Doc.recompute()
        self.assertEqual(len(self.Body.ViewObject.DiffuseColor), 1)
        colors = self.Body.ViewObject.DiffuseColor
        self.compareColors(colors[0], (1.0, 0.0, 0.0, 0.0))

        self.Box.ViewObject.DiffuseColor = [
            (1.0, 0.0, 0.0, 0.0),
            (0.9, 0.0, 0.0, 0.0),
            (0.8, 0.0, 0.0, 0.0),
            (0.7, 0.0, 0.0, 0.0),
            (0.6, 1.0, 0.0, 0.0),
            (0.5, 1.0, 0.0, 0.0),
        ]
        self.Doc.recompute()
        self.assertEqual(len(self.Box.ViewObject.DiffuseColor), 6)
        colors = self.Box.ViewObject.DiffuseColor
        self.compareColors(colors[0], (1.0, 0.0, 0.0, 0.0))
        self.compareColors(colors[1], (0.9, 0.0, 0.0, 0.0))
        self.compareColors(colors[2], (0.8, 0.0, 0.0, 0.0))
        self.compareColors(colors[3], (0.7, 0.0, 0.0, 0.0))
        self.compareColors(colors[4], (0.6, 1.0, 0.0, 0.0))
        self.compareColors(colors[5], (0.5, 1.0, 0.0, 0.0))

        self.Box.ViewObject.DiffuseColor = [(1.0, 0.0, 0.0, 0.0)]
        self.Doc.recompute()
        self.assertEqual(len(self.Box.ViewObject.DiffuseColor), 1)
        colors = self.Box.ViewObject.DiffuseColor
        self.compareColors(colors[0], (1.0, 0.0, 0.0, 0.0))

    def tearDown(self):
        # closing doc
        FreeCAD.closeDocument("PartDesignTestMaterial")
        # print ("omit closing document for debugging")