treegrowjs/branch.coffee at master · k2xl/treegrowjs · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
class Branch
  # properties
  @ids = 0
  constructor:(@props)->
    @vertices = []
    @branches = []
    # console.log("initializing with ",@props)
    @savePosition()
    @id = Branch.ids++
    @props.initial_health ||= 100
    @props.color ||= "#000000"
    @props.branch_thickness || = 7
    @props.end_branch_thickness = @props.branch_thickness*.8
    @health = @props.initial_health
    @props.post_step ||=(branch)->
      if (branch.props.initial_health > 10 && branch.branches.length < 1 && branch.health < branch.props.initial_health/2)
        new_props = jQuery.extend(true,{},branch.props)
        new_props.initial_health*=.8;
        new_props.branch_thickness = branch.props.end_branch_thickness
        new_props.color = averageColors(branch.props.color,"#"+(0xFFFFFF*Math.random()).toString(12),10,1)
        left = new Branch(jQuery.extend(true,{},new_props))
        right = new Branch(jQuery.extend(true,{},new_props))
        branch.spawn(left)
        branch.spawn(right)
        left.savePosition()
        right.savePosition()
        left.rotate(-Math.PI/12)
        right.rotate(Math.PI/12)

  spawn:(branch)->
    @branches.push(branch)
  rotate:(amount)->
    ang = @get_angle()
    mag = @get_speed()
    @props.velocity[0] = Math.cos(ang+amount)*mag
    @props.velocity[1] = Math.sin(ang+amount)*mag

  get_angle:->
    Math.atan2(@props.velocity[1],@props.velocity[0])
  get_speed:->
    Math.sqrt((@props.velocity[0]*@props.velocity[0])+(@props.velocity[1]*@props.velocity[1]))
  step:(count=1)->
    # console.log(@id,@get_speed())
    @rotate( (Math.random()-.5)/12)
    i = 0
    lng = @branches.length
    while (i < lng)
      @branches[i].step()
      i++
    return if @health <= 0
    i = 0
    lng = @props.position.length
    while (i < lng)
      @props.position[i]+=@props.velocity[i]
      i++
    @savePosition()
    @health--
    @percDone = 1-@health/@props.initial_health
    @props.post_step(this)
    if (@health <= 0)
      @on_branch_dead()

  on_branch_dead:->
    # TODO: dispatch to Tree that branch is dead
  draw:(ctx,drawLast=1)->
    i = 0
    lng = @branches.length
    while (i < lng)
      @branches[i].draw(ctx,drawLast)
      i++
    return if @health <= 0
    i = (@vertices.length-drawLast)
    tempS = @vertices.length
    while (i < tempS)
      vertex = @vertices[i]
      prevVertex = @vertices[i-1]
      ctx.lineWidth = (@props.branch_thickness-@props.end_branch_thickness*@percDone)
      ctx.beginPath();
      ctx.strokeStyle = @props.color
      ctx.moveTo(prevVertex[0],prevVertex[1]);
      ctx.lineTo(vertex[0],vertex[1]);
      ctx.stroke();
      i++
  savePosition:->
    @vertices.push(@props.position.concat())
window.Branch = Branch