/****************************************************************************
 **
 ** Copyright (C) 2011 Nokia Corporation and/or its subsidiary(-ies).
 ** All rights reserved.
 ** Contact: Nokia Corporation (qt-info@nokia.com)
 **
 ** This file is part of the examples of the Qt Toolkit.
 **
 ** $QT_BEGIN_LICENSE:BSD$
 ** You may use this file under the terms of the BSD license as follows:
 **
 ** "Redistribution and use in source and binary forms, with or without
 ** modification, are permitted provided that the following conditions are
 ** met:
 **   * Redistributions of source code must retain the above copyright
 **     notice, this list of conditions and the following disclaimer.
 **   * Redistributions in binary form must reproduce the above copyright
 **     notice, this list of conditions and the following disclaimer in
 **     the documentation and/or other materials provided with the
 **     distribution.
 **   * Neither the name of Nokia Corporation and its Subsidiary(-ies) nor
 **     the names of its contributors may be used to endorse or promote
 **     products derived from this software without specific prior written
 **     permission.
 **
 ** THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 ** "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 ** LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
 ** A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
 ** OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 ** SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
 ** LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 ** DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 ** THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 ** (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 ** OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE."
 ** $QT_END_LICENSE$
 **
 ****************************************************************************/

#ifndef IMAGEVIEWER_H
#define IMAGEVIEWER_H

#include <QtGlobal>
#include <QMainWindow>
#include <QPrinter>
#include <QtGui>
#include <QResizeEvent>
#include <QPushButton>
#include <QSlider>
#include <QColor>
#include <QDoubleSpinBox>

#include <math.h>
#include "fstream"
#include <iostream>

#include "lazy_image.cpp"
#include "filter_gui.cpp"
#include "canny_edge_machine.cpp"
#include "hough_machine.cpp"

class QAction;
class QLabel;
class QMenu;
class QScrollArea;
class QScrollBar;
class QTextEdit;
class QVBoxLayout;
class QTabWidget;
class QPushButton;
class QSpinBox;

class ImageViewer : public QMainWindow {
  Q_OBJECT

  private:
    // First task tab
    QWidget* task_tab1_widget;
    QVBoxLayout* task_tab1_vboxlayout;
    QWidget* task_tab1_scrollwidget;
    QScrollArea* task_tab1_scrollarea;
    QVBoxLayout* task_tab1_scrolllayout;

    QPushButton* draw_black_line;
    QPushButton* draw_rainbow_cross;
    QPushButton* diagonal_cross;
    QPushButton* monochrome;
    QPushButton* reset_working_copy;
    QPushButton* save_to_original;
    
    QSlider* line_slider;

    // Second task tab
    QWidget* task_tab2_widget;
    QVBoxLayout* task_tab2_vboxlayout;
    QWidget* task_tab2_scrollwidget;
    QScrollArea* task_tab2_scrollarea;
    QVBoxLayout* task_tab2_scrolllayout;
    
    QLabel* reference_stats;
    QLabel* original_stats;
    QLabel* stats;
    QLabel* reference_histogramm_label;
    QLabel* original_histogramm_label;
    QLabel* histogramm_label;
    QPushButton* analyze_image;
    
    QSlider* brightness_slider;
    QSlider* contrast_slider;
    
    QSlider* contrast_percentile_slider;
    
    // Third task tab
    QWidget* task_tab3_widget;
    QVBoxLayout* task_tab3_vboxlayout;
    QWidget* task_tab3_scrollwidget;
    QScrollArea* task_tab3_scrollarea;
    QVBoxLayout* task_tab3_scrolllayout;
	
    QLabel* reference_histogramm_cumulative_label;
    QLabel* original_histogramm_cumulative_label;
    QLabel* histogramm_cumulative_label;

    QSlider* partial_adaption_pieces_slider;
    
    QPushButton* linear_histogramm_adaption;
    QPushButton* partial_linear_histogramm_adaption;
    QPushButton* succeeding_histogramm_adaption;
    
    // Fourth task tabWidget
    QWidget* task_tab4_widget;
    QVBoxLayout* task_tab4_vboxlayout;
    QWidget* task_tab4_scrollwidget;
    QScrollArea* task_tab4_scrollarea;
    QVBoxLayout* task_tab4_scrolllayout;
    
    QSlider* filter_size_slider;
    FilterGui* filter_gui;
    QPushButton* apply_filter;
    QComboBox* filter_mode_combobox;
    QPushButton* apply_border_mode_filter;
    
    // Fifth task tabWidget
    QWidget* task_tab5_widget;
    QVBoxLayout* task_tab5_vboxlayout;
    QWidget* task_tab5_scrollwidget;
    QScrollArea* task_tab5_scrollarea;
    QVBoxLayout* task_tab5_scrolllayout;

    QDoubleSpinBox* gauss_filter_size;
    QDoubleSpinBox* gauss_filter_sigma;
    QDoubleSpinBox* t_low_spinbox;    
    QDoubleSpinBox* t_high_spinbox;
    QPushButton* run_canny_edge;
    QPushButton* run_canny_edge_first_step;
    QPushButton* run_canny_edge_second_step;
    QPushButton* run_canny_edge_third_step;
    QPushButton* run_usm;
    QDoubleSpinBox* usm_sharpening;
    QDoubleSpinBox* usm_gauss_width;
    QDoubleSpinBox* usm_gauss_sigma;
    
    // Sixth task tabWidget
    QWidget* task_tab6_widget;
    QVBoxLayout* task_tab6_vboxlayout;
    QWidget* task_tab6_scrollwidget;
    QScrollArea* task_tab6_scrollarea;
    QVBoxLayout* task_tab6_scrolllayout;
    
    QPushButton* run_hough_transformation;
    
    // "My" space for storing data/results
    LazyImage* original;
    LazyImage* working_copy;
    LazyImage* histogramm_reference;
    CannyEdgeMachine* canny_edge_machine;

  private slots:
    // Custom slots
    void drawBlackLine();
    void drawRainbowCross();
    void drawDiagonalCross();
    void analyzeImage();
    void adjustBrightness(int b);
    void adjustContrast(int c);
    void convertToMonochrome();
    void resetWorkingCopy();
    void saveToOriginal();
    void robustAutomaticContrastAdaption(int c_param);
    void linearHistogrammAdaption();
    void partialLinearHistogrammAdaption();
    void succeedingHistogrammAdaption();
    void changeFilterSize(int s);
    void applyBasicFilter();
    void applyBasicFilterWithOverflowHandling();
    void runCannyEdge(void);
    void runCannyEdgeFirstStep(void);
    void runCannyEdgeSecondStep(void);
    void runCannyEdgeThirdStep(void);
    void doUnsharpMasking(void);
    void runHoughTransformation(void);

    void open();
    void openReference();
    void print();
    void zoomIn();
    void zoomOut();
    void normalSize();
    void fitToWindow();
    void about();
     
  public:
    ImageViewer();
    bool loadFile(const QString &);
    void updateImageDisplay();

  protected:
    void resizeEvent(QResizeEvent * event);

  private:
    // Custom methods
    int calcIntensity(int r, int g, int b);
    void drawRainbowCross(int h);
    int getAverageIntensity();
    int getIntensityVariance();
    
    void generateControlPanels();
    void startLogging();
    void generateMainGui();
    void createActions();
    void createMenus();
    void updateActions();
    void scaleImage(double factor);
    void adjustScrollBar(QScrollBar *scrollBar, double factor);

    void initializeImage();
    void renewLogging();

    QTabWidget* tabWidget;
    QTextEdit* logBrowser;
    QWidget* centralwidget;
    QLabel* imageLabel;
    QScrollArea* scrollArea;
    double scaleFactor;

    std::fstream logFile;

#ifndef QT_NO_PRINTER
    QPrinter printer;
#endif

    QAction *openAct;
    QAction *openHistAct;
    QAction *printAct;
    QAction *exitAct;
    QAction *zoomInAct;
    QAction *zoomOutAct;
    QAction *normalSizeAct;
    QAction *fitToWindowAct;
    QAction *aboutAct;
    QAction *aboutQtAct;

    QMenu *fileMenu;
    QMenu *viewMenu;
    QMenu *helpMenu;
};
#endif