#!/usr/bin/env ruby
# frozen_string_literal: true

$stderr.print "starting.\r"

require 'bundler/setup'

$stderr.print "starting..\r"
require 'red_amber'
include RedAmber

$stderr.print "starting...\r"
require 'datasets-arrow'

$stderr.print "reading penguins...\r"
penguins = DataFrame.new(Datasets::Penguins.new)

$stderr.print "reading diamonds...\r"
diamonds = DataFrame.new(Datasets::Diamonds.new)

$stderr.print "reading starwars...\r"
starwars = DataFrame.new(Datasets::Rdataset.new('dplyr', 'starwars'))

$stderr.print "reading openintro/simpsons_paradox_covid...\r"
ds = Datasets::Rdataset.new('openintro', 'simpsons_paradox_covid')
simpsons_paradox_covid = DataFrame.new(ds.to_arrow)

$stderr.print "reading mtcars...                          \r"
mtcars = DataFrame.new(Datasets::Rdatasets.new('datasets', 'mtcars'))

$stderr.print "reading iris...  \r"
iris = DataFrame.new(Datasets::Iris.new)

$stderr.print "reading band_members...\r"
band_members = DataFrame.new(Datasets::Rdatasets.new('dplyr', 'band_members'))

$stderr.print "reading band_instruments...\r"
band_instruments = DataFrame.new(Datasets::Rdatasets.new('dplyr', 'band_instruments'))

$stderr.print "reading band_instruments2...\r"
band_instruments2 = DataFrame.new(Datasets::Rdatasets.new('dplyr', 'band_instruments2'))

$stderr.print "reading import_cars...      \r"
import_cars = DataFrame.load(Arrow::Buffer.new(<<~TSV), format: :tsv)
  Year	Audi	BMW	BMW_MINI	Mercedes-Benz	VW
  2017	28336	52527	25427	68221	49040
  2018	26473	50982	25984	67554	51961
  2019	24222	46814	23813	66553	46794
  2020	22304	35712	20196	57041	36576
  2021	22535	35905	18211	51722	35215
TSV

$stderr.print "reading comecome...   \r"
comecome = DataFrame.load(Arrow::Buffer.new(<<~CSV), format: :csv)
  name,age
  Yasuko,68
  Rui,49
  Hinata,28
CSV

$stderr.print "reading general dataframe and subframes...\r"
dataframe = DataFrame.new(
  x: [*1..6],
  y: %w[A A B B B C],
  z: [false, true, false, nil, true, false]
)
subframes = SubFrames.new(dataframe, [[0, 1], [2, 3, 4], [5]])

# Welcome to RedAmber example!
# This environment will offer these pre-loaded datasets:
#   penguins, diamonds, iris, starwars, simpsons_paradox_covid,
#   mtcars, band_members, band_instruments, band_instruments2
#   (original) import_cars, comecome, dataframe, subframes
binding.irb