X-Git-Url: http://www.sigaev.ru/git/gitweb.cgi?a=blobdiff_plain;f=gendata.c;h=fc7f3c09953d13cb37b48793ed565790d0502246;hb=b99712a22de6fb3f45b36580e63596aac7e6061c;hp=3b7ae6fe7fb4c51c993a394bbc44d5993f2a53a8;hpb=0efed821b858c3d372c03038bb26b503b0a0dd99;p=tedtools.git

diff --git a/gendata.c b/gendata.c
index 3b7ae6f..fc7f3c0 100644
--- a/gendata.c
+++ b/gendata.c
@@ -39,7 +39,7 @@ static void
 usage() {
         puts(
         "Usage:\n"
-        "tbtreetest  -b [-r -c COUNT] \n"
+        "tbtreetest  -b [-r -c COUNT -f] | -s [ -c COUNT ]\n"
         );
         exit(1);
 }
@@ -49,17 +49,25 @@ extern int opterr;
 
 #define NO_MODE		0
 #define MODE_BTREE	1
+#define MODE_SFX	2
 
+static char *symbols="qwertyuiopasdfghjklzxcvbnm1234567890";
+static int symbolslen=0;
 
 int
 main(int argn, char *argv[]) {
         int mode=NO_MODE;
-	int isrnd=0, count=10,i; 
+	int isrnd=0, count=10,i, isfind=0; 
 
         opentlog(TL_OPEN_STDERR,TL_DEBUG, NULL);
+	symbolslen=strlen(symbols);
+	
           
-        while((i=getopt(argn,argv,"brc:")) != EOF) {
+        while((i=getopt(argn,argv,"sbrc:f")) != EOF) {
                 switch(i) {
+                        case 'f':
+                                isfind=1;
+                                break;
                         case 'r':
                                 isrnd=1;
                                 break;
@@ -67,6 +75,10 @@ main(int argn, char *argv[]) {
 				if ( mode ) usage();
                                 mode = MODE_BTREE;
                                 break;
+                        case 's':
+				if ( mode ) usage();
+                                mode = MODE_SFX;
+                                break;
                         case 'c':
                                 count=atoi(optarg);
                                 break;
@@ -77,14 +89,24 @@ main(int argn, char *argv[]) {
 	}
 
 	if ( mode==MODE_BTREE ) {
-		int j, cnt,c;
-		for(i=0;i<count;i++) {
-			printf("I\t%d\t", (int)( (isrnd) ? random()%count : i ));
-			cnt=1+random()%512;
-			for(j=0;j<cnt;j++) {
-				do { c=random()%128; } while ( !isalnum(c) );	
-				fputc(c, stdout);
+		int cnt;
+		if ( isfind ) 
+			for(i=0;i<count;i++)
+				printf("S\t%d\n", (int)( (isrnd) ? random()%count : i )); 
+		else	
+			for(i=0;i<count;i++) {
+				printf("I\t%d\t", (int)( (isrnd) ? random()%count : i ));
+				cnt=1+random()%512;
+				while(cnt-->0)
+					fputc( symbols[ random()%symbolslen ], stdout);
+				fputc('\n', stdout);
 			}
+	} else if ( mode==MODE_SFX ) {
+		int cnt;
+		for(i=0;i<count;i++) {
+			cnt=1+random()%10;
+			while(cnt-->0)
+				fputc( tolower(symbols[ random()%symbolslen ]), stdout);
 			fputc('\n', stdout);
 		}
 	} else {