v/vlib/benchmark/benchmark.v

183 lines
4.2 KiB
V
Raw Normal View History

module benchmark
import time
import term
2019-12-19 22:29:37 +01:00
2019-12-30 05:23:54 +01:00
const (
2020-05-22 17:36:09 +02:00
b_ok = term.ok_message('OK ')
b_fail = term.fail_message('FAIL')
b_skip = term.warn_message('SKIP')
b_spent = term.ok_message('SPENT')
2019-12-30 05:23:54 +01:00
)
2019-12-19 22:29:37 +01:00
pub struct Benchmark {
pub mut:
bench_timer time.StopWatch
2020-05-18 21:38:06 +02:00
verbose bool
no_cstep bool
step_timer time.StopWatch
2019-12-19 22:29:37 +01:00
ntotal int
nok int
nfail int
nskip int
2019-12-30 05:23:54 +01:00
nexpected_steps int
cstep int
bok string
bfail string
}
2019-12-19 22:29:37 +01:00
pub fn new_benchmark() Benchmark {
return Benchmark{
bench_timer: time.new_stopwatch({})
verbose: true
}
}
2020-04-18 04:49:03 +02:00
pub fn new_benchmark_no_cstep() Benchmark {
return Benchmark{
bench_timer: time.new_stopwatch({})
2020-04-18 04:49:03 +02:00
verbose: true
no_cstep: true
}
}
pub fn new_benchmark_pointer() &Benchmark {
return &Benchmark{
bench_timer: time.new_stopwatch({})
verbose: true
}
}
2020-05-17 13:51:18 +02:00
pub fn (mut b Benchmark) set_total_expected_steps(n int) {
2019-12-30 05:23:54 +01:00
b.nexpected_steps = n
}
2020-05-17 13:51:18 +02:00
pub fn (mut b Benchmark) stop() {
b.bench_timer.stop()
}
2020-05-17 13:51:18 +02:00
pub fn (mut b Benchmark) step() {
b.step_timer.restart()
2020-04-18 04:49:03 +02:00
if !b.no_cstep {
b.cstep++
}
}
2020-05-17 13:51:18 +02:00
pub fn (mut b Benchmark) fail() {
b.step_timer.stop()
b.ntotal++
b.nfail++
}
2020-05-17 13:51:18 +02:00
pub fn (mut b Benchmark) ok() {
b.step_timer.stop()
b.ntotal++
b.nok++
}
2020-05-17 13:51:18 +02:00
pub fn (mut b Benchmark) skip() {
b.step_timer.stop()
b.ntotal++
b.nskip++
}
2020-05-17 13:51:18 +02:00
pub fn (mut b Benchmark) fail_many(n int) {
b.step_timer.stop()
2019-12-19 22:29:37 +01:00
b.ntotal += n
b.nfail += n
}
2020-05-17 13:51:18 +02:00
pub fn (mut b Benchmark) ok_many(n int) {
b.step_timer.stop()
2019-12-19 22:29:37 +01:00
b.ntotal += n
b.nok += n
}
2020-05-17 13:51:18 +02:00
pub fn (mut b Benchmark) neither_fail_nor_ok() {
b.step_timer.stop()
}
pub fn start() Benchmark {
mut b := new_benchmark()
b.step()
return b
}
2020-05-17 13:51:18 +02:00
pub fn (mut b Benchmark) measure(label string) i64 {
b.ok()
res := b.step_timer.elapsed().microseconds()
2020-05-22 17:36:09 +02:00
println(b.step_message_with_label(b_spent, 'in $label'))
b.step()
return res
}
2019-12-30 05:23:54 +01:00
pub fn (b &Benchmark) step_message_with_label(label string, msg string) string {
timed_line := b.tdiff_in_ms(msg, b.step_timer.elapsed().microseconds())
if b.nexpected_steps > 1 {
2019-12-30 05:23:54 +01:00
mut sprogress := ''
if b.nexpected_steps < 10 {
2020-04-18 04:49:03 +02:00
sprogress = if b.no_cstep { 'TMP1/${b.nexpected_steps:1d}' } else {
'${b.cstep:1d}/${b.nexpected_steps:1d}'
}
} else if b.nexpected_steps >= 10 && b.nexpected_steps < 100 {
2020-04-18 04:49:03 +02:00
sprogress = if b.no_cstep { 'TMP2/${b.nexpected_steps:2d}' } else {
'${b.cstep:2d}/${b.nexpected_steps:2d}'
}
} else if b.nexpected_steps >= 100 && b.nexpected_steps < 1000 {
2020-04-18 04:49:03 +02:00
sprogress = if b.no_cstep { 'TMP3/${b.nexpected_steps:3d}' } else {
'${b.cstep:3d}/${b.nexpected_steps:3d}'
}
} else {
sprogress = if b.no_cstep { 'TMP4/${b.nexpected_steps:4d}' } else {
'${b.cstep:4d}/${b.nexpected_steps:4d}'
}
2019-12-30 05:23:54 +01:00
}
return '${label:-5s} [${sprogress}] ${timed_line}'
2019-12-30 05:23:54 +01:00
}
return '${label:-5s}${timed_line}'
}
pub fn (b &Benchmark) step_message(msg string) string {
2019-12-30 05:23:54 +01:00
return b.step_message_with_label('', msg)
}
pub fn (b &Benchmark) step_message_ok(msg string) string {
2020-05-22 17:36:09 +02:00
return b.step_message_with_label(b_ok, msg)
2019-12-30 05:23:54 +01:00
}
pub fn (b &Benchmark) step_message_fail(msg string) string {
2020-05-22 17:36:09 +02:00
return b.step_message_with_label(b_fail, msg)
}
pub fn (b &Benchmark) step_message_skip(msg string) string {
2020-05-22 17:36:09 +02:00
return b.step_message_with_label(b_skip, msg)
}
pub fn (b &Benchmark) total_message(msg string) string {
mut tmsg := '${msg}\n ok, fail, skip, total = ' + term.ok_message('${b.nok:5d}') + ', ' + if b.nfail > 0 { term.red('${b.nfail:5d}') } else { '${b.nfail:5d}' } + ', ' + if b.nskip > 0 { term.bright_yellow('${b.nskip:5d}') } else { '${b.nskip:5d}' } + ', ' + '${b.ntotal:5d}'
if b.verbose {
tmsg = '<=== total time spent $tmsg'
}
mut spaces := ' '
if b.nexpected_steps > 1 {
// NB: the formula below accounts for the progress bar [step/total]
str_steps := '$b.nexpected_steps'
x := 4 + str_steps.len * 2 + 5
spaces = ' '.repeat(x)
}
return spaces + b.tdiff_in_ms(tmsg, b.bench_timer.elapsed().microseconds())
}
// .total_duration - returns the duration in ms
pub fn (b &Benchmark) total_duration() i64 {
return b.bench_timer.elapsed().milliseconds()
}
2019-12-19 22:29:37 +01:00
// //////////////////////////////////////////////////////////////////
fn (b &Benchmark) tdiff_in_ms(s string, tdiff i64) string {
if b.verbose {
return '${f64(tdiff)/1000.0:9.3f} ms $s'
}
return s
}