carray 1.4.0 → 1.5.1

data/lib/carray/ordering.rb

@@ -0,0 +1,157 @@
+# ----------------------------------------------------------------------------
+#
+#  carray/ordering.rb
+#
+#  This file is part of Ruby/CArray extension library.
+#  You can redistribute it and/or modify it under the terms of
+#  the Ruby Licence.
+#
+#  Copyright (C) 2005 Hiroki Motoyoshi
+#
+# ----------------------------------------------------------------------------
+
+class CArray
+  
+  # reversed
+
+  def reversed
+    return self[*([-1..0]*ndim)]
+  end
+  
+  # roll / shift
+
+  def shift! (*argv, &block)
+    self[] = self.shifted(*argv, &block)
+    return self
+  end
+
+  def shift (*argv, &block)
+    return self.shifted(*argv, &block).to_ca
+  end
+
+  def rolled (*argv)
+    argv.push({:roll => Array.new(ndim){1} })
+    return shifted(*argv)
+  end
+
+  def roll! (*argv)
+    self[] = self.rolled(*argv)
+    return self
+  end
+
+  def roll (*argv)
+    return self.rolled(*argv).to_ca
+  end
+
+    
+  # Returns the reference which elements are sorted by the comparison method
+  # given as block
+  def sorted_by (type=nil, opt={}, &block)
+    type, bytes =
+      CArray.guess_type_and_bytes(type||data_type, opt[:bytes]||bytes)
+    cmpary = convert(type, :bytes=>bytes, &block)
+    return self[cmpary.sort_addr]
+  end
+
+  # Returns the array which elements are sorted by the comparison method
+  # given as block
+  def sort_by (type=nil, opt={}, &block)
+    type, bytes =
+      CArray.guess_type_and_bytes(type||data_type, opt[:bytes]||bytes)
+    cmpary = convert(type, :bytes=>bytes, &block)
+    return self[cmpary.sort_addr].to_ca
+  end
+
+  def max_by (&block)
+    if empty?
+      return UNDEF
+    else
+      addr = convert(:object, &block).max_addr
+      return self[addr]
+    end
+  end
+
+  def min_by (&block)
+    if empty?
+      return UNDEF
+    else
+      addr = convert(:object, &block).min_addr
+      return self[addr]
+    end
+  end
+
+  def range 
+    return (self.min)..(self.max)
+  end
+
+  def nlargest (n)
+    obj = self.to_ca
+    list = []
+    n.times do |i|
+      k = obj.max_addr
+      list << obj[k]
+      obj[k] = UNDEF
+    end
+    list.to_ca.to_type(data_type)
+  end
+
+  def nlargest_addr (n)
+    obj = self.to_ca
+    list = []
+    n.times do |i|
+      k = obj.max_addr
+      list << k
+      obj[k] = UNDEF
+    end
+    CA_INT64(list)
+  end
+
+  def nsmallest (n)
+    obj = self.to_ca
+    list = []
+    n.times do |i|
+      k = obj.min_addr
+      list << obj[k]
+      obj[k] = UNDEF
+    end
+    list.to_ca.to_type(data_type)
+  end
+
+  def nsmallest_addr (n)
+    obj = self.to_ca
+    list = []
+    n.times do |i|
+      k = obj.min_addr
+      list << k
+      obj[k] = UNDEF
+    end
+    CA_INT64(list)
+  end
+
+  def order (dir = 1)
+    if dir >= 0   ### ascending order
+      if has_mask?
+        obj = template(:int32) { UNDEF }
+        sel = is_not_masked
+        obj[sel][self[sel].sort_addr].seq!
+        return obj
+      else
+        obj = template(:int32)
+        obj[sort_addr].seq!
+        return obj
+      end
+    else           ### descending order
+      if has_mask?
+        obj = template(:int32) { UNDEF}
+        sel = is_not_masked
+        obj[sel][self[sel].sort_addr.reversed].seq!
+        return obj
+      else  
+        obj = template(:int32)
+        obj[sort_addr.reversed].seq!
+        return obj
+      end
+    end
+  end
+
+end

data/lib/carray/{base/serialize.rb → serialize.rb}

@@ -1,6 +1,6 @@
 # ----------------------------------------------------------------------------
 #
-#  carray/base/serialize.rb
+#  carray/serialize.rb
 #
 #  This file is part of Ruby/CArray extension library.
 #  You can redistribute it and/or modify it under the terms of
@@ -33,7 +33,7 @@
 # offset 20 bytes
 #   data_type        : int32  : data type specifier
 #   bytes            : int32  : byte size of each data
-#   rank             : int32  : rank of array
+#   ndim             : int32  : ndim of array
 #   elements         : int32  : number of all elements
 #   has_mask         : int32  : 0 (not masked) or 1 (masked)
 # offset 40 bytes
@@ -54,13 +54,26 @@ class CArray::Serializer   # :nodoc:
     char_p    :endian,         :bytes=>4
     int32     :data_type
     int64     :bytes
-    int32     :rank
+    int32     :ndim
     int64     :elements
     int32     :has_mask
     array     :dim,            :type => CArray.int64(CA_RANK_MAX)
     int32     :has_attr
   }
 
+  Header_Legacy = CA.struct(:pack=>1, :size=>256) {
+    char_p    :magic_string,   :bytes=>8
+    char_p    :data_type_name, :bytes=>8
+    char_p    :endian,         :bytes=>4
+    int32     :data_type
+    int32     :bytes
+    int32     :ndim
+    int32     :elements
+    int32     :has_mask
+    array     :dim,            :type => CArray.int32(CA_RANK_MAX)
+    int32     :has_attr
+  }
+
   def initialize (io)
     case io
     when String
@@ -79,10 +92,10 @@ class CArray::Serializer   # :nodoc:
     header[:endian]         = ( endian == CA_LITTLE_ENDIAN ) ? "_LE_" : "_BE_"
     header[:data_type]      = ca.data_type
     header[:bytes]          = ca.bytes
-    header[:rank]           = ca.rank
+    header[:ndim]           = ca.ndim
     header[:elements]       = ca.elements
     header[:has_mask]       = ca.has_mask? ? 1 : 0
-    header[:dim][[0,ca.rank]] = ca.dim
+    header[:dim][[0,ca.ndim]] = ca.dim
     attr = nil
     if ca.attribute
       attr = ca.attribute.clone
@@ -114,8 +127,12 @@ class CArray::Serializer   # :nodoc:
     return ca
   end
   
-  def load (opt = {})
-    header = Header.decode(@io.read(256))
+  def load (opt = {}, legacy: false)
+    if legacy
+      header = Header_Legacy.decode(@io.read(256))
+    else
+      header = Header.decode(@io.read(256))
+    end
     if header[:magic_string] != "_CARRAY_"
       raise "not a CArray binary data"      
     end
@@ -130,10 +147,10 @@ class CArray::Serializer   # :nodoc:
     end
     data_type = header[:data_type]
     bytes     = header[:bytes]
-    rank      = header[:rank]
+    ndim      = header[:ndim]
     elements  = header[:elements]
     has_mask  = header[:has_mask] != 0 ? true : false
-    dim       = header[:dim][[0, rank]].to_a
+    dim       = header[:dim][[0, ndim]].to_a
     has_attr  = header[:has_attr]
     if data_type == 255
       data_type = header[:data_type_name].strip.to_sym
@@ -202,7 +219,8 @@ class CArray
 
   def marshal_dump ()
     if self.class != CArray and self.class != CScalar
-      raise TypeError, "can't dump a virtual or wrapped array."
+      return CArray.dump(self.to_ca)
+#      raise TypeError, "can't dump a virtual or wrapped array."
     end
     return CArray.dump(self)
   end
@@ -213,48 +231,5 @@ class CArray
     initialize_copy(ca)
   end
 
-  ### obsolete methods
-
-  def save_binary (filename, opt={})    # :nodoc: 
-    warn "CArray#save_binary will be obsolete, use CArray.save"
-    open(filename, "w") { |io|
-      return Serializer.new(io).save(self, opt)
-    }
-  end
-
-  def self.load_binary (filename, opt={})     # :nodoc: 
-    warn "CArray.load_binary will be obsolete, use CArray.load"
-    open(filename) { |io|
-      return Serializer.new(io).load(opt)
-    }
-  end
-
-  def save_binary_io (io, opt={})        # :nodoc:
-    warn "CArray#save_binary_io will be obsolete, use CArray.save"
-    return Serializer.new(io).save(self, opt) 
-  end
-
-  def self.load_binary_io (io, opt={})   # :nodoc:
-    warn "CArray#load_binary_io will be obsolete, use CArray.load"
-    return Serializer.new(io).load(opt)   
-  end 
-
-  def to_binary (io="", opt={})          # :nodoc:
-    warn "CArray#to_binary will be obsolete, use CArray.dump"
-    Serializer.new(io).save(self, opt) 
-    return io
-  end
-
-  def self.from_binary (io, opt={})      # :nodoc:
-    warn "CArray.from_binary will be obsolete, use CArray.load"
-    return Serializer.new(io).load(opt)   
-  end 
-
-  # depleted methods
-
-  def self.load_from_file (filename, data_type, dim, opt={}) # :nodoc:
-    raise "Sorry, CArray.load_from_file is depleted"
-  end
-
 end
 

data/lib/carray/{base/string.rb → string.rb}

@@ -1,11 +1,19 @@
+# ----------------------------------------------------------------------------
+#
+#  carray/string.rb
+#
+#  This file is part of Ruby/CArray extension library.
+#  You can redistribute it and/or modify it under the terms of
+#  the Ruby Licence.
+#
+#  Copyright (C) 2005 Hiroki Motoyoshi
+#
+# ----------------------------------------------------------------------------
+
 require "date"
 
 class CArray
 
-  def test (&block)
-    return convert(:boolean) {|v| yield(v) ? true : false }
-  end
-
   def str_len ()
     return convert(:int, &:length)
   end
@@ -178,65 +186,5 @@ class CArray
     return convert {|s| regexp.match(s) {|m| m[0].sub(regexp, replace) } || "" }
   end
     
-  def str_to_datetime (template = nil)
-    if template
-      return convert() {|v| DateTime.strptime(v, template) }
-    else
-      return convert() {|v| DateTime.parse(v) }
-    end
-  end
-
-  def str_to_time (template = nil)
-    if template
-      return str_strptime(template)
-    else
-      return convert() {|v| Time.parse(v) }
-    end
-  end
-
-  def time_format (template = nil)
-    if template
-      return str_strftime(template)
-    else
-      return convert(&:to_s)
-    end    
-  end
-
-  def time_year
-    return convert(:int, &:year)
-  end
-
-  def time_month
-    return convert(:int, &:month)
-  end
-
-  def time_day
-    return convert(:int, &:day)
-  end
-
-  def time_hour 
-    return convert(:int, &:hour)
-  end
-
-  def time_minute
-    return convert(:int, &:minute)
-  end
-
-  def time_second
-    return convert(:double) {|d| d.second + d.second_fraction }
-  end
-
-  def time_jd
-    return convert(:int, &:jd)
-  end
-
-  def time_ajd
-    return convert(:double, &:ajd)
-  end
-
-  def time_is_leap
-    return test(&:leap?)
-  end
-
 end
 

data/lib/carray/{base/struct.rb → struct.rb}

@@ -1,6 +1,6 @@
 # ----------------------------------------------------------------------------
 #
-#  carray/base/struct.rb
+#  carray/struct.rb
 #
 #  This file is part of Ruby/CArray extension library.
 #  You can redistribute it and/or modify it under the terms of
@@ -22,7 +22,7 @@
 # The implementation of other properties (cf. initialization, instance,
 # methods ...) are left free.
 #
-# CA::Struct and CA::Union are examples of such data class.
+# CAStruct and CAUnion are examples of such data class.
 #
 # option = {
 #   :pack => 1,   # nil for alignment, int for pack(n)
@@ -90,7 +90,7 @@ class CArray
 
 end
 
-class CA::Struct
+class CAStruct
 
   include Enumerable
 
@@ -264,7 +264,7 @@ class CA::Struct
 
 end
 
-class CA::Union < CA::Struct
+class CAUnion < CAStruct
   class << self
     def inspect
       return name.nil? ? "AnonUnion" : name
@@ -275,11 +275,11 @@ end
 module CA
 
   def self.struct (opt={}, &block)
-    return Struct::Builder.new(:struct, opt).define(&block)
+    return CAStruct::Builder.new(:struct, opt).define(&block)
   end
 
   def self.union (opt={}, &block)
-    return Struct::Builder.new(:union, opt).define(&block)
+    return CAStruct::Builder.new(:union, opt).define(&block)
   end
 
 end
@@ -290,7 +290,7 @@ end
 #
 # ---------------------------------------------------------------------
 
-class CA::Struct::Builder  # :nodoc:
+class CAStruct::Builder  # :nodoc:
 
   class Member # :nodoc:
 
@@ -358,9 +358,9 @@ class CA::Struct::Builder  # :nodoc:
     # ---
     case @type
     when :struct
-      klass = Class.new(CA::Struct)
+      klass = Class.new(CAStruct)
     when :union
-      klass = Class.new(CA::Union)
+      klass = Class.new(CAUnion)
     end
     # ---
     if @align.nil?
@@ -477,7 +477,7 @@ class CA::Struct::Builder  # :nodoc:
     if block
       opt = {:pack => @align}.update(opt)
       st = self.class.new(:struct, opt).define(&block)
-    elsif opt[:type] and opt[:type] <= CA::Struct
+    elsif opt[:type] and opt[:type] <= CAStruct
       st = opt[:type]
     else
       raise "type is not given for struct member of struct"
@@ -494,7 +494,7 @@ class CA::Struct::Builder  # :nodoc:
     if block
       opt = {:pack => @align}.update(opt)
       st = self.class.new(:union, opt).define(&block)
-    elsif opt[:type] and opt[:type] <= CA::Struct
+    elsif opt[:type] and opt[:type] <= CAStruct
       st = opt[:type]
     else
       raise "type is not given for union member of struct"
@@ -556,14 +556,14 @@ end
 
 
 
-class CA::Struct
+class CAStruct
 
   def self.spec
     output = ""
     table  = self::MEMBER_TABLE
     stlist = []
     if self.name.nil?
-      if self <= CA::Union
+      if self <= CAUnion
         prefix = "union"
       else
         prefix = "struct"
@@ -573,7 +573,7 @@ class CA::Struct
     else
       output << sprintf("%s = ", self.name)
     end
-    if self < CA::Union
+    if self < CAUnion
       output << sprintf("CA.union(:size=>%i) {\n", self::DATA_SIZE)
     else
       output << sprintf("CA.struct(:size=>%i) {\n", self::DATA_SIZE)
@@ -582,10 +582,10 @@ class CA::Struct
       offset, type, option = *table[member]
       case type
       when Class
-        if type < CA::Struct
+        if type < CAStruct
           stlist << type
           if type.name.nil?
-            if type <= CA::Union
+            if type <= CAUnion
               prefix = "union"
             else
               prefix = "struct"