#include "opencv2/highgui/highgui.hpp"

#include "opencv2/core/core.hpp"

#include "opencv2/imgproc/imgproc.hpp"

#include <iostream>

#include <stdio.h>

#define MAXSTRING 50 // 50 chars enough?

using namespace cv;

using namespace std;

// Class to hold the perceptual chunks.

class percepUnit {

        public:

                Mat image; // percept itself

                Mat mask; // alpha channel

                int x1, y1; // close edges TODO change the below values to opencv types? (Point, Rect?)

                int w, h; // width / height of patch

                // constructor method

                percepUnit(Mat ROI, Mat alpha, int ix, int iy, int iw, int ih) {

                        ROI.copyTo(image); // make copies since args are refs!!

                        alpha.copyTo(mask);

                        x1 = ix;

                        y1 = iy;

                        w = iw;        

                        h = ih;

                }

                // Destructor

                ~percepUnit() {

                        // Free memory

                        image.release();

                        mask.release();

                }

};

// pixel by pixel offset copy of perceptUnit into larger image.

// TODO try and dump the dimentions of the image and mask files, and also dump the bounding box sizes to check if they match.

int copyPercept(percepUnit &unit, Mat &dest) {

        // Loop through pixels in percept image

        for( int y = 0; y < unit.image.rows; y++ ) {

        for( int x = 0; x < unit.image.cols; x++ ) {

                        // Make sure this pixel is in the mask.

                        if (unit.mask.at<char>(y,x) != 0) {

                                // get pixels from src image:

                                Vec3b pixel = unit.image.at<Vec3b>(y,x); // Vec3b is a 3 element vector of unsigned chars.

                                // set pixels in dest image (offset for this percept)

                                dest.at<Vec3b>(y+unit.y1,x+unit.x1) = pixel;

                        }

                }

        }

        return(0);

}

int main(int argc, char* argv[])

{

        Mat reconstruction2 = Mat(1080,1920, CV_8UC3, Scalar(255,255,255)); // white background

        for (int s=1070; s<1920; s++) {

                Mat tmpImage = Mat(s,s,  CV_8UC3, Scalar(0,0,0));

                Mat tmpMask = Mat(s,s,  CV_8UC1, Scalar(255,255,255));

                percepUnit tmpUnit = percepUnit(tmpImage, tmpMask, 0, 0, s, s);

                cout << "size: " << s << endl;

                reconstruction2 = copyPercept(tmpUnit, reconstruction2);

        }

        imwrite("reconstruction2.png",reconstruction2);

        exit(0); // this was "return" before, did that not clean up anything?

}