lib/lisp.rb @@ -19,6 +19,7 @@ require 'active_record' #custom requires require 'kml' +require 'lib/lisp' def get_db_conn(env) env = env.to_sym initialize.rb @@ -126,7 +126,7 @@ class Co2ColorCodeBar < Processing::App end end - save "#{GTRON_ROOT}/output/co2_color_bar.png" + save $co2_color_bar_file # close window and quit the method # There might be a cleaner way lib/co2_color_code.rb @@ -15,8 +15,8 @@ class DataPoint < ActiveRecord::Base dp end - co2 = File.read(Dir.glob("#{dir_name}/lear*.txt").first).split(/\r?\n/) - co2 = co2[1...co2.size].map{|x| x.split(/,\s+/)[2].to_f } + co2 = cdr(File.read(Dir.glob("#{dir_name}/lear*.txt").first).split(/\r?\n/)) + co2 = co2.map{|x| x.split(/,\s+/)[2].to_f } data_points = data_point_coords.map do |dp| if dp.altitude > 0 dp.co2_ppm = co2.shift @@ -24,7 +24,8 @@ class DataPoint < ActiveRecord::Base dp end data_points.each{|dp| dp.save } - puts "DROPPING #{co2.size} PIECES OF VERY IMPORTANT DATA" if co2.size + ActiveRecord::Base.logger.warn( + "DROPPING #{co2.size} PIECES OF VERY IMPORTANT DATA") if co2.size data_points end models/data_point.rb @@ -1,5 +1,6 @@ desc "Tests out my shape drawing code by drawing simple shapes" task :draw_simple_shapes do + output_path = ENV["OUTPUT_PATH"] || "#{GTRON_ROOT}/output" kml = KMLFile.new doc = KML::Document.new(:name => "Simple Shapes") circle_p = KML::Placemark.new(:name => "Circle") @@ -24,5 +25,5 @@ task :draw_simple_shapes do doc.features << circle_p << cylinder_p << square_p << square1_p kml.objects << doc puts kml - File.open("output/simple_shapes.kml", "w") {|f| f.write kml.render } + File.open("#{output_path}/simple_shapes.kml", "w") {|f| f.write kml.render } end tasks/draw_simple_shapes.rake @@ -1,7 +1,9 @@ desc "Import data into the database" task :import do get_db_conn(ENV["GTRON_ENV"] || GTRON_ENV) + input_path = ENV["INPUT_PATH"] || "#{GTRON_ROOT}/input" + Flight.delete_all DataPoint.delete_all - Flight.load(Dir.glob("#{GTRON_ROOT}/input/*")) + Flight.load(Dir.glob("#{input_path}/*")) end tasks/import.rake @@ -1,7 +1,9 @@ desc "Creates Color Bar for Google Earth" task :make_color_bar do + output_path = ENV["OUTPUT_PATH"] || "#{GTRON_ROOT}/output" + $co2_color_bar_file = File.expand_path("#{output_path}/co2_color_bar.png") Co2ColorCode.make_color_bar - image_path = File.expand_path("#{GTRON_ROOT}/output/co2_color_bar.png") + image_path = $co2_color_bar_file kml = KMLFile.new doc = KML::Document.new(:name => "ASCENDS CO2 Color Bar") overlay = KML::ScreenOverlay.new(:name => 'CO2 Color Bar', @@ -11,5 +13,5 @@ task :make_color_bar do :href => "file:/#{image_path}")) doc.features << overlay kml.objects << doc - File.open("output/co2_color_bar.kml", "w") {|f| f.write kml.render } + File.open("#{output_path}/co2_color_bar.kml", "w") {|f| f.write kml.render } end tasks/make_color_bar.rake @@ -1,5 +1,6 @@ desc "Plots the flightpath in Google Earth" task :plot_datapoint_paths do + output_path = ENV["OUTPUT_PATH"] || "#{GTRON_ROOT}/output" get_db_conn(GTRON_ENV) kml = KMLFile.new doc = KML::Document.new(:name => "ASCENDS Flight Paths") @@ -16,5 +17,5 @@ task :plot_datapoint_paths do doc.features << style << placemark kml.objects << doc - File.open("output/datapoint_paths.kml","w") {|f| f.write kml.render } + File.open("#{output_path}/datapoint_paths.kml","w") {|f| f.write kml.render } end tasks/plot_datapoint_paths.rake @@ -1,14 +1,19 @@ -desc "Plot columns" +require 'benchmark' + +desc "Plot columns of CO2" task :plot_flightpath_with_co2_columns do + output_path = ENV["OUTPUT_PATH"] || "#{GTRON_ROOT}/output" get_db_conn(GTRON_ENV) kml = KMLFile.new doc = KML::Document.new(:name => "CO2 Columns") + dps = DataPoint.find(:all) heading = 0 column_coords = [] + #assemble datapoint tuples as [Column Pair coordinates, CO2 measure] dps.each_with_index do |dp, i| if dp.co2_ppm if i < (dps.size-1) @@ -20,6 +25,7 @@ task :plot_flightpath_with_co2_columns do end end + #create polygons from computed coordinates. column_coords.each_with_index do |c, i| if i < column_coords.size-1 sty = KML::Style.new(:poly_style => KML::PolyStyle.new( @@ -34,7 +40,7 @@ task :plot_flightpath_with_co2_columns do placemark.features << sty << col doc.features << placemark end - end.compact + end kml.objects << doc - File.open("output/co2_columns.kml", "w") {|f| f.write kml.render} + File.open("#{output_path}/co2_columns.kml", "w") {|f| f.write kml.render} end tasks/plot_flightpath_with_co2_columns.rake @@ -2,21 +2,25 @@ require File.dirname(__FILE__) + '/../test_helper.rb' class TestDrawSimpleShapes < Test::Unit::TestCase def setup + ENV["OUTPUT_PATH"] = TEST_TMP + bare_setup + get_db_conn(GTRON_ENV) @rake = Rake::Application.new Rake.application = @rake load File.dirname(__FILE__) + '/../../tasks/draw_simple_shapes.rake' end - should "be true" do - # Testing rake is a bit different - # http://blog.nicksieger.com/articles/2007/06/11/test-your-rake-tasks - # Example: - # @rake["task_name"].invoke - assert true + context "Drawing Shapes" do + setup { @rake["draw_simple_shapes"].invoke } + + should "make kml file" do + assert File.exists?("#{TEST_TMP}/simple_shapes.kml") + end end def teardown Rake.application = nil + bare_teardown end end test/tasks/test_draw_simple_shapes.rb @@ -5,16 +5,25 @@ class TestImport < Test::Unit::TestCase get_db_conn(GTRON_ENV) @rake = Rake::Application.new Rake.application = @rake + ENV["INPUT_PATH"] = "test/sample_data" load File.dirname(__FILE__) + '/../../tasks/import.rake' end - should "import data" do - # Testing rake is a bit different - # http://blog.nicksieger.com/articles/2007/06/11/test-your-rake-tasks - # Example: - # @rake["task_name"].invoke - #@rake["import"].invoke - assert true + context "Import data" do + setup { @rake["import"].invoke } + + should "create flights" do + assert_equal 1, Flight.find(:all).size + end + + should "create datapoints" do + assert_equal 10, DataPoint.find(:all).size + end + + should "associate datapoints with flight" do + assert_equal DataPoint.find(:all).size, + Flight.find(:first).data_points.size + end end def teardown test/tasks/test_import.rb @@ -21,4 +21,12 @@ module Test end end +TEST_TMP = "#{GTRON_ROOT}/test/tmp" +def bare_setup + FileUtils.mkdir TEST_TMP +end + +def bare_teardown + FileUtils.rm_rf TEST_TMP +end test/test_helper.rb ad624db758a3c559a6d325e666fd209ffa42d2d4 Ben Hughes ben@pixelmachine.org http://github.com/schleyfox/ascends_viz/commit/4b45b2acd21d65cca98708605fc3883530d9ac6f 4b45b2acd21d65cca98708605fc3883530d9ac6f 2008-06-30T13:57:05-07:00 2008-06-30T13:57:05-07:00 I did something, I don't really care about what * something involving lisp 00df524a9c3721bfda6b6ae58ea491ae49243efe Ben Hughes ben@pixelmachine.org