# -*- coding: utf-8 -*-
###############################################################################
#
# Chartsheet - A class for writing the Excel XLSX Chartsheet files.
#
# Used in conjunction with WriteXLSX
#
# Copyright 2000-2011, John McNamara, jmcnamara@cpan.org
# Convert to ruby by Hideo NAKAMURA, cxn03651@msj.biglobe.ne.jp
#

require 'write_xlsx/worksheet'

module Writexlsx
  class Chartsheet < Worksheet
    include Writexlsx::Utility

    attr_writer :chart

    def initialize(workbook, index, name)
      super
      @drawing           = Drawing.new
      @is_chartsheet     = true
      @chart             = nil
      @charts            = [1]
      @zoom_scale_normal = 0
      @print_style.orientation = false
    end

    #
    # Assemble and write the XML file.
    #
    def assemble_xml_file # :nodoc:
      return unless @writer
      @writer.xml_decl

      # Write the root chartsheet element.
      write_chartsheet

      # Write the worksheet properties.
      write_sheet_pr

      # Write the sheet view properties.
      write_sheet_views

      # Write the sheetProtection element.
      write_sheet_protection

      # Write the printOptions element.
      write_print_options

      # Write the worksheet page_margins.
      write_page_margins

      # Write the worksheet page setup.
      write_page_setup

      # Write the headerFooter element.
      write_header_footer

      # Write the drawing element.
      write_drawings

      # Close the worksheet tag.
      @writer.end_tag('chartsheet')

      # Close the XML writer object and filehandle.
      @writer.crlf
      @writer.close
    end

    def protect(password = '', options = {})
      @chart.protection = 1

      options[:sheet]     = 0
      options[:content]   = 1
      options[:scenarios] = 1

      super(password, options)
    end


    ###############################################################################
    #
    # Encapsulated Chart methods.
    #
    ###############################################################################

    def add_series(*args)
      @chart.add_series(*args)
    end

    def set_x_axis(*args)
      @chart.set_x_axis(*args)
    end

    def set_y_axis(*args)
      @chart.set_y_axis(*args)
    end

    def set_x2_axis(*args)
      @chart.set_x2_axis(*args)
    end

    def set_y2_axis(*args)
      @chart.set_y2_axis(*args)
    end

    def set_title(*args)
      @chart.set_title(*args)
    end

    def set_legend(*args)
      @chart.set_legend(*args)
    end

    def set_plotarea(*args)
      @chart.set_plotarea(*args)
    end

    def set_chartarea(*args)
      @chart.set_chartarea(*args)
    end

    def set_style(*args)
      @chart.set_style(*args)
    end

    def show_blanks_as(*args)
      @chart.show_blanks_as(*args)
    end

    def show_hidden_data(*args)
      @chart.show_hidden_data(*args)
    end

    #
    # Set up chart/drawings.
    #
    def prepare_chart(index, chart_id, drawing_id) # :nodoc:
      @chart.id = chart_id - 1

      drawing = Drawing.new
      @drawing = drawing
      @drawing.orientation = @print_style.orientation

      @external_drawing_links << [ '/drawing', "../drawings/drawing#{drawing_id}.xml" ]

      @drawing_links << [ '/chart', "../charts/chart#{chart_id}.xml"]
    end

    private

    #
    # Write the <chartsheet> element. This is the root element of Chartsheet.
    #
    def write_chartsheet # :nodoc:
      schema                 = 'http://schemas.openxmlformats.org/'
      xmlns                  = schema + 'spreadsheetml/2006/main'
      xmlns_r                = schema + 'officeDocument/2006/relationships'
      xmlns_mc               = schema + 'markup-compatibility/2006'
      xmlns_mv               = 'urn:schemas-microsoft-com:mac:vml'
      mc_ignorable           = 'mv'
      mc_preserve_attributes = 'mv:*'

      attributes = [
                    'xmlns',   xmlns,
                    'xmlns:r', xmlns_r
                   ]

      @writer.start_tag('chartsheet', attributes)
    end

    #
    # Write the <sheetPr> element for Sheet level properties.
    #
    def write_sheet_pr # :nodoc:

      attributes = []

      attributes << {'filterMode' => 1} if ptrue?(@filter_on)

      if ptrue?(@fit_page) || ptrue?(@tab_color)
        @writer.tag_elements('sheetPr', attributes) do
          write_tab_color
          write_page_set_up_pr
        end
      else
        @writer.empty_tag('sheetPr', attributes)
      end
    end
  end
end