Initial source commit

minorGems/util/random/testRandom.cpp

@@ -0,0 +1,215 @@
+#include "RandomSource.h"
+
+#include <stdio.h>
+
+
+// perform tests on a random source
+
+// tests found here:
+// NIST FIPS 140-1 U.S. Federal Standard
+// http://csrc.nist.gov/publications/fips/fips140-1/fips1401.pdf
+
+void testRandom( RandomSource *inSource ) {
+    
+    int bitsPerStep = 1;
+    
+
+    // 20000 bits
+    // must be a multiple of bitsPerStep
+    int numBits = 20000;
+    
+    int numSteps = numBits / bitsPerStep;
+
+    unsigned int *drawnNumbers = new unsigned int[ numSteps ];
+    
+    unsigned int *drawnBits = new unsigned int[ numBits ];
+    
+
+    unsigned int bitSum = 0;
+    
+    // 1 or 0 when we're in a run
+    unsigned int currentRunType = 2;
+    unsigned int currentRunLength = 0;
+    
+
+    unsigned int longestRunLength = 0;
+    
+    #define maxRunLengthToTrack 100
+
+    unsigned int runLengthCounts[2][ maxRunLengthToTrack ];
+        
+    int r;
+    for( r=0; r<maxRunLengthToTrack; r++ ) {
+        runLengthCounts[0][r] = 0;
+        runLengthCounts[1][r] = 0;
+        }
+    
+
+    int i;
+    int bitIndex = 0;
+    
+    for( i=0; i<numSteps; i++ ) {
+        drawnNumbers[i] = inSource->getRandomBoundedInt( 0, 1 );
+        //drawnNumbers[i] = inSource->getRandomBoolean();
+
+
+        for( int b=0; b<bitsPerStep; b++ ) {
+            unsigned int bit = ( drawnNumbers[i] >> b & 0x01 );
+            bitSum += bit;
+            
+            drawnBits[bitIndex] = bit;
+            bitIndex++;
+            
+
+            if( bit == currentRunType ) {
+                currentRunLength++;
+                
+
+                if( currentRunLength > longestRunLength ) {
+                    longestRunLength = currentRunLength;
+                    }
+                }
+            else {
+                // end of run
+
+                if( currentRunLength > 0 && 
+                    currentRunLength < maxRunLengthToTrack ) {
+                    // count it
+                    runLengthCounts[currentRunType][ currentRunLength ] ++;
+                    }
+                
+                currentRunType = bit;
+                currentRunLength = 1;
+                }
+            }
+        }
+
+    float mean = (float)bitSum / numBits;
+
+    printf( "Mean = %f\n", mean );
+
+    
+    float varSum = 0;
+
+    for( i=0; i<numBits; i++ ) {
+        unsigned int bit = drawnBits[i];
+        
+        varSum += (bit - mean) * (bit - mean);
+        }
+
+
+    float variance = varSum / numBits;
+    
+    printf( "Var = %f\n", variance );
+
+    printf( "Monobit count = %d  (Allowed [9654 .. 10346])\n", bitSum );
+    
+    printf( "Longest run = %d\n", longestRunLength );
+    
+    printf( "0 Run Length\tTimes Seen\n"
+            "------------\t----------\n" );
+    
+    for( r=0; r<maxRunLengthToTrack && r <= longestRunLength; r++ ) {
+        printf( "%d\t\t%d\n", r, runLengthCounts[0][r] );
+        }
+    printf( "1 Run Length\tTimes Seen\n"
+            "------------\t----------\n" );
+    
+    for( r=0; r<maxRunLengthToTrack && r <= longestRunLength; r++ ) {
+        printf( "%d\t\t%d\n", r, runLengthCounts[1][r] );
+        }
+
+
+    // poker test
+    int j;
+    int numCValues = numBits / 4;
+    
+    unsigned int *c = new unsigned int[numCValues];
+    unsigned int *f = new unsigned int[16];
+    
+    for( i=0; i<16; i++ ) {
+        f[i] = 0;
+        }
+    
+    
+    for( j=1; j<numCValues; j++ ) {
+        //printf( "bit[%d] = %d\n", j, drawnBits[j] );
+        
+        c[j] = 
+            8 * drawnBits[ 4 * j - 3]
+            +
+            4 * drawnBits[ 4 * j - 2]
+            +
+            2 * drawnBits[ 4 * j - 1]
+            +
+            1 * drawnBits[ 4 * j ];
+
+        f[ c[j] ] ++;
+        }
+    
+    int fSum = 0;
+    //printf( "Poker hand counts:\n" );
+    for( i=0; i<16; i++ ) {
+        //printf( "f[%d] = %d\n", i, f[i] );
+        
+        fSum += f[i] * f[i];
+        }
+    
+    float Y = 16.0f/numCValues * fSum - numCValues;
+    
+    printf( "Poker test = %f   (Allowed [1.03 .. 57.4])\n", Y );
+    
+    delete [] c;
+    delete [] f;
+    
+    
+
+    // autocorrelation
+    char failed = false;
+    
+    int numShifts = 5000;
+    unsigned int *Z = new unsigned int[ numShifts ];
+    
+    for( int t=1; t<numShifts; t++ ) {
+        Z[t] = 0;
+        
+
+        for( int j=0; j<5000; j++ ) {
+            Z[t] += drawnBits[j] ^ drawnBits[ j + t ];
+            }
+
+        if( Z[t] > 2326 && Z[t] < 2674 ) {
+            }
+        else {
+            failed = true;
+            //printf( "autocorrelation test failed for Z[%d] = %d\n", t, Z[t] );
+            }        
+        }
+    if( !failed ) {
+        printf( "Autocorrelation test passed.\n" );
+        }
+    
+    
+
+
+    delete [] drawnNumbers;
+    delete [] drawnBits;
+    }
+
+
+
+
+#include "CustomRandomSource.h"
+#include "StdRandomSource.h"
+
+int main() {
+    
+    CustomRandomSource randSource( 11234258 );
+    //StdRandomSource randSource( 11 );
+    
+    testRandom( &randSource );
+    
+
+    return 0;
+    }
+