/// \file

// RAN - 2014/06/10 : Clear the textEdit control image is loaded.

#include "mainwindow.h"
#include "ui_mainwindow.h"
#include <QtDebug>
#include <QLabel>
#include "steganography.h"

/// \fn StegaPanel::StegaPanel(QWidget *parent)
/// \~English
/// \brief GUI Constructor
/// \~Spanish
/// \brief Constructor del GUI
StegaPanel::StegaPanel(QWidget *parent) :
    QMainWindow(parent),
    ui(new Ui::StegaPanel)
{
    ui->setupUi(this); //

}

/// \fn void StegaPanel::hideMessage()
/// \~English
/// \brief Function that calls the message embedding function.
/// \~Spanish
/// \brief Funcion que llama la funcion de embedir el mensaje.
///
void StegaPanel::hideMessage()
{

    string binMsg ;         // Will be used to store the message in a binary String format

    ui->error_msg_label->clear(); //clear error text
    orig_msg = ui->textEdit->toPlainText(); //obtain msg from textEdit

    if(old_image.isNull()){
        qDebug() << "Load an image first!";
        ui->error_msg_label->setText("NO IMAGE LOADED!");
    }
    else if (orig_msg == NULL){
        ui->error_msg_label->setText("No message to hide!");
    }
    else{

        binMsg = messageToBinaryString(orig_msg.toStdString()) ;

        if(((old_image.width() * old_image.height()) * 3) < (int)binMsg.length())
        {
            ui->error_msg_label->setText("IMAGE NOT BIG ENOUGH FOR MESSAGE!");
        }
        else
        {
            EmbbedMessage(old_image, new_image, orig_msg.toStdString()) ;
        }

        ui->error_msg_label->setText("Message hidden in Stego Image with 1 stego bits.");
    }
}


/// \fn void StegaPanel::revealMessage()
/// \~English
/// \brief Function that calls the message extraction function.
/// \~Spanish
/// \brief Funcion que llama la funcion de extraer el mensaje.
///
void StegaPanel::revealMessage()
{

    string msg = ExtractMessage(new_image) ;

    if (msg.length() > 0){
        ui->textEdit->setText(QString::fromStdString(msg));
        ui->error_msg_label->setText("Message extracted!");
    }
    else{
        ui->error_msg_label->setText("Could not extract message!");
    }
}

/// \fn void StegaPanel::on_loadImage_clicked()
/// \~English
/// \brief Event function to load an image from the
///                    file system.
/// \~Spanish
/// \brief Funcion de evento que carga una imagen de el
///                    sistema de archivos.
///
void StegaPanel::on_loadImage_clicked()
{
    QString fname = QFileDialog::getOpenFileName(this, tr("Choose an image"), QDir::homePath());
        if (!fname.isEmpty()){
            QImage image(fname);
            if (image.isNull())
                QMessageBox::information(this, tr("Choose an image"),tr("Cannot load %1.").arg(fname));
            old_image=image;
            new_image=image;
            repaint();
        }
        ui->old_image->setPixmap(QPixmap::fromImage(old_image));
        ui->new_image->setPixmap(QPixmap::fromImage(new_image));
        ui->textEdit->setText("");
        ui->error_msg_label->setText("Image loaded!");
}

/// \fn void StegaPanel::on_hideMessage_clicked()
/// \~English
/// \brief Event function to hide message into
///                     the image.
/// \~Spanish
/// \brief Funcion de evento para esconder e
///         el mensaje en la imagen.
///
void StegaPanel::on_hideMessage_clicked()
{
    hideMessage();
    ui->new_image->setPixmap(QPixmap::fromImage(new_image));
}

/// \fn void StegaPanel::on_getMessage_clicked()
/// \~English
/// \brief Event function to extract the message hidden
///                     in the image.
/// \~Spanish
/// \brief Funcion de evento para estraer el mensaje
///                     escondido en la imagen.
///
void StegaPanel::on_getMessage_clicked()
{
    revealMessage();
}

/// \fn StegaPanel::~StegaPanel()
/// \~English
/// \brief GUI Destructor
/// \~Spanish
/// \brief Destructor del GUI
StegaPanel::~StegaPanel()
{
    delete ui;
}

/// \fn void StegaPanel::on_storeImage_clicked()
/// \~English
/// \brief on_storeImage_clicked - Event function to save the new image in the
///                             file system.
/// \~Spanish
/// \brief Funcion de evento para salvar la imagen nueva
///                             en el sistema de archivos.
///
void StegaPanel::on_storeImage_clicked()
{
    QPixmap out = QPixmap::grabWidget(this,361,10,481,481);
    QString fname = QFileDialog::getSaveFileName(this, tr("Save Edited Image"), (""), tr("PNG (*.png)" ));
    new_image.save(fname, "PNG");
    ui->error_msg_label->setText("Image saved to" + fname + " !");
}