write_xlsx 1.13.0 → 1.15.1

data/lib/write_xlsx/workbook/chart_data.rb

@@ -0,0 +1,188 @@
+# -*- coding: utf-8 -*-
+# frozen_string_literal: true
+
+module Writexlsx
+  class Workbook
+    module ChartData
+      private
+
+      #
+      # Convert a range formula such as Sheet1!$B$1:$B$5 into a sheet name and cell
+      # range such as ( 'Sheet1', 0, 1, 4, 1 ).
+      #
+      def get_chart_range(range) # :nodoc:
+        # Split the range formula into sheetname and cells at the last '!'.
+        pos = range.rindex('!')
+        return nil unless pos
+
+        if pos > 0
+          sheetname = range[0, pos]
+          cells = range[(pos + 1)..-1]
+        end
+
+        # Split the cell range into 2 cells or else use single cell for both.
+        if cells =~ /:/
+          cell_1, cell_2 = cells.split(":")
+        else
+          cell_1 = cells
+          cell_2 = cells
+        end
+
+        # Remove leading/trailing apostrophes and convert escaped quotes to single.
+        sheetname.sub!(/^'/, '')
+        sheetname.sub!(/'$/, '')
+        sheetname.gsub!("''", "'")
+
+        row_start, col_start = xl_cell_to_rowcol(cell_1)
+        row_end,   col_end   = xl_cell_to_rowcol(cell_2)
+
+        # Check that we have a 1D range only.
+        return nil if row_start != row_end && col_start != col_end
+
+        [sheetname, row_start, col_start, row_end, col_end]
+      end
+
+      #
+      # Add "cached" data to charts to provide the numCache and strCache data for
+      # series and title/axis ranges.
+      #
+      def add_chart_data # :nodoc:
+        worksheets = {}
+        seen_ranges = {}
+
+        # Map worksheet names to worksheet objects.
+        @worksheets.each { |worksheet| worksheets[worksheet.name] = worksheet }
+
+        # Build an array of the worksheet charts including any combined charts.
+        @charts.collect { |chart| [chart, chart.combined] }.flatten.compact
+          .each do |chart|
+            chart.formula_ids.each do |range, id|
+              # Skip if the series has user defined data.
+              if chart.formula_data[id]
+                seen_ranges[range] = chart.formula_data[id] unless seen_ranges[range]
+                next
+              # Check to see if the data is already cached locally.
+              elsif seen_ranges[range]
+                chart.formula_data[id] = seen_ranges[range]
+                next
+              end
+
+              # Convert the range formula to a sheet name and cell range.
+              sheetname, *cells = get_chart_range(range)
+
+              # Skip if we couldn't parse the formula.
+              next unless sheetname
+
+              # Handle non-contiguous ranges: (Sheet1!$A$1:$A$2,Sheet1!$A$4:$A$5).
+              # We don't try to parse the ranges. We just return an empty list.
+              if sheetname =~ /^\([^,]+,/
+                chart.formula_data[id] = []
+                seen_ranges[range] = []
+                next
+              end
+
+              # Raise if the name is unknown since it indicates a user error in
+              # a chart series formula.
+              raise "Unknown worksheet reference '#{sheetname}' in range '#{range}' passed to add_series()\n" unless worksheets[sheetname]
+
+              # Add the data to the chart.
+              # And store range data locally to avoid lookup if seen agein.
+              chart.formula_data[id] =
+                seen_ranges[range] = chart_data(worksheets[sheetname], cells)
+            end
+        end
+      end
+
+      def chart_data(worksheet, cells)
+        # Get the data from the worksheet table.
+        data = worksheet.get_range_data(*cells)
+
+        # Convert shared string indexes to strings.
+        data.collect do |token|
+          if token.is_a?(Hash)
+            string = @shared_strings.string(token[:sst_id])
+
+            # Ignore rich strings for now. Deparse later if necessary.
+            if string =~ /^<r>/ && string =~ %r{</r>$}
+              ''
+            else
+              string
+            end
+          else
+            token
+          end
+        end
+      end
+
+      #
+      # Sort internal and user defined names in the same order as used by Excel.
+      # This may not be strictly necessary but unsorted elements caused a lot of
+      # issues in the the Spreadsheet::WriteExcel binary version. Also makes
+      # comparison testing easier.
+      #
+      def sort_defined_names(names) # :nodoc:
+        names.sort do |a, b|
+          name_a  = normalise_defined_name(a[0])
+          name_b  = normalise_defined_name(b[0])
+          sheet_a = normalise_sheet_name(a[2])
+          sheet_b = normalise_sheet_name(b[2])
+          # Primary sort based on the defined name.
+          if name_a > name_b
+            1
+          elsif name_a < name_b
+            -1
+          elsif sheet_a >= sheet_b  # name_a == name_b
+            # Secondary sort based on the sheet name.
+            1
+          else
+            -1
+          end
+        end
+      end
+
+      # Used in the above sort routine to normalise the defined names. Removes any
+      # leading '_xmln.' from internal names and lowercases the strings.
+      def normalise_defined_name(name) # :nodoc:
+        name.sub(/^_xlnm./, '').downcase
+      end
+
+      # Used in the above sort routine to normalise the worksheet names for the
+      # secondary sort. Removes leading quote and lowercases the strings.
+      def normalise_sheet_name(name) # :nodoc:
+        name.sub(/^'/, '').downcase
+      end
+
+      #
+      # Extract the named ranges from the sorted list of defined names. These are
+      # used in the App.xml file.
+      #
+      def extract_named_ranges(defined_names) # :nodoc:
+        named_ranges = []
+
+        defined_names.each do |defined_name|
+          name, index, range = defined_name
+
+          # Skip autoFilter ranges.
+          next if name == '_xlnm._FilterDatabase'
+
+          # We are only interested in defined names with ranges.
+          next unless range =~ /^([^!]+)!/
+
+          sheet_name = ::Regexp.last_match(1)
+
+          # Match Print_Area and Print_Titles xlnm types.
+          if name =~ /^_xlnm\.(.*)$/
+            xlnm_type = ::Regexp.last_match(1)
+            name = "#{sheet_name}!#{xlnm_type}"
+          elsif index != -1
+            name = "#{sheet_name}!#{name}"
+          end
+
+          named_ranges << name
+        end
+
+        named_ranges
+      end
+    end
+  end
+end

data/lib/write_xlsx/workbook/format_preparation.rb

@@ -0,0 +1,199 @@
+# -*- coding: utf-8 -*-
+# frozen_string_literal: true
+
+module Writexlsx
+  class Workbook
+    module FormatPreparation
+      private
+
+      #
+      # Prepare all of the format properties prior to passing them to Styles.rb.
+      #
+      def prepare_format_properties # :nodoc:
+        # Separate format objects into XF and DXF formats.
+        prepare_formats
+
+        # Set the font index for the format objects.
+        prepare_fonts
+
+        # Set the number format index for the format objects.
+        prepare_num_formats
+
+        # Set the border index for the format objects.
+        prepare_borders
+
+        # Set the fill index for the format objects.
+        prepare_fills
+      end
+
+      #
+      # Iterate through the XF Format objects and separate them into XF and DXF
+      # formats.
+      #
+      def prepare_formats # :nodoc:
+        @formats.formats.each do |format|
+          xf_index  = format.xf_index
+          dxf_index = format.dxf_index
+
+          @xf_formats[xf_index] = format   if xf_index
+          @dxf_formats[dxf_index] = format if dxf_index
+        end
+      end
+
+      #
+      # Iterate through the XF Format objects and give them an index to non-default
+      # font elements.
+      #
+      def prepare_fonts # :nodoc:
+        fonts = {}
+
+        @xf_formats.each { |format| format.set_font_info(fonts) }
+
+        @font_count = fonts.size
+
+        # For the DXF formats we only need to check if the properties have changed.
+        @dxf_formats.each do |format|
+          # The only font properties that can change for a DXF format are: color,
+          # bold, italic, underline and strikethrough.
+          format.has_dxf_font(true) if format.color? || format.bold? || format.italic? || format.underline? || format.strikeout?
+        end
+      end
+
+      #
+      # Iterate through the XF Format objects and give them an index to non-default
+      # number format elements.
+      #
+      # User defined records start from index 0xA4.
+      #
+      def prepare_num_formats # :nodoc:
+        num_formats        = []
+        unique_num_formats = {}
+        index              = 164
+
+        (@xf_formats + @dxf_formats).each do |format|
+          num_format = format.num_format
+
+          # Check if num_format is an index to a built-in number format.
+          # Also check for a string of zeros, which is a valid number format
+          # string but would evaluate to zero.
+          #
+          if num_format.to_s =~ /^\d+$/ && num_format.to_s !~ /^0+\d/
+            # Number format '0' is indexed as 1 in Excel.
+            num_format = 1 if num_format == 0
+            # Index to a built-in number format.
+            format.num_format_index = num_format
+            next
+          elsif num_format.to_s == 'General'
+            # The 'General' format has an number format index of 0.
+            format.num_format_index = 0
+            next
+          end
+
+          if unique_num_formats[num_format]
+            # Number format has already been used.
+            format.num_format_index = unique_num_formats[num_format]
+          else
+            # Add a new number format.
+            unique_num_formats[num_format] = index
+            format.num_format_index = index
+            index += 1
+
+            # Only store/increase number format count for XF formats
+            # (not for DXF formats).
+            num_formats << num_format if ptrue?(format.xf_index)
+          end
+        end
+
+        @num_formats = num_formats
+      end
+
+      #
+      # Iterate through the XF Format objects and give them an index to non-default
+      # border elements.
+      #
+      def prepare_borders # :nodoc:
+        borders = {}
+
+        @xf_formats.each { |format| format.set_border_info(borders) }
+
+        @border_count = borders.size
+
+        # For the DXF formats we only need to check if the properties have changed.
+        @dxf_formats.each do |format|
+          key = format.get_border_key
+          format.has_dxf_border(true) if key =~ /[^0:]/
+        end
+      end
+
+      #
+      # Iterate through the XF Format objects and give them an index to non-default
+      # fill elements.
+      #
+      # The user defined fill properties start from 2 since there are 2 default
+      # fills: patternType="none" and patternType="gray125".
+      #
+      def prepare_fills # :nodoc:
+        fills = {}
+        index = 2    # Start from 2. See above.
+
+        # Add the default fills.
+        fills['0:0:0']  = 0
+        fills['17:0:0'] = 1
+
+        # Store the DXF colors separately since them may be reversed below.
+        @dxf_formats.each do |format|
+          next unless format.fill_style.pattern != 0 || format.fill_style.bg_color != 0 || format.fill_style.fg_color != 0
+
+          format.has_dxf_fill(true)
+          format.dxf_bg_color = format.fill_style.bg_color
+          format.dxf_fg_color = format.fill_style.fg_color
+        end
+
+        @xf_formats.each do |format|
+          # The following logical statements jointly take care of special cases
+          # in relation to cell colours and patterns:
+          # 1. For a solid fill (_pattern == 1) Excel reverses the role of
+          #    foreground and background colours, and
+          # 2. If the user specifies a foreground or background colour without
+          #    a pattern they probably wanted a solid fill, so we fill in the
+          #    defaults.
+          #
+          if format.fill_style.pattern == 1 && ne_0?(format.fill_style.bg_color) && ne_0?(format.fill_style.fg_color)
+            format.fill_style.fg_color, format.fill_style.bg_color = format.fill_style.bg_color, format.fill_style.fg_color
+          elsif format.fill_style.pattern <= 1 && ne_0?(format.fill_style.bg_color) && eq_0?(format.fill_style.fg_color)
+            format.fill_style.fg_color = format.fill_style.bg_color
+            format.fill_style.bg_color = 0
+            format.fill_style.pattern  = 1
+          elsif format.fill_style.pattern <= 1 && eq_0?(format.fill_style.bg_color) && ne_0?(format.fill_style.fg_color)
+            format.fill_style.bg_color = 0
+            format.fill_style.pattern  = 1
+          end
+
+          key = format.get_fill_key
+
+          if fills[key]
+            # Fill has already been used.
+            format.fill_index = fills[key]
+            format.has_fill(false)
+          else
+            # This is a new fill.
+            fills[key]        = index
+            format.fill_index = index
+            format.has_fill(true)
+            index += 1
+          end
+        end
+
+        @fill_count = index
+      end
+
+      def eq_0?(val)
+        !ptrue?(val)
+      end
+
+      def ne_0?(val)
+        !eq_0?(val)
+      end
+    end
+  end
+end

data/lib/write_xlsx/workbook/initialization.rb

@@ -0,0 +1,223 @@
+# -*- coding: utf-8 -*-
+# frozen_string_literal: true
+
+module Writexlsx
+  class Workbook
+    #
+    # Workbook initialization helpers extracted from Workbook to keep the main
+    # class focused on public API and orchestration.
+    #
+    module Initialization
+      private
+
+      def setup_core_state(file, options, default_formats)
+        @options = options.dup                 # for test
+        @default_formats = default_formats.dup # for test
+        @writer = Package::XMLWriterSimple.new
+
+        @file = file
+        @tempdir = options[:tempdir] ||
+                   File.join(
+                     Dir.tmpdir,
+                     Digest::MD5.hexdigest("#{Time.now.to_f}-#{Process.pid}")
+                   )
+
+        @date_1904       = options[:date_1904] || false
+        @activesheet     = 0
+        @firstsheet      = 0
+        @selected        = 0
+        @fileclosed      = false
+        @optimization    = options[:optimization] || 0
+        @excel2003_style = options[:excel2003_style] || false
+        @read_only       = 0
+
+        @strings_to_urls =
+          options[:strings_to_urls].nil? || options[:strings_to_urls] ? true : false
+
+        @max_url_length = if options[:max_url_length]
+                            [options[:max_url_length].to_i, 255].max
+                          else
+                            MAX_URL_LENGTH
+                          end
+      end
+
+      def setup_workbook_state(_options)
+        @worksheets        = Sheets.new
+        @charts            = []
+        @drawings          = []
+        @defined_names     = []
+        @named_ranges      = []
+        @custom_colors     = []
+        @doc_properties    = {}
+        @custom_properties = []
+        @image_types       = {}
+        @images            = []
+        @has_comments      = false
+        @has_metadata      = false
+        @has_embedded_images = false
+        @has_embedded_descriptions = false
+
+        @x_window      = 240
+        @y_window      = 15
+        @window_width  = 16_095
+        @window_height = 9_660
+        @tab_ratio     = 600
+      end
+
+      def setup_format_state(_default_formats)
+        @formats     = Formats.new
+        @xf_formats  = []
+        @dxf_formats = []
+        @num_formats = []
+      end
+
+      def setup_shared_strings
+        @shared_strings = Package::SharedStrings.new
+      end
+
+      def setup_embedded_assets
+        @embedded_image_indexes = {}
+        @embedded_images        = []
+      end
+
+      def setup_calculation_state
+        @calc_id      = 124519
+        @calc_mode    = 'auto'
+        @calc_on_load = true
+      end
+
+      def setup_default_formats
+        if @excel2003_style
+          add_format(
+            @default_formats.merge(
+              xf_index:    0,
+              font_family: 0,
+              font:        'Arial',
+              size:        10,
+              theme:       -1
+            )
+          )
+        else
+          add_format(@default_formats.merge(xf_index: 0))
+        end
+
+        # Add a default URL format.
+        @default_url_format = add_format(hyperlink: 1)
+      end
+
+      #
+      # Workbook の生成時のオプションハッシュを解析する
+      #
+      def process_workbook_options(*params)
+        case params.size
+        when 0
+          [{}, {}]
+        when 1 # one hash
+          options_keys = %i[tempdir date_1904 optimization excel2003_style strings_to_urls max_url_length]
+
+          hash = params.first
+          options = hash.select { |k, _v| options_keys.include?(k) }
+
+          default_format_properties =
+            hash[:default_format_properties] ||
+            hash.reject { |k, _v| options_keys.include?(k) }
+
+          [options, default_format_properties.dup]
+        when 2 # array which includes options and default_format_properties
+          options, default_format_properties = params
+          default_format_properties ||= {}
+
+          [options.dup, default_format_properties.dup]
+        end
+      end
+
+      def filename
+        setup_filename unless @filename
+        @filename
+      end
+
+      def fileobj
+        setup_filename unless @fileobj
+        @fileobj
+      end
+
+      def setup_filename # :nodoc:
+        if @file.respond_to?(:to_str) && @file != ''
+          @filename = @file
+          @fileobj  = nil
+        elsif @file.respond_to?(:write)
+          @filename = File.join(tempdir, Digest::MD5.hexdigest(Time.now.to_s) + '.xlsx.tmp')
+          @fileobj  = @file
+        else
+          raise "'#{@file}' must be valid filename String of IO object."
+        end
+      end
+
+      attr_reader :tempdir
+
+      #
+      # Sets the colour palette to the Excel defaults.
+      #
+      def set_color_palette # :nodoc:
+        @palette = [
+          [0x00, 0x00, 0x00, 0x00],    # 8
+          [0xff, 0xff, 0xff, 0x00],    # 9
+          [0xff, 0x00, 0x00, 0x00],    # 10
+          [0x00, 0xff, 0x00, 0x00],    # 11
+          [0x00, 0x00, 0xff, 0x00],    # 12
+          [0xff, 0xff, 0x00, 0x00],    # 13
+          [0xff, 0x00, 0xff, 0x00],    # 14
+          [0x00, 0xff, 0xff, 0x00],    # 15
+          [0x80, 0x00, 0x00, 0x00],    # 16
+          [0x00, 0x80, 0x00, 0x00],    # 17
+          [0x00, 0x00, 0x80, 0x00],    # 18
+          [0x80, 0x80, 0x00, 0x00],    # 19
+          [0x80, 0x00, 0x80, 0x00],    # 20
+          [0x00, 0x80, 0x80, 0x00],    # 21
+          [0xc0, 0xc0, 0xc0, 0x00],    # 22
+          [0x80, 0x80, 0x80, 0x00],    # 23
+          [0x99, 0x99, 0xff, 0x00],    # 24
+          [0x99, 0x33, 0x66, 0x00],    # 25
+          [0xff, 0xff, 0xcc, 0x00],    # 26
+          [0xcc, 0xff, 0xff, 0x00],    # 27
+          [0x66, 0x00, 0x66, 0x00],    # 28
+          [0xff, 0x80, 0x80, 0x00],    # 29
+          [0x00, 0x66, 0xcc, 0x00],    # 30
+          [0xcc, 0xcc, 0xff, 0x00],    # 31
+          [0x00, 0x00, 0x80, 0x00],    # 32
+          [0xff, 0x00, 0xff, 0x00],    # 33
+          [0xff, 0xff, 0x00, 0x00],    # 34
+          [0x00, 0xff, 0xff, 0x00],    # 35
+          [0x80, 0x00, 0x80, 0x00],    # 36
+          [0x80, 0x00, 0x00, 0x00],    # 37
+          [0x00, 0x80, 0x80, 0x00],    # 38
+          [0x00, 0x00, 0xff, 0x00],    # 39
+          [0x00, 0xcc, 0xff, 0x00],    # 40
+          [0xcc, 0xff, 0xff, 0x00],    # 41
+          [0xcc, 0xff, 0xcc, 0x00],    # 42
+          [0xff, 0xff, 0x99, 0x00],    # 43
+          [0x99, 0xcc, 0xff, 0x00],    # 44
+          [0xff, 0x99, 0xcc, 0x00],    # 45
+          [0xcc, 0x99, 0xff, 0x00],    # 46
+          [0xff, 0xcc, 0x99, 0x00],    # 47
+          [0x33, 0x66, 0xff, 0x00],    # 48
+          [0x33, 0xcc, 0xcc, 0x00],    # 49
+          [0x99, 0xcc, 0x00, 0x00],    # 50
+          [0xff, 0xcc, 0x00, 0x00],    # 51
+          [0xff, 0x99, 0x00, 0x00],    # 52
+          [0xff, 0x66, 0x00, 0x00],    # 53
+          [0x66, 0x66, 0x99, 0x00],    # 54
+          [0x96, 0x96, 0x96, 0x00],    # 55
+          [0x00, 0x33, 0x66, 0x00],    # 56
+          [0x33, 0x99, 0x66, 0x00],    # 57
+          [0x00, 0x33, 0x00, 0x00],    # 58
+          [0x33, 0x33, 0x00, 0x00],    # 59
+          [0x99, 0x33, 0x00, 0x00],    # 60
+          [0x99, 0x33, 0x66, 0x00],    # 61
+          [0x33, 0x33, 0x99, 0x00],    # 62
+          [0x33, 0x33, 0x33, 0x00]    # 63
+        ]
+      end
+    end
+  end
+end