Module: Trxl::AvgSumFunction2

Defined in:

lib/trxl/trxl_grammar.rb

Instance Method Summary

• #eval(env = Environment.new) ⇒ Object
• #expressions ⇒ Object

Instance Method Details

#eval(env = Environment.new) ⇒ Object

# File 'lib/trxl/trxl_grammar.rb', line 7496

def eval(env = Environment.new)
  strict = true
  nr_of_vals = 0
  values = expressions
  strict_flag = values[0].eval(env)
  if strict_flag.is_a?(TrueClass) || strict_flag.is_a?(FalseClass)
    values.shift
    strict = strict_flag
  end
  values.inject(0) do |sum, e|
    next_val = e.eval(env)
    sum + if next_val.is_a?(Array)
      nr_of_vals = 0
      res = next_val.inject(0) do |next_sum, val|
        if val.is_a?(Array)
          if (size = val.compact.size) > 0
            next_sum + val.inject(0) { |s, v| s + (v || 0) } / size
          else
            next_sum
          end
        else
          nr_of_vals += 1 if val && (strict || (!strict && val != 0))
          next_sum + (val || 0)
        end
      end
      nr_of_vals != 0 ? res / nr_of_vals : res
    else
      next_val || 0
    end
  end
end

#expressions ⇒ Object

# File 'lib/trxl/trxl_grammar.rb', line 7528

def expressions
  [ expression ] + more_expressions.elements.map { |e| e.expression }
end