Try to fix the INC in PASTHRU.
[p5sagit/p5-mst-13.2.git] / lib / Benchmark.pm
1 package Benchmark;
2
3 use strict;
4
5
6 =head1 NAME
7
8 Benchmark - benchmark running times of Perl code
9
10 =head1 SYNOPSIS
11
12     use Benchmark qw(:all) ;
13
14     timethis ($count, "code");
15
16     # Use Perl code in strings...
17     timethese($count, {
18         'Name1' => '...code1...',
19         'Name2' => '...code2...',
20     });
21
22     # ... or use subroutine references.
23     timethese($count, {
24         'Name1' => sub { ...code1... },
25         'Name2' => sub { ...code2... },
26     });
27
28     # cmpthese can be used both ways as well
29     cmpthese($count, {
30         'Name1' => '...code1...',
31         'Name2' => '...code2...',
32     });
33
34     cmpthese($count, {
35         'Name1' => sub { ...code1... },
36         'Name2' => sub { ...code2... },
37     });
38
39     # ...or in two stages
40     $results = timethese($count, 
41         {
42             'Name1' => sub { ...code1... },
43             'Name2' => sub { ...code2... },
44         },
45         'none'
46     );
47     cmpthese( $results ) ;
48
49     $t = timeit($count, '...other code...')
50     print "$count loops of other code took:",timestr($t),"\n";
51
52     $t = countit($time, '...other code...')
53     $count = $t->iters ;
54     print "$count loops of other code took:",timestr($t),"\n";
55
56     # enable hires wallclock timing if possible
57     use Benchmark ':hireswallclock';
58
59 =head1 DESCRIPTION
60
61 The Benchmark module encapsulates a number of routines to help you
62 figure out how long it takes to execute some code.
63
64 timethis - run a chunk of code several times
65
66 timethese - run several chunks of code several times
67
68 cmpthese - print results of timethese as a comparison chart
69
70 timeit - run a chunk of code and see how long it goes
71
72 countit - see how many times a chunk of code runs in a given time
73
74
75 =head2 Methods
76
77 =over 10
78
79 =item new
80
81 Returns the current time.   Example:
82
83     use Benchmark;
84     $t0 = new Benchmark;
85     # ... your code here ...
86     $t1 = new Benchmark;
87     $td = timediff($t1, $t0);
88     print "the code took:",timestr($td),"\n";
89
90 =item debug
91
92 Enables or disable debugging by setting the C<$Benchmark::Debug> flag:
93
94     debug Benchmark 1;
95     $t = timeit(10, ' 5 ** $Global ');
96     debug Benchmark 0;
97
98 =item iters
99
100 Returns the number of iterations.
101
102 =back
103
104 =head2 Standard Exports
105
106 The following routines will be exported into your namespace
107 if you use the Benchmark module:
108
109 =over 10
110
111 =item timeit(COUNT, CODE)
112
113 Arguments: COUNT is the number of times to run the loop, and CODE is
114 the code to run.  CODE may be either a code reference or a string to
115 be eval'd; either way it will be run in the caller's package.
116
117 Returns: a Benchmark object.
118
119 =item timethis ( COUNT, CODE, [ TITLE, [ STYLE ]] )
120
121 Time COUNT iterations of CODE. CODE may be a string to eval or a
122 code reference; either way the CODE will run in the caller's package.
123 Results will be printed to STDOUT as TITLE followed by the times.
124 TITLE defaults to "timethis COUNT" if none is provided. STYLE
125 determines the format of the output, as described for timestr() below.
126
127 The COUNT can be zero or negative: this means the I<minimum number of
128 CPU seconds> to run.  A zero signifies the default of 3 seconds.  For
129 example to run at least for 10 seconds:
130
131         timethis(-10, $code)
132
133 or to run two pieces of code tests for at least 3 seconds:
134
135         timethese(0, { test1 => '...', test2 => '...'})
136
137 CPU seconds is, in UNIX terms, the user time plus the system time of
138 the process itself, as opposed to the real (wallclock) time and the
139 time spent by the child processes.  Less than 0.1 seconds is not
140 accepted (-0.01 as the count, for example, will cause a fatal runtime
141 exception).
142
143 Note that the CPU seconds is the B<minimum> time: CPU scheduling and
144 other operating system factors may complicate the attempt so that a
145 little bit more time is spent.  The benchmark output will, however,
146 also tell the number of C<$code> runs/second, which should be a more
147 interesting number than the actually spent seconds.
148
149 Returns a Benchmark object.
150
151 =item timethese ( COUNT, CODEHASHREF, [ STYLE ] )
152
153 The CODEHASHREF is a reference to a hash containing names as keys
154 and either a string to eval or a code reference for each value.
155 For each (KEY, VALUE) pair in the CODEHASHREF, this routine will
156 call
157
158         timethis(COUNT, VALUE, KEY, STYLE)
159
160 The routines are called in string comparison order of KEY.
161
162 The COUNT can be zero or negative, see timethis().
163
164 Returns a hash of Benchmark objects, keyed by name.
165
166 =item timediff ( T1, T2 )
167
168 Returns the difference between two Benchmark times as a Benchmark
169 object suitable for passing to timestr().
170
171 =item timestr ( TIMEDIFF, [ STYLE, [ FORMAT ] ] )
172
173 Returns a string that formats the times in the TIMEDIFF object in
174 the requested STYLE. TIMEDIFF is expected to be a Benchmark object
175 similar to that returned by timediff().
176
177 STYLE can be any of 'all', 'none', 'noc', 'nop' or 'auto'. 'all' shows
178 each of the 5 times available ('wallclock' time, user time, system time,
179 user time of children, and system time of children). 'noc' shows all
180 except the two children times. 'nop' shows only wallclock and the
181 two children times. 'auto' (the default) will act as 'all' unless
182 the children times are both zero, in which case it acts as 'noc'.
183 'none' prevents output.
184
185 FORMAT is the L<printf(3)>-style format specifier (without the
186 leading '%') to use to print the times. It defaults to '5.2f'.
187
188 =back
189
190 =head2 Optional Exports
191
192 The following routines will be exported into your namespace
193 if you specifically ask that they be imported:
194
195 =over 10
196
197 =item clearcache ( COUNT )
198
199 Clear the cached time for COUNT rounds of the null loop.
200
201 =item clearallcache ( )
202
203 Clear all cached times.
204
205 =item cmpthese ( COUNT, CODEHASHREF, [ STYLE ] )
206
207 =item cmpthese ( RESULTSHASHREF, [ STYLE ] )
208
209 Optionally calls timethese(), then outputs comparison chart.  This:
210
211     cmpthese( -1, { a => "++\$i", b => "\$i *= 2" } ) ;
212
213 outputs a chart like:
214
215            Rate    b    a
216     b 2831802/s   -- -61%
217     a 7208959/s 155%   --
218
219 This chart is sorted from slowest to fastest, and shows the percent speed
220 difference between each pair of tests.
221
222 c<cmpthese> can also be passed the data structure that timethese() returns:
223
224     $results = timethese( -1, { a => "++\$i", b => "\$i *= 2" } ) ;
225     cmpthese( $results );
226
227 in case you want to see both sets of results.
228
229 Returns a reference to an ARRAY of rows, each row is an ARRAY of cells from the
230 above chart, including labels. This:
231
232     my $rows = cmpthese( -1, { a => '++$i', b => '$i *= 2' }, "none" );
233
234 returns a data structure like:
235
236     [
237         [ '',       'Rate',   'b',    'a' ],
238         [ 'b', '2885232/s',  '--', '-59%' ],
239         [ 'a', '7099126/s', '146%',  '--' ],
240     ]
241
242 B<NOTE>: This result value differs from previous versions, which returned
243 the C<timethese()> result structure.  If you want that, just use the two
244 statement C<timethese>...C<cmpthese> idiom shown above.
245
246 Incidently, note the variance in the result values between the two examples;
247 this is typical of benchmarking.  If this were a real benchmark, you would
248 probably want to run a lot more iterations.
249
250 =item countit(TIME, CODE)
251
252 Arguments: TIME is the minimum length of time to run CODE for, and CODE is
253 the code to run.  CODE may be either a code reference or a string to
254 be eval'd; either way it will be run in the caller's package.
255
256 TIME is I<not> negative.  countit() will run the loop many times to
257 calculate the speed of CODE before running it for TIME.  The actual
258 time run for will usually be greater than TIME due to system clock
259 resolution, so it's best to look at the number of iterations divided
260 by the times that you are concerned with, not just the iterations.
261
262 Returns: a Benchmark object.
263
264 =item disablecache ( )
265
266 Disable caching of timings for the null loop. This will force Benchmark
267 to recalculate these timings for each new piece of code timed.
268
269 =item enablecache ( )
270
271 Enable caching of timings for the null loop. The time taken for COUNT
272 rounds of the null loop will be calculated only once for each
273 different COUNT used.
274
275 =item timesum ( T1, T2 )
276
277 Returns the sum of two Benchmark times as a Benchmark object suitable
278 for passing to timestr().
279
280 =back
281
282 =head2 :hireswallclock
283
284 If the Time::HiRes module has been installed, you can specify the
285 special tag C<:hireswallclock> for Benchmark (if Time::HiRes is not
286 available, the tag will be silently ignored).  This tag will cause the
287 wallclock time to be measured in microseconds, instead of integer
288 seconds.  Note though that the speed computations are still conducted
289 in CPU time, not wallclock time.
290
291 =head1 NOTES
292
293 The data is stored as a list of values from the time and times
294 functions:
295
296       ($real, $user, $system, $children_user, $children_system, $iters)
297
298 in seconds for the whole loop (not divided by the number of rounds).
299
300 The timing is done using time(3) and times(3).
301
302 Code is executed in the caller's package.
303
304 The time of the null loop (a loop with the same
305 number of rounds but empty loop body) is subtracted
306 from the time of the real loop.
307
308 The null loop times can be cached, the key being the
309 number of rounds. The caching can be controlled using
310 calls like these:
311
312     clearcache($key);
313     clearallcache();
314
315     disablecache();
316     enablecache();
317
318 Caching is off by default, as it can (usually slightly) decrease
319 accuracy and does not usually noticably affect runtimes.
320
321 =head1 EXAMPLES
322
323 For example,
324
325     use Benchmark qw( cmpthese ) ;
326     $x = 3;
327     cmpthese( -5, {
328         a => sub{$x*$x},
329         b => sub{$x**2},
330     } );
331
332 outputs something like this:
333
334    Benchmark: running a, b, each for at least 5 CPU seconds...
335           Rate    b    a
336    b 1559428/s   -- -62%
337    a 4152037/s 166%   --
338
339
340 while 
341
342     use Benchmark qw( timethese cmpthese ) ;
343     $x = 3;
344     $r = timethese( -5, {
345         a => sub{$x*$x},
346         b => sub{$x**2},
347     } );
348     cmpthese $r;
349
350 outputs something like this:
351
352     Benchmark: running a, b, each for at least 5 CPU seconds...
353              a: 10 wallclock secs ( 5.14 usr +  0.13 sys =  5.27 CPU) @ 3835055.60/s (n=20210743)
354              b:  5 wallclock secs ( 5.41 usr +  0.00 sys =  5.41 CPU) @ 1574944.92/s (n=8520452)
355            Rate    b    a
356     b 1574945/s   -- -59%
357     a 3835056/s 144%   --
358
359
360 =head1 INHERITANCE
361
362 Benchmark inherits from no other class, except of course
363 for Exporter.
364
365 =head1 CAVEATS
366
367 Comparing eval'd strings with code references will give you
368 inaccurate results: a code reference will show a slightly slower
369 execution time than the equivalent eval'd string.
370
371 The real time timing is done using time(2) and
372 the granularity is therefore only one second.
373
374 Short tests may produce negative figures because perl
375 can appear to take longer to execute the empty loop
376 than a short test; try:
377
378     timethis(100,'1');
379
380 The system time of the null loop might be slightly
381 more than the system time of the loop with the actual
382 code and therefore the difference might end up being E<lt> 0.
383
384 =head1 SEE ALSO
385
386 L<Devel::DProf> - a Perl code profiler
387
388 =head1 AUTHORS
389
390 Jarkko Hietaniemi <F<jhi@iki.fi>>, Tim Bunce <F<Tim.Bunce@ig.co.uk>>
391
392 =head1 MODIFICATION HISTORY
393
394 September 8th, 1994; by Tim Bunce.
395
396 March 28th, 1997; by Hugo van der Sanden: added support for code
397 references and the already documented 'debug' method; revamped
398 documentation.
399
400 April 04-07th, 1997: by Jarkko Hietaniemi, added the run-for-some-time
401 functionality.
402
403 September, 1999; by Barrie Slaymaker: math fixes and accuracy and 
404 efficiency tweaks.  Added cmpthese().  A result is now returned from 
405 timethese().  Exposed countit() (was runfor()).
406
407 December, 2001; by Nicholas Clark: make timestr() recognise the style 'none'
408 and return an empty string. If cmpthese is calling timethese, make it pass the
409 style in. (so that 'none' will suppress output). Make sub new dump its
410 debugging output to STDERR, to be consistent with everything else.
411 All bugs found while writing a regression test.
412
413 September, 2002; by Jarkko Hietaniemi: add ':hireswallclock' special tag.
414
415 =cut
416
417 # evaluate something in a clean lexical environment
418 sub _doeval { no strict;  eval shift }
419
420 #
421 # put any lexicals at file scope AFTER here
422 #
423
424 use Carp;
425 use Exporter;
426
427 our(@ISA, @EXPORT, @EXPORT_OK, %EXPORT_TAGS, $VERSION);
428
429 @ISA=qw(Exporter);
430 @EXPORT=qw(timeit timethis timethese timediff timestr);
431 @EXPORT_OK=qw(timesum cmpthese countit
432               clearcache clearallcache disablecache enablecache);
433 %EXPORT_TAGS=( all => [ @EXPORT, @EXPORT_OK ] ) ;
434
435 $VERSION = 1.051;
436
437 # --- ':hireswallclock' special handling
438
439 my $hirestime;
440
441 sub mytime () { time }
442
443 &init;
444
445 sub BEGIN {
446     if (eval 'require Time::HiRes') {
447         import Time::HiRes qw(time);
448         $hirestime = \&Time::HiRes::time;
449     }
450 }
451
452 sub import {
453     my $class = shift;
454     if (grep { $_ eq ":hireswallclock" } @_) {
455         @_ = grep { $_ ne ":hireswallclock" } @_;
456         *mytime = $hirestime if defined $hirestime;
457     }
458     Benchmark->export_to_level(1, $class, @_);
459 }
460
461 our($Debug, $Min_Count, $Min_CPU, $Default_Format, $Default_Style,
462     %_Usage, %Cache, $Do_Cache);
463
464 sub init {
465     $Debug = 0;
466     $Min_Count = 4;
467     $Min_CPU   = 0.4;
468     $Default_Format = '5.2f';
469     $Default_Style = 'auto';
470     # The cache can cause a slight loss of sys time accuracy. If a
471     # user does many tests (>10) with *very* large counts (>10000)
472     # or works on a very slow machine the cache may be useful.
473     &disablecache;
474     &clearallcache;
475 }
476
477 sub debug { $Debug = ($_[1] != 0); }
478
479 sub usage { 
480     my $calling_sub = (caller(1))[3];
481     $calling_sub =~ s/^Benchmark:://;
482     return $_Usage{$calling_sub} || '';
483 }
484
485
486 # The cache needs two branches: 's' for strings and 'c' for code.  The
487 # emtpy loop is different in these two cases.
488
489 $_Usage{clearcache} = <<'USAGE';
490 usage: clearcache($count);
491 USAGE
492
493 sub clearcache    { 
494     die usage unless @_ == 1;
495     delete $Cache{"$_[0]c"}; delete $Cache{"$_[0]s"}; 
496 }
497
498 $_Usage{clearallcache} = <<'USAGE';
499 usage: clearallcache();
500 USAGE
501
502 sub clearallcache { 
503     die usage if @_;
504     %Cache = (); 
505 }
506
507 $_Usage{enablecache} = <<'USAGE';
508 usage: enablecache();
509 USAGE
510
511 sub enablecache   {
512     die usage if @_;
513     $Do_Cache = 1; 
514 }
515
516 $_Usage{disablecache} = <<'USAGE';
517 usage: disablecache();
518 USAGE
519
520 sub disablecache  {
521     die usage if @_;
522     $Do_Cache = 0; 
523 }
524
525
526 # --- Functions to process the 'time' data type
527
528 sub new { my @t = (mytime, times, @_ == 2 ? $_[1] : 0);
529           print STDERR "new=@t\n" if $Debug;
530           bless \@t; }
531
532 sub cpu_p { my($r,$pu,$ps,$cu,$cs) = @{$_[0]}; $pu+$ps         ; }
533 sub cpu_c { my($r,$pu,$ps,$cu,$cs) = @{$_[0]};         $cu+$cs ; }
534 sub cpu_a { my($r,$pu,$ps,$cu,$cs) = @{$_[0]}; $pu+$ps+$cu+$cs ; }
535 sub real  { my($r,$pu,$ps,$cu,$cs) = @{$_[0]}; $r              ; }
536 sub iters { $_[0]->[5] ; }
537
538
539 $_Usage{timediff} = <<'USAGE';
540 usage: $result_diff = timediff($result1, $result2);
541 USAGE
542
543 sub timediff {
544     my($a, $b) = @_;
545
546     die usage unless ref $a and ref $b;
547
548     my @r;
549     for (my $i=0; $i < @$a; ++$i) {
550         push(@r, $a->[$i] - $b->[$i]);
551     }
552     bless \@r;
553 }
554
555 $_Usage{timesum} = <<'USAGE';
556 usage: $sum = timesum($result1, $result2);
557 USAGE
558
559 sub timesum {
560     my($a, $b) = @_;
561
562     die usage unless ref $a and ref $b;
563
564     my @r;
565     for (my $i=0; $i < @$a; ++$i) {
566         push(@r, $a->[$i] + $b->[$i]);
567     }
568     bless \@r;
569 }
570
571
572 $_Usage{timestr} = <<'USAGE';
573 usage: $formatted_result = timestr($result1);
574 USAGE
575
576 sub timestr {
577     my($tr, $style, $f) = @_;
578
579     die usage unless ref $tr;
580
581     my @t = @$tr;
582     warn "bad time value (@t)" unless @t==6;
583     my($r, $pu, $ps, $cu, $cs, $n) = @t;
584     my($pt, $ct, $tt) = ($tr->cpu_p, $tr->cpu_c, $tr->cpu_a);
585     $f = $Default_Format unless defined $f;
586     # format a time in the required style, other formats may be added here
587     $style ||= $Default_Style;
588     return '' if $style eq 'none';
589     $style = ($ct>0) ? 'all' : 'noc' if $style eq 'auto';
590     my $s = "@t $style"; # default for unknown style
591     my $w = $hirestime ? "%2g" : "%2d";
592     $s=sprintf("$w wallclock secs (%$f usr %$f sys + %$f cusr %$f csys = %$f CPU)",
593                             $r,$pu,$ps,$cu,$cs,$tt) if $style eq 'all';
594     $s=sprintf("$w wallclock secs (%$f usr + %$f sys = %$f CPU)",
595                             $r,$pu,$ps,$pt) if $style eq 'noc';
596     $s=sprintf("$w wallclock secs (%$f cusr + %$f csys = %$f CPU)",
597                             $r,$cu,$cs,$ct) if $style eq 'nop';
598     $s .= sprintf(" @ %$f/s (n=$n)", $n / ( $pu + $ps )) if $n && $pu+$ps;
599     $s;
600 }
601
602 sub timedebug {
603     my($msg, $t) = @_;
604     print STDERR "$msg",timestr($t),"\n" if $Debug;
605 }
606
607 # --- Functions implementing low-level support for timing loops
608
609 $_Usage{runloop} = <<'USAGE';
610 usage: runloop($number, [$string | $coderef])
611 USAGE
612
613 sub runloop {
614     my($n, $c) = @_;
615
616     $n+=0; # force numeric now, so garbage won't creep into the eval
617     croak "negative loopcount $n" if $n<0;
618     confess usage unless defined $c;
619     my($t0, $t1, $td); # before, after, difference
620
621     # find package of caller so we can execute code there
622     my($curpack) = caller(0);
623     my($i, $pack)= 0;
624     while (($pack) = caller(++$i)) {
625         last if $pack ne $curpack;
626     }
627
628     my ($subcode, $subref);
629     if (ref $c eq 'CODE') {
630         $subcode = "sub { for (1 .. $n) { local \$_; package $pack; &\$c; } }";
631         $subref  = eval $subcode;
632     }
633     else {
634         $subcode = "sub { for (1 .. $n) { local \$_; package $pack; $c;} }";
635         $subref  = _doeval($subcode);
636     }
637     croak "runloop unable to compile '$c': $@\ncode: $subcode\n" if $@;
638     print STDERR "runloop $n '$subcode'\n" if $Debug;
639
640     # Wait for the user timer to tick.  This makes the error range more like 
641     # -0.01, +0.  If we don't wait, then it's more like -0.01, +0.01.  This
642     # may not seem important, but it significantly reduces the chances of
643     # getting a too low initial $n in the initial, 'find the minimum' loop
644     # in &countit.  This, in turn, can reduce the number of calls to
645     # &runloop a lot, and thus reduce additive errors.
646     my $tbase = Benchmark->new(0)->[1];
647     while ( ( $t0 = Benchmark->new(0) )->[1] == $tbase ) {} ;
648     &$subref;
649     $t1 = Benchmark->new($n);
650     $td = &timediff($t1, $t0);
651     timedebug("runloop:",$td);
652     $td;
653 }
654
655 $_Usage{timeit} = <<'USAGE';
656 usage: $result = timeit($count, 'code' );        or
657        $result = timeit($count, sub { code } );
658 USAGE
659
660 sub timeit {
661     my($n, $code) = @_;
662     my($wn, $wc, $wd);
663
664     die usage unless defined $code and
665                      (!ref $code or ref $code eq 'CODE');
666
667     printf STDERR "timeit $n $code\n" if $Debug;
668     my $cache_key = $n . ( ref( $code ) ? 'c' : 's' );
669     if ($Do_Cache && exists $Cache{$cache_key} ) {
670         $wn = $Cache{$cache_key};
671     } else {
672         $wn = &runloop($n, ref( $code ) ? sub { } : '' );
673         # Can't let our baseline have any iterations, or they get subtracted
674         # out of the result.
675         $wn->[5] = 0;
676         $Cache{$cache_key} = $wn;
677     }
678
679     $wc = &runloop($n, $code);
680
681     $wd = timediff($wc, $wn);
682     timedebug("timeit: ",$wc);
683     timedebug("      - ",$wn);
684     timedebug("      = ",$wd);
685
686     $wd;
687 }
688
689
690 my $default_for = 3;
691 my $min_for     = 0.1;
692
693
694 $_Usage{countit} = <<'USAGE';
695 usage: $result = countit($time, 'code' );        or
696        $result = countit($time, sub { code } );
697 USAGE
698
699 sub countit {
700     my ( $tmax, $code ) = @_;
701
702     die usage unless @_;
703
704     if ( not defined $tmax or $tmax == 0 ) {
705         $tmax = $default_for;
706     } elsif ( $tmax < 0 ) {
707         $tmax = -$tmax;
708     }
709
710     die "countit($tmax, ...): timelimit cannot be less than $min_for.\n"
711         if $tmax < $min_for;
712
713     my ($n, $tc);
714
715     # First find the minimum $n that gives a significant timing.
716     for ($n = 1; ; $n *= 2 ) {
717         my $td = timeit($n, $code);
718         $tc = $td->[1] + $td->[2];
719         last if $tc > 0.1;
720     }
721
722     my $nmin = $n;
723
724     # Get $n high enough that we can guess the final $n with some accuracy.
725     my $tpra = 0.1 * $tmax; # Target/time practice.
726     while ( $tc < $tpra ) {
727         # The 5% fudge is to keep us from iterating again all
728         # that often (this speeds overall responsiveness when $tmax is big
729         # and we guess a little low).  This does not noticably affect 
730         # accuracy since we're not couting these times.
731         $n = int( $tpra * 1.05 * $n / $tc ); # Linear approximation.
732         my $td = timeit($n, $code);
733         my $new_tc = $td->[1] + $td->[2];
734         # Make sure we are making progress.
735         $tc = $new_tc > 1.2 * $tc ? $new_tc : 1.2 * $tc;
736     }
737
738     # Now, do the 'for real' timing(s), repeating until we exceed
739     # the max.
740     my $ntot  = 0;
741     my $rtot  = 0;
742     my $utot  = 0.0;
743     my $stot  = 0.0;
744     my $cutot = 0.0;
745     my $cstot = 0.0;
746     my $ttot  = 0.0;
747
748     # The 5% fudge is because $n is often a few % low even for routines
749     # with stable times and avoiding extra timeit()s is nice for
750     # accuracy's sake.
751     $n = int( $n * ( 1.05 * $tmax / $tc ) );
752
753     while () {
754         my $td = timeit($n, $code);
755         $ntot  += $n;
756         $rtot  += $td->[0];
757         $utot  += $td->[1];
758         $stot  += $td->[2];
759         $cutot += $td->[3];
760         $cstot += $td->[4];
761         $ttot = $utot + $stot;
762         last if $ttot >= $tmax;
763
764         $ttot = 0.01 if $ttot < 0.01;
765         my $r = $tmax / $ttot - 1; # Linear approximation.
766         $n = int( $r * $ntot );
767         $n = $nmin if $n < $nmin;
768     }
769
770     return bless [ $rtot, $utot, $stot, $cutot, $cstot, $ntot ];
771 }
772
773 # --- Functions implementing high-level time-then-print utilities
774
775 sub n_to_for {
776     my $n = shift;
777     return $n == 0 ? $default_for : $n < 0 ? -$n : undef;
778 }
779
780 $_Usage{timethis} = <<'USAGE';
781 usage: $result = timethis($time, 'code' );        or
782        $result = timethis($time, sub { code } );
783 USAGE
784
785 sub timethis{
786     my($n, $code, $title, $style) = @_;
787     my($t, $forn);
788
789     die usage unless defined $code and
790                      (!ref $code or ref $code eq 'CODE');
791
792     if ( $n > 0 ) {
793         croak "non-integer loopcount $n, stopped" if int($n)<$n;
794         $t = timeit($n, $code);
795         $title = "timethis $n" unless defined $title;
796     } else {
797         my $fort  = n_to_for( $n );
798         $t     = countit( $fort, $code );
799         $title = "timethis for $fort" unless defined $title;
800         $forn  = $t->[-1];
801     }
802     local $| = 1;
803     $style = "" unless defined $style;
804     printf("%10s: ", $title) unless $style eq 'none';
805     print timestr($t, $style, $Default_Format),"\n" unless $style eq 'none';
806
807     $n = $forn if defined $forn;
808
809     # A conservative warning to spot very silly tests.
810     # Don't assume that your benchmark is ok simply because
811     # you don't get this warning!
812     print "            (warning: too few iterations for a reliable count)\n"
813         if     $n < $Min_Count
814             || ($t->real < 1 && $n < 1000)
815             || $t->cpu_a < $Min_CPU;
816     $t;
817 }
818
819
820 $_Usage{timethese} = <<'USAGE';
821 usage: timethese($count, { Name1 => 'code1', ... });        or
822        timethese($count, { Name1 => sub { code1 }, ... });
823 USAGE
824
825 sub timethese{
826     my($n, $alt, $style) = @_;
827     die usage unless ref $alt eq 'HASH';
828
829     my @names = sort keys %$alt;
830     $style = "" unless defined $style;
831     print "Benchmark: " unless $style eq 'none';
832     if ( $n > 0 ) {
833         croak "non-integer loopcount $n, stopped" if int($n)<$n;
834         print "timing $n iterations of" unless $style eq 'none';
835     } else {
836         print "running" unless $style eq 'none';
837     }
838     print " ", join(', ',@names) unless $style eq 'none';
839     unless ( $n > 0 ) {
840         my $for = n_to_for( $n );
841         print ", each" if $n > 1 && $style ne 'none';
842         print " for at least $for CPU seconds" unless $style eq 'none';
843     }
844     print "...\n" unless $style eq 'none';
845
846     # we could save the results in an array and produce a summary here
847     # sum, min, max, avg etc etc
848     my %results;
849     foreach my $name (@names) {
850         $results{$name} = timethis ($n, $alt -> {$name}, $name, $style);
851     }
852
853     return \%results;
854 }
855
856
857 $_Usage{cmpthese} = <<'USAGE';
858 usage: cmpthese($count, { Name1 => 'code1', ... });        or
859        cmpthese($count, { Name1 => sub { code1 }, ... });  or
860        cmpthese($result, $style);
861 USAGE
862
863 sub cmpthese{
864     my ($results, $style);
865
866     if( ref $_[0] ) {
867         ($results, $style) = @_;
868     }
869     else {
870         my($count, $code) = @_[0,1];
871         $style = $_[2] if defined $_[2];
872
873         die usage unless ref $code eq 'HASH';
874
875         $results = timethese($count, $code, ($style || "none"));
876     }
877
878     $style = "" unless defined $style;
879
880     # Flatten in to an array of arrays with the name as the first field
881     my @vals = map{ [ $_, @{$results->{$_}} ] } keys %$results;
882
883     for (@vals) {
884         # The epsilon fudge here is to prevent div by 0.  Since clock
885         # resolutions are much larger, it's below the noise floor.
886         my $rate = $_->[6] / ( $_->[2] + $_->[3] + 0.000000000000001 );
887         $_->[7] = $rate;
888     }
889
890     # Sort by rate
891     @vals = sort { $a->[7] <=> $b->[7] } @vals;
892
893     # If more than half of the rates are greater than one...
894     my $display_as_rate = $vals[$#vals>>1]->[7] > 1;
895
896     my @rows;
897     my @col_widths;
898
899     my @top_row = ( 
900         '', 
901         $display_as_rate ? 'Rate' : 's/iter', 
902         map { $_->[0] } @vals 
903     );
904
905     push @rows, \@top_row;
906     @col_widths = map { length( $_ ) } @top_row;
907
908     # Build the data rows
909     # We leave the last column in even though it never has any data.  Perhaps
910     # it should go away.  Also, perhaps a style for a single column of
911     # percentages might be nice.
912     for my $row_val ( @vals ) {
913         my @row;
914
915         # Column 0 = test name
916         push @row, $row_val->[0];
917         $col_widths[0] = length( $row_val->[0] )
918             if length( $row_val->[0] ) > $col_widths[0];
919
920         # Column 1 = performance
921         my $row_rate = $row_val->[7];
922
923         # We assume that we'll never get a 0 rate.
924         my $rate = $display_as_rate ? $row_rate : 1 / $row_rate;
925
926         # Only give a few decimal places before switching to sci. notation,
927         # since the results aren't usually that accurate anyway.
928         my $format = 
929            $rate >= 100 ? 
930                "%0.0f" : 
931            $rate >= 10 ?
932                "%0.1f" :
933            $rate >= 1 ?
934                "%0.2f" :
935            $rate >= 0.1 ?
936                "%0.3f" :
937                "%0.2e";
938
939         $format .= "/s"
940             if $display_as_rate;
941
942         my $formatted_rate = sprintf( $format, $rate );
943         push @row, $formatted_rate;
944         $col_widths[1] = length( $formatted_rate )
945             if length( $formatted_rate ) > $col_widths[1];
946
947         # Columns 2..N = performance ratios
948         my $skip_rest = 0;
949         for ( my $col_num = 0 ; $col_num < @vals ; ++$col_num ) {
950             my $col_val = $vals[$col_num];
951             my $out;
952             if ( $skip_rest ) {
953                 $out = '';
954             }
955             elsif ( $col_val->[0] eq $row_val->[0] ) {
956                 $out = "--";
957                 # $skip_rest = 1;
958             }
959             else {
960                 my $col_rate = $col_val->[7];
961                 $out = sprintf( "%.0f%%", 100*$row_rate/$col_rate - 100 );
962             }
963             push @row, $out;
964             $col_widths[$col_num+2] = length( $out )
965                 if length( $out ) > $col_widths[$col_num+2];
966
967             # A little wierdness to set the first column width properly
968             $col_widths[$col_num+2] = length( $col_val->[0] )
969                 if length( $col_val->[0] ) > $col_widths[$col_num+2];
970         }
971         push @rows, \@row;
972     }
973
974     return \@rows if $style eq "none";
975
976     # Equalize column widths in the chart as much as possible without
977     # exceeding 80 characters.  This does not use or affect cols 0 or 1.
978     my @sorted_width_refs = 
979        sort { $$a <=> $$b } map { \$_ } @col_widths[2..$#col_widths];
980     my $max_width = ${$sorted_width_refs[-1]};
981
982     my $total = @col_widths - 1 ;
983     for ( @col_widths ) { $total += $_ }
984
985     STRETCHER:
986     while ( $total < 80 ) {
987         my $min_width = ${$sorted_width_refs[0]};
988         last
989            if $min_width == $max_width;
990         for ( @sorted_width_refs ) {
991             last 
992                 if $$_ > $min_width;
993             ++$$_;
994             ++$total;
995             last STRETCHER
996                 if $total >= 80;
997         }
998     }
999
1000     # Dump the output
1001     my $format = join( ' ', map { "%${_}s" } @col_widths ) . "\n";
1002     substr( $format, 1, 0 ) = '-';
1003     for ( @rows ) {
1004         printf $format, @$_;
1005     }
1006
1007     return \@rows ;
1008 }
1009
1010
1011 1;