SWAMPy

✎ CARVE FLESH

EDITING: unit.rb

require 'strscan'
require 'bigdecimal'

module Tins::Unit
  Prefix = Struct.new(:name, :step, :multiplier, :fraction)

PREFIX_LC = [
    '', 'k', 'm', 'g', 't', 'p', 'e', 'z', 'y',
  ].each_with_index.map { |n, i| Prefix.new(n.freeze, 1000, 1000 ** i, false) }.freeze

PREFIX_UC = [
    '', 'K', 'M', 'G', 'T', 'P', 'E', 'Z', 'Y',
  ].each_with_index.map { |n, i| Prefix.new(n.freeze, 1024, 1024 ** i, false) }.freeze

PREFIX_F = [
    '', 'm', 'µ', 'n', 'p', 'f', 'a', 'z', 'y',
  ].each_with_index.map { |n, i| Prefix.new(n.freeze, 1000, 1000 ** -i, true) }.freeze

class ParserError < ArgumentError; end

module_function

def prefixes(identifier)
    case identifier
    when :uppercase, :uc, 1024
      PREFIX_UC
    when :lowercase, :lc, 1000
      PREFIX_LC
    when :fraction, :f, 0.001
      PREFIX_F
    when Array
      identifier
    end
  end

def format(value, format: '%f %U', prefix: 1024, unit: ?b)
    prefixes = prefixes(prefix)
    first_prefix = prefixes.first or
      raise ArgumentError, 'a non-empty array of prefixes is required'
    if value.zero?
      result = format.sub('%U', unit)
      result %= value
    else
      prefix = prefixes[
        (first_prefix.fraction ? -1 : 1) * Math.log(value.abs) / Math.log(first_prefix.step)
      ]
      result = format.sub('%U', "#{prefix.name}#{unit}")
      result %= (value / prefix.multiplier.to_f)
    end
  end

class UnitParser < StringScanner
    NUMBER = /([+-]?
               (?:0|[1-9]\d*)
               (?:
                 \.\d+(?i:e[+-]?\d+) |
                 \.\d+ |
                 (?i:e[+-]?\d+)
               )?
             )/x

def initialize(source, unit, prefixes = nil)
      super source
      if prefixes
        @unit_re    = unit_re(Tins::Unit.prefixes(prefixes), unit)
        @unit_lc_re = @unit_uc_re = nil
      else
        @unit_lc_re = unit_re(Tins::Unit.prefixes(:lc), unit)
        @unit_uc_re = unit_re(Tins::Unit.prefixes(:uc), unit)
        @unit_re    = nil
      end
      @number       = 1.0
    end

def unit_re(prefixes, unit)
      re = Regexp.new(
        "(#{prefixes.reverse.map { |pre| Regexp.quote(pre.name) } * ?|})(#{unit})"
      )
      re.singleton_class.class_eval do
        define_method(:prefixes) { prefixes }
      end
      re
    end

private :unit_re

attr_reader :number

def scan(re)
      re.nil? and return
      super
    end

def scan_number
      scan(NUMBER) or return
      @number *= BigDecimal(self[1])
    end

def scan_unit
      case
      when scan(@unit_re)
        prefix = @unit_re.prefixes.find { |pre| pre.name == self[1] } or return
        @number *= prefix.multiplier
      when scan(@unit_lc_re)
        prefix = @unit_lc_re.prefixes.find { |pre| pre.name == self[1] } or return
        @number *= prefix.multiplier
      when scan(@unit_uc_re)
        prefix = @unit_uc_re.prefixes.find { |pre| pre.name == self[1] } or return
        @number *= prefix.multiplier
      end
    end

def scan_char(char)
      scan(/#{char}/) or return
    end

def parse
      raise NotImplementedError
    end
  end

class FormatParser < StringScanner
    def initialize(format, unit_parser)
      super format
      @unit_parser = unit_parser
    end

def reset
      super
      @unit_parser.reset
    end

def location
      @unit_parser.peek(10).inspect
    end

private :location

def parse
      reset
      until eos? || @unit_parser.eos?
        case
        when scan(/%f/)
          @unit_parser.scan_number or
            raise ParserError, "\"%f\" expected at #{location}"
        when scan(/%U/)
          @unit_parser.scan_unit or
            raise ParserError, "\"%U\" expected at #{location}"
        when scan(/%%/)
          @unit_parser.scan_char(?%) or
            raise ParserError, "#{?%.inspect} expected at #{location}"
        else
          char = scan(/./)
          @unit_parser.scan_char(char) or
            raise ParserError, "#{char.inspect} expected at #{location}"
        end
      end
      unless eos? && @unit_parser.eos?
        raise ParserError,
          "format #{string.inspect} and string "\
            "#{@unit_parser.string.inspect} do not match"
      end
      @unit_parser.number
    end
  end

# Parse the string +string+ if it matches +format+ with the unit +unit+ and
  # the prefixes specified by +prefix+.
  def parse(string, format: '%f %U', unit: ?b, prefix: nil)
    prefixes = prefixes(prefix)
    FormatParser.new(format, UnitParser.new(string, unit, prefixes)).parse
  end

def parse?(string, **options)
    parse(string, **options)
  rescue ParserError
    nil
  end
end

CANCEL

Name	Type	Size	Modified	Actions
↩ ..	DIR	—	—
📁 xt/	DIR	—	2023-09-24 03:48
📄 alias.rb	RB	13 B	2023-09-24 03:48	EDIT
📄 annotate.rb	RB	1.2 KB	2023-09-24 03:48	EDIT
📄 ask_and_send.rb	RB	707 B	2023-09-24 03:48	EDIT
📄 attempt.rb	RB	3.6 KB	2023-09-24 03:48	EDIT
📄 bijection.rb	RB	995 B	2023-09-24 03:48	EDIT
📄 case_predicate.rb	RB	109 B	2023-09-24 03:48	EDIT
📄 complete.rb	RB	372 B	2023-09-24 03:48	EDIT
📄 concern.rb	RB	804 B	2023-09-24 03:48	EDIT
📄 count_by.rb	RB	145 B	2023-09-24 03:48	EDIT
📄 date_dummy.rb	RB	1011 B	2023-09-24 03:48	EDIT
📄 date_time_dummy.rb	RB	1011 B	2023-09-24 03:48	EDIT
📄 deep_const_get.rb	RB	1.4 KB	2023-09-24 03:48	EDIT
📄 deep_dup.rb	RB	164 B	2023-09-24 03:48	EDIT
📄 dslkit.rb	RB	18.2 KB	2023-09-24 03:48	EDIT
📄 duration.rb	RB	1.9 KB	2023-09-24 03:48	EDIT
📄 expose.rb	RB	878 B	2023-09-24 03:48	EDIT
📄 extract_last_argument_options.rb	RB	315 B	2023-09-24 03:48	EDIT
📄 file_binary.rb	RB	2.6 KB	2023-09-24 03:48	EDIT
📄 find.rb	RB	3.8 KB	2023-09-24 03:48	EDIT
📄 generator.rb	RB	2 KB	2023-09-24 03:48	EDIT
📄 go.rb	RB	2.3 KB	2023-09-24 03:48	EDIT
📄 hash_symbolize_keys_recursive.rb	RB	1.2 KB	2023-09-24 03:48	EDIT
📄 hash_union.rb	RB	265 B	2023-09-24 03:48	EDIT
📄 if_predicate.rb	RB	87 B	2023-09-24 03:48	EDIT
📄 implement.rb	RB	1.1 KB	2023-09-24 03:48	EDIT
📄 limited.rb	RB	1 KB	2023-09-24 03:48	EDIT
📄 lines_file.rb	RB	2.5 KB	2023-09-24 03:48	EDIT
📄 lru_cache.rb	RB	912 B	2023-09-24 03:48	EDIT
📄 memoize.rb	RB	3.1 KB	2023-09-24 03:48	EDIT
📄 method_description.rb	RB	3 KB	2023-09-24 03:48	EDIT
📄 minimize.rb	RB	1.7 KB	2023-09-24 03:48	EDIT
📄 module_group.rb	RB	212 B	2023-09-24 03:48	EDIT
📄 named_set.rb	RB	151 B	2023-09-24 03:48	EDIT
📄 null.rb	RB	1.3 KB	2023-09-24 03:48	EDIT
📄 once.rb	RB	559 B	2023-09-24 03:48	EDIT
📄 p.rb	RB	627 B	2023-09-24 03:48	EDIT
📄 partial_application.rb	RB	859 B	2023-09-24 03:48	EDIT
📄 proc_compose.rb	RB	276 B	2023-09-24 03:48	EDIT
📄 proc_prelude.rb	RB	1.6 KB	2023-09-24 03:48	EDIT
📄 range_plus.rb	RB	112 B	2023-09-24 03:48	EDIT
📄 require_maybe.rb	RB	162 B	2023-09-24 03:48	EDIT
📄 responding.rb	RB	395 B	2023-09-24 03:48	EDIT
📄 secure_write.rb	RB	676 B	2023-09-24 03:48	EDIT
📄 sexy_singleton.rb	RB	1.6 KB	2023-09-24 03:48	EDIT
📄 string_byte_order_mark.rb	RB	714 B	2023-09-24 03:48	EDIT
📄 string_camelize.rb	RB	353 B	2023-09-24 03:48	EDIT
📄 string_underscore.rb	RB	293 B	2023-09-24 03:48	EDIT
📄 string_version.rb	RB	2.1 KB	2023-09-24 03:48	EDIT
📄 subhash.rb	RB	1.3 KB	2023-09-24 03:48	EDIT
📄 temp_io.rb	RB	648 B	2023-09-24 03:48	EDIT
📄 temp_io_enum.rb	RB	658 B	2023-09-24 03:48	EDIT
📄 terminal.rb	RB	699 B	2023-09-24 03:48	EDIT
📄 thread_local.rb	RB	1.5 KB	2023-09-24 03:48	EDIT
📄 time_dummy.rb	RB	1.4 KB	2023-09-24 03:48	EDIT
📄 timed_cache.rb	RB	1.2 KB	2023-09-24 03:48	EDIT
📄 to.rb	RB	156 B	2023-09-24 03:48	EDIT
📄 to_proc.rb	RB	178 B	2023-09-24 03:48	EDIT
📄 token.rb	RB	1.2 KB	2023-09-24 03:48	EDIT
📄 uniq_by.rb	RB	169 B	2023-09-24 03:48	EDIT
📄 unit.rb	RB	4.5 KB	2023-09-24 03:48	EDIT
📄 version.rb	RB	264 B	2023-09-24 03:48	EDIT
📄 write.rb	RB	460 B	2023-09-24 03:48	EDIT
📄 xt.rb	RB	1.3 KB	2023-09-24 03:48	EDIT