rb-daspk 0.0.1-universal-darwin-10

data/lib/rb-daspk.rb

@@ -0,0 +1,205 @@
+# Copyright (c) 2009 Eric Todd Meyers
+#
+# Permission is hereby granted, free of charge, to any person
+# obtaining a copy of this software and associated documentation
+# files (the "Software"), to deal in the Software without
+# restriction, including without limitation the rights to use,
+# copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the
+# Software is furnished to do so, subject to the following
+# conditions:
+#
+# The above copyright notice and this permission notice shall be
+# included in all copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+# EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES
+# OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+# NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
+# HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
+# WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
+# FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
+# OTHER DEALINGS IN THE SOFTWARE.
+#
+#
+# More information, source code and gems @ http://rubyforge.org/projects/rb-daspk/
+#
+#
+
+require 'rubygems'
+require 'ffi'
+
+module FFI
+  class Pointer
+    def read_double
+      get_float64(0)
+    end
+
+    def write_double(obj)
+      put_float64(0, obj)
+    end
+
+    def read_array_of_double(length)
+      get_array_of_double(0, length)
+    end
+
+    def write_array_of_double(ary)
+      put_array_of_double(0, ary)
+    end
+  end
+end
+
+
+module DASPK
+ extend FFI::Library
+ ffi_lib File.dirname(__FILE__) +  '/libdaspk.dylib'
+ callback :jac, [:pointer,:pointer,:pointer,:pointer,:pointer,:pointer,:pointer,:pointer,:pointer], :void
+ callback :res, [:pointer,:pointer,:pointer,:pointer,:pointer,:pointer,:pointer,:pointer,:pointer], :void
+ attach_function 'ddaspk_', [:res,    :pointer,:pointer,:pointer,:pointer,:pointer,:pointer,:pointer,:pointer,:pointer, 
+                             :pointer,:pointer,:pointer,:pointer,:pointer,:pointer,:jac,:pointer,:pointer,:pointer, 
+                             :pointer,:pointer,:pointer], :void
+
+ class Solver
+
+    include DASPK
+
+    attr_accessor :rtol,:atol,:ialgs,:jac,:maxstep
+
+    def initialize (neq, y, yprime, inter=nil, jac=nil, &res)
+       @lwr   = 5000
+       @lwi   = 5000
+       @plwr  = FFI::MemoryPointer.new(:int)
+       @plwr.write_int(@lwr)
+       @plwi  = FFI::MemoryPointer.new(:int)
+       @plwi.write_int(@lwi)
+       @work = [0]*@lwr
+       @pwork = FFI::MemoryPointer.new(:double,@lwr)
+       @piwork = FFI::MemoryPointer.new(:int,@lwi)
+       @info = [0]*30
+       @pinfo  = FFI::MemoryPointer.new(:int,30)
+       @ialgs = [1]*neq
+       @res = res
+       @inter = inter
+       @rtol = 0.000001
+       @atol = 0.000001
+       @jac  = jac
+       @maxstep =0
+
+       if (inter) then
+         @info[2]=1
+       end
+
+       @neq = neq
+       @pneq  = FFI::MemoryPointer.new(:int)
+       @pneq.write_int(neq)
+       @y=y
+       @yprime=yprime
+
+       @py       = FFI::MemoryPointer.new(:double,neq)
+       @py.write_array_of_double(y)
+       @pyprime  = FFI::MemoryPointer.new(:double,neq)
+       @pyprime.write_array_of_double(yprime)
+
+       @prtol  = FFI::MemoryPointer.new(:double)
+       @patol  = FFI::MemoryPointer.new(:double)
+
+    end
+
+    def solve (time, tout)
+
+       @prtol.write_double(@rtol)
+       @patol.write_double(@atol)
+
+#
+#  residual proc called from the Fortran. Let's pass it onto the user's residual procedure
+#  in a easier to use form
+#
+       resproc = Proc.new do |p1,p2,p3,p4,p5,p6,p7,p8,p9|
+          delta = p5.read_array_of_double(@neq) 
+          @res.call(p1.read_double,p2.read_array_of_double(@neq),p3.read_array_of_double(@neq),delta)
+          p5.write_array_of_double(delta)
+       end
+
+
+       if (@jac) then
+         @info[4]=1  
+       else
+         @info[4]=0  
+       end
+#
+#  jacobian proc called from the Fortran. Let's pass it onto the user's jacboian procedure
+#  in a easier to use form
+#
+       jacproc = Proc.new do |p1,p2,p3,p4,p5,p6,p7,p8,p9|
+         all = p4.read_array_of_double(@neq*@neq)
+         jacm = Array.new
+         for i in 0..(@neq-1) do
+           arow = Array.new
+           for j in 0..(@neq-1) do
+              arow << all[j*@neq+i]   # Fortran stores in column major, though references in row major     
+           end
+           jacm << arow
+         end
+
+         @jac.call(p1.read_double,p2.read_array_of_double(@neq),p3.read_array_of_double(@neq),jacm,p5.read_double)
+         
+         for i in 0..(@neq-1) do
+           arow = jacm[i]
+           for j in 0..(@neq-1) do
+              all[j*@neq+i] = arow[j] # Fortran stores in column major, though references in row major     
+           end
+         end
+         p4.write_array_of_double(all)
+       end
+
+       if (@maxstep!=0) then
+          @info[6]=1
+          @work[1]=@maxstep
+       else
+          @info[6]=0
+          @work[1]=@maxstep
+       end        
+       
+       ptime = FFI::MemoryPointer.new(:double)
+       ptime.write_double(time)
+
+       ptout  = FFI::MemoryPointer.new(:double)
+       ptout.write_double(tout)
+       
+       idid=1
+       pidid  = FFI::MemoryPointer.new(:int)
+       pidid.write_int(idid)
+       
+       @pwork.write_array_of_double(@work)
+       @pinfo.write_array_of_int(@info)
+
+      
+       while (idid==1) do
+          ddaspk_(resproc,@pneq,ptime,@py,@pyprime,ptout,@pinfo,@prtol,@patol,pidid,@pwork,@plwr,@piwork,@plwi,nil,nil,jacproc,nil,nil,nil,nil,nil,nil)
+          idid = pidid.read_int
+#
+# report intermediate values
+#
+          if ((idid==1)&&(@inter)) then
+             @inter.call(ptime.read_double,@py.read_array_of_double(@neq),@pyprime.read_array_of_double(@neq))
+          end
+
+       end
+
+       @work=@pwork.read_array_of_double(@lwr) # save the working ram
+       @info[0]=1
+
+       [ptime.read_double,@py.read_array_of_double(@neq),@pyprime.read_array_of_double(@neq),idid]
+
+    end
+
+    def reset
+       @info[0]=0
+       @py.write_array_of_double(@y)
+       @pyprime.write_array_of_double(@yprime)
+    end
+
+
+ end
+
+end

metadata

@@ -0,0 +1,65 @@
+--- !ruby/object:Gem::Specification 
+name: rb-daspk
+version: !ruby/object:Gem::Version 
+  version: 0.0.1
+platform: universal-darwin-10
+authors: 
+- Eric Meyers
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2009-11-27 00:00:00 -05:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: ffi
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: 0.5.3
+    version: 
+description: Allows users to solve differential algebraic equations in Ruby, using Ruby constructs. Interfaces with the DASPK Fortran library. DASPK solves DAE's of the form G(t,y,y',p) = 0
+email: etm@ericmeyers.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- lib/libdaspk.dylib
+- lib/rb-daspk.rb
+has_rdoc: true
+homepage: 
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+requirements: []
+
+rubyforge_project: rb-daspk
+rubygems_version: 1.3.5
+signing_key: 
+specification_version: 3
+summary: DASPK Ruby Interface. Allows users to solve differential algebraic equations in Ruby, using Ruby constructs. Interfaces with the DASPK Fortran library. DASPK solves DAE's of the form G(t,y,y',p) = 0
+test_files: []
+