80 lines
3.7 KiB
Python
80 lines
3.7 KiB
Python
# ***************************************************************************
|
|
# * (c) 2019 Eliud Cabrera Castillo <e.cabrera-castillo@tum.de> *
|
|
# * *
|
|
# * This file is part of the FreeCAD CAx development system. *
|
|
# * *
|
|
# * 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. *
|
|
# * *
|
|
# * 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 Library General Public License for more details. *
|
|
# * *
|
|
# * You should have received a copy of the GNU Library General Public *
|
|
# * License along with FreeCAD; if not, write to the Free Software *
|
|
# * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 *
|
|
# * USA *
|
|
# * *
|
|
# ***************************************************************************
|
|
"""Provides the viewprovider code for the Array object."""
|
|
## @package view_array
|
|
# \ingroup draftviewproviders
|
|
# \brief Provides the viewprovider code for the Array object.
|
|
|
|
## \addtogroup draftviewproviders
|
|
# @{
|
|
from draftviewproviders.view_base import ViewProviderDraft
|
|
from draftutils import gui_utils
|
|
|
|
|
|
class ViewProviderDraftArray(ViewProviderDraft):
|
|
"""a view provider that displays a Array icon instead of a Draft icon"""
|
|
|
|
def __init__(self,vobj):
|
|
super(ViewProviderDraftArray, self).__init__(vobj)
|
|
|
|
def getIcon(self):
|
|
if hasattr(self.Object, "ArrayType"):
|
|
if self.Object.ArrayType == 'ortho':
|
|
return ":/icons/Draft_Array.svg"
|
|
elif self.Object.ArrayType == 'polar':
|
|
return ":/icons/Draft_PolarArray.svg"
|
|
elif self.Object.ArrayType == 'circular':
|
|
return ":/icons/Draft_CircularArray.svg"
|
|
elif hasattr(self.Object, "PointObject"):
|
|
return ":/icons/Draft_PointArray.svg"
|
|
elif hasattr(self.Object, "RotationFactor"):
|
|
return ":/icons/Draft_PathTwistedArray.svg"
|
|
else:
|
|
return ":/icons/Draft_PathArray.svg"
|
|
|
|
def resetColors(self, vobj):
|
|
obj = vobj.Object
|
|
if obj.Base is not None:
|
|
colors = gui_utils.get_diffuse_color(obj.Base)
|
|
if not colors:
|
|
return
|
|
if len(colors) > 1:
|
|
n = 1
|
|
if hasattr(obj, "ArrayType"):
|
|
if obj.ArrayType == "ortho":
|
|
n = obj.NumberX * obj.NumberY * obj.NumberZ
|
|
elif obj.ArrayType == "polar":
|
|
n = obj.NumberPolar
|
|
else: # "circular"
|
|
n = obj.Count
|
|
elif hasattr(obj, "Count"):
|
|
n = obj.Count
|
|
colors = colors * n
|
|
vobj.DiffuseColor = colors
|
|
|
|
|
|
# Alias for compatibility with v0.18 and earlier
|
|
_ViewProviderDraftArray = ViewProviderDraftArray
|
|
|
|
## @}
|