seis_ruby 0.0.1 → 0.1.0

data/README.rdoc

@@ -0,0 +1,5 @@
+= SeisRuby
+== Executables
+=== seis_ruby
+Please see details by typing:
+  > seis_ruby help

data/bin/seis_ruby

@@ -1,17 +1,4 @@
 #!/usr/bin/env ruby
 
-require 'thor'
-class SeisRubyRunner < Thor
-  require 'seis_ruby'
-  require 'ruby_patch'
-  require 'yaml'
-
-  desc 'gcmt_catalog_scrape URL', 'scrape cmtsolution data from GCMT catalog search result pages.'
-  def gcmt_catalog_scrape(url)
-    file = "#{__METHOD__}_#{Time.now.ymdhms}.yaml"
-    open(file, 'w').write(SeisRuby::Io::GcmtCatalog.scrape(url).to_yaml)
-    puts file
-  end
-end
-
-SeisRubyRunner.start
+require 'seis_ruby'
+::SeisRuby::Bin::SeisRubyRunner.start

data/lib/seis_ruby/bin/seis_ruby_runner.rb

@@ -0,0 +1,58 @@
+module SeisRuby
+  module Bin
+    require 'thor'
+
+    class SeisRubyRunner < Thor
+      require 'yaml'
+      require 'fileutils'
+      require 'ruby_patch'
+      extend ::RubyPatch::AutoLoad
+
+      COMMAND_NAME = 'seis_ruby'
+
+      desc 'gcmt_catalog_scrape URL', 'scrape cmtsolution data from GCMT catalog search result pages.'
+      def gcmt_catalog_scrape(url)
+        file = "#{__METHOD__}_#{Time.now.ymdhms}.yaml"
+        open(file, 'w').write(SeisRuby::Io::GcmtCatalog.scrape(url).to_yaml)
+        puts file
+      end
+
+      desc "generate_completion", "Generate bash completion of seis_ruby's commands."
+      def generate_completion
+        commands = `#{COMMAND_NAME} --help`.split("\n").select{|l| l =~ / *#{COMMAND_NAME} /}.map{|l| l.split[1]}
+        function = <<-EOS
+_#{COMMAND_NAME}()
+{
+  local commands current previous
+  commands="#{commands.join(' ')}"
+  current="${COMP_WORDS[COMP_CWORD]}"
+  previous="${COMP_WORDS[COMP_CWORD-1]}"
+
+  case "${previous}" in
+    #{COMMAND_NAME}|help)
+      COMPREPLY=( $(compgen -W "${commands}" ${current}) );;
+    * )
+      COMPREPLY=( $(compgen -f ${current}) );;
+  esac
+}
+complete -F _#{COMMAND_NAME} #{COMMAND_NAME}
+        EOS
+
+        file = "#{ENV['HOME']}/.#{COMMAND_NAME}.d/completion.bash"
+        FileUtils.mkdir_p(File.dirname(file))
+        FileUtils.cp(file, "#{file}.#{Time.now.ymdhms}.bak") if File.file?(file)
+        open(file, 'w').write(function)
+
+        tilde_file = file.sub(/\A#{ENV['HOME']}/, '~')
+        puts <<-EOS
+Please add following code to your ~/.bashrc if necessary.
+
+# enable seis_ruby completion.
+if [ -f #{tilde_file} ]; then
+       . #{tilde_file}
+fi
+        EOS
+      end
+    end
+  end
+end

data/lib/seis_ruby/bin.rb

@@ -0,0 +1,6 @@
+module SeisRuby
+  module Bin
+    require 'ruby_patch'
+    extend ::RubyPatch::AutoLoad
+  end
+end

data/lib/seis_ruby/data/cmtsolution.rb

@@ -1,8 +1,8 @@
 module SeisRuby
   module Data
     module Cmtsolution
-      require 'rainbow'
-      require 'ruby_patch/string'
+      require 'ruby_patch'
+      extend ::RubyPatch::AutoLoad
 
       ParseError = Class.new(StandardError)
 
@@ -42,20 +42,34 @@ module SeisRuby
         h
       end
 
-      def parse(text_for_one_event)
+      def _parse_one_event(text_for_one_event)
         begin
           lines = _preprocess_for_parse(text_for_one_event)
           _parse_lines(lines)
         rescue
-          raise ParseError, text_for_one_event\
-            .split("\n")\
-            .map{|line| line.quote('|'.color(:red))}\
-            .join("\n")
+          raise ParseError, text_for_one_event
         end
       end
 
-      def parse_list(events)
-        events.map{|ev| parse(ev)}
+      # @param [String, Array<String>] events List of CMTSOLUTION format texts.
+      # @return [Array<Hash>] You can access :event_name, :hypocenter, :centroid.
+      def parse(*events)
+        h = events.flatten.map{|ev|
+          begin
+            _parse_one_event(ev)
+          rescue ParseError => e
+            e
+          end
+        }.classify{|o| o.exception?}
+        unless (error_list = h[true]).empty?
+          error_message = error_list.map{|e| e.message}.join("\n\n")
+          raise ParseError, <<-EOS
+Could not parse following events:
+#{error_message}
+          EOS
+        end
+
+        h[false]
       end
     end
   end

data/lib/seis_ruby/data.rb

@@ -1,6 +1,6 @@
 module SeisRuby
   module Data
-    require 'seis_ruby/data/cmtsolution'
-    require 'seis_ruby/data/sac'
+    require 'ruby_patch'
+    extend ::RubyPatch::AutoLoad
   end
 end

data/lib/seis_ruby/io/gcmt_catalog/custom_html_parser.rb

@@ -4,6 +4,9 @@ module SeisRuby
       require 'mechanize'
 
       class CustomHtmlParser < ::Mechanize::Page
+        require 'ruby_patch'
+        extend ::RubyPatch::AutoLoad
+
         def initialize(uri = nil, response = nil, body = nil, code = nil, mech = nil)
           super(uri, response, body.gsub(/<=/, '&lt;='), code, mech)
         end

data/lib/seis_ruby/io/gcmt_catalog.rb

@@ -3,30 +3,38 @@ module SeisRuby
     module GcmtCatalog
       require 'rainbow'
       require 'mechanize'
-      require 'seis_ruby/io/gcmt_catalog/custom_html_parser'
-      require 'seis_ruby/data/cmtsolution'
+      require 'ruby_patch'
+      extend ::RubyPatch::AutoLoad
 
       module_function
 
+      # @param [String] url URL of a first page of Global CMT Catalog search result.
+      # @return [Array<Hash>] {::SeisRuby::Data::Cmtsolution#parse}
       def scrape(url)
         agent = Mechanize.new{|a|
           a.pluggable_parser.html = ::SeisRuby::Io::GcmtCatalog::CustomHtmlParser
         }
 
-        event_list = []
+        result = {
+          query: url,
+          result: [],
+        }
+
         while url
           agent.get(url)
-          agent.page.search('pre')\
+          event_text_list = agent.page.search('pre')\
             .find{|e| e.text =~ /event name:/}\
             .text.split("\n\n")\
-            .each{|event|
-            begin
-              event_list << ::SeisRuby::Data::Cmtsolution.parse(event) unless event.strip.empty?
-            rescue ::SeisRuby::Data::Cmtsolution::ParseError => e
-              $stderr.puts e.class.to_s.color(:red)
-              $stderr.puts e.message
-            end
-          }
+            .delete_if{|t| t.strip.empty?}
+
+          begin
+            event_list = ::SeisRuby::Data::Cmtsolution.parse(event_text_list)
+          rescue ::SeisRuby::Data::Cmtsolution::ParseError => e
+            $stderr.puts e.class.to_s.color(:red)
+            $stderr.puts e.message
+          ensure
+            result[:result].concat event_list
+          end
 
           url = nil
           if next_link = agent.page.links\
@@ -35,7 +43,7 @@ module SeisRuby
           end
         end
 
-        event_list
+        result
       end
     end
   end

data/lib/seis_ruby/io.rb

@@ -1,5 +1,6 @@
 module SeisRuby
   module Io
-    require 'seis_ruby/io/gcmt_catalog'
+    require 'ruby_patch'
+    extend ::RubyPatch::AutoLoad
   end
 end

data/lib/seis_ruby/version.rb

@@ -1,3 +1,3 @@
 module SeisRuby
-  VERSION = '0.0.1'
+  VERSION = '0.1.0'
 end

data/lib/seis_ruby.rb

@@ -1,5 +1,5 @@
 module SeisRuby
   require 'seis_ruby/version'
-  require 'seis_ruby/data'
-  require 'seis_ruby/io'
+  require 'ruby_patch'
+  extend ::RubyPatch::AutoLoad
 end

data/seis_ruby.gemspec

@@ -1,19 +1,39 @@
 gem_name = File.basename(Dir.pwd)
-require "./lib/#{gem_name}/version"
+require 'rainbow'
+require 'ruby_patch'
+
+dir = File.join(__DIR__, 'lib')
+$LOAD_PATH.unshift(File.join(dir)) unless $LOAD_PATH.include?(dir)
+require "#{gem_name}"
 
 Gem::Specification.new do |s|
   s.files = `git ls-files`.split
   s.name = gem_name
   s.summary = "Ruby library for earthquake science."
   s.version = SeisRuby::VERSION
-  s.add_development_dependency 'rspec', '~> 2'
-  s.add_development_dependency 'simplecov', '~> 0'
-  s.add_runtime_dependency 'rainbow', '~>1'
-  s.add_runtime_dependency 'mechanize', '~>2'
-  s.add_runtime_dependency 'ruby_patch', ['>= 0.1.0', '< 1.0.0']
+  s.add_development_dependency 'rspec', '~> 2.10'
+  s.add_development_dependency 'simplecov', '~> 0.6'
+  s.add_runtime_dependency 'rainbow', '~> 1.1'
+  s.add_runtime_dependency 'thor', '~> 0.15'
+  s.add_runtime_dependency 'mechanize', '~> 2.5'
+  s.add_runtime_dependency 'ruby_patch', '~> 0.0'
   s.author = 'kshramt'
   s.description = "Ruby library for earthquake science. Provides Parser for various data formats used in seismology."
   s.executables << 'seis_ruby'
-  s.required_ruby_version = '>= 1.9.0'
+  s.post_install_message = (<<-EOS
+
+
+SeisRuby #{::SeisRuby::VERSION}
+Thank you for installing.
+Please execute following command:
+
+#{::SeisRuby::Bin::SeisRubyRunner::COMMAND_NAME} generate_completion
+
+Enjoy!
+
+
+  EOS
+  ).color(:green)
+  s.required_ruby_version = '~> 1.9'
   s.test_files.concat `git ls-files spec`.split.select{|path| path =~ /_spec\.rb/}
 end

data/spec/seis_ruby/data/cmtsolution_spec.rb

@@ -1,5 +1,4 @@
 require 'spec_helper'
-require 'seis_ruby/data/cmtsolution'
 
 describe ::SeisRuby::Data::Cmtsolution do
   before :each do
@@ -71,27 +70,13 @@ Mtp:      -1.491000e+24
   describe '.parse' do
     context 'normal case' do
       it do
-        ::SeisRuby::Data::Cmtsolution.parse(@valid_data).should == @parsed_valid_data
+        ::SeisRuby::Data::Cmtsolution.parse(@valid_data, @valid_data, [@valid_data, @valid_data]).should == [@parsed_valid_data, @parsed_valid_data, @parsed_valid_data, @parsed_valid_data]
       end
     end
 
     context 'wrong case' do
       it do
-        lambda{::SeisRuby::Data::Cmtsolution.parse(@wrong_data)}.should raise_error(::SeisRuby::Data::Cmtsolution::ParseError)
-      end
-    end
-  end
-
-  describe '.parse_list' do
-    context 'normal case' do
-      it do
-        ::SeisRuby::Data::Cmtsolution.parse_list([@valid_data, @valid_data]).should == [@parsed_valid_data, @parsed_valid_data]
-      end
-    end
-
-    context 'wrong case' do
-      it do
-        lambda{::SeisRuby::Data::Cmtsolution.parse_list([@valid_data, @wrong_data])}.should raise_error(::SeisRuby::Data::Cmtsolution::ParseError)
+        lambda{::SeisRuby::Data::Cmtsolution.parse([@valid_data, @wrong_data])}.should raise_error(::SeisRuby::Data::Cmtsolution::ParseError)
       end
     end
   end

data/spec/seis_ruby/io/gcmt_catalog_spec.rb

@@ -1,7 +1,6 @@
 require 'yaml'
 require 'ruby_patch'
 require 'spec_helper'
-require 'seis_ruby'
 
 describe ::SeisRuby::Io::GcmtCatalog do
   describe '.scrape' do

data/spec/seis_ruby/io/scrape.yaml

@@ -1,4 +1,6 @@
 ---
+:query: http://www.globalcmt.org/cgi-bin/globalcmt-cgi-bin/CMT4/form?itype=ymd&yr=1976&mo=1&day=1&oyr=1976&omo=1&oday=1&jyr=1976&jday=1&ojyr=1&ojday=1&otype=nd&nday=400&lmw=0&umw=10&lms=0&ums=10&lmb=0&umb=10&llat=-90&ulat=90&llon=-180&ulon=180&lhd=0&uhd=1000&lts=-9999&uts=9999&lpe1=0&upe1=90&lpe2=0&upe2=90&list=4
+:result:
 - :hypocenter:
     :data_source: MLI
     :year: 1976
@@ -157,7 +159,7 @@
     :mrr: 4.78e+27
     :mtt: -4.9e+26
     :mpp: -4.3e+27
-    :mrt: 830000000000000100000000000.0
+    :mrt: 8.3e+26
     :mrp: 3.62e+27
     :mtp: -1.32e+27
   :event_name: 011476A

data/spec/spec_helper.rb

@@ -14,3 +14,5 @@ RSpec.configure do |config|
   config.run_all_when_everything_filtered = true
   config.filter_run :focus
 end
+
+require 'seis_ruby'

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: seis_ruby
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.1.0
   prerelease: 
 platform: ruby
 authors:
@@ -9,66 +9,74 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-06-26 00:00:00.000000000 Z
+date: 2012-07-07 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rspec
-  requirement: &79465380 !ruby/object:Gem::Requirement
+  requirement: &86475830 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ~>
       - !ruby/object:Gem::Version
-        version: '2'
+        version: '2.10'
   type: :development
   prerelease: false
-  version_requirements: *79465380
+  version_requirements: *86475830
 - !ruby/object:Gem::Dependency
   name: simplecov
-  requirement: &79465130 !ruby/object:Gem::Requirement
+  requirement: &86475580 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ~>
       - !ruby/object:Gem::Version
-        version: '0'
+        version: '0.6'
   type: :development
   prerelease: false
-  version_requirements: *79465130
+  version_requirements: *86475580
 - !ruby/object:Gem::Dependency
   name: rainbow
-  requirement: &79464900 !ruby/object:Gem::Requirement
+  requirement: &86475350 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ~>
       - !ruby/object:Gem::Version
-        version: '1'
+        version: '1.1'
   type: :runtime
   prerelease: false
-  version_requirements: *79464900
+  version_requirements: *86475350
+- !ruby/object:Gem::Dependency
+  name: thor
+  requirement: &86475120 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '0.15'
+  type: :runtime
+  prerelease: false
+  version_requirements: *86475120
 - !ruby/object:Gem::Dependency
   name: mechanize
-  requirement: &79464670 !ruby/object:Gem::Requirement
+  requirement: &86474890 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ~>
       - !ruby/object:Gem::Version
-        version: '2'
+        version: '2.5'
   type: :runtime
   prerelease: false
-  version_requirements: *79464670
+  version_requirements: *86474890
 - !ruby/object:Gem::Dependency
   name: ruby_patch
-  requirement: &79464420 !ruby/object:Gem::Requirement
+  requirement: &86474660 !ruby/object:Gem::Requirement
     none: false
     requirements:
-    - - ! '>='
-      - !ruby/object:Gem::Version
-        version: 0.1.0
-    - - <
+    - - ~>
       - !ruby/object:Gem::Version
-        version: 1.0.0
+        version: '0.0'
   type: :runtime
   prerelease: false
-  version_requirements: *79464420
+  version_requirements: *86474660
 description: Ruby library for earthquake science. Provides Parser for various data
   formats used in seismology.
 email: 
@@ -77,8 +85,11 @@ executables:
 extensions: []
 extra_rdoc_files: []
 files:
+- README.rdoc
 - bin/seis_ruby
 - lib/seis_ruby.rb
+- lib/seis_ruby/bin.rb
+- lib/seis_ruby/bin/seis_ruby_runner.rb
 - lib/seis_ruby/data.rb
 - lib/seis_ruby/data/cmtsolution.rb
 - lib/seis_ruby/data/sac.rb
@@ -94,16 +105,17 @@ files:
 - spec/spec_helper.rb
 homepage: 
 licenses: []
-post_install_message: 
+post_install_message: ! "\e[32m\n\nSeisRuby 0.1.0\nThank you for installing.\nPlease
+  execute following command:\n\nseis_ruby generate_completion\n\nEnjoy!\n\n\n\e[0m"
 rdoc_options: []
 require_paths:
 - lib
 required_ruby_version: !ruby/object:Gem::Requirement
   none: false
   requirements:
-  - - ! '>='
+  - - ~>
     - !ruby/object:Gem::Version
-      version: 1.9.0
+      version: '1.9'
 required_rubygems_version: !ruby/object:Gem::Requirement
   none: false
   requirements: