forecaster 0.0.1 → 0.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: bc80587e750da30fa99b3a4b1309c3a67d792f86
-  data.tar.gz: 42f6e928fc8c09c02ea0de356642884e16e10f30
+  metadata.gz: 2a5197a183958f87c3231d9794568fd71794c5b5
+  data.tar.gz: d33426c83219a0a7d813be7dea525a8426cc61ad
 SHA512:
-  metadata.gz: aeb8bf4f69a32551e3c2e8cbebb96378979644eefcb8bb239c7d8035c79d489c4031d1a451de812b592140cd37daeee006bfe826578e5b381aa69ed742b42dd1
-  data.tar.gz: 4ec0004fc10aad0a25b675c041eb975636dda8f8978952173da92f0a8f41938fa382730c60aa573747a1b3d04c688e1229ec7f4c24c1683f22ab321228552a3c
+  metadata.gz: f7c2cda2ff2a646779c9326ef40b93c8e12593753dd62c3418f7def70b4bc9104640a1353652259a613b17dfb686bce95486bf9f6b286511a0a15f1da966b2d3
+  data.tar.gz: 973f053a188acd33f48f4305280f5106181749d65942c86bff946c98921fe8b18377d9b35c3687fb1c0107bf286ac8082e77aa36fac4b98539c66a72cfa5d009

data/lib/forecaster/configuration.rb

@@ -1,17 +1,23 @@
 module Forecaster
   class Configuration
-    attr_accessor :server, :wgrib2, :cache_dir, :records
+    attr_accessor :server, :cache_dir, :curl_path, :wgrib2_path, :records
 
     def initialize
       @server = "http://www.ftp.ncep.noaa.gov/data/nccf/com/gfs/prod"
-      @wgrib2 = "/usr/bin/wgrib2"
       @cache_dir = "/tmp/forecaster"
+      @curl_path = "curl"
+      @wgrib2_path = "wgrib2"
+
+      # See: http://www.nco.ncep.noaa.gov/pmb/products/gfs/gfs_upgrade/gfs.t06z.pgrb2.0p25.f006.shtml
+      # See: http://www.cpc.ncep.noaa.gov/products/wesley/wgrib2/
+      # Use `variable` and `level` attributes separated by colons to identify
+      # the records to download and read.
       @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"
+        :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
 

data/lib/forecaster/forecast.rb

@@ -10,63 +10,45 @@ module Forecaster
       @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%" % [
+    def dirname
+      subdir = "%04d%02d%02d%02d" % [
         @year, @month, @day, @hour_of_run
       ]
-      path = File.join(cachename, pathname, filename)
-
-      raise "'#{path}' not found" unless File.exists?(path)
+      File.join(Forecaster.configuration.cache_dir, subdir)
+    end
 
-      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]
+    def filename
+      "gfs.t%02dz.pgrb2.0p25.f%03d" % [
+        @hour_of_run, @hour_of_forecast
+      ]
+    end
 
-      params["val"]
+    def url
+      server = Forecaster.configuration.server
+      "%s/gfs.%04d%02d%02d%02d/%s" % [
+        server, @year, @month, @day, @hour_of_run, filename
+      ]
     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))
+      File.exist?(File.join(dirname, filename))
     end
 
+    # This method will save the forecast file in the cache directory.
+    # But only the parts of the file containing the fields defined in
+    # the configuration will be downloaded.
     def fetch
       return self if fetched?
 
-      server = Forecaster.configuration.server
-      cachename = Forecaster.configuration.cache_dir
-      curl = "curl -f -s -S"
+      curl = Forecaster.configuration.curl_path
 
-      pathname = "%04d%02d%02d%02d%" % [
-        @year, @month, @day, @hour_of_run
-      ]
-      FileUtils.mkpath(File.join(cachename, pathname))
+      FileUtils.mkpath(File.join(dirname))
 
-      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)
+      path = File.join(dirname, filename)
 
       # puts "Downloading '#{url}.idx' ..."
-      cmd = "#{curl} -o #{path}.idx #{url}.idx"
-      return self unless system(cmd)
+      cmd = "#{curl} -fsS -o #{path}.idx #{url}.idx"
+      raise "Download of '#{url}.idx' failed" unless system(cmd)
 
       lines = IO.readlines("#{path}.idx")
       n = lines.count
@@ -90,10 +72,28 @@ module Forecaster
       system("rm #{path}.idx")
 
       # puts "Downloading '#{url}' ..."
-      cmd = "#{curl} -r #{ranges.join(",")} -o #{path} #{url}"
-      return self unless system(cmd)
+      cmd = "#{curl} -fsS -r #{ranges.join(",")} -o #{path} #{url}"
+      raise "Download of '#{url}' failed" unless system(cmd)
 
       self
     end
+
+    def read(field, latitude: 0.0, longitude: 0.0)
+      wgrib2 = Forecaster.configuration.wgrib2_path
+      record = Forecaster.configuration.records[field]
+      path = File.join(dirname, filename)
+
+      raise "'#{path}' not found" unless File.exists?(path)
+
+      coords = "#{longitude} #{latitude}"
+      output = `#{wgrib2} #{path} -lon #{coords} -match "#{record}"`
+
+      raise "Could not read '#{record}' in '#{path}'" if output.empty?
+
+      fields = output.split("\n").first.split(":")
+      params = Hash[*fields.last.split(",").map { |s| s.split("=") }.flatten]
+
+      params["val"]
+    end
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: forecaster
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
 platform: ruby
 authors:
 - Vincent Ollivier
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-05-04 00:00:00.000000000 Z
+date: 2016-06-09 00:00:00.000000000 Z
 dependencies: []
 description: Wrapper around curl and wgrib2 to fetch and read GFS data
 email: v@vinc.cc
@@ -39,7 +39,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 2.2.2
+rubygems_version: 2.4.5.1
 signing_key: 
 specification_version: 4
 summary: Wrapper around curl and wgrib2 to fetch and read GFS data