#include "CFDMainTreeWidget.h"

#include "GUIWidget/TreeWidget.h"
#include "GUIFrame/PropertyWidget.h"
#include "CFDTreeWidget.h"

namespace GUI{

    CFDMainTreeWidget::CFDMainTreeWidget(MainWindow *parent, PropertyWidget* prowidget) : MainTreeWidget(parent)
    {
        this->m_PropertyWidget = prowidget;
//        this->setWidget(_treeWidget);
        init();
    }

    void CFDMainTreeWidget::init()
    {
        QTabWidget* tabWidget = new QTabWidget;
        this->setWidget(tabWidget);

//        tabWidget->addTab(_treeWidget,tr("Solution Analysis"));

        addTreePreprocessing(tabWidget);
        addTreeSolutionAnalysis(tabWidget);
        addTreePostprocessing(tabWidget);
    }

    void CFDMainTreeWidget::addTreePreprocessing(QTabWidget *tabWidget)
    {

        _treePreprocessing = new TreeWidget(this);
        tabWidget->addTab(_treePreprocessing,tr("Preprocessing"));


        _treePreprocessing->updateTree();
    }

    void CFDMainTreeWidget::addTreeSolutionAnalysis(QTabWidget *tabWidget)
    {
        _treeSolutionAnalysis = new CFDTreeWidget(this,m_PropertyWidget);
        tabWidget->addTab(_treeSolutionAnalysis,tr("SolutionAnalysis"));

        _treeSolutionAnalysis->updateTree("SolutionAnalysis");
    }

    void CFDMainTreeWidget::addTreePostprocessing(QTabWidget *tabWidget)
    {
        _treePostprocessing = new CFDTreeWidget(this,m_PropertyWidget);
        tabWidget->addTab(_treePostprocessing,tr("Postprocessing"));

        _treePostprocessing->updateTree("Postprocessing");
    }

}