na 4 lab

#include "stdafx.h"

#include <opencv2/imgproc/imgproc.hpp>  // Gaussian Blur
#include <opencv2/core/core.hpp>        // Basic OpenCV structures (cv::Mat, Scalar)
#include <opencv2/highgui/highgui.hpp>
#include <stdio.h>
#include <math.h>
#include <iostream>

#define PI (3.141592653589793)

using namespace std;
using namespace cv;

int _tmain(int argc, _TCHAR* argv[])
{
    // Stworzenie okna w którym przechwycone obrazy będą wyświetlane
    cvNamedWindow("JPEG", CV_WINDOW_AUTOSIZE);

    // Pobranie obrazu
    Mat imageFruits = imread("fruits2.jpg",CV_LOAD_IMAGE_GRAYSCALE);

    // Uzyskanie macierzy pikseli na podstawie obiektów Mat
    Mat_<uchar> fruitsPixels = imageFruits;

    const int blockCols = 320/8;
    const int blockRows = 240/8;

    // Utworzenie bloków przechowujących odpowiednio: oryginalny obraz, obraz po DCT, obraz po IDCT
    float block[blockRows][blockCols][8][8];
    float blockDCT[blockRows][blockCols][8][8];
    float blockIDCT[blockRows][blockCols][8][8];


    // TODO
        // podział obrazu na bloki

    for(int i=0; i< fruitsPixels.rows; i++){
        for(int j=0; j< fruitsPixels.cols; j++){
            block[i/8][j/8][i%8][j%8]=fruitsPixels[i][j];
        }
    }
        // zerowanie blockDCT i blockIDCT
    for(int i=0; i< fruitsPixels.rows; i++){
        for(int j=0; j< fruitsPixels.cols; j++){
            blockDCT[i/8][j/8][i%8][j%8]=0;
            blockIDCT[i/8][j/8][i%8][j%8]=0;
        }
    }


    for(int i=0; i< 8 ;i++){
        for(int j=0; j< 8; j++){
            block[0][0][i][j]=0;
        }
    }

    // TODO
        // DCT
    for (int i=0; i<blockRows; i++) {
        for (int j=0; j<blockCols; j++) {
            for (int u=0; u<8; u++) {
                for (int v=0; v<8; v++) {
                    for (int m=0; m<8; m++) {
                        for (int n=0; n<8; n++) {
                            blockDCT[i][j][u][v]+=1.0/64*block[i][j][m][n]*cos((2*m+1)*u*PI/16)*cos((2*n+1)*v*PI/16);
                        }
                    }
                }
            }
        }
    }

        // kwantyzacja
        // dekwantyzacja
        // IDCT

    for (int i=0; i<blockRows; i++) {
        for (int j=0; j<blockCols; j++) {
            for (int u=0; u<8; u++) {
                for (int v=0; v<8; v++) {
                    for (int m=0; m<8; m++) {
                        for (int n=0; n<8; n++) {
                            int lambda1=0;
                            if(u==0) lambda1=1;
                            else lambda1=2;
                            int lambda2=0;
                            if(v==0) lambda2=1;
                            else lambda2=2;
                            blockIDCT[i][j][m][n]+=lambda1*lambda2*blockDCT[i][j][u][v]*cos((2*m+1)*u*PI/16)*cos((2*n+1)*v*PI/16);
                        }
                    }
                }
            }
        }
    }

    // TODO
        // "składanie" obrazu z bloków

    for(int i=0; i< fruitsPixels.rows; i++){
        for(int j=0; j< fruitsPixels.cols; j++){
            fruitsPixels[i][j]=blockIDCT[i/8][j/8][i%8][j%8];
        }
    }

    // Wyświetlenie obrazu
    imshow("JPEG", imageFruits);

    // Oczekiwanie na wciśnięcie klawisza Esc lub Enter
    char key;
    do  key = cvWaitKey(1);
    while(key != 27 && key != 13);

    // Niszczenie okna
    cvDestroyWindow("JPEG");
    return 0;
}