1 files changed, 113 insertions, 0 deletions

diff --git a/src/tests/pic.cpp b/src/tests/pic.cpp
index 95bb523..4f4139f 100644
--- a/src/tests/pic.cpp
+++ b/src/tests/pic.cpp
@@ -3,8 +3,20 @@
 #include "neural/row.h"
 #include "neural/neuron.h"
 
+#include <time.h>
+
+#include <bu/random.h>
+#include <bu/sio.h>
+using namespace Bu;
+
+#include <stdio.h>
+#include <png.h>
+#include <zlib.h>
+
 int main( int argc, char *argv[] )
 {
+        Bu::Random::seed( time( NULL ) );
+
         Neural::Column<float> *c = new Neural::Column<float>();
         Neural::Row<float> *r1 = new Neural::Row<float>();
         r1->addNode( new Neural::Neuron<float>() );
@@ -27,6 +39,107 @@ int main( int argc, char *argv[] )
         r3->addNode( new Neural::Neuron<float>() );
         r3->addNode( new Neural::Neuron<float>() );
         c->addNode( r3 );
+
+        c->finalize( 2 );
+        sio << "Total weights:   " << c->getNumWeights() << sio.nl;
+        sio << "Total biases:    " << c->getNumBiases() << sio.nl;
+        sio << "Network inputs:  " << c->getNumInputs() << sio.nl;
+        sio << "Network outputs: " << c->getNumOutputs() << sio.nl;
+
+        float *pWeights = new float[c->getNumWeights()];
+        float *pBiases = new float[c->getNumBiases()];
+
+        for( int j = 0; j < c->getNumWeights(); j++ )
+                pWeights[j] = (Bu::Random::randNorm()*2.0)-1.0;
+        for( int j = 0; j < c->getNumBiases(); j++ )
+                pBiases[j] = (Bu::Random::randNorm()*2.0)-1.0;
+
+        c->setWeights( pWeights );
+        c->setBiases( pBiases );
+        delete pWeights;
+        delete pBiases;
+
+        float *pIn = new float[c->getNumInputs()];
+        float *pOut = new float[c->getNumOutputs()];
+
+        FILE *fp = fopen("test.png", "wb");
+
+    if (!fp)
+       return 1;
+
+    png_structp png_ptr = png_create_write_struct
+       (PNG_LIBPNG_VER_STRING, NULL, NULL, NULL );
+
+    if (!png_ptr)
+       return 1;
+
+    png_infop info_ptr = png_create_info_struct(png_ptr);
+    if (!info_ptr)
+    {
+       png_destroy_write_struct(&png_ptr,
+           (png_infopp)NULL);
+       return 1;
+    }
+
+    png_set_IHDR(png_ptr, info_ptr, 500, 500, 8, PNG_COLOR_TYPE_RGB,
+                PNG_INTERLACE_NONE, PNG_COMPRESSION_TYPE_DEFAULT,
+                PNG_FILTER_TYPE_DEFAULT
+                );
+   
+    /* Set the zlib compression level */
+    png_set_compression_level(png_ptr,
+        Z_BEST_COMPRESSION);
+
+    /* Set other zlib parameters for compressing IDAT */
+    png_set_compression_mem_level(png_ptr, 8);
+    png_set_compression_strategy(png_ptr,
+        Z_DEFAULT_STRATEGY);
+    png_set_compression_window_bits(png_ptr, 15);
+    png_set_compression_method(png_ptr, 8);
+    png_set_compression_buffer_size(png_ptr, 8192);
+
+    /* Set zlib parameters for text compression
+     * If you don't call these, the parameters
+     * fall back on those defined for IDAT chunks
+     */
+    png_set_text_compression_mem_level(png_ptr, 8);
+    png_set_text_compression_strategy(png_ptr,
+        Z_DEFAULT_STRATEGY);
+    png_set_text_compression_window_bits(png_ptr, 15);
+    png_set_text_compression_method(png_ptr, 8);
+
+        png_bytep *row_pointers;
+        row_pointers = new png_bytep[500];
+        for( int y = 0; y < 500; y++ )
+        {
+                row_pointers[y] = new png_byte[500*3];
+                for( int x = 0; x < 500; x++ )
+                {
+                        pIn[0] = (x/499.0)*2.0-1.0;
+                        pIn[1] = (y/499.0)*2.0-1.0;
+                        c->process( pIn, pOut );
+                        row_pointers[y][x*3+0] = pOut[0]*127+127;
+                        row_pointers[y][x*3+1] = pOut[1]*127+127;
+                        row_pointers[y][x*3+2] = pOut[2]*127+127;
+                }
+        }
+
+        png_init_io( png_ptr, fp );
+
+        png_set_rows( png_ptr, info_ptr, row_pointers );
+
+        png_write_png(png_ptr, info_ptr, PNG_TRANSFORM_IDENTITY, NULL);
+
+        if (setjmp(png_jmpbuf(png_ptr)))
+    {
+        png_destroy_write_struct(&png_ptr, &info_ptr);
+               fclose(fp);
+           return 1;
+    }
+        png_destroy_write_struct( &png_ptr, &info_ptr );
+        fclose( fp );
+
+        delete c;
         
         return 0;
 }