window_grayscale_batch_test.cpp

Go to the documentation of this file.

#include <boost/thread/thread.hpp>
#include <boost/bind.hpp>

#include <types/MNISTTypes.hpp>

#include <logger/Log.hpp>
#include <logger/ConsoleOutput.hpp>
using namespace mic::logger;

#include <opengl/visualization/WindowManager.hpp>
#include <opengl/visualization/WindowGrayscaleBatch.hpp>
using namespace mic::opengl::visualization;

WindowGrayscaleBatch<float>* w_batch;

void test_thread_body (void) {

    // Generate a batch that will be used for returning minibatches of size 5.
    mic::types::MNISTBatch<float> batch(6);
    for(size_t i=0; i< 18; i++) {
        // Generate "data".
        MatrixXfPtr data (new MatrixXf(3, 6));
        data->setZero();
        (*data)(i)=(float)(i-9.0)/18.0;
        batch.data().push_back(data);

        // Generate "label".
        batch.labels().push_back(std::make_shared<unsigned int>(i));

        // Add index.
        batch.indices().push_back(i);
    }//: for

    // Main application loop.
    while (!APP_STATE->Quit()) {

        // If not paused.
        if (!APP_STATE->isPaused()) {

            // If single step mode - pause after the step.
            if (APP_STATE->isSingleStepModeOn())
                APP_STATE->pressPause();

            { // Enter critical section - with the use of scoped lock from AppState!
                APP_DATA_SYNCHRONIZATION_SCOPED_LOCK();

                // Retrieve the next minibatch.
                mic::types::MNISTBatch<float> bt = batch.getNextBatch();

                // Set sample to be displayed.
                w_batch->setBatchUnsynchronized(bt.data());
            }//: end of critical section

        }//: if

        // Sleep.
        APP_SLEEP();
    }//: while

}//: image_encoder_and_visualization_test



int main(int argc, char* argv[]) {
    // Set console output to logger.
    LOGGER->addOutput(new ConsoleOutput());
    LOG(LINFO) << "Logger initialized. Starting application";

    // Initialize GLUT! :]
    VGL_MANAGER->initializeGLUT(argc, argv);

    // Create batch visualization window.
    w_batch = new WindowGrayscaleBatch<float>("Grayscale batch", Grayscale::Norm_HotCold, Grayscale::Grid_Both, 0, 0, 512, 512);

    boost::thread test_thread(boost::bind(&test_thread_body));

    // Start visualization thread.
    VGL_MANAGER->startVisualizationLoop();

    LOG(LINFO) << "Waiting for threads to join...";
    // End test thread.
    test_thread.join();
    LOG(LINFO) << "Threads joined - ending application";
}//: main