rust 0.3 → 0.9

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 985f7e940ab123fa452dae63792bde90613b134176ff4eaf5b59a719dd8a1ed5
-  data.tar.gz: bef0bb5028c99cb43c8e5453fdbbaf687bad65f8e6b54b06f949e5df6fb61bda
+  metadata.gz: aaa404db11033ff42b529516ae4a3f3e252bdddf7677d082847ee06625144f8e
+  data.tar.gz: f72ebc2c95385b87a445f3fb8de3e517579b002e0606209903acd2327616befd
 SHA512:
-  metadata.gz: 418181b9357665ecc654e9a765e24aa792d6287dd5998b1b1bd8f3278e6951d785fff8afe9594c6c95e1f7a384cb08f844939728b62a96ab416259de1c14512b
-  data.tar.gz: 810f14821924bd1b0cebf4fbf9f52e510abc0e935be6cc2852294fd374d54411bcd2a1943f5b60d2ce93501abcaff9cac25dd458e1a8b354b665aab199705a4a
+  metadata.gz: 00567c9b6216f7e9dc1a4135dbea263eecae2178273851c023980233531729313e15147e1bec8e9521df2d4fee15e2c4c8bf9bd4a0e2d1475f1d383339a17c21
+  data.tar.gz: 89800fff95be559e6f6bbbc05b6b5f5b52b7d59c54c61cd76ef679876ce5de0c08b405d88eadf0b254c95306ad98f38120e5960a92ce8358fca19ed6cc5c181d

data/bin/ruby-rust

@@ -0,0 +1,3 @@
+#!/usr/bin/ruby
+
+system "irb -r \"rust\""

data/lib/{rust-csv.rb → rust/core/csv.rb}

@@ -1,9 +1,10 @@
-require_relative 'rust-core'
+require_relative '../core'
+require 'csv'
 
 module Rust
     class CSV
         def self.read_all(pattern, **options)
-            result = {}
+            result = DataFrameHash.new
             Dir.glob(pattern).each do |filename|
                 result[filename] = CSV.read(filename, **options)
             end
@@ -49,10 +50,9 @@ module Rust
             raise TypeError, "Expected Rust::DataFrame" unless dataframe.is_a?(Rust::DataFrame)
             
             write_headers = options[:headers] != false
-            options[:headers] = dataframe.column_names if options[:headers] == nil
+            options[:headers] = dataframe.column_names unless options[:headers]
             
             hash = {}
-            labels = nil
             ::CSV.open(filename, 'w', write_headers: write_headers, **options) do |csv|
                 dataframe.each do |row|
                     csv << row
@@ -93,3 +93,13 @@ module Rust
         end
     end
 end
+
+module Rust::RBindings
+    def read_csv(filename, **options)
+        Rust::CSV.read(filename, **options)
+    end
+    
+    def write_csv(filename, dataframe, **options)
+        Rust::CSV.write(filename, dataframe, **options)
+    end
+end

data/lib/rust/core/rust.rb

@@ -0,0 +1,157 @@
+require 'code-assertions'
+require 'stringio'
+require 'rinruby'
+
+module Rust
+    CLIENT_MUTEX = Mutex.new
+    R_MUTEX      = Mutex.new
+    
+    R_ENGINE     = RinRuby.new(echo: false)
+    
+    private_constant    :R_ENGINE
+    private_constant    :R_MUTEX
+    private_constant    :CLIENT_MUTEX
+    
+    @@debugging = $RUST_DEBUG || false
+    @@in_client_mutex = false
+    
+    def self.debug
+        @@debugging = true
+    end
+    
+    def self.debug?
+        return @@debugging
+    end
+    
+    def self.exclusive
+        result = nil
+        CLIENT_MUTEX.synchronize do
+            @@in_client_mutex = true
+            result = yield
+            @@in_client_mutex = false
+        end
+        return result
+    end
+    
+    def self.[]=(variable, value)
+        if value.is_a?(RustDatatype)
+            value.load_in_r_as(variable.to_s)
+        elsif value.is_a?(String) || value.is_a?(Numeric) || value.is_a?(Array) || value.is_a?(::Matrix)
+            R_ENGINE.assign(variable, value)
+        else
+            raise "Trying to assign #{variable} with #{value.class}; expected RustDatatype, String, Numeric, or Array"
+        end
+        
+    end
+    
+    def self.[](variable)
+        return RustDatatype.pull_variable(variable)
+    end
+    
+    def self._eval_big(r_command, return_warnings = false)
+        r_command = r_command.join("\n") if r_command.is_a?(Array)
+        
+        self._rexec(r_command, return_warnings) do |cmd|
+            result = true
+            instructions = cmd.lines
+            
+            while instructions.size > 0
+                current_command = ""
+                
+                while (instructions.size > 0) && (current_command.length + instructions.first.length < 10000)
+                    current_command << instructions.shift
+                end
+                
+                result &= R_ENGINE.eval(current_command)
+            end
+            
+            result
+        end
+    end
+    
+    def self._pull(r_command, return_warnings = false)
+        self._rexec(r_command, return_warnings) { |cmd| R_ENGINE.pull(cmd) }
+    end
+    
+    def self._eval(r_command, return_warnings = false)
+        self._rexec(r_command, return_warnings) { |cmd| R_ENGINE.eval(cmd) }
+    end
+    
+    def self._rexec(r_command, return_warnings = false)
+        puts "Calling _rexec with command: #{r_command}" if @@debugging
+        R_MUTEX.synchronize do
+            assert("This command must be executed in an exclusive block") { @@in_client_mutex }
+            
+            result = nil
+            begin
+                $stdout = StringIO.new
+                if return_warnings
+                    R_ENGINE.echo(true, true)
+                else
+                    R_ENGINE.echo(false, false)
+                end
+                result = yield(r_command)
+            ensure
+                R_ENGINE.echo(false, false)
+                warnings = $stdout.string
+                $stdout = STDOUT
+            end
+            
+            if return_warnings
+                puts " Got #{warnings.size} warnings, with result #{result.inspect[0...100]}" if @@debugging
+                return result, warnings.lines.map { |w| w.strip.chomp }
+            else
+                puts " Result: #{result.inspect[0...100]}" if @@debugging
+                return result
+            end
+        end
+    end
+    
+    def self.check_library(name)
+        self.exclusive do
+            result, _ = self._pull("require(\"#{name}\", character.only = TRUE)", true)
+            return result
+        end
+    end
+    
+    def self.load_library(name)
+        self.exclusive do
+            self._eval("library(\"#{name}\", character.only = TRUE)")
+        end
+        
+        return nil
+    end
+    
+    def self.install_library(name)
+        self.exclusive do
+            self._eval("install.packages(\"#{name}\", dependencies = TRUE)")
+        end
+        
+        return nil
+    end
+    
+    def self.prerequisite(library)
+        self.install_library(library) unless self.check_library(library)
+        self.load_library(library)
+    end
+end
+
+module Rust::RBindings
+    def data_frame(*args)
+        Rust::DataFrame.new(*args)
+    end
+end
+
+module Rust::TestCases
+    def self.sample_dataframe(columns, size=100)
+        result = Rust::DataFrame.new(columns)
+        size.times do |i|
+            result << columns.map { |c| yield i, c }
+        end
+        return result
+    end
+end
+
+def bind_r!
+    include Rust::RBindings
+end

data/lib/rust/core/types/all.rb

@@ -0,0 +1,4 @@
+self_path = File.expand_path(__FILE__)
+Dir.glob(File.dirname(self_path) + "/*.rb").each do |lib|
+    require_relative lib unless lib == self_path
+end