注册会员 登录
Project1 返回首页

QQEat https://rpg.blue/?326184 [收藏] [复制] [分享] [RSS] 这个人很懒，什么也没留下。

日志

【RM】Benchmark v1.0 脚本基准测试

已有 205 次阅读2023-1-15 11:40 |个人分类:脚本| Benchmark

  1. #encoding:utf-8
  2. #==============================================================================
  3. # ■ Benchmark v1.0 By QQEat
  4. #------------------------------------------------------------------------------
  5. # 　基准测试(优雅 + 完美强迫症打印)
  6. #==============================================================================
  7. #
  8. #  # 打印代码块执行时间(count 默认 1, 可在下面设置更改)
  9. #  Benchmark.measure(count){|r| ... }
  10. #
  11. #  # 测试代码快效率
  12. #  Benchmark.measure(10_000) { [*1..1000].shuffle }
  13. #
  14. #  # 汇报打印
  15. #  Benchmark.measure do |x|
  16. #    a = (1..100).to_a.shuffle!
  17. #    x.report('Array#sort', 100000)  { a.sort  }
  18. #    x.report('Array#sort!', 100000) { a.sort! }
  19. #    10.times{ x.report{ a.sort! } }
  20. #  end
  21. #

  23. module Benchmark
  24.   #--------------------------------------------------------------------------
  25.   # ● 常量
  26.   #--------------------------------------------------------------------------
  27.   ITERATIONS = 1 # 默认执行次数
  28.   MARGIN = 3 # 列间距
  29.   SPLIT_LINE = true # 是否显示列分割线
  30.   COLS = [
  31.     [7, '#', :R],
  32.     [10, 'user', :R],   # 用户时间
  33.     [10, 'system', :R], # 系统时间
  34.     [10, 'real', :R],   # 真实用时
  35.     [10, 'avg', :R],    # 平均用时
  36.     [10, 'times', :R],  # 执行次数
  37.   ].map.with_index{|v, i|
  38.     [v[0], v[1].send(v[2] == :L ? :ljust : :rjust, v[0]) + ' ' * MARGIN, v[2]]
  39.   } # [列宽, 列名, 对齐方向]
  40.   
  41.   #==============================================================================
  42.   # ■ Reporter
  43.   #==============================================================================
  44.   class Reporter
  45.     #--------------------------------------------------------------------------
  46.     # ● 初始化
  47.     #--------------------------------------------------------------------------
  48.     def initialize
  49.       @total = [0, 0, 0, 0]
  50.     end
  51.     #--------------------------------------------------------------------------
  52.     # ● 汇报执行(标签名, 执行次数)
  53.     #--------------------------------------------------------------------------
  54.     def report(label = nil, iterate = ITERATIONS)
  55.       result = Benchmark.send(:time, iterate) { yield }
  56.       result << result[-1] / iterate
  57.       Benchmark.send(:output, label, *result.map{|v| '%0.6f' % v }, iterate)
  58.       @total.map!.with_index { |t, i| t += result[i].to_f }
  59.     end
  60.   end
  61.   #--------------------------------------------------------------------------
  62.   # ● measure 测量脚本执行时间
  63.   #--------------------------------------------------------------------------
  64.   def self.measure(iterate = ITERATIONS, &block)
  65.     @row = 0
  66.     width = COLS.inject(0){|a,b| a+b[1].length } / 2 - 5
  67.     puts "#{'-'*width} BENCHMARK #{'-'*width}"
  68.     puts COLS.map{|v| v[1] }.join
  69.     reporter = Reporter.new
  70.     if block.arity == 0
  71.       result = time(iterate) { yield }
  72.     else
  73.       yield reporter
  74.       result = reporter.instance_variable_get(:@total)
  75.     end
  76.     print "\n"
  77.     output('Total', *result.map{|v| '%0.6f' % v }, '')
  78.     print "\n"
  79.   end
  80.   
  81.   class << self
  82.     private
  83.     #--------------------------------------------------------------------------
  84.     # ● 执行脚本(执行次数)
  85.     #   return：[用户时间, 系统时间, 真实时间]
  86.     #--------------------------------------------------------------------------
  87.     def time(iterate = ITERATIONS)
  88.       t = Process.times
  89.       initial = [t.utime, t.stime, Time.now]
  90.       iterate.times { yield }
  91.       t2 = Process.times
  92.       [t2.utime, t2.stime, Time.now].map!.with_index do |time, index|
  93.         time - initial[index]
  94.       end
  95.     end
  96.     #--------------------------------------------------------------------------
  97.     # ● 打印输出
  98.     #--------------------------------------------------------------------------
  99.     def output(*args)
  100.       args[-1] = args[-1].to_s.gsub(/(?=(?!\b)(\d{3})+$)/, ',')
  101.       (args[0] ||= @row.to_s) << ':'
  102.       args.map!.with_index{|v,i| v.to_s.scan(/.{1,#{COLS[i][0]}}/) }
  103.       row = args.max_by{|v| v.length }.length
  104.       row.times do |i|
  105.         bottom = i == row-1
  106.         args.each_with_index do |arr, n|
  107.           begin_ = n == 0
  108.           end_ = n == args.length - 1
  109.           col = COLS[n]
  110.           diff = args[0].length - arr.length
  111.           diff = 0 if diff < 0
  112.           str = (i < diff ? '' : arr[i-diff]) || ''
  113.           str = str.send(col[2] == :L ? :ljust : :rjust, col[0])
  114.           if SPLIT_LINE and !end_
  115.             hmar = (MARGIN - 1) / 2
  116.             str << ' ' * hmar << '|' << ' ' * (MARGIN - hmar - 1)
  117.           else
  118.             str << ' ' * MARGIN
  119.           end
  120.           str << "\n" if end_
  121.           print str
  122.         end
  123.       end
  124.       @row += 1
  125.     end
  126.   end
  127. end

鸡蛋
鲜花
收藏分享邀请 举报

全部作者的其他最新日志

评论 (0 个评论)

facelist doodle 涂鸦笔

您需要登录后才可以评论 登录 | 注册会员

Archiver|手机版|小黑屋|Project1游戏制作

GMT+8, 2026-10-10 03:27

Powered by Discuz! X5.0

© 2001-2026 Discuz! Team.

返回顶部