File: wedge.rb

package info (click to toggle)
ruby-rubyvis 0.6.1%2Bdfsg1-2
  • links: PTS, VCS
  • area: main
  • in suites: buster, stretch
  • size: 1,808 kB
  • ctags: 679
  • sloc: ruby: 11,114; makefile: 2
file content (145 lines) | stat: -rw-r--r-- 4,280 bytes parent folder | download | duplicates (3)
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
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
module Rubyvis
  # Alias for Rubyvis::Wedge
  def self.Wedge
    Rubyvis::Wedge
  end
  class Wedge < Mark
    def type
      "wedge"
    end
    @properties=Mark.properties.dup
    
    attr_accessor_dsl :start_angle, :end_angle, :angle, :inner_radius, :outer_radius, :line_width, [:stroke_style, lambda {|d| Rubyvis.color(d)}], [:fill_style, lambda {|d| Rubyvis.color(d)}]
    
    def self.defaults
      a=Rubyvis.Colors.category20()
      Wedge.new.mark_extend(Mark.defaults).start_angle(lambda  {s=self.sibling; s ? s.end_angle: -Math::PI.quo(2) } ).inner_radius( 0 ).line_width( 1.5 ).stroke_style( nil ).fill_style( lambda {a.scale(self.index)})
    end
    def mid_radius
      (inner_radius+outer_radius) / 2.0
    end
    def mid_angle
      (start_angle+end_angle) / 2.0
    end
    
    def anchor(name)
      that=self
      partial=lambda {|s| s.inner_radius!=0 ? true : s.angle < 2*Math::PI}
      mid_radius=lambda {|s| (s.inner_radius+s.outer_radius) / 2.0}
      mid_angle=lambda {|s| (s.start_angle+s.end_angle) / 2.0 }
      
      mark_anchor(name).left(lambda {
        s = self.scene.target[self.index];
        if (partial.call(s))
          case (self.name()) 
          when "outer"
            s.left + s.outer_radius * Math.cos(mid_angle.call(s))
          when "inner"
            s.left + s.inner_radius * Math.cos(mid_angle.call(s))
          when "start"
            s.left + mid_radius.call(s) * Math.cos(s.start_angle)
          when "center"
            s.left + mid_radius.call(s) * Math.cos(mid_angle.call(s))
          when "end"
            s.left + mid_radius.call(s) * Math.cos(s.end_angle)
          else
            s.left
          end
        else
          s.left
        end        
      }).top(lambda {
        s = self.scene.target[self.index];
        if (partial.call(s))
          case (self.name()) 
          when "outer"
            s.top + s.outer_radius * Math.sin(mid_angle.call(s))
          when "inner"
            s.top + s.inner_radius * Math.sin(mid_angle.call(s))
          when "start"
            s.top + mid_radius.call(s) * Math.sin(s.start_angle)
          when "center"
            s.top + mid_radius.call(s) * Math.sin(mid_angle.call(s))
          when "end"
            s.top + mid_radius.call(s) * Math.sin(s.end_angle)
          else
            s.top
          end
        else
          s.top
        end
          
      }).text_align(lambda {
      s = self.scene.target[self.index];
        if (partial.call(s))
          case (self.name()) 
          when 'outer'
            that.upright(mid_angle.call(s)) ? 'right':'left'
          when 'inner'
            that.upright(mid_angle.call(s)) ? 'left':'right'
          else 
            'center'
          end
        else
          'center'
        end
      }).text_baseline(lambda {
      s = self.scene.target[self.index];
        if (partial.call(s))
          case (self.name()) 
          when 'start'
            that.upright(s.start_angle) ? 'top':'bottom'
          when 'end'
            that.upright(s.end_angle) ? 'bottom':'top'
          else
            'middle'
          end
        else
          'middle'
        end
      }).text_angle(lambda {
        s = self.scene.target[self.index];
        a=0
        if (partial.call(s))
          case (self.name()) 
          when 'center'
            a=mid_angle.call(s)
          when 'inner'
            a=mid_angle.call(s)
          when 'outer'
            a=mid_angle.call(s)
          when 'start'
            a=s.start_angle
          when 'end'
            a=s.end_angle
          end
        end
        that.upright(a) ? a: (a+Math::PI) 
      })
      
    end
    
    def self.upright(angle)
      angle=angle % (2*Math::PI)
      angle=(angle<0) ? (2*Math::PI+angle) : angle
      (angle < Math::PI/2.0) or (angle>=3*Math::PI / 2.0)  
      
    end
    def upright(angle)
      angle=angle % (2*Math::PI)
      angle=(angle<0) ? (2*Math::PI+angle) : angle
      (angle < Math::PI/2.0) or (angle>=3*Math::PI / 2.0)  
    end
    
    def build_implied(s)
      
      if s.angle.nil?
        s.angle = s.end_angle - s.start_angle 
      elsif s.end_angle.nil?
        s.end_angle = s.start_angle + s.angle
      end
      
      mark_build_implied(s)
    end
  end
end