mathpack 0.4.7 → 0.4.8

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: ebeda1905dee9fe22d4a6469200537d76d21c432
-  data.tar.gz: b3b71ff00ef7a9ced9dd6855f8ef912329167f0e
+  metadata.gz: 10ff066c9adc6d22256567faf50bb86bfa41b919
+  data.tar.gz: dd387c24f4f687771eed4119b9d4e843be181e2b
 SHA512:
-  metadata.gz: 89c4e920ab56d6dc9ba263d0f04dc6682efba2f2374d01c3afc8f56b809e0a40ec96b9987e64c8ec8e5881b9e29bdd73162b32a15fa18c0bab6b43c0bbb14572
-  data.tar.gz: e19cd9fe1cf1bdc670e5b0ad2f1a2b47f9e6254cf6e4ba9cad9742a7f66ec43bfecf722f7f99919775196ca22a1f9144226762b3c5547e72495c688b9065f8a0
+  metadata.gz: 6a4147c4f2be3f383dd001c9092e61ee127bb5cfbfd7300ec58ad4fa430b91dc358f561440fa70c4f1e86e1749519a93e332d2a1c04e27e112cc0547d5049fcb
+  data.tar.gz: fb1dfcefe66bd0dc9e8496353d57eb10288b394551cdcf07d2f83a5ff32e85dfac65b752f2ffa128346cfafbfa31b6a57dd13b15f42b1b87a287f5def36ea707

data/lib/mathpack.rb

@@ -8,3 +8,4 @@ require 'mathpack/io'
 require 'mathpack/functions'
 require 'mathpack/functional'
 require 'mathpack/integral_equations'
+require 'mathpack/differential_equations'

data/lib/mathpack/differential_equations.rb

@@ -0,0 +1,66 @@
+require 'mathpack'
+
+module Mathpack
+  module DifferentialEquations
+    def self.runge_kutta_4(params = {})
+      nodes = Mathpack::Approximation.generate_nodes(from: params[:from], to: params[:to], step: params[:step])
+      y = Array.new(params[:system].length) { Array.new(nodes.length) }
+      y.each_index do |i|
+        y[i][0] = params[:y0][i]
+      end
+      fi0 = Array.new(y.length)
+      fi1 = Array.new(y.length)
+      fi2 = Array.new(y.length)
+      fi3 = Array.new(y.length)
+      for i in 0...nodes.length-1 do
+        column = y.transpose[i]
+        y_volna = Array.new(y.length)
+        y.each_index do |j|
+          fi0[j] = params[:system][j].call(nodes[i], *column)
+        end
+        y.each_index do |j|
+          y_volna.each_index { |k| y_volna[k] = column[k] + 0.5 * params[:step] * fi0[k] }
+          fi1[j] = params[:system][j].call(nodes[i] + 0.5 * params[:step], *y_volna)
+        end
+        y.each_index do |j|
+          y_volna.each_index { |k| y_volna[k] = column[k] + 0.5 * params[:step] * fi1[k] }
+          fi2[j] = params[:system][j].call(nodes[i] + 0.5 * params[:step], *y_volna)
+        end
+        y.each_index do |j|
+          y_volna.each_index { |k| y_volna[k] = column[k] + params[:step] * fi2[k] }
+          fi3[j] = params[:system][j].call(nodes[i + 1], *y_volna)
+        end
+        y.each_index do |j|
+          y[j][i + 1] = y[j][i] + params[:step]/6.0 * (fi0[j] + 2.0 * fi1[j] + 2.0 * fi2[j] + fi3[j])
+          y[j][i + 1] = y[j][i] + params[:step]/6.0 * (fi0[j] + 2.0 * fi1[j] + 2.0 * fi2[j] + fi3[j])
+        end
+      end
+      y
+    end
+
+    def self.solve_cauchie_system(params = {})
+      params[:method] ||= :runge_kutta_4
+      params[:m] = 2
+      step = (params[:to] - params[:from]) / 2.0
+      result = Mathpack::DifferentialEquations.send(params[:method], from: params[:from], to: params[:to], step: step, y0: params[:y0], system: params[:system], eps: params[:eps])
+      u_next = Array.new(params[:system].length) { |i| result[i] }
+      loop do
+        u_prev = Array.new(u_next.length) { |i| u_next[i].dup }
+        step /= 2.0
+        result = Mathpack::DifferentialEquations.send(params[:method], from: params[:from], to: params[:to], step: step, y0: params[:y0], system: params[:system], eps: params[:eps])
+        u_next.each_index do |i|
+          u_next[i] = result[i]
+        end
+        is_finished = true
+        u_next.each_index do |i|
+          if Mathpack::IO.count_diff(u_next[i], u_prev[i]) / (2**params[:m] - 1) > params[:eps]
+            is_finished = false
+            break
+          end
+        end
+        break if is_finished
+      end
+      { x: Mathpack::Approximation.generate_nodes(from: params[:from], to: params[:to], step: step), y: u_next.first }
+    end
+  end
+end

data/lib/mathpack/version.rb

@@ -1,3 +1,3 @@
 module Mathpack
-  VERSION = '0.4.7'
+  VERSION = '0.4.8'
 end

data/spec/differential_equations_spec.rb

@@ -0,0 +1,14 @@
+describe 'DifferentialEquation' do
+  require 'mathpack/differential_equations'
+
+  describe '#solve_cauchie_system' do
+    let(:eps) { 1e-6 }
+
+    it 'solves cauchie system correctly' do
+      cauchie_problem = [ ->(x, u1, u2) { u2 }, -> (x, u1, u2) { - 1.0 / x  * u2 + 1.0 / x**2 * u1 + 3.0 }]
+      result = Mathpack::DifferentialEquations.solve_cauchie_system(from: 1.0, to: 2.0, eps: eps, system: cauchie_problem, y0: [1.0, 1.0])
+      correct_result = Mathpack::Approximation.fill_f(result[:x]){ |t| t**2 - 0.5 * t + 1.0/(2.0 * t) }
+      expect(Mathpack::IO.count_diff(result[:y], correct_result) <= eps).to eq(true)
+    end
+  end
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: mathpack
 version: !ruby/object:Gem::Version
-  version: 0.4.7
+  version: 0.4.8
 platform: ruby
 authors:
 - maxmilan
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-06-05 00:00:00.000000000 Z
+date: 2015-10-03 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -53,6 +53,7 @@ files:
 - Rakefile
 - lib/mathpack.rb
 - lib/mathpack/approximation.rb
+- lib/mathpack/differential_equations.rb
 - lib/mathpack/functional.rb
 - lib/mathpack/functions.rb
 - lib/mathpack/integral_equations.rb
@@ -64,6 +65,7 @@ files:
 - lib/mathpack/version.rb
 - mathpack.gemspec
 - spec/approximation_spec.rb
+- spec/differential_equations_spec.rb
 - spec/functional_spec.rb
 - spec/functions_spec.rb
 - spec/integral_equations_spec.rb