#include "OperGraphPickPreview.h"

// VTK
#include <vtkActor.h>
#include <vtkDataSet.h>
#include <vtkUnstructuredGrid.h>

// App
#include "FITK_Kernel/FITKAppFramework/FITKAppFramework.h"
#include "FITK_Kernel/FITKAppFramework/FITKGlobalData.h"

// Global data
#include "FITK_Kernel/FITKCore/FITKDataRepo.h"

// Data
#include "FITK_Interface/FITKInterfaceModel/FITKUnstructuredMesh.h"

// Graph
#include "FITK_Interface/FITKVTKAlgorithm/FITKGraphActor.h"
#include "FITK_Interface/FITKVTKAlgorithm/FITKGraphActor2D.h"
#include "FITK_Component/FITKFluidVTKGraphAdaptor/FITKFluidVTKGraphObject3D.h"
#include "FITK_Component/FITKFluidVTKGraphAdaptor/FITKFluidVTKGraphObjectModel.h"
#include "FITK_Component/FITKFluidVTKGraphAdaptor/FITKFluidVTKGraphObjectPickPreview.h"

// Render
#include "FITK_Component/FITKRenderWindowVTK/FITKGraph3DWindowVTK.h"
#include "FITK_Component/FITKRenderWindowVTK/FITKGraphRender.h"
#include "FITK_Component/FITKRenderWindowVTK/FITKGraphObjectVTK.h"

// Pick data provider
#include "GUIWidget/PickedDataProvider.h"

// GUI
#include "GUIFrame/MainWindow.h"
#include "GUIFrame/RenderWidget.h"
#include "FITK_Component/FITKWidget/FITKMdiArea.h"
#include "GUIWidget/GUIPickInfo.h"

namespace GUIOper
{
    OperGraphPickPreview::OperGraphPickPreview()
    {
        // 实例化预选高亮可视化对象。
        m_graphObjPreview = new Exchange::FITKFluidVTKGraphObjectPickPreview;

        // 初始化拾取数据管理器。
        m_pickedProvider = GraphData::PickedDataProvider::getInstance();

        // 新版本：根据当前拾取对象类型与计算方法预选。
        //@{
        // 初始化可视化对象三维数据。
        m_graphObjPreview->setPickedData(m_pickedProvider->getPickedPreviewDataSet());
        //@}
    }

    OperGraphPickPreview::~OperGraphPickPreview()
    {
        // 析构。
        if (m_graphObjPreview)
        {
            m_graphObjPreview->removeFromGraphWidget();

            delete m_graphObjPreview;
            m_graphObjPreview = nullptr;
        }
    }

    void OperGraphPickPreview::picked(Comp::FITKGraph3DWindowVTK* graphWindow, vtkActor2D* actor)
    {
        if (!graphWindow || !actor)
        {
            return;
        }

        bool needUpdate = m_pickedProvider->addPreviewPicked(actor);
        if (!needUpdate)
        {
            return;
        }

        // 添加到渲染窗口。
        m_graphObjPreview->setVisible(true);
        addGraphObjectToWidget(m_graphObjPreview, graphWindow);

        // 刷新窗口。
        graphWindow->reRender();
    }

    void OperGraphPickPreview::picked(Comp::FITKGraph3DWindowVTK* graphWindow, vtkActor* actor, int index, double* pickedWorldPos)
    {
        if (!graphWindow || !actor || !m_graphObjPreview)
        {
            return;
        }

        auto pickType = GUI::GUIPickInfo::GetPickInfo()._pickObjType;

        // 新版本：根据当前拾取对象类型与计算方法预选。
        //@{
        bool needUpdate = m_pickedProvider->addPreviewPicked(actor, index, pickedWorldPos);
        if (!needUpdate)
        {
            return;
        }

        // 设置演员样式。
        //@{
        switch (pickType)
        {
        case GUI::GUIPickInfo::POBJVert:
            m_graphObjPreview->setPickedType(1);
            break;
        case GUI::GUIPickInfo::POBJEdge:
            m_graphObjPreview->setPickedType(2);
            break;
        case GUI::GUIPickInfo::POBJFace:
        case GUI::GUIPickInfo::POBJSolid:
            m_graphObjPreview->setPickedType(3);
            break;
        default:
            m_graphObjPreview->setVisible(false);
            return;
        }
        //@}

        // 添加到渲染窗口。
        m_graphObjPreview->setVisible(true);
        addGraphObjectToWidget(m_graphObjPreview, graphWindow);

        // 刷新窗口。
        graphWindow->reRender();
    }

    void OperGraphPickPreview::clear(Comp::FITKGraph3DWindowVTK* graphWindow)
    {
        // 清空预选数据。
        GraphData::PickedDataProvider::getInstance()->clearPickedDataPreview();

        // 取消预选高亮并隐藏可视化对象。
        if (m_graphObjPreview)
        {
            m_graphObjPreview->disHighlight();
            m_graphObjPreview->setVisible(false);

            if (!graphWindow)
            {
                graphWindow = dynamic_cast<Comp::FITKGraph3DWindowVTK*>(m_graphObjPreview->getGraphWidget());
            }
        }

        // 刷新窗口。
        if (graphWindow)
        {
            graphWindow->reRender();
        }
    }

    void OperGraphPickPreview::addGraphObjectToWidget(Exchange::FITKFluidVTKGraphObject3D* obj, Comp::FITKGraph3DWindowVTK* graphWidget)
    {
        if (!obj || !graphWidget)
        {
            return;
        }

        // 添加可视化对象。
        //@{
        obj->removeFromGraphWidget();
        graphWidget->addObject(obj->getRenderLayer(), obj, true);
        //}

        // 刷新窗口。
        //@{
        graphWidget->reRender();
        //@}
    }
}  // namespace GUIOper