1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
|
// Test the RNGs in cpptraj with TestU01
#include <string.h>
#include <stdlib.h>
#include "TestU01.h"
#include "cpptraj_rng.h"
static void* rngptr = 0;
unsigned int test_rng(void)
{
return num_cpptraj_rng( rngptr );
}
int argIs(const char* arg, const char* key)
{
if (strcmp(arg, key) == 0) return 1;
return 0;
}
int main(int argc, char** argv) {
int it = 0;
int iseed = 0;
int icount = 0;
int itype = 0;
int imax = 10;
int imode = 1; // 1 = small crush
// Parse options
for (int iarg = 1; iarg < argc; iarg++) {
if (argIs(argv[iarg], "-S")) {
iseed = atoi( argv[++iarg] );
} else if (argIs(argv[iarg], "-t")) {
icount = atoi( argv[++iarg] );
} else if (argIs(argv[iarg], "-r")) {
itype = atoi( argv[++iarg] );
} else if (argIs(argv[iarg], "--max")) {
imax = atoi( argv[++iarg] );
} else if (argIs(argv[iarg], "--mode")) {
imode = atoi( argv[++iarg] );
}/* else if (arg == "--mode") {
std::string marg(argv[++iarg]);
if (marg == "diehard")
mode = DIEHARD;
else if (marg == "shuffle")
mode = SHUFFLE;
else if (marg == "range")
mode = RANGE;
else {
fprintf(stderr,"Error: Unrecognized mode: %s\n", marg.c_str());
return 1;
}
}*/
}
// Setup RNG
rngptr = get_cpptraj_rng( itype, iseed );
if (imode == 0) {
for (it = 0; it < icount; it++)
printf("%10u\n", num_cpptraj_rng( rngptr ));
} else {
// Create TestU01 PRNG object for our generator
unif01_Gen* gen = unif01_CreateExternGenBits((char*)"CPPTRAJ", test_rng);
// Run the tests.
if (imode == 1)
bbattery_SmallCrush(gen);
else if (imode == 2)
bbattery_Crush(gen);
// Clean up.
unif01_DeleteExternGenBits(gen);
}
destroy_cpptraj_rng( rngptr );
return 0;
}
|
