$fn=32;
m = 3;
n = 2;

thickness = 1.3;
nodeSize = 2;


width=15;

angle_factor = 0.7;
length_factor = 0.6;



// skip_first_branch: do not draw the first branch for top, bottom, or none

module level(number= 3 *m*n, angle=70, width=20, length=10, skip_first_branch="none") {
    step_n = width/n;
    step_m = width/m;
    
    
    // top
    
    for(k=[0:number/n-1]){
    translate([k*n*step_n,0,0]){
        // tree and balls
        for(i=[0:n-1]){
            if(skip_first_branch == "top" && i==0)
            {}
            else{
                translate([i*step_n,0,0])
                rotate(a=[-90+(180-angle)/2+i/(n-1)*angle,0,0]){
                    cylinder(length, d=thickness);
                    
                    // m balls at the top
                    if(k<number/n-1)
                    translate([0,0,length])
                    for(j=[0:m-1]){
                        translate([j*step_m,0,0])
                        sphere(d=nodeSize);
                    }
                    else
                        translate([0,0,length])
                        sphere(d=nodeSize);
                }
            }
        }
        
        // Horizontal bars
        if(k<number/n-1){
        for(i=[0:n-1]){
            if(skip_first_branch == "top" && i==0)
            {}
            else{
                translate([i*step_n,0,0])
                rotate(a=[-90+(180-angle)/2+i/(n-1)*angle,0,0]){
                    rotate(a=[0,90,0])
                    translate([-length,0,-thickness/2])
                    cylinder(n*step_n+thickness, d=thickness);
                }
            }
        }
        }
    }
    }
    
    
    
    // bottom
     color("Magenta")
    
    
    for(k=[0:number/m-1]){
    translate([k*m*step_n,0,0]){
        // tree and balls
        for(i=[0:m-1]){
            if(skip_first_branch == "bottom" && i==0)
            {}
            else{
                translate([i*step_n,0,0])
                rotate(a=[90+(180-angle)/2+i/(m-1)*angle,0,0]){
                    cylinder(length, d=thickness);
                    
                    // m balls at the top
                    if(k<number/m-1)
                    translate([0,0,length])
                    for(j=[0:n-1]){
                        new_step = step_n*m/n;
                        translate([j*new_step,0,0])
                        sphere(d=nodeSize);
                    }
                    else
                        translate([0,0,length])
                        sphere(d=nodeSize);
                }
            }
        }
        
        // Horizontal bars
        if(k<number/m-1){
        for(i=[0:m-1]){
            if(skip_first_branch == "bottom" && i==0)
            {}
            else{
                translate([i*step_n,0,0])
                rotate(a=[90+(180-angle)/2+i/(m-1)*angle,0,0]){
                    rotate(a=[0,90,0])
                    translate([-length,0,-thickness/2])
                    cylinder(step_n*m+thickness, d=thickness);
                }
            }
        }
        }
    }
    }
    
}


module recurs(depth, number=m*n*m*n, angle=80, width=20, length=20, skip_first_branch = "none"){
    step_n = width/n;
    step_m = width/m;
    
    
    level(number, angle, width, length, skip_first_branch);
    
    if(depth <= 1){}
    else{
        
        /// top
        newnumber = number*m/n;
        newwidth = width/m*n;
        newangle = angle*angle_factor;
        newlength = length*length_factor;
        
        for(i=[0:n-1]){
            translate([i*step_n,0,0]){
                a1= [90+-angle/2+i/(n-1)*angle,0,0];
                rotate(a1)
                translate([0,length,0])
                rotate(-a1)
                {
                    color("Blue")//sphere(2);
                    if(i<n/2)
                        recurs(depth=depth-1,number=newnumber, angle=newangle, width=newwidth, length=newlength, skip_first_branch= "bottom");
                    else // second half: mirror
                    mirror([0,1,0])
                    recurs(depth=depth-1,number=newnumber, angle=newangle, width=newwidth, length=newlength, skip_first_branch= "bottom");
                } 
            }
        }
        
        /// bottom
        newnumber2 = number*n/m;
        newwidth2 = width/n*m;
        newangle2 = angle*angle_factor;
        newlength2 = length*length_factor;
        
        for(i=[0:m-1]){
            translate([i*step_n,0,0]){
                a1= [-90+-angle/2+i/(m-1)*angle,0,0];
                rotate(a1)
                translate([0,length,0])
                rotate(-a1)
                {
                    color("Green")//sphere(2);
                    if(i<m/2)
                    recurs(depth=depth-1,number=newnumber2, angle=newangle2, width=newwidth2, length=newlength2, skip_first_branch= "top");
                    else // second half: mirror
                    mirror([0,1,0])
                    recurs(depth=depth-1,number=newnumber2, angle=newangle2, width=newwidth2, length=newlength2, skip_first_branch= "top");
                } 
            }
        }
        
    }
}

// TODO: quand on récurse, il faut enlever les branches qui existent déjà par récurrence...


recurs(2, m*n*m*n, width=width);

// initial bar and balls
rotate([0,90,0])cylinder(width*m*n*m-width/n, d=thickness);
for(i=[0:m*n*m*n-1]){
    translate([i*width/n, 0, 0])
    sphere(d=nodeSize);
}