X-Git-Url: https://projects.mako.cc/source/selectricity-live/blobdiff_plain/09c67d9323c7d379d4c6de5dc0457b6de16cff14..45d47676ca3cc1c19b457f83a9482ba8b906f829:/app/controllers/graph_controller.rb diff --git a/app/controllers/graph_controller.rb b/app/controllers/graph_controller.rb index 7b593cc..17e1578 100644 --- a/app/controllers/graph_controller.rb +++ b/app/controllers/graph_controller.rb @@ -1,24 +1,52 @@ require 'date' class GraphController < ApplicationController + + class GruffGraff + + def initialize(options) + size = "700x400" + @graph = options[:graph_type].new(size) + + @graph.theme = { :background_colors => ['#73BF26', '#ffffff'] } + @graph.font = File.expand_path('/usr/X11R6/lib/X11/fonts/TTF/Vera.ttf', + RAILS_ROOT) + + # fill in the data with the optional data name + @graph.data( options.fetch(:data_name, "Data"), options[:data] ) + + # set the labels or create an empty hash + @graph.labels = options[:interval_labels] \ + if options.has_key?(:interval_labels) and \ + options[:interval_labels].class == Hash + @graph.x_axis_label = options[:x_axis_label] \ + if options.has_key?(:x_axis_label) + @graph.y_axis_label = options[:y_axis_label] \ + if options.has_key?(:y_axis_label) + @graph.title = options[:title] if options.has_key?(:title) + + @graph.minimum_value = 0.0 + + end + + def output + return([@graph.to_blob, {:disposition => 'inline', :type => 'image/png'}]) + end + + end + # produce a graph of votes per day during an election def votes_per_day @election = Election.find(params[:id]) data, labels = get_votes_per_day_data(@election) - line = Gruff::Line.new("700x400") - line.theme = { :background_colors => ['#73BF26', '#ffffff'] } - line.title = "Voters Per Day" - line.font = File.expand_path('/usr/X11R6/lib/X11/fonts/TTF/Vera.ttf', - RAILS_ROOT) - - line.data( "#{@election.name}", data ) - line.labels = labels - - line.x_axis_label = "Date" - line.y_axis_label = "Number of Votes" - line.minimum_value = 0.0 - - send_data(line.to_blob, :disposition => 'inline', :type => 'image/png') + graph = GruffGraff.new( :graph_type => Gruff::Line, + :data_name => @election.name, + :data => data, + :interval_labels => labels, + :title => "Voters Per Day", + :x_axis_label => "Data", + :y_axis_label =>"Number of Votes") + send_data(*graph.output) end #will place votes in a fixed number of intervals, and shows votes over time @@ -26,26 +54,16 @@ class GraphController < ApplicationController @election = Election.find(params[:id]) data, labels, scale = get_votes_per_interval_data(@election) - line = Gruff::Line.new("700x400") - line.theme = { :background_colors => ['#73BF26', '#ffffff'] } - line.title = "Voters Over Time" - line.font = File.expand_path('/usr/X11R6/lib/X11/fonts/TTF/Vera.ttf', - RAILS_ROOT) - - line.data("#{@election.name}", data ) - line.labels = labels - - line.x_axis_label = scale - line.y_axis_label = "Number of Votes" - line.minimum_value = 0.0 - - send_data(line.to_blob, :disposition => 'inline', :type => 'image/png') - end - - def quickvote_bar - @election = Election.find(params[:id]) + graph = GruffGraff.new( :graph_type => Gruff::Line, + :data_name => @election.name, + :data => data, + :interval_labels => labels, + :title => "Voters Over Time", + :x_axis_label => scale, + :y_axis_label => "Number of Votes") + send_data(*graph.output) end - + def borda_bar @election = Election.find(params[:id]) pref_tally = make_preference_tally(@election) @@ -53,22 +71,79 @@ class GraphController < ApplicationController @borda_result = BordaVote.new(pref_tally).result data, labels = get_borda_points(@borda_result) - bar = Gruff::Bar.new("700x400") - bar.theme = { :background_colors => ['#73BF26', '#ffffff'] } - bar.title = "Points Per Candidate" - bar.font = File.expand_path('/usr/X11R6/lib/X11/fonts/TTF/Vera.ttf', - RAILS_ROOT) - - bar.data("#{@election.name}", data) - bar.labels = labels - - bar.y_axis_label = "Points" - bar.x_axis_label = "Candidate" - bar.minimum_value = 0.0 - send_data(bar.to_blob, :disposition => 'inline', :type => 'image/png') + graph = GruffGraff.new( :graph_type => Gruff::Bar, + :data_name => @election.name, + :data => data, + :interval_labels => labels, + :title => "Points Per Candidate", + :y_axis_label => "Points", + :x_axis_label => "Candidate") + send_data(*graph.output) end + def choices_positions + @election = Election.find(params[:id]) + pref_tally = make_preference_tally(@election) + + fulldata = get_positions_info(@election) + (0...@election.candidates.size).each do |i| + sbar.data("Candidate#{i+1}", fulldata[i]) + end + + sbar = Gruff::Bar.new + sbar.title = "Times Voted in Each Position" + #sbar.theme = { :background_colors => ['#73BF26', '#ffffff'] } + sbar.font = File.expand_path('/usr/X11R6/lib/X11/fonts/TTF/Vera.ttf', + RAILS_ROOT) + + graph = GruffGraff.new( :graph_type => Gruff::Bar, + :data_name => @election.name, + :data => data, + :interval_labels => labels, + :title => "Times Candidate Was Voted Each Position", + :x_axis_label => "Rank", + :y_axis_label => "Number of Times") + + + + + sbar.x_axis_label = "Rank" + sbar.y_axis_label = "Number of Times Ranked" + + sbar.minimum_value = 0.0 + send_data(sbar.to_blob, :disposition => 'inline', :type => 'image/png') + + + #data, labels = get_positions_info(@election) + + end private + + def get_positions_info(election) + buckets = Hash.new + buckets2= Hash.new + ranks = Hash.new + + election.candidates.each do |candidate| + buckets[candidate.id] = [] + buckets2[candidate.id] = [] + end + + election.votes.each do |vote| + vote.rankings.each do |ranking| + buckets[ranking.candidate_id] << ranking.rank + end + end + + buckets.each_pair do |id, array| + (1..election.candidates.size).each do |i| + buckets2[id] << (array.find_all {|rank| rank == i}).size + end + end + + return buckets2.values + + end # generate the data and labels for each graph def get_votes_per_day_data(election) @@ -132,7 +207,7 @@ class GraphController < ApplicationController # Put votes into bucket according to the time interval to which they belong, # referenced by their key - # Will build a graph over time, as each successive interval wil lhave more + # Will build a graph over time, as each successive interval will have more # vote objects election.votes.each do |vote| buckets.keys.sort.each do |inter| @@ -150,17 +225,17 @@ class GraphController < ApplicationController labels_hash[0] = starttime.min.to_s labels_hash[(numcols/2)-1] = (starttime + (timedelta/2)).min.to_s labels_hash[numcols-1] = Time.now.min.to_s - interval_type = "Minutes" + interval_type = "Minute of the Hour" elsif timedelta < 2.days #more than 2 hours means use hours for labels labels_hash[0] = starttime.hour.to_s labels_hash[(numcols/2)-1] = (starttime + (timedelta/2)).hour.to_s labels_hash[numcols-1] = Time.now.hour.to_s - interval_type = "Hours" + interval_type = "Hour of the Day on 24 hour scale" else #more than 2 days means use dates for labels labels_hash[0] = (Date.parse(starttime.to_s)).to_s - labels_hash[(numcols/2)-1] = (Date.parse(starttime + (timedelta/2))).to_s + labels_hash[(numcols/2)-1] = (Date.parse((starttime + (timedelta/2)).to_s)).to_s labels_hash[numcols-1] = (Date.today).to_s - interval_type = "Days" + interval_type = "The Date" end # Make sure to return an array for data and hash for labels