class Branch
{
  int _x,_y,_col;
  Vector branches;
  int[] myPixels;
  int[] drawingPixels;
  
  TreeRect leftRect;
  int _level;
  float myBrightness;

  Branch(int $x, int $y, color $c, int[] $pixels, TreeRect $leftRect, int $level){
    _x=$x;
    _y=$y;
    _col=$c;
    myPixels=$pixels;
    leftRect=$leftRect;
    _level=$level;
    if(_level>nLevels) nLevels=_level*1.0;
    branches=new Vector();
    _col=colors[_level%colors.length];
  }
  void addBranch(int $x, int $y, color $c, int[] $p, TreeRect $leftRect){
    branches.addElement(new Branch($x,$y,$c,$p,$leftRect,_level+1));
  }
  void drawMe(){
    int len=branches.size();
    if(len>0){
      ((Branch) branches.get(0)).drawMe();
      ((Branch) branches.get(1)).drawMe();
    }
    else{
      color t_col=color((255/nLevels)*_level);
      loadPixels();
      int pos;
      for(int y=0; y<h; y++){
        for(int x=w2; x<w; x++){
          pos=w*y+x;
          if(myPixels[pos]==1){
            pixels[pos]=t_col;
          }else if(y>1 && y<h-1 && x>0 && x<w-1 && (myPixels[(w*(y+1))+x]==1 || myPixels[(w*(y-1))+x]==1 || myPixels[(w*y)+x+1]==1 || myPixels[(w*y)+x-1]==1)){
            pixels[pos]=bg1;
          }
        }
      }
      updatePixels();
      _col=colors[_level%colors.length];
      fill(_col);
      rect(leftRect._x,leftRect._y,leftRect._w,leftRect._h);
    }
  }
  void splitMe(int $x, int $y, color $c){
    int[] pixels_1= new int[npixels];
    int[] pixels_2= new int[npixels];
    int pos;
    float d_1;
    float d_2;
    float count_1=0;
    float count_2=0;
    for(int y=0; y<h; y++){
      for(int x=0; x<w; x++){
        pos=w*y+x;
        if(myPixels[pos]==1){
          d_1=distance(x,y,_x,_y);
          d_2=distance(x,y,$x,$y);
          if(d_1<d_2){
            pixels_1[pos]=1;
            count_1++;
          }else{
            pixels_2[pos]=1;
            count_2++;
          }
        }
      }
    }
    int r1_x,r2_x,r1_y,r2_y;
    int r1_w,r2_w,r1_h,r2_h;
    if(leftRect._w>leftRect._h){
      // cut width
      r1_w=int((leftRect._w/(count_1+count_2))*count_1);
      r2_w=leftRect._w-r1_w;
      r1_h=r2_h=leftRect._h;
      r1_x=leftRect._x;
      r2_x=leftRect._x+r1_w;
      r1_y=r2_y=leftRect._y;
    }else{
      // cut height
      r1_h=int((leftRect._h/(count_1+count_2))*count_1);
      r2_h=leftRect._h-r1_h;
      r1_w=r2_w=leftRect._w;
      r1_y=leftRect._y;
      r2_y=leftRect._y+r1_h;
      r1_x=r2_x=leftRect._x;
    }
    TreeRect rect_1=new TreeRect(r1_x,r1_y,r1_w,r1_h);
    TreeRect rect_2=new TreeRect(r2_x,r2_y,r2_w,r2_h);
    myPixels=null;
    branches.addElement(new Branch(_x,_y,$c,pixels_1,rect_1,_level+1));
    branches.addElement(new Branch($x,$y,$c,pixels_2,rect_2,_level+1));
    blendedRects.addElement(rect_1);
    blendedRects.addElement(rect_2);
  }
  Branch getBranch(int $x, int $y){
    Branch res=null;
    Branch t_res;
    if(branches.size()>0){
      Branch child_1=(Branch) branches.get(0);
      Branch child_2=(Branch) branches.get(1);
      t_res=child_1.getBranch($x,$y);
      if(t_res!=null){
        res=t_res;
      }
      else{
        t_res=child_2.getBranch($x,$y);
        if(t_res!=null) res=t_res;
      }
    }
    else{
      if(myPixels[w*$y+$x]==1) res=this;
    }
    return res;
  }
  Branch getRect(int $x, int $y){
    Branch res=null;
    Branch t_res;
    if(branches.size()>0){
      Branch child_1=(Branch) branches.get(0);
      Branch child_2=(Branch) branches.get(1);
      t_res=child_1.getRect($x,$y);
      if(t_res!=null){
        res=t_res;
      }
      else{
        t_res=child_2.getRect($x,$y);
        if(t_res!=null) res=t_res;
      }
    }
    else{
      if($x>=leftRect._x && $x<leftRect._x+leftRect._w && $y>=leftRect._y && $y<leftRect._y+leftRect._h) res=this;
    }
    return res;
  }
  void highLightBranch(int $x, int $y){
    if($x>=leftRect._x && $x<leftRect._x+leftRect._w && $y>=leftRect._y && $y<leftRect._y+leftRect._h){
      fill(color(0,126,169,255/nLevels*_level));
      rect(leftRect._x,leftRect._y,leftRect._w,leftRect._h);
      if(branches.size()>0){
        ((Branch) branches.get(0)).highLightBranch($x,$y);
        ((Branch) branches.get(1)).highLightBranch($x,$y);
      }
    }
  }
}