[RFC]: add a C benchmark harness #13480

#include "stdlib/blas/ext/base/zindex_of_truthy.h"
#include "stdlib/complex/float64/ctor.h"
#include <stdlib.h>
#include <stdio.h>
#include <math.h>
#include <time.h>
#include <sys/time.h>

#define NAME "zindex_of_truthy"
#define ITERATIONS 10000000
#define REPEATS 3
#define MIN 1
#define MAX 6

/**
* Prints the TAP version.
*/
static void print_version( void ) {
	printf( "TAP version 13\n" );
}

/**
* Prints the TAP summary.
*
* @param total     total number of tests
* @param passing   total number of passing tests
*/
static void print_summary( int total, int passing ) {
	printf( "#\n" );
	printf( "1..%d\n", total ); // TAP plan
	printf( "# total %d\n", total );
	printf( "# pass  %d\n", passing );
	printf( "#\n" );
	printf( "# ok\n" );
}

/**
* Prints benchmarks results.
*
* @param iterations   number of iterations
* @param elapsed      elapsed time in seconds
*/
static void print_results( int iterations, double elapsed ) {
	double rate = (double)iterations / elapsed;
	printf( "  ---\n" );
	printf( "  iterations: %d\n", iterations );
	printf( "  elapsed: %0.9f\n", elapsed );
	printf( "  rate: %0.9f\n", rate );
	printf( "  ...\n" );
}

/**
* Returns a clock time.
*
* @return clock time
*/
static double tic( void ) {
	struct timeval now;
	gettimeofday( &now, NULL );
	return (double)now.tv_sec + (double)now.tv_usec/1.0e6;
}

/**
* Generates a random number on the interval [0,1).
*
* @return random number
*/
static double rand_double( void ) {
	int r = rand();
	return (double)r / ( (double)RAND_MAX + 1.0 );
}

/**
* Runs a benchmark.
*
* @param iterations   number of iterations
* @param len          array length
* @return             elapsed time in seconds
*/
static double benchmark1( int iterations, int len ) {
	double elapsed;
	double *x;
	double t;
	int idx;
	int i;

	x = (double *)malloc( (size_t)( len*2 ) * sizeof( double ) );
	for ( i = 0; i < len*2; i++ ) {
		x[ i ] = 0.0;
	}
	x[ (len*2)-1 ] = 1.0;
	idx = -1;
	t = tic();
	for ( i = 0; i < iterations; i++ ) {
		x[ (len*2)-4 ] = (double)( i % 4 );
		idx = stdlib_strided_zindex_of_truthy( len, (const stdlib_complex128_t *)x, 1 );
		if ( idx < 0 ) {
			printf( "unexpected result\n" );
			break;
		}
	}
	elapsed = tic() - t;
	if ( idx < 0 ) {
		printf( "unexpected result\n" );
	}
	free( x );
	return elapsed;
}

/**
* Runs a benchmark.
*
* @param iterations   number of iterations
* @param len          array length
* @return             elapsed time in seconds
*/
static double benchmark2( int iterations, int len ) {
	double elapsed;
	double *x;
	double t;
	int idx;
	int i;

	x = (double *)malloc( (size_t)( len*2 ) * sizeof( double ) );
	for ( i = 0; i < len*2; i++ ) {
		x[ i ] = 0.0;
	}
	x[ (len*2)-1 ] = 1.0;
	idx = -1;
	t = tic();
	for ( i = 0; i < iterations; i++ ) {
		x[ (len*2)-4 ] = (double)( i % 4 );
		idx = stdlib_strided_zindex_of_truthy_ndarray( len, (const stdlib_complex128_t *)x, 1, 0 );
		if ( idx < 0 ) {
			printf( "unexpected result\n" );
			break;
		}
	}
	elapsed = tic() - t;
	if ( idx < 0 ) {
		printf( "unexpected result\n" );
	}
	free( x );
	return elapsed;
}

/**
* Main execution sequence.
*/
int main( void ) {
	double elapsed;
	int count;
	int iter;
	int len;
	int i;
	int j;

	// Use the current time to seed the random number generator:
	srand( time( NULL ) );

	print_version();
	count = 0;
	for ( i = MIN; i <= MAX; i++ ) {
		len = pow( 10, i );
		iter = ITERATIONS / pow( 10, i-1 );
		for ( j = 0; j < REPEATS; j++ ) {
			count += 1;
			printf( "# c::%s:len=%d\n", NAME, len );
			elapsed = benchmark1( iter, len );
			print_results( iter, elapsed );
			printf( "ok %d benchmark finished\n", count );
		}
		for ( j = 0; j < REPEATS; j++ ) {
			count += 1;
			printf( "# c::%s:ndarray:len=%d\n", NAME, len );
			elapsed = benchmark2( iter, len );
			print_results( iter, elapsed );
			printf( "ok %d benchmark finished\n", count );
		}
	}
	print_summary( count, count );
}

#include "stdlib/blas/ext/base/zindex_of_truthy.h"
#include "stdlib/complex/float64/ctor.h"
#include "stdlib/bench.h"

#ifndef NAME
#define NAME "zindex_of_truthy"
#endif

/**
* Runs a benchmark.
*
* @param iterations   number of iterations
* @param len          array length
* @return             elapsed time in seconds
*/
STDLIB_BENCHMARK( benchmark1 ) {
	int idx = -1;

	STDLIB_BENCHMARK_MALLOC_ARRAY_FLOAT64( x, len*2 );
	STDLIB_BENCHMARK_FILL_STRIDED_ARRAY_FLOAT64( x, len*2, 0.0 );
	x[ (len*2)-1 ] = 1.0;
	
	STDLIB_BENCHMARK_LOOP_PREAMBLE {
		x[ (len*2)-4 ] = (double)( i % 4 );
		idx = stdlib_strided_zindex_of_truthy( len, (const stdlib_complex128_t *)x, 1 );
		if ( idx < 0 ) {
			printf( "unexpected result\n" );
			break;
		}
	}
	STDLIB_BENCHMARK_LOOP_EPILOGUE;
	if ( idx < 0 ) {
		printf( "unexpected result\n" );
	}
	STDLIB_BENCHMARK_FREE( x );
	STDLIB_BENCHMARK_EPILOGUE;
}

/**
* Runs a benchmark.
*
* @param iterations   number of iterations
* @param len          array length
* @return             elapsed time in seconds
*/
STDLIB_BENCHMARK( benchmark2 ) {
	int idx = -1;

	STDLIB_BENCHMARK_MALLOC_ARRAY_FLOAT64( x, len*2 );
	STDLIB_BENCHMARK_FILL_STRIDED_ARRAY_FLOAT64( x, len*2, 0.0 );
	x[ (len*2)-1 ] = 1.0;
	
	STDLIB_BENCHMARK_LOOP_PREAMBLE {
		x[ (len*2)-4 ] = (double)( i % 4 );
		idx = stdlib_strided_zindex_of_truthy_ndarray( len, (const stdlib_complex128_t *)x, 1, 0 );
		if ( idx < 0 ) {
			printf( "unexpected result\n" );
			break;
		}
	}
	STDLIB_BENCHMARK_LOOP_EPILOGUE;
	if ( idx < 0 ) {
		printf( "unexpected result\n" );
	}
	STDLIB_BENCHMARK_FREE( x );
	STDLIB_BENCHMARK_EPILOGUE;
}

/**
* Main execution sequence.
*/
STDLIB_BENCH {
	int iterations = 10000000;
	int repeats = 3;
	int min = 1;
	int max = 6;

	STDLIB_BENCH_LENGTH_PREAMBLE( repeats, iterations, min, max ) {
		printf( "# c::%s:len=%d\n", NAME, len );
		STDLIB_RUN_BENCHMARK( benchmark1 );
	}
	STDLIB_BENCH_LENGTH_PREAMBLE( repeats, iterations, min, max ) {
		printf( "# c::%s:ndarray:len=%d\n", NAME, len );
		STDLIB_RUN_BENCHMARK( benchmark2 );
	}
	STDLIB_BENCH_EPILOGUE;
}