// 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 <cmath>
#include <QColor>
#include <QDebug>
#include <QColorDialog>
#include <iostream>
#include <float.h> // for float,double macros
#include <string.h>

#define PI 3.1415

// 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();

    segment_count = ui->segmentSlider->value();
    QString str(("Ilosc segmentow: " + std::to_string(segment_count)).c_str());
    ui->segmentLabel->setText(str);

    // 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();
}


void MyWindow::on_segmentSlider_valueChanged(int val)
{
    segment_count = val;
    QString str(("Ilosc segmentow: " + std::to_string(segment_count)).c_str());
    ui->segmentLabel->setText(str);
}


// 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;

    mode = event->button() == Qt::LeftButton ? Circle : event->button() == Qt::RightButton ? Ellipse : Other;

    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;
    }

    if(mode == Circle){
        DrawCircle(startX, startY, x, y);
    }else if(mode == Ellipse){
        DrawEllipse(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 DrawPixel(unsigned char* ptr, int szer, int wys, QColor color, int x, int y){
    if(x >= szer || y >= wys || x < 0 || y < 0)
        return;

    ptr[szer*4*y + 4*x] = color.blue();
    ptr[szer*4*y + 4*x + 1] = color.green();
    ptr[szer*4*y + 4*x + 2] = color.red();
    ptr[szer*4*y + 4*x + 3] = color.alpha();
}

void MyWindow::DrawLine(int x1, int y1, int x2, int y2, bool update_temp){
    if(update_temp)
        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;

    QColor color(255, 255, 255, 255);

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

            DrawPixel(ptr, szer, wys, color, 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;

            DrawPixel(ptr, szer, wys, color, x, y);
        }
    }
}


void MyWindow::DrawCircle(int x1, int y1, int x2, int y2){
    if(x2 >= szer || y2 >= wys || x2 < 0 || y2 < 0)
        return;

    UpdateTempImage();

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

    float R = sqrt(pow(x2 - x1, 2) + pow(y2 - y1, 2));
    int q_last_x = R / sqrt(2) + 1;

    QColor color(255, 255, 255, 255);

    for(int x = 0; x <= q_last_x; x++){
        float y = sqrt(pow(R, 2) - pow(x, 2));

        for(int k = 0; k < 2; k++){
            for(int i = 0; i < 2; i++){
                for(int j = 0; j < 2; j++){
                    DrawPixel(ptr, szer, wys, color, x1 + x, y1 - y);
                    y *= -1;
                }
                x *= -1;
            }
            int tmp = x;
            x = y;
            y = tmp;
        }
    }
}
void MyWindow::DrawEllipse(int x1, int y1, int x2, int y2){
    if(x2 >= szer || y2 >= wys || x2 < 0 || y2 < 0)
        return;

    UpdateTempImage();

    float last_x = 0;
    float last_y = 0;

    float Rx = abs(x2 - x1);
    float Ry = abs(y2 - y1);

    for(int i = 0; i <= segment_count; i++){
        double angle = 2.0 * PI * (float(i) / segment_count);

        float x = sin(angle) * Rx;
        float y = cos(angle) * Ry;

        if(i > 0)
            DrawLine(last_x + x1, last_y + y1, x + x1, y + y1, false);

        last_x = x;
        last_y = y;
    }

    return;

}