map_formulae_to_ruby.rb

require_relative 'map_values_to_ruby'

class MapFormulaeToRuby < MapValuesToRuby
  
  attr_accessor :sheet_names
  attr_accessor :worksheet
  
  FUNCTIONS = {
    '*' => 'multiply',
    '+' => 'add',
    '-' => 'subtract',
    '/' => 'divide',
    '<' => 'less_than?',
    '<=' => 'less_than_or_equal?',
    '<>' => 'not_equal?',
    '=' => 'excel_equal?',
    '>' => 'more_than?',
    '>=' => 'more_than_or_equal?',
    'ABS' => 'abs',
    'AND' => 'excel_and',
    'AVERAGE' => 'average',
    'CHOOSE' => 'choose',
    'COSH' => 'cosh',
    'COUNT' => 'count',
    'COUNTA' => 'counta',
    'FIND' => 'find',
    'IF' => 'excel_if',
    'IFERROR' => 'iferror',
    'INDEX' => 'index',
    'LEFT' => 'left',
    'MATCH' => 'excel_match',
    'MAX' => 'max',
    'MIN' => 'min',
    'MOD' => 'mod',
<<<<<<< HEAD
<<<<<<< HEAD
    'MONTH' => 'month',
    'OR' => 'excel_or',
=======
>>>>>>> parent of d46088e... Implemented EXP, INT, and MONTH functions
=======
>>>>>>> parent of d46088e... Implemented EXP, INT, and MONTH functions
    'PI' => 'pi',
    'PMT' => 'pmt',
    'ROUND' => 'round',
    'ROUNDDOWN' => 'rounddown',
    'ROUNDUP' => 'roundup',
    'SUBTOTAL' => 'subtotal',
    'SUM' => 'sum',
    'SUMIF' => 'sumif',
    'SUMIFS' => 'sumifs',
    'SUMPRODUCT' => 'sumproduct',
    'TODAY' => 'today',
    'VLOOKUP' => 'vlookup',
    '^' => 'power'
  }
  
  def prefix(symbol,ast)
    return map(ast) if symbol == "+"
    return "negative(#{map(ast)})"
  end
  
  def brackets(*contents)
    "(#{contents.map { |a| map(a) }.join(',')})"
  end
  
  def arithmetic(left,operator,right)
    "#{FUNCTIONS[operator.last]}(#{map(left)},#{map(right)})"
  end
  
  def string_join(*strings)
    "string_join(#{strings.map {|a| map(a)}.join(',')})"
  end
  
  def comparison(left,operator,right)
    "#{FUNCTIONS[operator.last]}(#{map(left)},#{map(right)})"
  end
  
  def function(function_name,*arguments)
    if FUNCTIONS.has_key?(function_name)
      "#{FUNCTIONS[function_name]}(#{arguments.map { |a| map(a) }.join(",")})"
    else
      raise NotSupportedException.new("Function #{function_name} not supported")
    end
  end
  
  def cell(reference)
    reference.downcase.gsub('$','')
  end
  
  def sheet_reference(sheet,reference)
    "#{sheet_names[sheet]}_#{map(reference)}"
  end
  
  def array(*rows)
    "[#{rows.map {|r| map(r)}.join(",")}]"
  end
  
  def row(*cells)
    "[#{cells.map {|r| map(r)}.join(",")}]"
  end
  
end