//##################################################################################################
//
//   Custom Visualization Core library
//   Copyright (C) 2011-2012 Ceetron AS
//    
//   This library 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. 
//    
//   This library 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. 
//
//##################################################################################################

#include "cafPdmUiTreeViewEditor.h"

#include "cafPdmObject.h"
#include "cafPdmField.h"
#include "cafPdmUiEditorHandle.h"
#include "cafUiTreeModelPdm.h"

#include <QWidget>
#include <QGridLayout>
#include <QTreeView>



namespace caf
{


//--------------------------------------------------------------------------------------------------
/// 
//--------------------------------------------------------------------------------------------------
PdmUiTreeViewEditor::PdmUiTreeViewEditor()
{
}


//--------------------------------------------------------------------------------------------------
/// 
//--------------------------------------------------------------------------------------------------
PdmUiTreeViewEditor::~PdmUiTreeViewEditor()
{
    if (m_proxyEditor)
    {
        delete m_proxyEditor;
        m_proxyEditor = NULL;
    }
}

//--------------------------------------------------------------------------------------------------
/// 
//--------------------------------------------------------------------------------------------------
QWidget* PdmUiTreeViewEditor::createWidget(QWidget* parent)
{
    m_mainWidget = new QWidget(parent);
    m_layout     = new QVBoxLayout();
    m_layout->setContentsMargins(0, 0, 0, 0);
    m_mainWidget->setLayout(m_layout);

    m_treeModelPdm = new caf::UiTreeModelPdm(m_mainWidget);
    m_treeView = new QTreeView(m_mainWidget);
    m_treeView->setHeaderHidden(true);
    m_treeView->setModel(m_treeModelPdm);
    
    m_layout->addWidget(m_treeView);

    m_proxyEditor = new PdmUiProxyEditorHandle(this);
    
    return m_mainWidget;
}

//--------------------------------------------------------------------------------------------------
/// 
//--------------------------------------------------------------------------------------------------
void PdmUiTreeViewEditor::configureAndUpdateUi(const QString& uiConfigName)
{
    if (this->pdmObject())
    {
        this->pdmObject()->objectEditorAttribute(uiConfigName, &m_editorAttributes);
    }

    if (!m_treeModelPdm->treeItemRoot() || m_treeModelPdm->treeItemRoot()->dataObject() != this->pdmObject())
    {
        caf::PdmUiTreeItem* treeItemRoot = caf::UiTreeItemBuilderPdm::buildViewItems(NULL, -1, this->pdmObject());
        m_treeModelPdm->setTreeItemRoot(treeItemRoot);
    }

    // Update tree model, and set the proxy editor for all Pdm child objects
    if (m_treeModelPdm->treeItemRoot() && m_treeModelPdm->treeItemRoot()->dataObject())
    {
        m_treeModelPdm->updateUiSubTree(m_treeModelPdm->treeItemRoot()->dataObject());

        std::vector<PdmObject*> children;
        childObjects(m_treeModelPdm->treeItemRoot()->dataObject(), children);

        size_t cIdx;
        for (cIdx = 0; cIdx < children.size(); ++cIdx)
        {
            addEditorRecursively(children[cIdx], m_proxyEditor);
        }
    }


    PdmUiTreeViewEditorAttribute leab;


    // Notify all connected views that the complete model is updated
    m_treeModelPdm->notifyModelChanged();
}

//--------------------------------------------------------------------------------------------------
/// 
//--------------------------------------------------------------------------------------------------
void PdmUiTreeViewEditor::addEditorRecursively(PdmObject* pdmObject, PdmUiEditorHandle* editorHandle)
{
    if (!pdmObject) return;

    std::vector<PdmObject*> children;
    childObjects(pdmObject, children);

    size_t cIdx;
    for (cIdx = 0; cIdx < children.size(); ++cIdx)
    {
        addEditorRecursively(children[cIdx], editorHandle);
    }

    pdmObject->addFieldEditor(editorHandle);
}

//--------------------------------------------------------------------------------------------------
/// 
//--------------------------------------------------------------------------------------------------
void PdmUiTreeViewEditor::childObjects(PdmObject* pdmObject, std::vector<PdmObject*>& children)
{
    if (!pdmObject) return;

    std::vector<PdmFieldHandle*> fields;
    pdmObject->fields(fields);

    size_t fIdx;
    for (fIdx = 0; fIdx < fields.size(); ++fIdx)
    {
        if (fields[fIdx]) fields[fIdx]->childObjects(&children);
    }
}

//--------------------------------------------------------------------------------------------------
/// 
//--------------------------------------------------------------------------------------------------
QTreeView* PdmUiTreeViewEditor::treeView()
{
    return m_treeView;
}



} // end namespace caf