#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include "minwise.h"

void printbits(char *name, char *buf,int len){
	for(int i=0;i<len;i++){
		buf[i]+='0';
	}
	buf[len]='\0';
	printf("%s=%d:%s\n", name, len,buf);
	for(int i=0;i<len;i++){
		buf[i]-='0';
	}

}
void testip(int n){ // testing irreducible Polynomial
	printf("Testing Irreducible Polynomial class\n");
	//irreduciblePolynomial *ip=new irreduciblePolynomial(n);
	for(int i=0;i<10;i++){
		irreduciblePolynomial ip(n);
		printbits("f=",ip.f,n);
		/*
		irreduciblePolynomial *ip=new irreduciblePolynomial(n);
		delete ip;
		*/
	}
}
void testlfsr(int m, int pos, int len){
	printf("lfsr(%d),pos=%d,len=%d\n",m,pos,len);
	char *buf=new char[len+1];
	
	for(int i=0;i<10;i++){
	LFSR lfsr(m);
		printbits("s=", lfsr.s,m);
		printbits("f=", lfsr.f,m);
		lfsr.getBits(pos,len,buf);
		printf("pos(%d)",pos);
		printbits("", buf,len);
	}
	delete[] buf;
}

void testkwhf(int n, int r, int k, int e){
	kwHashFunction h(n,r,k,e);

	int R=(1<<r);
	int *count=new int[R];
	for(int i=0;i<R;i++) count[i]=0;
	int N=1000;
	for(int i=0;i<N;i++){
		int t=h.hash(RAND(N));
		count[t]++;
	}
	for(int i=0;i<R;i++) printf("%d ",count[i]);
	printf("\n");
	delete []count;
}

void testmw(int n, int k, int e, int N){
	int K=(1<<k)-1;
	int count=0;
	printf("# restricted min-wise indepedent permutaton for [2^%d] k=2^%d=%d,eps=2^-%d=%f \n ",n,k,(1<<k),e,1.0/(1<<e)); 
	for(int i=0;i<N;i++){
		mwPermutation p(n,k,e);
		INTTYPE p0=p.map(0);
		INTTYPE min=p0; 
		for(int j=1;j<K;j++){
			INTTYPE t=p.map(j);
			if(t<0){
				printf("#negative result %llX %llX\n", t,j);
			}
			if(min>t)min=t;
		}
		if(min==p0) 
			count++;
//		printf("trial %d: count=%d min=%d\n",i,count,min); 

		if(i>100&&i%10==9){
			double eps= (1.0*count*K)/(i+1)-1.0;
			//eps=|X|*(count/(i+1)-1/|X|)
//			printf("After %d trials: eps=%f for |X|=%d\n",i+1,eps,K); 
			if(eps<0)
			      eps=-eps;
			printf("%d %g\n",(i+1),eps); 

		}
	} 
}

int main(int c, char **argv){
	global_rng=new RNG;
	global_rng->set_seed(RNG::RAW_SEED_SOURCE,1);
	c--;argv++;
	if(c&&strstr(*argv,"test")){//test mode
		if(c>=2&&strcmp(*argv,"testip")==0){
			argv++;
			testip(atoi(*(argv++)));
			c-=2;
		}else
		if(c>=4&&strcmp(*argv,"testlfsr")==0){
			argv++;
			testlfsr(
				atoi(*(argv)),
				atoi(*(argv+1)),
				atoi(*(argv+2))
			 );
			argv+=3;
			c-=4;
		}else
		if(c>=5&&strcmp(*argv,"testkwhf")==0){
			argv++;
			testkwhf(
				atoi(*(argv)),
				atoi(*(argv+1)),
				atoi(*(argv+2)),
				atoi(*(argv+3))
			 );
			argv+=4;
			c-=5;
		}else
		if(strcmp(*argv,"testmw")==0){
			argv++;
			testmw(
				atoi(*(argv)),
				atoi(*(argv+1)),
				atoi(*(argv+2)),
				atoi(*(argv+3))
			 );
			argv+=4;
			c-=5;

		}else
		{
		}
	}
}