forecaster 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: bc80587e750da30fa99b3a4b1309c3a67d792f86
+  data.tar.gz: 42f6e928fc8c09c02ea0de356642884e16e10f30
+SHA512:
+  metadata.gz: aeb8bf4f69a32551e3c2e8cbebb96378979644eefcb8bb239c7d8035c79d489c4031d1a451de812b592140cd37daeee006bfe826578e5b381aa69ed742b42dd1
+  data.tar.gz: 4ec0004fc10aad0a25b675c041eb975636dda8f8978952173da92f0a8f41938fa382730c60aa573747a1b3d04c688e1229ec7f4c24c1683f22ab321228552a3c

data/lib/forecaster/configuration.rb

@@ -0,0 +1,24 @@
+module Forecaster
+  class Configuration
+    attr_accessor :server, :wgrib2, :cache_dir, :records
+
+    def initialize
+      @server = "http://www.ftp.ncep.noaa.gov/data/nccf/com/gfs/prod"
+      @wgrib2 = "/usr/bin/wgrib2"
+      @cache_dir = "/tmp/forecaster"
+      @records = {
+        prate: "PRATE:surface",
+        tmp:   "TMP:2 m above ground",
+        ugrd:  "UGRD:10 m above ground",
+        vgrd:  "VGRD:10 m above ground",
+        tcdc:  "TCDC:entire atmosphere"
+      }
+    end
+
+    def self.configure(options)
+      options.each do |option, value|
+        self[option] = value
+      end
+    end
+  end
+end

data/lib/forecaster/forecast.rb

@@ -0,0 +1,99 @@
+require "fileutils"
+
+module Forecaster
+  class Forecast
+    def initialize(year, month, day, hour_of_run, hour_of_forecast)
+      @year = year
+      @month = month
+      @day = day
+      @hour_of_run = hour_of_run
+      @hour_of_forecast = hour_of_forecast
+    end
+
+    def read(field, latitude: 0.0, longitude: 0.0)
+      wgrib2 = Forecaster.configuration.wgrib2
+      record = Forecaster.configuration.records[field]
+      cachename = Forecaster.configuration.cache_dir
+
+      filename = "gfs.t%02dz.pgrb2.0p25.f%03d" % [
+        @hour_of_run, @hour_of_forecast
+      ]
+      pathname = "%04d%02d%02d%02d%" % [
+        @year, @month, @day, @hour_of_run
+      ]
+      path = File.join(cachename, pathname, filename)
+
+      raise "'#{path}' not found" unless File.exists?(path)
+
+      out = `#{wgrib2} #{path} -lon #{longitude} #{latitude} -match "#{record}"`
+      lines = out.split("\n")
+      fields = lines.first.split(":")
+      params = Hash[*fields.last.split(",").map { |s| s.split("=") }.flatten]
+
+      params["val"]
+    end
+
+    def fetched?
+      cachename = Forecaster.configuration.cache_dir
+      pathname = "%04d%02d%02d%02d%" % [
+        @year, @month, @day, @hour_of_run
+      ]
+      filename = "gfs.t%02dz.pgrb2.0p25.f%03d" % [
+        @hour_of_run, @hour_of_forecast
+      ]
+      File.exist?(File.join(cachename, pathname, filename))
+    end
+
+    def fetch
+      return self if fetched?
+
+      server = Forecaster.configuration.server
+      cachename = Forecaster.configuration.cache_dir
+      curl = "curl -f -s -S"
+
+      pathname = "%04d%02d%02d%02d%" % [
+        @year, @month, @day, @hour_of_run
+      ]
+      FileUtils.mkpath(File.join(cachename, pathname))
+
+      filename = "gfs.t%02dz.pgrb2.0p25.f%03d" % [
+        @hour_of_run, @hour_of_forecast
+      ]
+      url = "%s/gfs.%04d%02d%02d%02d/%s" % [
+        server, @year, @month, @day, @hour_of_run, filename
+      ]
+      path = File.join(cachename, pathname, filename)
+
+      # puts "Downloading '#{url}.idx' ..."
+      cmd = "#{curl} -o #{path}.idx #{url}.idx"
+      return self unless system(cmd)
+
+      lines = IO.readlines("#{path}.idx")
+      n = lines.count
+      ranges = lines.each_index.reduce([]) do |r, i|
+        records = Forecaster.configuration.records
+        if records.values.any? { |record| lines[i].include?(record) }
+          first = lines[i].split(":")[1].to_i
+          last = ""
+
+          j = i
+          while (j += 1) < n
+            last = lines[j].split(":")[1].to_i - 1
+            break if last != first - 1
+          end
+
+          r << "#{first}-#{last}" # cURL syntax for a range
+        else
+          r
+        end
+      end
+      system("rm #{path}.idx")
+
+      # puts "Downloading '#{url}' ..."
+      cmd = "#{curl} -r #{ranges.join(",")} -o #{path} #{url}"
+      return self unless system(cmd)
+
+      self
+    end
+  end
+end

data/lib/forecaster.rb

@@ -0,0 +1,17 @@
+require "forecaster/configuration"
+require "forecaster/forecast"
+
+module Forecaster
+  class << self
+    attr_accessor :configuration
+  end
+
+  def self.configure
+    self.configuration ||= Forecaster::Configuration.new
+    yield(configuration)
+  end
+
+  def self.fetch(year, month, day, hour_of_run, hour_of_forecast)
+    Forecaster::Forecast.new(year, month, day, hour_of_run, hour_of_forecast).fetch
+  end
+end

metadata

@@ -0,0 +1,46 @@
+--- !ruby/object:Gem::Specification
+name: forecaster
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- Vincent Ollivier
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2015-05-04 00:00:00.000000000 Z
+dependencies: []
+description: Wrapper around curl and wgrib2 to fetch and read GFS data
+email: v@vinc.cc
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/forecaster.rb
+- lib/forecaster/configuration.rb
+- lib/forecaster/forecast.rb
+homepage: https://github.com/vinc/forecaster
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.2.2
+signing_key: 
+specification_version: 4
+summary: Wrapper around curl and wgrib2 to fetch and read GFS data
+test_files: []