145 lines
5.0 KiB
C
145 lines
5.0 KiB
C
/*
|
|
datagencli.c
|
|
compressible data command line generator
|
|
Copyright (C) Yann Collet 2012-2015
|
|
|
|
GPL v2 License
|
|
|
|
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 2 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, write to the Free Software Foundation, Inc.,
|
|
51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
|
|
|
|
You can contact the author at :
|
|
- ZSTD source repository : https://github.com/Cyan4973/zstd
|
|
- Public forum : https://groups.google.com/forum/#!forum/lz4c
|
|
*/
|
|
|
|
/*-************************************
|
|
* Includes
|
|
**************************************/
|
|
#include "util.h" /* Compiler options */
|
|
#include <stdio.h> /* fprintf, stderr */
|
|
#include "datagen.h" /* RDG_generate */
|
|
|
|
|
|
/*-************************************
|
|
* Constants
|
|
**************************************/
|
|
#define KB *(1 <<10)
|
|
#define MB *(1 <<20)
|
|
#define GB *(1U<<30)
|
|
|
|
#define SIZE_DEFAULT ((64 KB) + 1)
|
|
#define SEED_DEFAULT 0
|
|
#define COMPRESSIBILITY_DEFAULT 50
|
|
|
|
|
|
/*-************************************
|
|
* Macros
|
|
**************************************/
|
|
#define DISPLAY(...) fprintf(stderr, __VA_ARGS__)
|
|
#define DISPLAYLEVEL(l, ...) if (displayLevel>=l) { DISPLAY(__VA_ARGS__); }
|
|
static unsigned displayLevel = 2;
|
|
|
|
|
|
/*-*******************************************************
|
|
* Command line
|
|
*********************************************************/
|
|
static int usage(const char* programName)
|
|
{
|
|
DISPLAY( "Compressible data generator\n");
|
|
DISPLAY( "Usage :\n");
|
|
DISPLAY( " %s [args]\n", programName);
|
|
DISPLAY( "\n");
|
|
DISPLAY( "Arguments :\n");
|
|
DISPLAY( " -g# : generate # data (default:%i)\n", SIZE_DEFAULT);
|
|
DISPLAY( " -s# : Select seed (default:%i)\n", SEED_DEFAULT);
|
|
DISPLAY( " -P# : Select compressibility in %% (default:%i%%)\n", COMPRESSIBILITY_DEFAULT);
|
|
DISPLAY( " -h : display help and exit\n");
|
|
return 0;
|
|
}
|
|
|
|
|
|
int main(int argc, const char** argv)
|
|
{
|
|
double proba = (double)COMPRESSIBILITY_DEFAULT / 100;
|
|
double litProba = 0.0;
|
|
U64 size = SIZE_DEFAULT;
|
|
U32 seed = SEED_DEFAULT;
|
|
const char* const programName = argv[0];
|
|
|
|
int argNb;
|
|
for(argNb=1; argNb<argc; argNb++) {
|
|
const char* argument = argv[argNb];
|
|
|
|
if(!argument) continue; /* Protection if argument empty */
|
|
|
|
/* Handle commands. Aggregated commands are allowed */
|
|
if (*argument=='-') {
|
|
argument++;
|
|
while (*argument!=0) {
|
|
switch(*argument)
|
|
{
|
|
case 'h':
|
|
return usage(programName);
|
|
case 'g':
|
|
argument++;
|
|
size=0;
|
|
while ((*argument>='0') && (*argument<='9'))
|
|
size *= 10, size += *argument++ - '0';
|
|
if (*argument=='K') { size <<= 10; argument++; }
|
|
if (*argument=='M') { size <<= 20; argument++; }
|
|
if (*argument=='G') { size <<= 30; argument++; }
|
|
if (*argument=='B') { argument++; }
|
|
break;
|
|
case 's':
|
|
argument++;
|
|
seed=0;
|
|
while ((*argument>='0') && (*argument<='9'))
|
|
seed *= 10, seed += *argument++ - '0';
|
|
break;
|
|
case 'P':
|
|
argument++;
|
|
proba=0.0;
|
|
while ((*argument>='0') && (*argument<='9'))
|
|
proba *= 10, proba += *argument++ - '0';
|
|
if (proba>100.) proba=100.;
|
|
proba /= 100.;
|
|
break;
|
|
case 'L': /* hidden argument : Literal distribution probability */
|
|
argument++;
|
|
litProba=0.;
|
|
while ((*argument>='0') && (*argument<='9'))
|
|
litProba *= 10, litProba += *argument++ - '0';
|
|
if (litProba>100.) litProba=100.;
|
|
litProba /= 100.;
|
|
break;
|
|
case 'v':
|
|
displayLevel = 4;
|
|
argument++;
|
|
break;
|
|
default:
|
|
return usage(programName);
|
|
}
|
|
} } } /* for(argNb=1; argNb<argc; argNb++) */
|
|
|
|
DISPLAYLEVEL(4, "Data Generator \n");
|
|
DISPLAYLEVEL(3, "Seed = %u \n", seed);
|
|
if (proba!=COMPRESSIBILITY_DEFAULT) DISPLAYLEVEL(3, "Compressibility : %i%%\n", (U32)(proba*100));
|
|
|
|
RDG_genStdout(size, proba, litProba, seed);
|
|
DISPLAYLEVEL(1, "\n");
|
|
|
|
return 0;
|
|
}
|