#include "horizontalslicetoimagemapper.h"
#include <iostream>
#include <cmath>


HorizontalSliceToImageMapper::HorizontalSliceToImageMapper()
{
    magnitudeBool = false;
    std::cout << "DEBUG: MAPPER CONSTRUCTOR CALLED" << std::endl;
}

HorizontalSliceToImageMapper::~HorizontalSliceToImageMapper()
{
    std::cout << "MAPPER DESTRUCTOR CALLED!\n";
}

void HorizontalSliceToImageMapper::getSlice(float *source, int x, int y) {
    slice = source;
    xs = x;
    ys = y;
}

void HorizontalSliceToImageMapper::getSlice(float *source) {
    slice = source;
}

void HorizontalSliceToImageMapper::setMagnitude(bool x) {
    magnitudeBool = x;
}

float HorizontalSliceToImageMapper::totalWindSpeed(float x, float y, float z) {
    return sqrt(x*x + y*y + z*z);
}

QImage HorizontalSliceToImageMapper::mapSliceToImage() {
    // We loop through the windspeed value and map a red colour to it, if positive
    // otherwise it's a blue colour, the brightness determines the intensity
    // We amplify the intensity by a factor 3 for a better visualization


    //std::cout << xs << " hier kommt noch ys:" << ys << std::endl;
    QImage image(xs, ys, QImage::Format_RGB32);
    QRgb colour;
    int value;

    if (!magnitudeBool) {
        for (int i = 0; i < xs; i++) {
            for (int j = 0; j < ys; j++) {
                value = int(255*slice[3*(j*xs+i)]);
                //std::cout << slice[3*(i*xs+j)] << " ";
                //std::cout << value << std::endl;
                if (value >= 0) {
                    colour = qRgb(3*value, 20, 0);
                } else {
                    colour = qRgb(0, 20, -3*value);
                }
                image.setPixel(j, i, colour);
            }
        }
    } else {
        for (int i = 0; i < xs; i++) {
            for (int j = 0; j < ys; j++) {
                value = int(HorizontalSliceToImageMapper::totalWindSpeed(255*slice[3*(j*xs+i)+2], 255*slice[3*(j*xs+i)+1], 255*slice[3*(j*xs+i)]));
                //std::cout << value << std::endl;
                //always positive so no need for extra checks
                colour = qRgb(3*value, 20, 20);
                image.setPixel(j, i, colour);
            }
        }
    }




    // FOR DEBUG
    if (image.isNull()) {
        std::cout << "IMAGE LOADING ERROR IN MAPPER";
    }
    return image;
}

QImage HorizontalSliceToImageMapper::mapSliceToImage(QString fileName) {
    // CREATED FOR LEARNING PURPOSES
    // maps the UHH logo to a frame
    QImage image;
    if (!image.load(fileName)) {
        std::cout << "ERROR! Image could not be read!" << std::endl;
    }
    return image;
}