#!/usr/bin/env ruby
# -*- coding: utf-8 -*-

#######################################################################
#
# A demo of a Stock chart in Excel::Writer::XLSX.
#
# reverse(c), March 2011, John McNamara, jmcnamara@cpan.org
# convert to ruby by Hideo NAKAMURA, cxn03651@msj.biglobe.ne.jp
#

require 'rubygems'
require 'write_xlsx'

workbook    = WriteXLSX.new('chart_stock.xlsx')
worksheet   = workbook.add_worksheet
bold        = workbook.add_format(:bold => 1)
date_format = workbook.add_format(:num_format => 'dd/mm/yyyy')
chart       = workbook.add_chart(:type => 'stock', :embedded => 1)


# Add the worksheet data that the charts will refer to.
headings = [ 'Date', 'High', 'Low', 'Close' ]
data = [
    [ '2007-01-01T', '2007-01-02T', '2007-01-03T', '2007-01-04T', '2007-01-05T' ],
    [ 27.2,  25.03, 19.05, 20.34, 18.5 ],
    [ 23.49, 19.55, 15.12, 17.84, 16.34 ],
    [ 25.45, 23.05, 17.32, 20.45, 17.34 ]
]

worksheet.write('A1', headings, bold)

(0 .. 4).each do |row|
    worksheet.write_date_time(row + 1, 0, data[0][row], date_format)
    worksheet.write(row + 1, 1, data[1][row])
    worksheet.write(row + 1, 2, data[2][row])
    worksheet.write(row + 1, 3, data[3][row])
end

worksheet.set_column('A:D', 11)

# Add a series for each of the High-Low-Close columns.
chart.add_series(
  :categories => '=Sheet1!$A$2:$A$6',
  :values     => '=Sheet1!$B$2:$B$6'
)

chart.add_series(
  :categories => '=Sheet1!$A$2:$A$6',
  :values     => '=Sheet1!$C$2:$C$6'
)

chart.add_series(
  :categories => '=Sheet1!$A$2:$A$6',
  :values     => '=Sheet1!$D$2:$D$6'
)

# Add a chart title and some axis labels.
chart.set_title(:name => 'High-Low-Close')
chart.set_x_axis(:name => 'Date')
chart.set_y_axis(:name => 'Share price')

worksheet.insert_chart('E9', chart)

workbook.close