message.rb in infobar-0.0.2

- old
+ new

@@ -1,8 +1,9 @@
 require 'infobar/spinner'
 require 'infobar/duration'
 require 'infobar/number'
+require 'infobar/rate'
 
 class Infobar::Message
   class << self
     prepend Tins::Delegate
 
@@ -86,15 +87,11 @@
       Infobar.counter.eta
     end
   end
 
   # rate with or without units
-  register('%r', unit: nil, prefix: 1000, format: '%f %U') do |directive, opts|
-    if opts[:unit]
-      Tins::Unit.format(Infobar.counter.rate, **opts)
-    else
-      Infobar.counter.rate
-    end
+  register('%r', unit: nil, prefix: 1000, format: '%.3f%U%t') do |directive, opts|
+    Infobar::Rate.new(Infobar.counter.rate, Infobar.counter.fifo_rate, **opts)
   end
 
   # average time as a duration
   register('%a', format: '%m:%s.%f') do |directive, opts|
     Infobar::Duration.new(Infobar.counter.average_time, **opts)