Draft: split array function from Draft.py
This commit is contained in:
committed by
Yorik van Havre
parent
c53f447f58
commit
d124bfce4a
@@ -60,6 +60,7 @@ SET(Draft_utilities
|
||||
|
||||
SET(Draft_functions
|
||||
draftfunctions/__init__.py
|
||||
draftfunctions/array.py
|
||||
draftfunctions/cut.py
|
||||
draftfunctions/downgrade.py
|
||||
draftfunctions/draftify.py
|
||||
|
||||
@@ -178,6 +178,8 @@ from draftutils.gui_utils import load_texture
|
||||
# Draft functions
|
||||
#---------------------------------------------------------------------------
|
||||
|
||||
from draftfunctions.array import array
|
||||
|
||||
from draftfunctions.cut import cut
|
||||
|
||||
from draftfunctions.downgrade import downgrade
|
||||
@@ -523,63 +525,6 @@ def makePointArray(base, ptlst):
|
||||
return obj
|
||||
|
||||
|
||||
def array(objectslist,arg1,arg2,arg3,arg4=None,arg5=None,arg6=None):
|
||||
"""array(objectslist,xvector,yvector,xnum,ynum) for rectangular array,
|
||||
array(objectslist,xvector,yvector,zvector,xnum,ynum,znum) for rectangular array,
|
||||
or array(objectslist,center,totalangle,totalnum) for polar array: Creates an array
|
||||
of the objects contained in list (that can be an object or a list of objects)
|
||||
with, in case of rectangular array, xnum of iterations in the x direction
|
||||
at xvector distance between iterations, and same for y and z directions with yvector
|
||||
and ynum and zvector and znum. In case of polar array, center is a vector, totalangle
|
||||
is the angle to cover (in degrees) and totalnum is the number of objects, including
|
||||
the original.
|
||||
|
||||
This function creates an array of independent objects. Use makeArray() to create a
|
||||
parametric array object."""
|
||||
|
||||
def rectArray(objectslist,xvector,yvector,xnum,ynum):
|
||||
typecheck([(xvector,Vector), (yvector,Vector), (xnum,int), (ynum,int)], "rectArray")
|
||||
if not isinstance(objectslist,list): objectslist = [objectslist]
|
||||
for xcount in range(xnum):
|
||||
currentxvector=Vector(xvector).multiply(xcount)
|
||||
if not xcount==0:
|
||||
move(objectslist,currentxvector,True)
|
||||
for ycount in range(ynum):
|
||||
currentxvector=FreeCAD.Base.Vector(currentxvector)
|
||||
currentyvector=currentxvector.add(Vector(yvector).multiply(ycount))
|
||||
if not ycount==0:
|
||||
move(objectslist,currentyvector,True)
|
||||
def rectArray2(objectslist,xvector,yvector,zvector,xnum,ynum,znum):
|
||||
typecheck([(xvector,Vector), (yvector,Vector), (zvector,Vector),(xnum,int), (ynum,int),(znum,int)], "rectArray2")
|
||||
if not isinstance(objectslist,list): objectslist = [objectslist]
|
||||
for xcount in range(xnum):
|
||||
currentxvector=Vector(xvector).multiply(xcount)
|
||||
if not xcount==0:
|
||||
move(objectslist,currentxvector,True)
|
||||
for ycount in range(ynum):
|
||||
currentxvector=FreeCAD.Base.Vector(currentxvector)
|
||||
currentyvector=currentxvector.add(Vector(yvector).multiply(ycount))
|
||||
if not ycount==0:
|
||||
move(objectslist,currentyvector,True)
|
||||
for zcount in range(znum):
|
||||
currentzvector=currentyvector.add(Vector(zvector).multiply(zcount))
|
||||
if not zcount==0:
|
||||
move(objectslist,currentzvector,True)
|
||||
def polarArray(objectslist,center,angle,num):
|
||||
typecheck([(center,Vector), (num,int)], "polarArray")
|
||||
if not isinstance(objectslist,list): objectslist = [objectslist]
|
||||
fraction = float(angle)/num
|
||||
for i in range(num):
|
||||
currangle = fraction + (i*fraction)
|
||||
rotate(objectslist,currangle,center,copy=True)
|
||||
if arg6:
|
||||
rectArray2(objectslist,arg1,arg2,arg3,arg4,arg5,arg6)
|
||||
elif arg4:
|
||||
rectArray(objectslist,arg1,arg2,arg3,arg4)
|
||||
else:
|
||||
polarArray(objectslist,arg1,arg2,arg3)
|
||||
|
||||
|
||||
def getDXF(obj,direction=None):
|
||||
"""getDXF(object,[direction]): returns a DXF entity from the given
|
||||
object. If direction is given, the object is projected in 2D."""
|
||||
|
||||
108
src/Mod/Draft/draftfunctions/array.py
Normal file
108
src/Mod/Draft/draftfunctions/array.py
Normal file
@@ -0,0 +1,108 @@
|
||||
# ***************************************************************************
|
||||
# * Copyright (c) 2009, 2010 Yorik van Havre <yorik@uncreated.net> *
|
||||
# * Copyright (c) 2009, 2010 Ken Cline <cline@frii.com> *
|
||||
# * Copyright (c) 2020 FreeCAD Developers *
|
||||
# * *
|
||||
# * 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 object code for Draft array function."""
|
||||
## @package array
|
||||
# \ingroup DRAFT
|
||||
# \brief Provides the object code for Draft array.
|
||||
|
||||
import FreeCAD as App
|
||||
|
||||
import draftutils.utils as utils
|
||||
|
||||
from draftfunctions.move import move
|
||||
from draftfunctions.rotate import rotate
|
||||
|
||||
|
||||
def array(objectslist, arg1, arg2, arg3, arg4=None, arg5=None, arg6=None):
|
||||
"""
|
||||
This function creates an array of independent objects.
|
||||
Use makeArray() to create a parametric array object.
|
||||
|
||||
Creates an array of the given objects (that can be an object or a list
|
||||
of objects).
|
||||
|
||||
In case of rectangular array, xnum of iterations in the x direction
|
||||
at xvector distance between iterations, and same for y and z directions
|
||||
with yvector and ynum and zvector and znum.
|
||||
|
||||
In case of polar array, center is a vector, totalangle is the angle
|
||||
to cover (in degrees) and totalnum is the number of objects, including
|
||||
the original.
|
||||
|
||||
Use
|
||||
---
|
||||
array(objectslist, xvector, yvector, xnum, ynum) for rectangular array
|
||||
|
||||
array(objectslist, xvector, yvector, zvector, xnum, ynum, znum) for rectangular array
|
||||
|
||||
array(objectslist, center, totalangle, totalnum) for polar array
|
||||
"""
|
||||
|
||||
def rectArray(objectslist,xvector,yvector,xnum,ynum):
|
||||
utils.type_check([(xvector, App.Vector),
|
||||
(yvector, App.Vector),
|
||||
(xnum,int), (ynum,int)],
|
||||
"rectArray")
|
||||
if not isinstance(objectslist,list): objectslist = [objectslist]
|
||||
for xcount in range(xnum):
|
||||
currentxvector=App.Vector(xvector).multiply(xcount)
|
||||
if not xcount==0:
|
||||
move(objectslist,currentxvector,True)
|
||||
for ycount in range(ynum):
|
||||
currentxvector=App.Vector(currentxvector)
|
||||
currentyvector=currentxvector.add(App.Vector(yvector).multiply(ycount))
|
||||
if not ycount==0:
|
||||
move(objectslist,currentyvector,True)
|
||||
|
||||
def rectArray2(objectslist,xvector,yvector,zvector,xnum,ynum,znum):
|
||||
utils.type_check([(xvector,App.Vector), (yvector,App.Vector), (zvector,App.Vector),(xnum,int), (ynum,int),(znum,int)], "rectArray2")
|
||||
if not isinstance(objectslist,list): objectslist = [objectslist]
|
||||
for xcount in range(xnum):
|
||||
currentxvector=App.Vector(xvector).multiply(xcount)
|
||||
if not xcount==0:
|
||||
move(objectslist,currentxvector,True)
|
||||
for ycount in range(ynum):
|
||||
currentxvector=App.Vector(currentxvector)
|
||||
currentyvector=currentxvector.add(App.Vector(yvector).multiply(ycount))
|
||||
if not ycount==0:
|
||||
move(objectslist,currentyvector,True)
|
||||
for zcount in range(znum):
|
||||
currentzvector=currentyvector.add(App.Vector(zvector).multiply(zcount))
|
||||
if not zcount==0:
|
||||
move(objectslist,currentzvector,True)
|
||||
|
||||
def polarArray(objectslist,center,angle,num):
|
||||
utils.type_check([(center,App.Vector), (num,int)], "polarArray")
|
||||
if not isinstance(objectslist,list): objectslist = [objectslist]
|
||||
fraction = float(angle)/num
|
||||
for i in range(num):
|
||||
currangle = fraction + (i*fraction)
|
||||
rotate(objectslist,currangle,center,copy=True)
|
||||
if arg6:
|
||||
rectArray2(objectslist,arg1,arg2,arg3,arg4,arg5,arg6)
|
||||
elif arg4:
|
||||
rectArray(objectslist,arg1,arg2,arg3,arg4)
|
||||
else:
|
||||
polarArray(objectslist,arg1,arg2,arg3)
|
||||
Reference in New Issue
Block a user