from maya.cmds import curve
from math import cos,sin

class FractalTree():
    '''Every branch is a curve (exact, 20.000+ curves in it-8)'''
    def __init__(self):
        self.it = 6      # tree iterations
        self.r = .7      # branch length factor
        self.s = .5      # branch spreading (0<x<pi/2)
        self.d3 = True   # 3D or 2D tree
        self.branch()
    def branch(self,
               x=0,y=0,z=0, # branch position
               a=0,b=0,     # branch angle
               l=10,        # branch length
               lvl=0        # branch iteration
               ):
        if(lvl<self.it):
            x2 = x + l*cos(a)*cos(b)
            y2 = y + l*cos(a)*sin(b)
            z2 = z + l*sin(a)
            curve(p=[(x,y,z),(x2,y2,z2)],d=1)
            self.branch(x2,y2,z2,a+self.s,b,l*self.r,lvl+1)
            self.branch(x2,y2,z2,a-self.s,b,l*self.r,lvl+1)
            if(self.d3):
                self.branch(x2,y2,z2,a,b+self.s,l*self.r,lvl+1)
                self.branch(x2,y2,z2,a,b-self.s,l*self.r,lvl+1)

tree = FractalTree()