/////////////////////////////////////////////////////////////////////////////////
//
//  Copyright (C) 2011-     Statoil ASA
//  Copyright (C) 2013-     Ceetron Solutions AS
//  Copyright (C) 2011-2012 Ceetron AS
//
//  ResInsight is free software: you can redistribute it and/or modify
//  it under the terms of the GNU General Public License as published by
//  the Free Software Foundation, either version 3 of the License, or
//  (at your option) any later version.
//
//  ResInsight is distributed in the hope that it will be useful, but WITHOUT ANY
//  WARRANTY; without even the implied warranty of MERCHANTABILITY or
//  FITNESS FOR A PARTICULAR PURPOSE.
//
//  See the GNU General Public License at <http://www.gnu.org/licenses/gpl.html>
//  for more details.
//
/////////////////////////////////////////////////////////////////////////////////

#pragma once

#include "RiuMainWindowBase.h"
#include "RiuMdiArea.h"

#include "cafPdmObjectHandle.h"
#include "cafPdmUiDragDropInterface.h"

#include <QEvent>
#include <QLabel>
#include <QPointer>

#include <memory>
#include <vector>

class QActionGroup;
class QMdiSubWindow;
class QToolButton;
class QSpinBox;
class QTimer;
class QUndoView;

class Rim3dView;
class RimCase;
class RimViewWindow;

class RiuMessagePanel;
class RiuProcessMonitor;
class RiuResultInfoPanel;
class RiuResultQwtPlot;
class RiuRelativePermeabilityPlotPanel;
class RiuPvtPlotPanel;
class RiuMohrsCirclePlot;

struct RimMdiWindowGeometry;

namespace caf
{
class PdmUiTreeView;
class AnimationToolBar;
class PdmObject;
class PdmUiPropertyView;
class PdmUiItem;
} // namespace caf

namespace ssihub
{
class Interface;
}

//==================================================================================================
//
//
//
//==================================================================================================
class RiuMainWindow : public RiuMainWindowBase
{
    Q_OBJECT

public:
    RiuMainWindow();
    ~RiuMainWindow() override;

    static RiuMainWindow* instance();

    QString mainWindowName() override;

    void initializeGuiNewProjectLoaded();
    void cleanupGuiCaseClose();
    void cleanupGuiBeforeProjectClose();

    void removeViewer( QWidget* viewer ) override;
    void initializeViewer( QMdiSubWindow* subWindow, QWidget* viewer, const RimMdiWindowGeometry& windowsGeometry ) override;
    void setActiveViewer( QWidget* subWindow ) override;

    void setResultInfo( const QString& info ) const;

    void refreshViewActions();
    void refreshAnimationActions();
    void updateScaleValue();

    RiuProcessMonitor* processMonitor();

    void selectedCases( std::vector<RimCase*>& cases );

    void setDefaultWindowSize();

    void refreshDrawStyleActions();

    void tileSubWindows() override;
    void storeSubWindowTiling( bool tiled ) override;
    void clearWindowTiling() override;

    bool                  subWindowsAreTiled() const override;
    bool                  isAnyMdiSubWindowVisible();
    QMdiSubWindow*        findMdiSubWindow( QWidget* viewer ) override;
    RimViewWindow*        findViewWindowFromSubWindow( QMdiSubWindow* lhs );
    QList<QMdiSubWindow*> subWindowList( QMdiArea::WindowOrder order );

    RiuResultQwtPlot*                 resultPlot();
    RiuRelativePermeabilityPlotPanel* relativePermeabilityPlotPanel();
    RiuPvtPlotPanel*                  pvtPlotPanel();
    RiuMohrsCirclePlot*               mohrsCirclePlot();
    RiuMessagePanel*                  messagePanel();

    void showProcessMonitorDockPanel();
    void setDefaultToolbarVisibility();
    void applyFontSizesToDockedPlots();

protected:
    void closeEvent( QCloseEvent* event ) override;

private:
    void createActions();
    void createMenus();
    void createToolBars();
    void createDockPanels();

    void restoreTreeViewState();

    void updateUiFieldsFromActiveResult( caf::PdmObjectHandle* objectToUpdate );

private:
    // Edit actions
    QAction* m_newPropertyView;

    // View actions
    QAction* m_viewFromNorth;
    QAction* m_viewFromSouth;
    QAction* m_viewFromEast;
    QAction* m_viewFromWest;
    QAction* m_viewFromAbove;
    QAction* m_viewFromBelow;

    // Mock actions
    QAction* m_mockModelAction;
    QAction* m_mockResultsModelAction;
    QAction* m_mockLargeResultsModelAction;
    QAction* m_mockModelCustomizedAction;
    QAction* m_mockInputModelAction;

    QAction* m_snapshotAllViewsToFile;

    QAction* m_createCommandObject;
    QAction* m_showRegressionTestDialog;
    QAction* m_executePaintEventPerformanceTest;

    caf::AnimationToolBar* m_animationToolBar;

    RiuMdiArea*               m_mdiArea;
    RiuResultInfoPanel*       m_resultInfoPanel;
    RiuProcessMonitor*        m_processMonitor;
    QPointer<RiuMessagePanel> m_messagePanel;

    RiuResultQwtPlot*                 m_resultQwtPlot;
    RiuMohrsCirclePlot*               m_mohrsCirclePlot;
    RiuRelativePermeabilityPlotPanel* m_relPermPlotPanel;
    RiuPvtPlotPanel*                  m_pvtPlotPanel;

    QMenu*       m_windowMenu;
    QLabel*      m_memoryCriticalWarning;
    QToolButton* m_memoryUsedButton;
    QLabel*      m_memoryTotalStatus;
    QTimer*      m_memoryRefreshTimer;

    // Menu and action slots
private slots:

    friend class RiuMdiSubWindow;

    // Memory update slot
    void updateMemoryUsage();

    // File slots
    void slotRefreshFileActions();

    // Edit slots
    void slotNewObjectPropertyView();

    // View slots
    void slotRefreshViewActions();
    void slotViewFromNorth();
    void slotViewFromSouth();
    void slotViewFromEast();
    void slotViewFromWest();
    void slotViewFromAbove();
    void slotViewFromBelow();
    void slotScaleChanged( int scaleValue );

    void slotDrawStyleChanged( QAction* activatedAction );
    void slotToggleHideGridCellsAction( bool );
    void slotToggleFaultLabelsAction( bool );
    void slotToggleLightingAction( bool );

    void slotShowWellCellsAction( bool doAdd );

    // Debug slots
    void slotSnapshotAllViewsToFile();

    void slotCreateCommandObject();

    void slotShowRegressionTestDialog();
    void slotExecutePaintEventPerformanceTest();

    // Mock models
    void slotMockModel();
    void slotMockResultsModel();
    void slotMockLargeResultsModel();
    void slotMockModelCustomized();
    void slotInputMockModel();

    // Windows slots
    void slotBuildWindowActions();
    void slotSubWindowActivated( QMdiSubWindow* subWindow );

    void selectedObjectsChanged();
    void customMenuRequested( const QPoint& pos );

    void slotWorkaroundForQwtDockWidgets();

private:
    void selectViewInProjectTreePreservingSubItemSelection( const Rim3dView* previousActiveReservoirView,
                                                            Rim3dView*       activatedView );

    // Pdm System :
public:
    void setPdmRoot( caf::PdmObject* pdmRoot );

private:
    std::unique_ptr<caf::PdmUiDragDropInterface> m_dragDropInterface;

    caf::PdmObject*         m_pdmRoot;
    caf::PdmUiPropertyView* m_pdmUiPropertyView;

    QSpinBox* m_scaleFactor;

    QActionGroup* m_dsActionGroup;
    QAction*      m_enableLightingAction;
    QAction*      m_drawStyleHideGridCellsAction;
    QAction*      m_toggleFaultsLabelAction;
    QAction*      m_drawStyleLinesAction;
    QAction*      m_drawStyleLinesSolidAction;
    QAction*      m_drawStyleFaultLinesSolidAction;
    QAction*      m_drawStyleSurfOnlyAction;
    QAction*      m_showWellCellsAction;

    QToolBar* m_holoLensToolBar;

    std::vector<QPointer<QDockWidget>> m_additionalProjectViews;
};