ceph/src/arrow/ruby/red-arrow/lib/arrow/raw-table-converter.rb

   1 # Licensed to the Apache Software Foundation (ASF) under one
   2 # or more contributor license agreements.  See the NOTICE file
   3 # distributed with this work for additional information
   4 # regarding copyright ownership.  The ASF licenses this file
   5 # to you under the Apache License, Version 2.0 (the
   6 # "License"); you may not use this file except in compliance
   7 # with the License.  You may obtain a copy of the License at
   8 #
   9 #   http://www.apache.org/licenses/LICENSE-2.0
  10 #
  11 # Unless required by applicable law or agreed to in writing,
  12 # software distributed under the License is distributed on an
  13 # "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
  14 # KIND, either express or implied.  See the License for the
  15 # specific language governing permissions and limitations
  16 # under the License.
  17
  18 module Arrow
  19   class RawTableConverter
  20     attr_reader :n_rows
  21     attr_reader :schema
  22     attr_reader :values
  23     def initialize(raw_table)
  24       @raw_table = raw_table
  25       convert
  26     end
  27
  28     private
  29     def convert
  30       if @raw_table.is_a?(::Array) and @raw_table[0].is_a?(Column)
  31         fields = @raw_table.collect(&:field)
  32         @schema = Schema.new(fields)
  33         @values = @raw_table.collect(&:data)
  34       else
  35         fields = []
  36         @values = []
  37         @raw_table.each do |name, array|
  38           array = ArrayBuilder.build(array) if array.is_a?(::Array)
  39           fields << Field.new(name.to_s, array.value_data_type)
  40           @values << array
  41         end
  42         @schema = Schema.new(fields)
  43       end
  44       @n_rows = @values[0].length
  45     end
  46   end
  47 end