/* 
$Name:  $ 
$Id: mem-test.cpp,v 1.3 2005/12/30 23:09:55 dstrat Exp $

mem-test tests how much RAM an application can get.
Usage: mem-test <# chunks> <chunk size in MB>.
*/

#include <iostream>
#include <cstdlib>

using namespace std;

int main(int, char* argv[])
{
	int *p, i, iMax=0, j, k, numChunks, chunkSize;

	numChunks=atoi(argv[1]);
	chunkSize=atoi(argv[2]);

	cout << "mem-test: numChunks=" << numChunks << ", chunkSize=" << 
		chunkSize << "." << endl;
	cout << "mem-test: Allocating: " << flush;

	for (i=0;i<numChunks;i++)
	{
		cout << i+1 << flush;
		p=new int [chunkSize*1024*1024/4];
		if (p) 
		{
			cout << "+" << flush;
			iMax=i+1;

			for (k=0;k<8;k++)
			    for (j=0;j<chunkSize*1024*1024/4;j++)
				     p[j]=k+j;

			cout << "d " << flush;

		}
		else cout << "!" << flush;
	}

	cout << "\nmem-test: Done." << endl;
	cout << "mem-test: Successfully allocated " << iMax*chunkSize << 
		"MB, in " << iMax << " chunks of " << chunkSize << "MB each." << 
		endl;
	return 0;
}

// End.