#include <iostream>
#include <opencv2/highgui/highgui.hpp>

using namespace std;
using namespace cv;

int main(int argc, char ** argv)
{
    const char *fileName = "../baboon.jpg";

    if (argc > 1)
    {
        fileName = argv[1];
    }

    //
    // For your reference: Some image types of interest:
    //
    cout << "CV_8UC1  = " << CV_8UC1  << endl;
    cout << "CV_8UC3  = " << CV_8UC3  << endl;
    cout << "CV_32FC1 = " << CV_32FC1 << endl;
    cout << "CV_32FC3 = " << CV_32FC3 << endl;
    cout << "CV_64FC1 = " << CV_64FC1 << endl;
    cout << "CV_64FC3 = " << CV_64FC3 << endl;


    Mat originalImage = imread(fileName, 1); 
    // Check to make sure it's valid
    if (originalImage.empty() || (originalImage.data == NULL))
    {
        cout << "Can't load image from " << fileName << "!" << endl;
        exit(EXIT_FAILURE);
    }
    cout << "Working on image from " << fileName << endl;

    //
    // Show what we have so far
    //
    int x = 0;
    int y = 0;

    namedWindow("Original", CV_WINDOW_AUTOSIZE);
    moveWindow("Original", x, y);
    imshow("Original", originalImage);
    x += 100;
    y += 100;

    int width = originalImage.size().width;
    int height = originalImage.size().width;

    cout << "Original image Width x Height is " << width << "x" << height << endl;

    // Leave original alone, work on a copy
    Mat dctImage = originalImage.clone();

    // Step through the copied image with rectangles size 8x8
    // For each block, split into planes, do dct, and merge back
    // into the block.  (This will affect the image from
    // which the block is selected each time.)

    for (int i = 0; i < height; i += 8) {
        for (int j = 0; j < width; j+= 8) {
            Mat block = dctImage(Rect(i, j, 8, 8));
            vector<Mat> planes;
            split(block, planes);
            vector<Mat> outplanes(planes.size());
            for (size_t k = 0; k < planes.size(); k++)
            {
                planes[k].convertTo(planes[k], CV_32FC1);
                dct(planes[k], outplanes[k]);
                outplanes[k].convertTo(outplanes[k], CV_8UC1);
            }
            merge(outplanes, block);
        }
    }
    namedWindow("dctBlockImage");
    moveWindow("dctBlockImage", x, y);
    imshow("dctBlockImage", dctImage);
    x += 100;
    y += 100;
.
. // Whatever else you want to do
.
    
    waitKey(0); 

    destroyAllWindows();
}