Sha256: fbfa80792b3cfc77dc76ecd36e8c103b550dc796e8de135bbab5a5abaa3aaf03
Contents?: true
Size: 1.99 KB
Versions: 6
Compression:
Stored size: 1.99 KB
Contents
#
# Loading Tabular Data
# after Daniel Shiffman, by Martin Prout.
#
# This example demonstrates how to use CSV
# to retrieve data from a CSV file and make objects
# from that data.
#
# Here is what the CSV looks like:
#
# x,y,diameter,name
# 160,103,43.19838,Happy
# 372,137,52.42526,Sad
# 273,235,61.14072,Joyous
# 121,179,44.758068,Melancholy
#
require 'csv'
load_library 'bubble'
attr_reader :bubbles, :data
def setup
size(640, 360)
load_data
end
def draw
background(255)
# Display all bubbles
bubbles.each do |b|
b.display
b.rollover(mouse_x, mouse_y)
end
text_align(LEFT)
fill(0)
text('Click to add bubbles.', 10, height - 10)
end
def load_data
# Load CSV file into an Array of Hash objects
# :headers option indicates the file has a header row
@data = CSV.read('data/data.csv', :headers => true).map{|row| row.to_hash}
# The size of the array of Bubble objects is determined by the total number of 'rows' in the CSV
@bubbles = []
data.each do |row|
# You access the values via their column name (set by using headers option above)
x = row['x'].to_f
y = row['y'].to_f
d = row['diameter'].to_f
n = row['name']
# Make a Bubble object out of the data read
bubbles << Bubble.new(x, y, d, n)
end
end
def mouse_pressed
# Create a new 'row' hash
row = {'x' => mouse_x.to_s, 'y' => mouse_y.to_s, 'diameter' => random(40, 80).to_s, 'name' => 'Blah'}
# add the row to the existing data array
data << row
# If the table has more than 10 rows
if (data.size > 10)
# Delete the oldest row
data.shift
end
# read column names from data, and generate csv 'string' that can be written to file
column_names = data.first.keys
s = CSV.generate do |csv|
csv << column_names
data.each do |row|
csv << row.values
end
end
# Writing the csv data back to the same file, (also specify UTF-8 format)
File.open('data/data.csv', 'w:UTF-8') { |file| file.write(s)}
# And reloading it
load_data
end
Version data entries
6 entries across 6 versions & 1 rubygems