#include "img_handler.hpp"
#include "puzzle.hpp"

#define STB_IMAGE_IMPLEMENTATION
#define STB_IMAGE_WRITE_IMPLEMENTATION
#define STB_IMAGE_RESIZE_IMPLEMENTATION
#include "lib/stb_image.h"
#include "lib/stb_image_resize2.h"
#include "lib/stb_image_write.h"

#include <stdlib.h>
#include <filesystem>
#include <stdio.h>
#include <iostream>
#include <string>

ImageHandler::ImageHandler(const char *p) : path_to_file(p) {
    img_data = nullptr;
    resized_data = nullptr;
    /*
     * creating resources directory (if it didn't exist)
     */
    std::filesystem::path res_path = std::string("resources");
    if(!std::filesystem::exists(res_path))
        std::filesystem::create_directory(res_path);

#if defined(_WIN32)
    auto it = path_to_file.find_last_of('\\');
#else
    auto it = path_to_file.find_last_of('/');
#endif
    std::string file_name = path_to_file.substr(it + 1);
    std::string fn_withot_ext =
        file_name.substr(0, file_name.find_first_of('.'));
#if defined(_WIN32)
    path_to_save = "resources\\" + fn_withot_ext + "\\";
#else
    path_to_save = "resources/" + fn_withot_ext + '/';
#endif
    std::filesystem::create_directory(path_to_save);
}

void ImageHandler::load_img()
{
    img_data = stbi_load(path_to_file.c_str(), &in_width, &in_height,
                         &channel, 0);
}

void ImageHandler::resize_img()
{
    resized_data =
        reinterpret_cast<unsigned char*>(malloc(in_width * in_height *
                                                                    channel));
    stbir_resize_uint8_linear(img_data, in_width, in_height, 0, resized_data,
                              width, height, 0, (stbir_pixel_layout) channel);
}

void ImageHandler::save_img()
{
    int x, y, i, j, k = 0;
    for(i = 0; i < puzzles_per_side; ++i)
        for(j = 0; j < puzzles_per_side; ++j, ++k) {
            x = i * puzzle_size;
            y = j * puzzle_size;
            std::string tmp_name =
                path_to_save +
                std::to_string(k / puzzles_per_side) +
                std::to_string(k % puzzles_per_side) + ".png";
            stbi_write_png(tmp_name.c_str(), puzzle_size, puzzle_size, channel,
                           resized_data + channel * (x + y * width),
                           width * channel);
        }
    stbi_image_free(resized_data);
    stbi_image_free(img_data);
}