red_amber 0.1.5 → 0.1.6

data/lib/red_amber/{vector_compensable.rb → vector_updatable.rb}

@@ -6,7 +6,52 @@
 module RedAmber
   # mix-ins for class Vector
   # Functions to make up some data (especially missing) for new data.
-  module VectorCompensable
+  module VectorUpdatable
+    # Replace data
+    # @param arg [Array, Vector, Arrow::Array] index specifier
+    # @param replacer [Array, Vector, Arrow::Array] new data to replace for.
+    # @return [Vector] Replaced new Vector
+    def replace(args, replacer)
+      args = args.is_a?(Array) ? args : Array(args)
+      replacer = Array(replacer)
+      return self if args.empty? || args[0].nil?
+
+      replacer = nil if replacer.empty?
+      vector = parse_to_vector(args)
+      booleans =
+        if vector.boolean?
+          vector
+        elsif vector.numeric?
+          Vector.new(indices).is_in(vector)
+        else
+          raise VectorArgumentError, "Invalid data type #{args}"
+        end
+      replace_with(booleans, replacer)
+    end
+
+    # (related functions)
+    # fill_null_backward, fill_null_forward
+
+    # [Ternary element-wise]: boolean_vector.func(if_true, else) => vector
+    def if_else(true_choice, false_choice)
+      true_choice = true_choice.data if true_choice.is_a? Vector
+      false_choice = false_choice.data if false_choice.is_a? Vector
+      raise VectorTypeError, 'Reciever must be a boolean' unless boolean?
+
+      datum = find(:if_else).execute([data, true_choice, false_choice])
+      Vector.new(datum.value)
+    end
+
+    # same behavior as Ruby's invert
+    # ![true, false, nil] #=> [false, true, true]
+    def primitive_invert
+      raise VectorTypeError, "Not a boolean Vector: #{self}" unless boolean?
+
+      is_nil.if_else(false, self).invert
+    end
+
+    private
+
     # [Ternary]: replace_with(booleans, replacements) => vector
     # Replace items selected with a boolean mask
     #
@@ -18,7 +63,7 @@ module RedAmber
     # or with null if the mask is null.
     # Hence, for replacement arrays, len(replacements) == sum(mask == true).
 
-    def replace_with(booleans, replacements = nil)
+    def replace_with(booleans, replacer = nil)
       specifier =
         if booleans.is_a?(Arrow::BooleanArray)
           booleans
@@ -32,14 +77,14 @@ module RedAmber
       raise VectorArgumentError, 'Booleans size unmatch' if specifier.length != size
       raise VectorArgumentError, 'Booleans not have any `true`' unless specifier.any?
 
-      r = Array(replacements) # scalar to [scalar]
+      r = Array(replacer) # scalar to [scalar]
       r = [nil] if r.empty?
 
       replacer =
         if r.size == 1
-          case replacements
-          when Arrow::Array then replacements
-          when Vector then replacements.data
+          case replacer
+          when Arrow::Array then replacer
+          when Vector then replacer.data
           else
             Arrow::Array.new(r * specifier.to_a.count(true)) # broadcast
           end
@@ -53,16 +98,7 @@ module RedAmber
       values = replacer.class.new(data)
 
       datum = find('replace_with_mask').execute([values, specifier, replacer])
-      take_out_element_wise(datum)
-    end
-
-    # (related functions)
-    # fill_null_backward, fill_null_forward
-
-    private
-
-    def booleans?(enum)
-      enum.all? { |e| e.is_a?(TrueClass) || e.is_a?(FalseClass) || e.is_a?(NilClass) }
+      Vector.new(datum.value)
     end
   end
 end

data/lib/red_amber/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module RedAmber
-  VERSION = '0.1.5'
+  VERSION = '0.1.6'
 end

data/lib/red_amber.rb

@@ -1,26 +1,3 @@
 # frozen_string_literal: true
 
-require 'arrow'
-require 'rover-df'
-
-require_relative 'red_amber/data_frame_displayable'
-require_relative 'red_amber/data_frame_helper'
-require_relative 'red_amber/data_frame_indexable'
-require_relative 'red_amber/data_frame_selectable'
-require_relative 'red_amber/data_frame_observation_operation'
-require_relative 'red_amber/data_frame_variable_operation'
-require_relative 'red_amber/data_frame'
-require_relative 'red_amber/vector_compensable'
-require_relative 'red_amber/vector_functions'
-require_relative 'red_amber/vector'
-require_relative 'red_amber/version'
-
-module RedAmber
-  class Error < StandardError; end
-
-  class DataFrameArgumentError < ArgumentError; end
-  class DataFrameTypeError < TypeError; end
-
-  class VectorArgumentError < ArgumentError; end
-  class VectorTypeError < TypeError; end
-end
+require_relative 'red-amber'

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: red_amber
 version: !ruby/object:Gem::Version
-  version: 0.1.5
+  version: 0.1.6
 platform: ruby
 authors:
 - Hirokazu SUZUKI (heronshoes)
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2022-06-12 00:00:00.000000000 Z
+date: 2022-06-26 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: red-arrow
@@ -90,17 +90,19 @@ files:
 - doc/image/vector/unary_element_wise.png
 - doc/tdr.md
 - doc/tdr_ja.md
+- lib/red-amber.rb
 - lib/red_amber.rb
 - lib/red_amber/data_frame.rb
 - lib/red_amber/data_frame_displayable.rb
-- lib/red_amber/data_frame_helper.rb
 - lib/red_amber/data_frame_indexable.rb
 - lib/red_amber/data_frame_observation_operation.rb
 - lib/red_amber/data_frame_selectable.rb
 - lib/red_amber/data_frame_variable_operation.rb
+- lib/red_amber/helper.rb
 - lib/red_amber/vector.rb
-- lib/red_amber/vector_compensable.rb
 - lib/red_amber/vector_functions.rb
+- lib/red_amber/vector_selectable.rb
+- lib/red_amber/vector_updatable.rb
 - lib/red_amber/version.rb
 - red_amber.gemspec
 - sig/red_amber.rbs

data/lib/red_amber/data_frame_helper.rb

@@ -1,64 +0,0 @@
-# frozen_string_literal: true
-
-module RedAmber
-  # mix-in for the class DataFrame
-  module DataFrameHelper
-    private
-
-    def expand_range(args)
-      ary = args.each_with_object([]) do |e, a|
-        e.is_a?(Range) ? a.concat(normalized_array(e)) : a.append(e)
-      end
-      ary.map do |e|
-        if e.is_a?(Integer) && e.negative?
-          e + size
-        else
-          e
-        end
-      end
-    end
-
-    def normalized_array(range)
-      both_end = [range.begin, range.end]
-      both_end[1] -= 1 if range.exclude_end? && range.end.is_a?(Integer)
-
-      if both_end.any?(Integer) || both_end.all?(&:nil?)
-        if both_end.any? { |e| e&.>=(size) || e&.<(-size) }
-          raise DataFrameArgumentError, "Index out of range: #{range} for 0..#{size - 1}"
-        end
-
-        (0...size).to_a[range]
-      else
-        range.to_a
-      end
-    end
-
-    def out_of_range?(indeces)
-      indeces.max >= size || indeces.min < -size
-    end
-
-    def integers?(enum)
-      enum.all?(Integer)
-    end
-
-    def sym_or_str?(enum)
-      enum.all? { |e| e.is_a?(Symbol) || e.is_a?(String) }
-    end
-
-    def booleans?(enum)
-      enum.all? { |e| e.is_a?(TrueClass) || e.is_a?(FalseClass) || e.is_a?(NilClass) }
-    end
-
-    def create_dataframe_from_vector(key, vector)
-      DataFrame.new(key => vector.data)
-    end
-
-    def select_obs_by_boolean(array)
-      DataFrame.new(@table.filter(array))
-    end
-
-    def keys_by_booleans(booleans)
-      keys.select.with_index { |_, i| booleans[i] }
-    end
-  end
-end