Contents

# frozen_string_literal: true

require_relative 'generator'

module Dagger
  module Generate
    # Abstract base for generating numeric values
    class NumericGenerator < Dagger::Generator
      def process(args)
        array(args).each do |arg|
          result = process_recurse(arg)
          yield result unless result.nil?
        end
      end

      private

      def process_recurse(arg)
        case arg
        when Hash
          process_hash(arg)
        when ::String
          str = format_string(arg)
          from_s(str) unless str.nil?
        else
          from_s(arg)
        end
      end

      def process_hash(hash)
        hash.each do |key, args|
          result = process_hash_item(key, args)
          return result unless result.nil?
        end
      end

      def process_hash_item(key, args)
        case args
        when ::String
          send("numeric_#{key}", dictionary[args])
        else
          send("numeric_#{key}", process_args(args))
        end
      end

      def process_args(args)
        args = array(args).flat_map { |arg| process_recurse(arg) }
        args.delete(nil)
        args
      end

      def numeric_values(args)
        args.map { |value| from_s(value) }
      end

      def numeric_sum(args)
        args.sum
      end

      def numeric_product(args)
        args.reduce(:*)
      end

      def numeric_arithmetic_mean(args)
        return if args.empty?

        numeric_sum(args) / args.length
      end

      def numeric_geometric_mean(args)
        return if args.empty?

        numeric_product(args)**(1.0 / args.length)
      end

      def numeric_harmonic_mean(args)
        return if args.empty?

        args.length / args.map { |n| 1.0 / n }.sum
      end
    end
  end
end

Version data entries

4 entries across 4 versions & 1 rubygems

Version	Path
ruby-dagger-0.4.0	lib/dagger/numeric_generator.rb
ruby-dagger-0.3.3	lib/dagger/numeric_generator.rb
ruby-dagger-0.3.2	lib/dagger/numeric_generator.rb
ruby-dagger-0.3.1	lib/dagger/numeric_generator.rb