#include <iostream>
#include <string>
#include <cstdlib>
#include <SDL/SDL.h>
#include <png.h>

int* readPNG(const std::string& file) {
        FILE* fp = fopen(file.c_str(), "rb");
        png_byte header[8];
        png_structp png_ptr = png_create_read_struct(PNG_LIBPNG_VER_STRING,
                                                     NULL, NULL, NULL);
        png_infop info_ptr = png_create_info_struct(png_ptr);

        if (!png_ptr) {
                std::cout << "Error! png struct not ok." << std::endl;
                return NULL;
        }

        if (!info_ptr) {
                std::cout << "Error: info struct not ok." << std::endl;
                return NULL;
        }

        if (!fp) {
                std::cout << file << " not found!" << std::endl;
                return NULL;
        }

        fread(header, 1, 8, fp);
        bool is_png = !png_sig_cmp(header, 0, 8);

        if (!is_png) {
                std::cout << file << " is not a PNG file!" << std::endl;
                return NULL;
        }

        png_init_io(png_ptr, fp);
        png_set_sig_bytes(png_ptr, 8);

        png_read_png(png_ptr, info_ptr, PNG_TRANSFORM_IDENTITY, NULL);

        int width = info_ptr->width;
        int height = info_ptr->height;
        int ch = info_ptr->channels;

        int* image = new int[width * height];

        png_bytepp rows = png_get_rows(png_ptr, info_ptr);

        for (int y = 0; y < height; ++y) {
                for (int x = 0; x < width; ++x) {
                        image[y * width + x] = (rows[height-y-1][x*ch+0]<<16) |
                                                                (rows[height-y-1][x*ch+1]<<8) |
                                                                 rows[height-y-1][x*ch+2];
                }
        }
        return image;
}

int main() {
        SDL_Event e;
        int* image;

        if (!(image = readPNG("image.png"))) {
                return EXIT_FAILURE;
        }

        SDL_Surface* surface = SDL_SetVideoMode(640, 480, 32, SDL_DOUBLEBUF);
        Uint32* fbuf = (Uint32*)surface->pixels;

        do {
                SDL_PollEvent(&e);

                // Tässä esimerkissä kuvan koko on tiedettävä etukäteen ja se on 256x256.
                for (int y = 0; y < 256; ++y) {
                        for (int x = 0; x < 256; ++x) {
                                fbuf[y * 640 + x] = image[y * 256 + x];
                        }
                }
                SDL_Flip(surface);
        } while (e.type != SDL_KEYDOWN);
        return EXIT_SUCCESS;
}