aboutsummaryrefslogtreecommitdiffstats
path: root/room-schedule.rb
blob: 40e94aabd1f04cb083979b556abe65729e4615d5 (plain) (blame)
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
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
require 'date'
require 'time'
require 'SVG/Graph/Plot'

module SVG
  module Graph
    class RoomSchedule < Graph
      # In addition to the defaults set by Graph::initialize and
      # Plot::set_defaults, sets:
      # [x_label_format] '%Y-%m-%d %H:%M:%S'
      # [popup_format]  '%Y-%m-%d %H:%M:%S'
      def set_defaults
        init_with(
          :x_label_format     => '%Y-%m-%d %H:%M:%S',
          :popup_format       => '%Y-%m-%d %H:%M:%S',
          :scale_x_divisions  => false,
          :scale_x_integers   => false,
          :bar_gap            => true
        )
      end

      # The format string use do format the X axis labels.
      # See Time::strformat
      attr_accessor :x_label_format
      # Use this to set the spacing between dates on the axis.  The value
      # must be of the form
      # "\d+ ?(days|weeks|months|years|hours|minutes|seconds)?"
      #
      # EG:
      #
      #   graph.timescale_divisions = "2 weeks"
      #
      # will cause the chart to try to divide the X axis up into segments of
      # two week periods.
      attr_accessor :timescale_divisions
      # The formatting used for the popups.  See x_label_format
      attr_accessor :popup_format
      attr_accessor :scale_x_divisions
      attr_accessor :scale_x_integers
      attr_accessor :bar_gap

      protected

      def format x, y
        Time.at( x ).strftime( popup_format )
      end

      def get_x_labels
        rv = get_x_values.collect { |v| v.strftime( x_label_format ) }
      end

      def y_label_offset( height )
        height / -2.0
      end

      def get_y_labels
        @data.map {|room| room[:title] }
      end

      def draw_data
        bargap = bar_gap ? (field_height < 10 ? field_height / 2 : 10) : 0
        subbar_height = field_height - bargap

        field_count = 1
        y_mod = (subbar_height / 2) + (font_size / 2)
        min,max,div = x_range
        scale = (@graph_width) / (max - min).to_f
        @data.each do |room|
          room[:data].each do |entry|
            x_start = (entry.start_time - min).to_f
            x_end = (entry.end_time - min).to_f
            y = @graph_height - (field_height * field_count)
            bar_width = (x_end - x_start) * scale
            bar_start = x_start * scale

            @graph.add_element( "rect", {
              "x" => bar_start.to_s,
              "y" => y.to_s,
              "width" => bar_width.to_s,
              "height" => subbar_height.to_s,
              "class" => "fill#{field_count+1}"
            })
            t = @graph.add_element("text", {
              "x" => (bar_start + @font_size / 2).to_s,
              "y" => (y + 2 * @font_size).to_s,
            })
            t.add_text(entry.title)
          end
          field_count += 1
        end
      end

      def get_css
        return <<EOL
/* default fill styles for multiple datasets (probably only use a single dataset on this graph though) */
.key1,.fill1{
	fill: #ff0000;
	fill-opacity: 0.5;
	stroke: none;
	stroke-width: 0.5px;
}
.key2,.fill2{
	fill: #0000ff;
	fill-opacity: 0.5;
	stroke: none;
	stroke-width: 1px;
}
.key3,.fill3{
	fill: #00ff00;
	fill-opacity: 0.5;
	stroke: none;
	stroke-width: 1px;
}
.key4,.fill4{
	fill: #ffcc00;
	fill-opacity: 0.5;
	stroke: none;
	stroke-width: 1px;
}
.key5,.fill5{
	fill: #00ccff;
	fill-opacity: 0.5;
	stroke: none;
	stroke-width: 1px;
}
.key6,.fill6{
	fill: #ff00ff;
	fill-opacity: 0.5;
	stroke: none;
	stroke-width: 1px;
}
.key7,.fill7{
	fill: #00ffff;
	fill-opacity: 0.5;
	stroke: none;
	stroke-width: 1px;
}
.key8,.fill8{
	fill: #ffff00;
	fill-opacity: 0.5;
	stroke: none;
	stroke-width: 1px;
}
.key9,.fill9{
	fill: #cc6666;
	fill-opacity: 0.5;
	stroke: none;
	stroke-width: 1px;
}
.key10,.fill10{
	fill: #663399;
	fill-opacity: 0.5;
	stroke: none;
	stroke-width: 1px;
}
.key11,.fill11{
	fill: #339900;
	fill-opacity: 0.5;
	stroke: none;
	stroke-width: 1px;
}
.key12,.fill12{
	fill: #9966FF;
	fill-opacity: 0.5;
	stroke: none;
	stroke-width: 1px;
}
EOL
      end

      private

      def timescale_unit
        if timescale_divisions
          timescale_divisions =~ /(\d+) ?(days|weeks|hours|minutes|seconds)?/
          division_units = $2 ? $2 : "days"
          amount = $1.to_i
          unit =  nil
          case division_units
            when "weeks"
              unit = 7
            when "days"
              unit = 1
            when "hours"
              unit = (1.0 / 24)
            when "minutes"
              unit = (1.0 / 1440)
            when "seconds"
              unit = (1.0 / 86400)
          end
        end
        [ amount, unit ]
      end

      def x_range
        max_value = @data.map {|room| room[:data].map {|e| e.end_time}.max}.max
        min_value = @data.map {|room| room[:data].map {|e| e.start_time}.min}.min

        amount, unit = timescale_unit
        max_value += unit * (amount - (max_value.min % amount))
        range = max_value - min_value
        right_pad = 0 #range == 0 ? 10 : range / 20.0
        scale_range = (max_value + right_pad) - min_value

        scale_division = scale_x_divisions || (scale_range / 10.0)

        if scale_x_integers
          scale_division = scale_division < 1 ? 1 : scale_division.round
        end

        [min_value, max_value, scale_division]
      end

      def get_x_values
        rv = []
        min, max, scale_division = x_range
        if timescale_divisions
          amount, unit = timescale_unit
          min.step( max, amount * unit) {|v| rv << v} if amount
          return rv
        end
        min.step(max, scale_division) {|v| rv << v}
        return rv
      end
    end
  end
end