save frequency data as csv

Author: digininja

checkers_available/frequency.rb

@@ -9,23 +9,28 @@ class Frequency_Checker < Checker
 
 	def initialize
 		super
-		@description = "Frequency Checks"
+		@description = "Count the frequency of characters in each position, output as either CSV or text"
 
 		@frequencies = []
 		@position_count = []
 
 		@show_empty = false
+		@as_csv = false
 
-		@cli_params = [['--freq.show_empty', GetoptLong::NO_ARGUMENT]]
+		@cli_params = [['--freq.show_empty', GetoptLong::NO_ARGUMENT], ['--freq.as_csv', GetoptLong::NO_ARGUMENT]]
 	end
 
 	def usage
-		return "\t--freq.show_empty: show empty frequencies"
+		ret_str = "\t--freq.show_empty: show empty frequencies - does not apply to CSV\n"
+		ret_str << "\t--freq.as_csv: output in CSV format"
+		return ret_str
 	end
 
 	def parse_params opts
 		opts.each do |opt, arg|
 			case opt
+				when '--freq.as_csv'
+					@as_csv = true
 				when '--freq.show_empty'
 					@show_empty = true
 			end
@@ -36,20 +41,8 @@ def parse_params opts
 	def setup_new_frequency position 
 		@frequencies[position] = {}
 
-		"a".upto("z") do |a|
-			@frequencies[position][a] = 0
-		end
-
-		"A".upto("Z") do |a|
-			@frequencies[position][a] = 0
-		end
-
-		"0".upto("9") do |a|
-			@frequencies[position][a] = 0
-		end
-
-		punct = "čćžšđČĆŽŠĐ!\"$%^&*()-=_+[]{};':@,.<>"
-		punct.each_char do |a|
+		alphabet = "0123456789abcčćđdefghijklmnopqrsštuvwxyzžABCČĆDĐEFGHIJKLMNOPQRSŠTUVWXYZŽ !\"\#$%&'()*+,-./:;<=>?@[\\]^_`{|}~äüęé"
+		alphabet.each_char do |a|
 			@frequencies[position][a] = 0
 		end
 
@@ -80,22 +73,56 @@ def process_word (word, extras = nil)
 	end
 
 	def get_results()
-		ret_str = "Frequency Results\n"
-		ret_str << "-----------------\n"
-		
-		position = 0
-		@frequencies.each do |data|
-			ret_str << "position: #{position} - #{@position_count[position]} character#{(@position_count[position]>1)?'s':''}\n"
-			@frequencies[position].each_pair do |key, count|
-				if (@show_empty or count != 0)
-					ret_str << "#{key}: #{count} (#{(count.to_f/@position_count[position]) * 100}%) "
+		if @as_csv then
+			if @frequencies.count == 0 then
+				ret_str = "No data collected"
+			else
+				ret_str = ""
+				
+				header = '"Position","' + @frequencies[0].keys.join('","') + '"'
+				header.gsub! /"""/,'"<quote>"'
+				ret_str << header + "\n"
+				ret_str << "Count\n"
+
+				position = 0
+				@frequencies.each do |data|
+					ret_str << position.to_s + "," + data.values.join(',')
+					ret_str << "\n"
+					position += 1
+				end
+				ret_str << "\n\n\n"
+				ret_str << "Percentage\n"
+
+				position = 0
+				@frequencies.each do |data|
+					ret_str << position.to_s + ","
+					@frequencies[position].each_pair do |key, count|
+						ret_str << ((count.to_f/@position_count[position]) * 100).round(2).to_s + ","
+					end
+					ret_str << "\n"
+					position += 1
 				end
 			end
-		#	ret_str << @frequencies[position].inspect
+		else
+			ret_str = "Frequency Results\n"
+			ret_str << "-----------------\n"
+			
+			position = 0
+			@frequencies.each do |data|
+				ret_str << "position: #{position} - #{@position_count[position]} character#{(@position_count[position]>1)?'s':''}\n"
+				@frequencies[position].each_pair do |key, count|
+					if (@show_empty or count != 0)
+						ret_str << "#{key}: #{count} (#{(count.to_f/@position_count[position]) * 100}%) "
+					end
+				end
+			#	ret_str << @frequencies[position].inspect
+				ret_str << "\n"
+				position += 1
+			end
 			ret_str << "\n"
-			position += 1
 		end
-		ret_str << "\n"
+
+		return ret_str
 
 		return ret_str
 	end