// Dolaczamy plik naglowkowy naszej klasy MyWindow
#include "mywindow.h"

// Dolaczamy plik naglowkowy zawierajacy definicje GUI
// Plik ten jest generowany automatycznie
// z pliku XML "mywindow.ui"
#include "ui_mywindow.h"
#include <QColor>
#include <QDebug>
#include <iostream>
#include<float.h> // for float,double macros
// Definicja konstruktora, wywolujemy najpierw
// konstruktor klasy nadrzednej, nastepnie tworzymy
// obiekt klasy Ui_MyWindow reprezentujacy GUI
MyWindow::MyWindow(QWidget *parent) :
    QMainWindow(parent),
    ui(new Ui::MyWindow)
{
    // Wywolujemy funkcje tworzaca elementy GUI
    // Jej definicja znajduje sie w pliku "ui_mywindow.h"
    ui->setupUi(this);

    // Pobieramy wymiary i wspolrzedne lewego gornego naroznika ramki
    // i ustawiamy wartosci odpowiednich pol
    // Uwaga: ramke "rysujFrame" wykorzystujemy tylko do
    // wygodnego ustaiwenia tych wymiarow. Rysunek bedziemy wyswietlac
    // bezposrednio w glownym oknie aplikacji.
    szer = ui->rysujFrame->width();
    wys = ui->rysujFrame->height();
    poczX = ui->rysujFrame->x();
    poczY = ui->rysujFrame->y();

    // Tworzymy obiekt klasy QImage, o odpowiedniej szerokosci
    // i wysokosci. Ustawiamy format bitmapy na 32 bitowe RGB
    // (0xffRRGGBB).
    img = new QImage(szer,wys,QImage::Format_RGB32);
    img_tmp = new QImage(szer,wys,QImage::Format_RGB32);
    active_img = img;
}

// Definicja destruktora
MyWindow::~MyWindow()
{
    delete ui;
}

// Funkcja (slot) wywolywana po nacisnieciu przycisku "Wyjscie" (exitButton)
// Uwaga: polaczenie tej funkcji z sygnalem "clicked"
// emitowanym przez przycisk jest realizowane
// za pomoca funkcji QMetaObject::connectSlotsByName(MyWindow)
// znajdujacej sie w automatycznie generowanym pliku "ui_mywindow.h"
// Nie musimy wiec sami wywolywac funkcji "connect"
void MyWindow::on_exitButton_clicked()
{
    // qApp to globalny wskaznik do obiektu reprezentujacego aplikacje
    // quit() to funkcja (slot) powodujaca zakonczenie aplikacji z kodem 0 (brak bledu)
    qApp->quit();
}


// Funkcja "odmalowujaca" komponent
void MyWindow::paintEvent(QPaintEvent*)
{
    // Obiekt klasy QPainter pozwala nam rysowac na komponentach
    QPainter p(this);

    // Rysuje obrazek "img" w punkcie (poczX,poczY)
    // (tu bedzie lewy gorny naroznik)
    p.drawImage(poczX,poczY,*active_img);
}


// Funkcja (slot) wywolywana po nacisnieciu przycisku "Czysc" (cleanButton)
void MyWindow::on_cleanButton_clicked()
{
    // Funkcja czysci (zamalowuje na bialo) obszar rysowania
    // definicja znajduje sie ponizej
    czysc();

    // Funkcja "update()" powoduje ponowne "namalowanie" calego komponentu
    // Wywoluje funkcje "paintEvent"
    update();
}


// Funkcja powoduje wyczyszczenie (zamalowanie na bialo)
// obszaru rysowania
void MyWindow::czysc()
{
    // Wskaznik za pomoca, ktorego bedziemy modyfikowac obraz
    unsigned char *ptr;

    // Funkcja "bits()" zwraca wskaznik do pierwszego piksela danych
    ptr = img->bits();

    int i,j;

    // Przechodzimy po wszystkich wierszach obrazu
    for(i=0; i<wys; i++)
    {
        // Przechodzimy po pikselach danego wiersza
        // W kazdym wierszu jest "szer" pikseli (tzn. 4 * "szer" bajtow)
        for(j=0; j<szer; j++)
        {
            ptr[szer*4*i + 4*j]=0; // Skladowa BLUE
            ptr[szer*4*i + 4*j + 1] = 0; // Skladowa GREEN
            ptr[szer*4*i + 4*j + 2] = 0; // Skladowa RED
            ptr[szer*4*i + 4*j + 3] = 255; // Skladowa RED
        }
    }
}


// Funkcja (slot) wywolywana po nacisnieciu przycisku myszy (w glownym oknie)
void MyWindow::mousePressEvent(QMouseEvent *event)
{
    UpdateTempImage();

    // Pobieramy wspolrzedne punktu klikniecia
    int x = event->x();
    int y = event->y();

    // Sa to wspolrzedne wzgledem glownego okna,
    // Musimy odjac od nich wpolrzedne lewego gornego naroznika rysunku

    x -= poczX;
    y -= poczY;

    startX = x;
    startY = y;

    active_img = img_tmp;

    draw_finished = false;

    update();
}

void MyWindow::mouseMoveEvent(QMouseEvent *event)
{
    // Pobieramy wspolrzedne punktu klikniecia
    int x = event->x();
    int y = event->y();

    // Sa to wspolrzedne wzgledem glownego okna,
    // Musimy odjac od nich wpolrzedne lewego gornego naroznika rysunku

    x -= poczX;
    y -= poczY;

    if(x >= szer || y >= wys || x < 0 || y < 0 || draw_finished){
        ApplyTempImage();

        active_img = img;

        draw_finished = true;

        update();

        return;
    }

    DrawLine(startX, startY, x, y);

    // Odswiezamy komponent
    update();
}

void MyWindow::mouseReleaseEvent(QMouseEvent *event){
    if(draw_finished)
        return;

    ApplyTempImage();

    active_img = img;

    draw_finished = true;

    update();
}

void MyWindow::ApplyTempImage(){
    memcpy(img->bits(), img_tmp->bits(), szer * wys * 4);
}

void MyWindow::UpdateTempImage(){
     memcpy(img_tmp->bits(), img->bits(), szer * wys * 4);
}

void MyWindow::DrawLine(int x1, int y1, int x2, int y2){

    if(x2 >= szer || y2 >= wys || x2 < 0 || y2 < 0)
        return;

    UpdateTempImage();

    unsigned char *ptr = img_tmp->bits();

    if(x1 > x2){
        std::swap(x1, x2);
        std::swap(y1, y2);
    }
    float diff = x2 - x1;
    float a = diff != 0 ? (y2 - y1) / diff : FLT_MAX;

    if(abs(a) < 0.5f){
        for(int x = x1; x <= x2; x++){
            int x_form = x - x1;
            int y = a * x_form + y1;

            SetColor(ptr, QColor(255, 255, 255), x, y);
        }
    }
    else{
        if(y1 > y2){
            std::swap(x1, x2);
            std::swap(y1, y2);
        }
        float diff = x2 - x1;
        float a = diff != 0 ? (y2 - y1) / diff : FLT_MAX;
        for(int y = y1; y <= y2; y++){
            int y_form = y - y1;
            int x = ((float)(y_form) / a) + x1;

            SetColor(ptr, QColor(255, 255, 255), x, y);
        }
    }
}

void MyWindow::SetColor(unsigned char *ptr, QColor color, int x, int y){
    ptr[szer*4*y + 4*x] = color.red();
    ptr[szer*4*y + 4*x + 1] = color.green();
    ptr[szer*4*y + 4*x + 2] = color.blue();
    ptr[szer*4*y + 4*x + 3] = 255;
}