/*

 * MoUT.cpp

 *

 *  Created on: Mar 25, 2011

 *      Author: ams

 *

 *  A Museum of Useful Things to work with images using OpenCV. I ended up

 *  having to the same file conversions, etc. over and over again.

 *

 */

#include "MoUT.hpp"

using namespace std;

using namespace cv;

//Unsafe way to get pixel values from http://stackoverflow.com/questions/4747920/opencv-rgb-value-for-cvpoint-in-cvmat

RGB& GetRGB(cv::Mat &mat, int x, int y)

{

  assert((mat.step/mat.cols) == sizeof(RGB));

  RGB *data = (RGB*)mat.data;

  data += y * mat.cols + x;

  return *data;

}

/* Load an image, downsample, and convert to a neural net training vector */

string img2NNVect(string imgPath)

{

        // For downsampling

        vector<cv::Mat> smaller;

        int levels = 4;

        //For number to string conversions

        char buffer[45];

        // load file and set up new image

        // last parameter forces loading as RGB color

        cv::Mat original = imread(imgPath, 1);

        cv::buildPyramid(original, smaller, levels);

        //convert the smallest version into a row

        string imgRow = "";

        cv::Mat tiny = smaller[levels-1];

        for(int xx = 0; xx < tiny.cols; xx++)

        {

                for(int yy = 0; yy < tiny.rows; yy++)

                {

                        RGB pixel = GetRGB(tiny, xx, yy);

                        sprintf(buffer, "%d", pixel.r);

                        imgRow += buffer;

                        imgRow += " ";

                        sprintf(buffer, "%d", pixel.g);

                        imgRow += buffer;

                        imgRow += " ";

                        sprintf(buffer, "%d", pixel.b);

                        imgRow += buffer;

                        imgRow += " ";

                }

        }

}

void Tokenize(const string& str, vector<string>& tokens, const string& delimiters)

{

    // Skip delimiters at beginning.

    string::size_type lastPos = str.find_first_not_of(delimiters, 0);

    // Find first "non-delimiter".

    string::size_type pos     = str.find_first_of(delimiters, lastPos);

    while (string::npos != pos || string::npos != lastPos)

    {

        // Found a token, add it to the vector.

        tokens.push_back(str.substr(lastPos, pos - lastPos));

        // Skip delimiters.  Note the "not_of"

        lastPos = str.find_first_not_of(delimiters, pos);

        // Find next "non-delimiter"

        pos = str.find_first_of(delimiters, lastPos);

    }

}