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

class FractalTree():
    '''Tree in one curve (inexact but light)'''
    def __init__(self):
        self.pts = []
        self.it = 7      # 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()
        curve(p=self.pts,d=1)
    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):
            self.pts.append((x,y,z))
            x2 = x + l*cos(a)*cos(b)
            y2 = y + l*cos(a)*sin(b)
            z2 = z + l*sin(a)
            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()