To check out this repository please hg clone the following URL, or open the URL using EasyMercurial or your preferred Mercurial client.

Statistics Download as Zip
| Branch: | Tag: | Revision:

root / lib / redmine / helpers / gantt.rb @ 441:cbce1fd3b1b7

History | View | Annotate | Download (31 KB)

1 0:513646585e45 Chris
# Redmine - project management software
2 441:cbce1fd3b1b7 Chris
# Copyright (C) 2006-2011  Jean-Philippe Lang
3 0:513646585e45 Chris
#
4
# This program is free software; you can redistribute it and/or
5
# modify it under the terms of the GNU General Public License
6
# as published by the Free Software Foundation; either version 2
7
# of the License, or (at your option) any later version.
8 441:cbce1fd3b1b7 Chris
#
9 0:513646585e45 Chris
# This program is distributed in the hope that it will be useful,
10
# but WITHOUT ANY WARRANTY; without even the implied warranty of
11
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12
# GNU General Public License for more details.
13 441:cbce1fd3b1b7 Chris
#
14 0:513646585e45 Chris
# You should have received a copy of the GNU General Public License
15
# along with this program; if not, write to the Free Software
16
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
17
18
module Redmine
19
  module Helpers
20
    # Simple class to handle gantt chart data
21
    class Gantt
22 22:40f7cfd4df19 chris
      include ERB::Util
23
      include Redmine::I18n
24
25
      # :nodoc:
26
      # Some utility methods for the PDF export
27
      class PDF
28
        MaxCharactorsForSubject = 45
29
        TotalWidth = 280
30
        LeftPaneWidth = 100
31
32
        def self.right_pane_width
33
          TotalWidth - LeftPaneWidth
34
        end
35
      end
36
37 119:8661b858af72 Chris
      attr_reader :year_from, :month_from, :date_from, :date_to, :zoom, :months, :truncated, :max_rows
38 22:40f7cfd4df19 chris
      attr_accessor :query
39
      attr_accessor :project
40
      attr_accessor :view
41 441:cbce1fd3b1b7 Chris
42 0:513646585e45 Chris
      def initialize(options={})
43
        options = options.dup
44 441:cbce1fd3b1b7 Chris
45 0:513646585e45 Chris
        if options[:year] && options[:year].to_i >0
46
          @year_from = options[:year].to_i
47
          if options[:month] && options[:month].to_i >=1 && options[:month].to_i <= 12
48
            @month_from = options[:month].to_i
49
          else
50
            @month_from = 1
51
          end
52
        else
53
          @month_from ||= Date.today.month
54
          @year_from ||= Date.today.year
55
        end
56 441:cbce1fd3b1b7 Chris
57 0:513646585e45 Chris
        zoom = (options[:zoom] || User.current.pref[:gantt_zoom]).to_i
58 441:cbce1fd3b1b7 Chris
        @zoom = (zoom > 0 && zoom < 5) ? zoom : 2
59 0:513646585e45 Chris
        months = (options[:months] || User.current.pref[:gantt_months]).to_i
60
        @months = (months > 0 && months < 25) ? months : 6
61 441:cbce1fd3b1b7 Chris
62 0:513646585e45 Chris
        # Save gantt parameters as user preference (zoom and months count)
63
        if (User.current.logged? && (@zoom != User.current.pref[:gantt_zoom] || @months != User.current.pref[:gantt_months]))
64
          User.current.pref[:gantt_zoom], User.current.pref[:gantt_months] = @zoom, @months
65
          User.current.preference.save
66
        end
67 441:cbce1fd3b1b7 Chris
68 0:513646585e45 Chris
        @date_from = Date.civil(@year_from, @month_from, 1)
69
        @date_to = (@date_from >> @months) - 1
70 441:cbce1fd3b1b7 Chris
71 119:8661b858af72 Chris
        @subjects = ''
72
        @lines = ''
73
        @number_of_rows = nil
74 441:cbce1fd3b1b7 Chris
75 119:8661b858af72 Chris
        @issue_ancestors = []
76 441:cbce1fd3b1b7 Chris
77 119:8661b858af72 Chris
        @truncated = false
78
        if options.has_key?(:max_rows)
79
          @max_rows = options[:max_rows]
80
        else
81
          @max_rows = Setting.gantt_items_limit.blank? ? nil : Setting.gantt_items_limit.to_i
82
        end
83 0:513646585e45 Chris
      end
84 22:40f7cfd4df19 chris
85
      def common_params
86
        { :controller => 'gantts', :action => 'show', :project_id => @project }
87 0:513646585e45 Chris
      end
88 441:cbce1fd3b1b7 Chris
89 0:513646585e45 Chris
      def params
90 22:40f7cfd4df19 chris
        common_params.merge({  :zoom => zoom, :year => year_from, :month => month_from, :months => months })
91 0:513646585e45 Chris
      end
92 441:cbce1fd3b1b7 Chris
93 0:513646585e45 Chris
      def params_previous
94 22:40f7cfd4df19 chris
        common_params.merge({:year => (date_from << months).year, :month => (date_from << months).month, :zoom => zoom, :months => months })
95 0:513646585e45 Chris
      end
96 441:cbce1fd3b1b7 Chris
97 0:513646585e45 Chris
      def params_next
98 22:40f7cfd4df19 chris
        common_params.merge({:year => (date_from >> months).year, :month => (date_from >> months).month, :zoom => zoom, :months => months })
99 0:513646585e45 Chris
      end
100 22:40f7cfd4df19 chris
101
      # Returns the number of rows that will be rendered on the Gantt chart
102
      def number_of_rows
103 119:8661b858af72 Chris
        return @number_of_rows if @number_of_rows
104 441:cbce1fd3b1b7 Chris
105
        rows = projects.inject(0) {|total, p| total += number_of_rows_on_project(p)}
106 119:8661b858af72 Chris
        rows > @max_rows ? @max_rows : rows
107 22:40f7cfd4df19 chris
      end
108
109
      # Returns the number of rows that will be used to list a project on
110
      # the Gantt chart.  This will recurse for each subproject.
111
      def number_of_rows_on_project(project)
112 441:cbce1fd3b1b7 Chris
        return 0 unless projects.include?(project)
113 22:40f7cfd4df19 chris
114
        count = 1
115 441:cbce1fd3b1b7 Chris
        count += project_issues(project).size
116
        count += project_versions(project).size
117 22:40f7cfd4df19 chris
        count
118
      end
119
120
      # Renders the subjects of the Gantt chart, the left side.
121
      def subjects(options={})
122 119:8661b858af72 Chris
        render(options.merge(:only => :subjects)) unless @subjects_rendered
123
        @subjects
124 22:40f7cfd4df19 chris
      end
125
126
      # Renders the lines of the Gantt chart, the right side
127
      def lines(options={})
128 119:8661b858af72 Chris
        render(options.merge(:only => :lines)) unless @lines_rendered
129
        @lines
130
      end
131 441:cbce1fd3b1b7 Chris
132
      # Returns issues that will be rendered
133
      def issues
134
        @issues ||= @query.issues(
135
          :include => [:assigned_to, :tracker, :priority, :category, :fixed_version],
136
          :order => "#{Project.table_name}.lft ASC, #{Issue.table_name}.id ASC",
137
          :limit => @max_rows
138
        )
139
      end
140
141
      # Return all the project nodes that will be displayed
142
      def projects
143
        return @projects if @projects
144
145
        ids = issues.collect(&:project).uniq.collect(&:id)
146
        if ids.any?
147
          # All issues projects and their visible ancestors
148
          @projects = Project.visible.all(
149
            :joins => "LEFT JOIN #{Project.table_name} child ON #{Project.table_name}.lft <= child.lft AND #{Project.table_name}.rgt >= child.rgt",
150
            :conditions => ["child.id IN (?)", ids],
151
            :order => "#{Project.table_name}.lft ASC"
152
          ).uniq
153
        else
154
          @projects = []
155
        end
156
      end
157
158
      # Returns the issues that belong to +project+
159
      def project_issues(project)
160
        @issues_by_project ||= issues.group_by(&:project)
161
        @issues_by_project[project] || []
162
      end
163
164
      # Returns the distinct versions of the issues that belong to +project+
165
      def project_versions(project)
166
        project_issues(project).collect(&:fixed_version).compact.uniq
167
      end
168
169
      # Returns the issues that belong to +project+ and are assigned to +version+
170
      def version_issues(project, version)
171
        project_issues(project).select {|issue| issue.fixed_version == version}
172
      end
173
174 119:8661b858af72 Chris
      def render(options={})
175 441:cbce1fd3b1b7 Chris
        options = {:top => 0, :top_increment => 20, :indent_increment => 20, :render => :subject, :format => :html}.merge(options)
176
        indent = options[:indent] || 4
177
178 119:8661b858af72 Chris
        @subjects = '' unless options[:only] == :lines
179
        @lines = '' unless options[:only] == :subjects
180
        @number_of_rows = 0
181 441:cbce1fd3b1b7 Chris
182
        Project.project_tree(projects) do |project, level|
183
          options[:indent] = indent + level * options[:indent_increment]
184
          render_project(project, options)
185
          break if abort?
186 22:40f7cfd4df19 chris
        end
187 441:cbce1fd3b1b7 Chris
188 119:8661b858af72 Chris
        @subjects_rendered = true unless options[:only] == :lines
189
        @lines_rendered = true unless options[:only] == :subjects
190 441:cbce1fd3b1b7 Chris
191 119:8661b858af72 Chris
        render_end(options)
192 22:40f7cfd4df19 chris
      end
193
194
      def render_project(project, options={})
195 119:8661b858af72 Chris
        subject_for_project(project, options) unless options[:only] == :lines
196
        line_for_project(project, options) unless options[:only] == :subjects
197 441:cbce1fd3b1b7 Chris
198 22:40f7cfd4df19 chris
        options[:top] += options[:top_increment]
199
        options[:indent] += options[:indent_increment]
200 119:8661b858af72 Chris
        @number_of_rows += 1
201
        return if abort?
202 441:cbce1fd3b1b7 Chris
203
        issues = project_issues(project).select {|i| i.fixed_version.nil?}
204 119:8661b858af72 Chris
        sort_issues!(issues)
205 22:40f7cfd4df19 chris
        if issues
206 119:8661b858af72 Chris
          render_issues(issues, options)
207
          return if abort?
208 22:40f7cfd4df19 chris
        end
209
210 441:cbce1fd3b1b7 Chris
        versions = project_versions(project)
211
        versions.each do |version|
212
          render_version(project, version, options)
213 22:40f7cfd4df19 chris
        end
214
215
        # Remove indent to hit the next sibling
216
        options[:indent] -= options[:indent_increment]
217
      end
218
219
      def render_issues(issues, options={})
220 119:8661b858af72 Chris
        @issue_ancestors = []
221 441:cbce1fd3b1b7 Chris
222 22:40f7cfd4df19 chris
        issues.each do |i|
223 119:8661b858af72 Chris
          subject_for_issue(i, options) unless options[:only] == :lines
224
          line_for_issue(i, options) unless options[:only] == :subjects
225 441:cbce1fd3b1b7 Chris
226 22:40f7cfd4df19 chris
          options[:top] += options[:top_increment]
227 119:8661b858af72 Chris
          @number_of_rows += 1
228
          break if abort?
229 22:40f7cfd4df19 chris
        end
230 441:cbce1fd3b1b7 Chris
231 119:8661b858af72 Chris
        options[:indent] -= (options[:indent_increment] * @issue_ancestors.size)
232 22:40f7cfd4df19 chris
      end
233
234 441:cbce1fd3b1b7 Chris
      def render_version(project, version, options={})
235 22:40f7cfd4df19 chris
        # Version header
236 119:8661b858af72 Chris
        subject_for_version(version, options) unless options[:only] == :lines
237
        line_for_version(version, options) unless options[:only] == :subjects
238 441:cbce1fd3b1b7 Chris
239 22:40f7cfd4df19 chris
        options[:top] += options[:top_increment]
240 119:8661b858af72 Chris
        @number_of_rows += 1
241
        return if abort?
242 441:cbce1fd3b1b7 Chris
243
        issues = version_issues(project, version)
244 22:40f7cfd4df19 chris
        if issues
245 119:8661b858af72 Chris
          sort_issues!(issues)
246 22:40f7cfd4df19 chris
          # Indent issues
247
          options[:indent] += options[:indent_increment]
248 119:8661b858af72 Chris
          render_issues(issues, options)
249 22:40f7cfd4df19 chris
          options[:indent] -= options[:indent_increment]
250
        end
251 119:8661b858af72 Chris
      end
252 441:cbce1fd3b1b7 Chris
253 119:8661b858af72 Chris
      def render_end(options={})
254
        case options[:format]
255 441:cbce1fd3b1b7 Chris
        when :pdf
256 119:8661b858af72 Chris
          options[:pdf].Line(15, options[:top], PDF::TotalWidth, options[:top])
257
        end
258 22:40f7cfd4df19 chris
      end
259
260
      def subject_for_project(project, options)
261
        case options[:format]
262
        when :html
263 119:8661b858af72 Chris
          subject = "<span class='icon icon-projects #{project.overdue? ? 'project-overdue' : ''}'>"
264
          subject << view.link_to_project(project)
265
          subject << '</span>'
266
          html_subject(options, subject, :css => "project-name")
267 22:40f7cfd4df19 chris
        when :image
268 119:8661b858af72 Chris
          image_subject(options, project.name)
269 22:40f7cfd4df19 chris
        when :pdf
270 119:8661b858af72 Chris
          pdf_new_page?(options)
271
          pdf_subject(options, project.name)
272 22:40f7cfd4df19 chris
        end
273
      end
274
275
      def line_for_project(project, options)
276 37:94944d00e43c chris
        # Skip versions that don't have a start_date or due date
277
        if project.is_a?(Project) && project.start_date && project.due_date
278 22:40f7cfd4df19 chris
          options[:zoom] ||= 1
279
          options[:g_width] ||= (self.date_to - self.date_from + 1) * options[:zoom]
280 441:cbce1fd3b1b7 Chris
281 119:8661b858af72 Chris
          coords = coordinates(project.start_date, project.due_date, nil, options[:zoom])
282
          label = h(project)
283 441:cbce1fd3b1b7 Chris
284 22:40f7cfd4df19 chris
          case options[:format]
285
          when :html
286 119:8661b858af72 Chris
            html_task(options, coords, :css => "project task", :label => label, :markers => true)
287 22:40f7cfd4df19 chris
          when :image
288 119:8661b858af72 Chris
            image_task(options, coords, :label => label, :markers => true, :height => 3)
289 22:40f7cfd4df19 chris
          when :pdf
290 119:8661b858af72 Chris
            pdf_task(options, coords, :label => label, :markers => true, :height => 0.8)
291 22:40f7cfd4df19 chris
          end
292
        else
293
          ActiveRecord::Base.logger.debug "Gantt#line_for_project was not given a project with a start_date"
294
          ''
295
        end
296
      end
297
298
      def subject_for_version(version, options)
299
        case options[:format]
300
        when :html
301 119:8661b858af72 Chris
          subject = "<span class='icon icon-package #{version.behind_schedule? ? 'version-behind-schedule' : ''} #{version.overdue? ? 'version-overdue' : ''}'>"
302
          subject << view.link_to_version(version)
303
          subject << '</span>'
304
          html_subject(options, subject, :css => "version-name")
305 22:40f7cfd4df19 chris
        when :image
306 119:8661b858af72 Chris
          image_subject(options, version.to_s_with_project)
307 22:40f7cfd4df19 chris
        when :pdf
308 119:8661b858af72 Chris
          pdf_new_page?(options)
309
          pdf_subject(options, version.to_s_with_project)
310 22:40f7cfd4df19 chris
        end
311
      end
312
313
      def line_for_version(version, options)
314
        # Skip versions that don't have a start_date
315 37:94944d00e43c chris
        if version.is_a?(Version) && version.start_date && version.due_date
316 22:40f7cfd4df19 chris
          options[:zoom] ||= 1
317
          options[:g_width] ||= (self.date_to - self.date_from + 1) * options[:zoom]
318 441:cbce1fd3b1b7 Chris
319 119:8661b858af72 Chris
          coords = coordinates(version.start_date, version.due_date, version.completed_pourcent, options[:zoom])
320
          label = "#{h version } #{h version.completed_pourcent.to_i.to_s}%"
321
          label = h("#{version.project} -") + label unless @project && @project == version.project
322 22:40f7cfd4df19 chris
323
          case options[:format]
324
          when :html
325 119:8661b858af72 Chris
            html_task(options, coords, :css => "version task", :label => label, :markers => true)
326 22:40f7cfd4df19 chris
          when :image
327 119:8661b858af72 Chris
            image_task(options, coords, :label => label, :markers => true, :height => 3)
328 22:40f7cfd4df19 chris
          when :pdf
329 119:8661b858af72 Chris
            pdf_task(options, coords, :label => label, :markers => true, :height => 0.8)
330 22:40f7cfd4df19 chris
          end
331
        else
332
          ActiveRecord::Base.logger.debug "Gantt#line_for_version was not given a version with a start_date"
333
          ''
334
        end
335
      end
336
337
      def subject_for_issue(issue, options)
338 119:8661b858af72 Chris
        while @issue_ancestors.any? && !issue.is_descendant_of?(@issue_ancestors.last)
339
          @issue_ancestors.pop
340
          options[:indent] -= options[:indent_increment]
341
        end
342 441:cbce1fd3b1b7 Chris
343 119:8661b858af72 Chris
        output = case options[:format]
344 22:40f7cfd4df19 chris
        when :html
345 119:8661b858af72 Chris
          css_classes = ''
346
          css_classes << ' issue-overdue' if issue.overdue?
347
          css_classes << ' issue-behind-schedule' if issue.behind_schedule?
348
          css_classes << ' icon icon-issue' unless Setting.gravatar_enabled? && issue.assigned_to
349 441:cbce1fd3b1b7 Chris
350 119:8661b858af72 Chris
          subject = "<span class='#{css_classes}'>"
351
          if issue.assigned_to.present?
352
            assigned_string = l(:field_assigned_to) + ": " + issue.assigned_to.name
353 245:051f544170fe Chris
            subject << view.avatar(issue.assigned_to, :class => 'gravatar icon-gravatar', :size => 10, :title => assigned_string).to_s
354 119:8661b858af72 Chris
          end
355
          subject << view.link_to_issue(issue)
356
          subject << '</span>'
357 245:051f544170fe Chris
          html_subject(options, subject, :css => "issue-subject", :title => issue.subject) + "\n"
358 119:8661b858af72 Chris
        when :image
359
          image_subject(options, issue.subject)
360
        when :pdf
361
          pdf_new_page?(options)
362
          pdf_subject(options, issue.subject)
363
        end
364 22:40f7cfd4df19 chris
365 119:8661b858af72 Chris
        unless issue.leaf?
366
          @issue_ancestors << issue
367
          options[:indent] += options[:indent_increment]
368
        end
369 441:cbce1fd3b1b7 Chris
370 119:8661b858af72 Chris
        output
371 22:40f7cfd4df19 chris
      end
372
373
      def line_for_issue(issue, options)
374
        # Skip issues that don't have a due_before (due_date or version's due_date)
375
        if issue.is_a?(Issue) && issue.due_before
376 119:8661b858af72 Chris
          coords = coordinates(issue.start_date, issue.due_before, issue.done_ratio, options[:zoom])
377
          label = "#{ issue.status.name } #{ issue.done_ratio }%"
378 441:cbce1fd3b1b7 Chris
379 22:40f7cfd4df19 chris
          case options[:format]
380
          when :html
381 119:8661b858af72 Chris
            html_task(options, coords, :css => "task " + (issue.leaf? ? 'leaf' : 'parent'), :label => label, :issue => issue, :markers => !issue.leaf?)
382 22:40f7cfd4df19 chris
          when :image
383 119:8661b858af72 Chris
            image_task(options, coords, :label => label)
384 22:40f7cfd4df19 chris
          when :pdf
385 119:8661b858af72 Chris
            pdf_task(options, coords, :label => label)
386
        end
387 22:40f7cfd4df19 chris
        else
388
          ActiveRecord::Base.logger.debug "GanttHelper#line_for_issue was not given an issue with a due_before"
389
          ''
390
        end
391
      end
392
393 0:513646585e45 Chris
      # Generates a gantt image
394
      # Only defined if RMagick is avalaible
395 22:40f7cfd4df19 chris
      def to_image(format='PNG')
396 441:cbce1fd3b1b7 Chris
        date_to = (@date_from >> @months)-1
397 0:513646585e45 Chris
        show_weeks = @zoom > 1
398
        show_days = @zoom > 2
399 441:cbce1fd3b1b7 Chris
400 1:cca12e1c1fd4 Chris
        subject_width = 400
401 441:cbce1fd3b1b7 Chris
        header_height = 18
402 0:513646585e45 Chris
        # width of one day in pixels
403
        zoom = @zoom*2
404
        g_width = (@date_to - @date_from + 1)*zoom
405 22:40f7cfd4df19 chris
        g_height = 20 * number_of_rows + 30
406 441:cbce1fd3b1b7 Chris
        headers_height = (show_weeks ? 2*header_height : header_height)
407
        height = g_height + headers_height
408
409 0:513646585e45 Chris
        imgl = Magick::ImageList.new
410
        imgl.new_image(subject_width+g_width+1, height)
411
        gc = Magick::Draw.new
412 441:cbce1fd3b1b7 Chris
413 0:513646585e45 Chris
        # Subjects
414 119:8661b858af72 Chris
        gc.stroke('transparent')
415 441:cbce1fd3b1b7 Chris
        subjects(:image => gc, :top => (headers_height + 20), :indent => 4, :format => :image)
416
417 0:513646585e45 Chris
        # Months headers
418
        month_f = @date_from
419
        left = subject_width
420 441:cbce1fd3b1b7 Chris
        @months.times do
421 0:513646585e45 Chris
          width = ((month_f >> 1) - month_f) * zoom
422
          gc.fill('white')
423
          gc.stroke('grey')
424
          gc.stroke_width(1)
425
          gc.rectangle(left, 0, left + width, height)
426
          gc.fill('black')
427
          gc.stroke('transparent')
428
          gc.stroke_width(1)
429
          gc.text(left.round + 8, 14, "#{month_f.year}-#{month_f.month}")
430
          left = left + width
431
          month_f = month_f >> 1
432
        end
433 441:cbce1fd3b1b7 Chris
434 0:513646585e45 Chris
        # Weeks headers
435
        if show_weeks
436
                left = subject_width
437 441:cbce1fd3b1b7 Chris
                height = header_height
438 0:513646585e45 Chris
                if @date_from.cwday == 1
439
                    # date_from is monday
440
                week_f = date_from
441
                else
442
                    # find next monday after date_from
443
                        week_f = @date_from + (7 - @date_from.cwday + 1)
444
                        width = (7 - @date_from.cwday + 1) * zoom
445
                gc.fill('white')
446
                gc.stroke('grey')
447
                gc.stroke_width(1)
448 441:cbce1fd3b1b7 Chris
                gc.rectangle(left, header_height, left + width, 2*header_height + g_height-1)
449 0:513646585e45 Chris
                        left = left + width
450
                end
451
                while week_f <= date_to
452
                        width = (week_f + 6 <= date_to) ? 7 * zoom : (date_to - week_f + 1) * zoom
453
                gc.fill('white')
454
                gc.stroke('grey')
455
                gc.stroke_width(1)
456 441:cbce1fd3b1b7 Chris
                gc.rectangle(left.round, header_height, left.round + width, 2*header_height + g_height-1)
457 0:513646585e45 Chris
                gc.fill('black')
458
                gc.stroke('transparent')
459
                gc.stroke_width(1)
460 441:cbce1fd3b1b7 Chris
                gc.text(left.round + 2, header_height + 14, week_f.cweek.to_s)
461 0:513646585e45 Chris
                        left = left + width
462
                        week_f = week_f+7
463
                end
464
        end
465 441:cbce1fd3b1b7 Chris
466 0:513646585e45 Chris
        # Days details (week-end in grey)
467
        if show_days
468
                left = subject_width
469 441:cbce1fd3b1b7 Chris
                height = g_height + header_height - 1
470 0:513646585e45 Chris
                wday = @date_from.cwday
471 441:cbce1fd3b1b7 Chris
                (date_to - @date_from + 1).to_i.times do
472 0:513646585e45 Chris
              width =  zoom
473
              gc.fill(wday == 6 || wday == 7 ? '#eee' : 'white')
474 119:8661b858af72 Chris
              gc.stroke('#ddd')
475 0:513646585e45 Chris
              gc.stroke_width(1)
476 441:cbce1fd3b1b7 Chris
              gc.rectangle(left, 2*header_height, left + width, 2*header_height + g_height-1)
477 0:513646585e45 Chris
              left = left + width
478
              wday = wday + 1
479
              wday = 1 if wday > 7
480
                end
481
        end
482 441:cbce1fd3b1b7 Chris
483 0:513646585e45 Chris
        # border
484
        gc.fill('transparent')
485
        gc.stroke('grey')
486
        gc.stroke_width(1)
487 441:cbce1fd3b1b7 Chris
        gc.rectangle(0, 0, subject_width+g_width, headers_height)
488 0:513646585e45 Chris
        gc.stroke('black')
489 441:cbce1fd3b1b7 Chris
        gc.rectangle(0, 0, subject_width+g_width, g_height+ headers_height-1)
490
491 0:513646585e45 Chris
        # content
492 441:cbce1fd3b1b7 Chris
        top = headers_height + 20
493 119:8661b858af72 Chris
494
        gc.stroke('transparent')
495 22:40f7cfd4df19 chris
        lines(:image => gc, :top => top, :zoom => zoom, :subject_width => subject_width, :format => :image)
496 441:cbce1fd3b1b7 Chris
497 0:513646585e45 Chris
        # today red line
498
        if Date.today >= @date_from and Date.today <= date_to
499
          gc.stroke('red')
500
          x = (Date.today-@date_from+1)*zoom + subject_width
501 441:cbce1fd3b1b7 Chris
          gc.line(x, headers_height, x, headers_height + g_height-1)
502
        end
503
504 0:513646585e45 Chris
        gc.draw(imgl)
505
        imgl.format = format
506
        imgl.to_blob
507
      end if Object.const_defined?(:Magick)
508 22:40f7cfd4df19 chris
509
      def to_pdf
510 441:cbce1fd3b1b7 Chris
        pdf = ::Redmine::Export::PDF::ITCPDF.new(current_language)
511 22:40f7cfd4df19 chris
        pdf.SetTitle("#{l(:label_gantt)} #{project}")
512 441:cbce1fd3b1b7 Chris
        pdf.alias_nb_pages
513 22:40f7cfd4df19 chris
        pdf.footer_date = format_date(Date.today)
514
        pdf.AddPage("L")
515
        pdf.SetFontStyle('B',12)
516
        pdf.SetX(15)
517 441:cbce1fd3b1b7 Chris
        pdf.RDMCell(PDF::LeftPaneWidth, 20, project.to_s)
518 22:40f7cfd4df19 chris
        pdf.Ln
519
        pdf.SetFontStyle('B',9)
520 441:cbce1fd3b1b7 Chris
521 22:40f7cfd4df19 chris
        subject_width = PDF::LeftPaneWidth
522 441:cbce1fd3b1b7 Chris
        header_height = 5
523
524
        headers_height = header_height
525 22:40f7cfd4df19 chris
        show_weeks = false
526
        show_days = false
527 441:cbce1fd3b1b7 Chris
528 22:40f7cfd4df19 chris
        if self.months < 7
529
          show_weeks = true
530 441:cbce1fd3b1b7 Chris
          headers_height = 2*header_height
531 22:40f7cfd4df19 chris
          if self.months < 3
532
            show_days = true
533 441:cbce1fd3b1b7 Chris
            headers_height = 3*header_height
534 22:40f7cfd4df19 chris
          end
535
        end
536 441:cbce1fd3b1b7 Chris
537 22:40f7cfd4df19 chris
        g_width = PDF.right_pane_width
538
        zoom = (g_width) / (self.date_to - self.date_from + 1)
539
        g_height = 120
540 441:cbce1fd3b1b7 Chris
        t_height = g_height + headers_height
541
542 22:40f7cfd4df19 chris
        y_start = pdf.GetY
543 441:cbce1fd3b1b7 Chris
544 22:40f7cfd4df19 chris
        # Months headers
545
        month_f = self.date_from
546
        left = subject_width
547 441:cbce1fd3b1b7 Chris
        height = header_height
548
        self.months.times do
549
          width = ((month_f >> 1) - month_f) * zoom
550 22:40f7cfd4df19 chris
          pdf.SetY(y_start)
551
          pdf.SetX(left)
552 441:cbce1fd3b1b7 Chris
          pdf.RDMCell(width, height, "#{month_f.year}-#{month_f.month}", "LTR", 0, "C")
553 22:40f7cfd4df19 chris
          left = left + width
554
          month_f = month_f >> 1
555 441:cbce1fd3b1b7 Chris
        end
556
557 22:40f7cfd4df19 chris
        # Weeks headers
558
        if show_weeks
559
          left = subject_width
560 441:cbce1fd3b1b7 Chris
          height = header_height
561 22:40f7cfd4df19 chris
          if self.date_from.cwday == 1
562
            # self.date_from is monday
563
            week_f = self.date_from
564
          else
565
            # find next monday after self.date_from
566
            week_f = self.date_from + (7 - self.date_from.cwday + 1)
567
            width = (7 - self.date_from.cwday + 1) * zoom-1
568 441:cbce1fd3b1b7 Chris
            pdf.SetY(y_start + header_height)
569 22:40f7cfd4df19 chris
            pdf.SetX(left)
570 441:cbce1fd3b1b7 Chris
            pdf.RDMCell(width + 1, height, "", "LTR")
571 22:40f7cfd4df19 chris
            left = left + width+1
572
          end
573
          while week_f <= self.date_to
574
            width = (week_f + 6 <= self.date_to) ? 7 * zoom : (self.date_to - week_f + 1) * zoom
575 441:cbce1fd3b1b7 Chris
            pdf.SetY(y_start + header_height)
576 22:40f7cfd4df19 chris
            pdf.SetX(left)
577 441:cbce1fd3b1b7 Chris
            pdf.RDMCell(width, height, (width >= 5 ? week_f.cweek.to_s : ""), "LTR", 0, "C")
578 22:40f7cfd4df19 chris
            left = left + width
579
            week_f = week_f+7
580
          end
581
        end
582 441:cbce1fd3b1b7 Chris
583 22:40f7cfd4df19 chris
        # Days headers
584
        if show_days
585
          left = subject_width
586 441:cbce1fd3b1b7 Chris
          height = header_height
587 22:40f7cfd4df19 chris
          wday = self.date_from.cwday
588
          pdf.SetFontStyle('B',7)
589 441:cbce1fd3b1b7 Chris
          (self.date_to - self.date_from + 1).to_i.times do
590 22:40f7cfd4df19 chris
            width = zoom
591 441:cbce1fd3b1b7 Chris
            pdf.SetY(y_start + 2 * header_height)
592 22:40f7cfd4df19 chris
            pdf.SetX(left)
593 441:cbce1fd3b1b7 Chris
            pdf.RDMCell(width, height, day_name(wday).first, "LTR", 0, "C")
594 22:40f7cfd4df19 chris
            left = left + width
595
            wday = wday + 1
596
            wday = 1 if wday > 7
597
          end
598
        end
599 441:cbce1fd3b1b7 Chris
600 22:40f7cfd4df19 chris
        pdf.SetY(y_start)
601
        pdf.SetX(15)
602 441:cbce1fd3b1b7 Chris
        pdf.RDMCell(subject_width+g_width-15, headers_height, "", 1)
603
604 22:40f7cfd4df19 chris
        # Tasks
605 441:cbce1fd3b1b7 Chris
        top = headers_height + y_start
606 119:8661b858af72 Chris
        options = {
607
          :top => top,
608
          :zoom => zoom,
609
          :subject_width => subject_width,
610
          :g_width => g_width,
611
          :indent => 0,
612
          :indent_increment => 5,
613
          :top_increment => 5,
614
          :format => :pdf,
615
          :pdf => pdf
616
        }
617
        render(options)
618 22:40f7cfd4df19 chris
        pdf.Output
619
      end
620 441:cbce1fd3b1b7 Chris
621 0:513646585e45 Chris
      private
622 441:cbce1fd3b1b7 Chris
623 119:8661b858af72 Chris
      def coordinates(start_date, end_date, progress, zoom=nil)
624
        zoom ||= @zoom
625 441:cbce1fd3b1b7 Chris
626 119:8661b858af72 Chris
        coords = {}
627
        if start_date && end_date && start_date < self.date_to && end_date > self.date_from
628
          if start_date > self.date_from
629
            coords[:start] = start_date - self.date_from
630
            coords[:bar_start] = start_date - self.date_from
631
          else
632
            coords[:bar_start] = 0
633
          end
634
          if end_date < self.date_to
635
            coords[:end] = end_date - self.date_from
636
            coords[:bar_end] = end_date - self.date_from + 1
637
          else
638
            coords[:bar_end] = self.date_to - self.date_from + 1
639
          end
640 441:cbce1fd3b1b7 Chris
641 119:8661b858af72 Chris
          if progress
642 441:cbce1fd3b1b7 Chris
            progress_date = start_date + (end_date - start_date + 1) * (progress / 100.0)
643 119:8661b858af72 Chris
            if progress_date > self.date_from && progress_date > start_date
644
              if progress_date < self.date_to
645 441:cbce1fd3b1b7 Chris
                coords[:bar_progress_end] = progress_date - self.date_from
646 119:8661b858af72 Chris
              else
647
                coords[:bar_progress_end] = self.date_to - self.date_from + 1
648
              end
649
            end
650 441:cbce1fd3b1b7 Chris
651 119:8661b858af72 Chris
            if progress_date < Date.today
652
              late_date = [Date.today, end_date].min
653
              if late_date > self.date_from && late_date > start_date
654
                if late_date < self.date_to
655
                  coords[:bar_late_end] = late_date - self.date_from + 1
656
                else
657
                  coords[:bar_late_end] = self.date_to - self.date_from + 1
658
                end
659
              end
660
            end
661
          end
662
        end
663 441:cbce1fd3b1b7 Chris
664 119:8661b858af72 Chris
        # Transforms dates into pixels witdh
665
        coords.keys.each do |key|
666
          coords[key] = (coords[key] * zoom).floor
667
        end
668
        coords
669
      end
670 22:40f7cfd4df19 chris
671 119:8661b858af72 Chris
      # Sorts a collection of issues by start_date, due_date, id for gantt rendering
672
      def sort_issues!(issues)
673
        issues.sort! { |a, b| gantt_issue_compare(a, b, issues) }
674
      end
675 441:cbce1fd3b1b7 Chris
676 119:8661b858af72 Chris
      # TODO: top level issues should be sorted by start date
677
      def gantt_issue_compare(x, y, issues)
678
        if x.root_id == y.root_id
679
          x.lft <=> y.lft
680 0:513646585e45 Chris
        else
681 119:8661b858af72 Chris
          x.root_id <=> y.root_id
682
        end
683
      end
684 441:cbce1fd3b1b7 Chris
685 119:8661b858af72 Chris
      def current_limit
686
        if @max_rows
687
          @max_rows - @number_of_rows
688
        else
689
          nil
690
        end
691
      end
692 441:cbce1fd3b1b7 Chris
693 119:8661b858af72 Chris
      def abort?
694
        if @max_rows && @number_of_rows >= @max_rows
695
          @truncated = true
696
        end
697
      end
698 441:cbce1fd3b1b7 Chris
699 119:8661b858af72 Chris
      def pdf_new_page?(options)
700
        if options[:top] > 180
701
          options[:pdf].Line(15, options[:top], PDF::TotalWidth, options[:top])
702
          options[:pdf].AddPage("L")
703
          options[:top] = 15
704
          options[:pdf].Line(15, options[:top] - 0.1, PDF::TotalWidth, options[:top] - 0.1)
705
        end
706
      end
707 441:cbce1fd3b1b7 Chris
708 119:8661b858af72 Chris
      def html_subject(params, subject, options={})
709 245:051f544170fe Chris
        style = "position: absolute;top:#{params[:top]}px;left:#{params[:indent]}px;"
710
        style << "width:#{params[:subject_width] - params[:indent]}px;" if params[:subject_width]
711 441:cbce1fd3b1b7 Chris
712 245:051f544170fe Chris
        output = view.content_tag 'div', subject, :class => options[:css], :style => style, :title => options[:title]
713 119:8661b858af72 Chris
        @subjects << output
714
        output
715
      end
716 441:cbce1fd3b1b7 Chris
717 119:8661b858af72 Chris
      def pdf_subject(params, subject, options={})
718
        params[:pdf].SetY(params[:top])
719
        params[:pdf].SetX(15)
720 441:cbce1fd3b1b7 Chris
721 119:8661b858af72 Chris
        char_limit = PDF::MaxCharactorsForSubject - params[:indent]
722 441:cbce1fd3b1b7 Chris
        params[:pdf].RDMCell(params[:subject_width]-15, 5, (" " * params[:indent]) +  subject.to_s.sub(/^(.{#{char_limit}}[^\s]*\s).*$/, '\1 (...)'), "LR")
723
724 119:8661b858af72 Chris
        params[:pdf].SetY(params[:top])
725
        params[:pdf].SetX(params[:subject_width])
726 441:cbce1fd3b1b7 Chris
        params[:pdf].RDMCell(params[:g_width], 5, "", "LR")
727 119:8661b858af72 Chris
      end
728 441:cbce1fd3b1b7 Chris
729 119:8661b858af72 Chris
      def image_subject(params, subject, options={})
730
        params[:image].fill('black')
731
        params[:image].stroke('transparent')
732
        params[:image].stroke_width(1)
733
        params[:image].text(params[:indent], params[:top] + 2, subject)
734
      end
735 441:cbce1fd3b1b7 Chris
736 119:8661b858af72 Chris
      def html_task(params, coords, options={})
737
        output = ''
738
        # Renders the task bar, with progress and late
739
        if coords[:bar_start] && coords[:bar_end]
740
          output << "<div style='top:#{ params[:top] }px;left:#{ coords[:bar_start] }px;width:#{ coords[:bar_end] - coords[:bar_start] - 2}px;' class='#{options[:css]} task_todo'>&nbsp;</div>"
741 441:cbce1fd3b1b7 Chris
742 119:8661b858af72 Chris
          if coords[:bar_late_end]
743
            output << "<div style='top:#{ params[:top] }px;left:#{ coords[:bar_start] }px;width:#{ coords[:bar_late_end] - coords[:bar_start] - 2}px;' class='#{options[:css]} task_late'>&nbsp;</div>"
744 0:513646585e45 Chris
          end
745 119:8661b858af72 Chris
          if coords[:bar_progress_end]
746
            output << "<div style='top:#{ params[:top] }px;left:#{ coords[:bar_start] }px;width:#{ coords[:bar_progress_end] - coords[:bar_start] - 2}px;' class='#{options[:css]} task_done'>&nbsp;</div>"
747
          end
748
        end
749
        # Renders the markers
750
        if options[:markers]
751
          if coords[:start]
752
            output << "<div style='top:#{ params[:top] }px;left:#{ coords[:start] }px;width:15px;' class='#{options[:css]} marker starting'>&nbsp;</div>"
753
          end
754
          if coords[:end]
755
            output << "<div style='top:#{ params[:top] }px;left:#{ coords[:end] + params[:zoom] }px;width:15px;' class='#{options[:css]} marker ending'>&nbsp;</div>"
756
          end
757
        end
758
        # Renders the label on the right
759
        if options[:label]
760
          output << "<div style='top:#{ params[:top] }px;left:#{ (coords[:bar_end] || 0) + 8 }px;' class='#{options[:css]} label'>"
761
          output << options[:label]
762
          output << "</div>"
763
        end
764
        # Renders the tooltip
765
        if options[:issue] && coords[:bar_start] && coords[:bar_end]
766
          output << "<div class='tooltip' style='position: absolute;top:#{ params[:top] }px;left:#{ coords[:bar_start] }px;width:#{ coords[:bar_end] - coords[:bar_start] }px;height:12px;'>"
767
          output << '<span class="tip">'
768
          output << view.render_issue_tooltip(options[:issue])
769
          output << "</span></div>"
770
        end
771
        @lines << output
772
        output
773
      end
774 441:cbce1fd3b1b7 Chris
775 119:8661b858af72 Chris
      def pdf_task(params, coords, options={})
776
        height = options[:height] || 2
777 441:cbce1fd3b1b7 Chris
778 119:8661b858af72 Chris
        # Renders the task bar, with progress and late
779
        if coords[:bar_start] && coords[:bar_end]
780
          params[:pdf].SetY(params[:top]+1.5)
781
          params[:pdf].SetX(params[:subject_width] + coords[:bar_start])
782
          params[:pdf].SetFillColor(200,200,200)
783 441:cbce1fd3b1b7 Chris
          params[:pdf].RDMCell(coords[:bar_end] - coords[:bar_start], height, "", 0, 0, "", 1)
784
785 119:8661b858af72 Chris
          if coords[:bar_late_end]
786
            params[:pdf].SetY(params[:top]+1.5)
787
            params[:pdf].SetX(params[:subject_width] + coords[:bar_start])
788
            params[:pdf].SetFillColor(255,100,100)
789 441:cbce1fd3b1b7 Chris
            params[:pdf].RDMCell(coords[:bar_late_end] - coords[:bar_start], height, "", 0, 0, "", 1)
790 119:8661b858af72 Chris
          end
791
          if coords[:bar_progress_end]
792
            params[:pdf].SetY(params[:top]+1.5)
793
            params[:pdf].SetX(params[:subject_width] + coords[:bar_start])
794
            params[:pdf].SetFillColor(90,200,90)
795 441:cbce1fd3b1b7 Chris
            params[:pdf].RDMCell(coords[:bar_progress_end] - coords[:bar_start], height, "", 0, 0, "", 1)
796 119:8661b858af72 Chris
          end
797
        end
798
        # Renders the markers
799
        if options[:markers]
800
          if coords[:start]
801
            params[:pdf].SetY(params[:top] + 1)
802
            params[:pdf].SetX(params[:subject_width] + coords[:start] - 1)
803
            params[:pdf].SetFillColor(50,50,200)
804 441:cbce1fd3b1b7 Chris
            params[:pdf].RDMCell(2, 2, "", 0, 0, "", 1)
805 119:8661b858af72 Chris
          end
806
          if coords[:end]
807
            params[:pdf].SetY(params[:top] + 1)
808
            params[:pdf].SetX(params[:subject_width] + coords[:end] - 1)
809
            params[:pdf].SetFillColor(50,50,200)
810 441:cbce1fd3b1b7 Chris
            params[:pdf].RDMCell(2, 2, "", 0, 0, "", 1)
811 119:8661b858af72 Chris
          end
812
        end
813
        # Renders the label on the right
814
        if options[:label]
815
          params[:pdf].SetX(params[:subject_width] + (coords[:bar_end] || 0) + 5)
816 441:cbce1fd3b1b7 Chris
          params[:pdf].RDMCell(30, 2, options[:label])
817 0:513646585e45 Chris
        end
818
      end
819 22:40f7cfd4df19 chris
820 119:8661b858af72 Chris
      def image_task(params, coords, options={})
821
        height = options[:height] || 6
822 441:cbce1fd3b1b7 Chris
823 119:8661b858af72 Chris
        # Renders the task bar, with progress and late
824
        if coords[:bar_start] && coords[:bar_end]
825
          params[:image].fill('#aaa')
826
          params[:image].rectangle(params[:subject_width] + coords[:bar_start], params[:top], params[:subject_width] + coords[:bar_end], params[:top] - height)
827 441:cbce1fd3b1b7 Chris
828 119:8661b858af72 Chris
          if coords[:bar_late_end]
829
            params[:image].fill('#f66')
830
            params[:image].rectangle(params[:subject_width] + coords[:bar_start], params[:top], params[:subject_width] + coords[:bar_late_end], params[:top] - height)
831
          end
832
          if coords[:bar_progress_end]
833
            params[:image].fill('#00c600')
834
            params[:image].rectangle(params[:subject_width] + coords[:bar_start], params[:top], params[:subject_width] + coords[:bar_progress_end], params[:top] - height)
835
          end
836
        end
837
        # Renders the markers
838
        if options[:markers]
839
          if coords[:start]
840
            x = params[:subject_width] + coords[:start]
841
            y = params[:top] - height / 2
842
            params[:image].fill('blue')
843
            params[:image].polygon(x-4, y, x, y-4, x+4, y, x, y+4)
844
          end
845
          if coords[:end]
846
            x = params[:subject_width] + coords[:end] + params[:zoom]
847
            y = params[:top] - height / 2
848
            params[:image].fill('blue')
849
            params[:image].polygon(x-4, y, x, y-4, x+4, y, x, y+4)
850
          end
851
        end
852
        # Renders the label on the right
853
        if options[:label]
854
          params[:image].fill('black')
855
          params[:image].text(params[:subject_width] + (coords[:bar_end] || 0) + 5,params[:top] + 1, options[:label])
856
        end
857
      end
858 0:513646585e45 Chris
    end
859
  end
860
end