/*
 * Copyright (C) 2011 Andrea Mazzoleni
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

#include "portable.h"

#include "snapraid.h"
#include "util.h"
#include "raid.h"
#include "cpu.h"

void speed(void)
{
	struct timeval start;
	struct timeval stop;
	int64_t ds;
	int64_t dt;
	unsigned i, j;
	unsigned char digest[HASH_SIZE];

	unsigned count = 5000;
	unsigned block_size = 256 * 1024;
	unsigned diskmax = 4;
	void* buffer_alloc;
	unsigned char* buffer_aligned;
	unsigned char** buffer;

	buffer_aligned = malloc_nofail_align((diskmax + 2) * block_size, &buffer_alloc);
	buffer = malloc_nofail((diskmax + 2) * sizeof(void*));
	for(i=0;i<diskmax+2;++i) {
		buffer[i] = buffer_aligned + i * block_size;
		memset(buffer[i], i, block_size);
	}

#ifdef __GNUC__
	printf("Compiled with gcc " __VERSION__ "\n");
#endif

	printf("Speed test with %d disk and %d buffer size...\n", diskmax, block_size);

	gettimeofday(&start, 0);
	for(i=0;i<count;++i) {
		for(j=0;j<diskmax;++j) {
			memset(buffer[j], j, block_size);
		}
	}
	gettimeofday(&stop, 0);

	ds = block_size * (int64_t)count * diskmax;
	dt = (int64_t)1000000 * (stop.tv_sec - start.tv_sec) + (stop.tv_usec - start.tv_usec);

	printf("memset %llu [MB/s]\n", ds / dt);

	gettimeofday(&start, 0);
	for(i=0;i<count;++i) {
		for(j=0;j<diskmax;++j) {
			memhash(HASH_MD5, digest, buffer[j], block_size);
		}
	}
	gettimeofday(&stop, 0);

	ds = block_size * (int64_t)count * diskmax;
	dt = (int64_t)1000000 * (stop.tv_sec - start.tv_sec) + (stop.tv_usec - start.tv_usec);

	printf("MD5 %llu [MB/s]\n", ds / dt);

	gettimeofday(&start, 0);
	for(i=0;i<count;++i) {
		for(j=0;j<diskmax;++j) {
			memhash(HASH_MURMUR3, digest, buffer[j], block_size);
		}
	}
	gettimeofday(&stop, 0);

	ds = block_size * (int64_t)count * diskmax;
	dt = (int64_t)1000000 * (stop.tv_sec - start.tv_sec) + (stop.tv_usec - start.tv_usec);

	printf("Murmur3 %llu [MB/s]\n", ds / dt);

	gettimeofday(&start, 0);
	for(j=0;j<count;++j) {
		raid5_int32r2(buffer, diskmax, block_size);
	}
	gettimeofday(&stop, 0);

	ds = block_size * (int64_t)count * diskmax;
	dt = (int64_t)1000000 * (stop.tv_sec - start.tv_sec) + (stop.tv_usec - start.tv_usec);

	printf("RAID5 int32x2 %llu [MB/s]\n", ds / dt);

#if defined(__i386__) || defined(__x86_64__)
	if (cpu_has_mmx()) {
		gettimeofday(&start, 0);
		for(j=0;j<count;++j) {
			raid5_mmxr2(buffer, diskmax, block_size);
		}
		gettimeofday(&stop, 0);

		ds = block_size * (int64_t)count * diskmax;
		dt = (int64_t)1000000 * (stop.tv_sec - start.tv_sec) + (stop.tv_usec - start.tv_usec);

		printf("RAID5 mmxx2 %llu [MB/s]\n", ds / dt);
	}

	if (cpu_has_sse2()) {
		gettimeofday(&start, 0);
		for(j=0;j<count;++j) {
			raid5_sse2r2(buffer, diskmax, block_size);
		}
		gettimeofday(&stop, 0);

		ds = block_size * (int64_t)count * diskmax;
		dt = (int64_t)1000000 * (stop.tv_sec - start.tv_sec) + (stop.tv_usec - start.tv_usec);

		printf("RAID5 sse2x2 %llu [MB/s]\n", ds / dt);
	}
#endif

	gettimeofday(&start, 0);
	for(j=0;j<count;++j) {
		raid6_int32r2(buffer, diskmax, block_size);
	}
	gettimeofday(&stop, 0);

	ds = block_size * (int64_t)count * diskmax;
	dt = (int64_t)1000000 * (stop.tv_sec - start.tv_sec) + (stop.tv_usec - start.tv_usec);

	printf("RAID6 int32x2 %llu [MB/s]\n", ds / dt);

#if defined(__i386__) || defined(__x86_64__)
	if (cpu_has_mmx()) {
		gettimeofday(&start, 0);
		for(j=0;j<count;++j) {
			raid6_mmxr2(buffer, diskmax, block_size);
		}
		gettimeofday(&stop, 0);

		ds = block_size * (int64_t)count * diskmax;
		dt = (int64_t)1000000 * (stop.tv_sec - start.tv_sec) + (stop.tv_usec - start.tv_usec);

		printf("RAID6 mmxx2 %llu [MB/s]\n", ds / dt);
	}

	if (cpu_has_sse2()) {
		gettimeofday(&start, 0);
		for(j=0;j<count;++j) {
			raid6_sse2r2(buffer, diskmax, block_size);
		}
		gettimeofday(&stop, 0);

		ds = block_size * (int64_t)count * diskmax;
		dt = (int64_t)1000000 * (stop.tv_sec - start.tv_sec) + (stop.tv_usec - start.tv_usec);

		printf("RAID6 sse2x2 %llu [MB/s]\n", ds / dt);
	}
#endif

	free(buffer_alloc);
	free(buffer);
}