// Created on: 2017-06-16
// Created by: Natalia ERMOLAEVA
// Copyright (c) 2017 OPEN CASCADE SAS
//
// This file is part of Open CASCADE Technology software library.
//
// This library is free software; you can redistribute it and/or modify it under
// the terms of the GNU Lesser General Public License version 2.1 as published
// by the Free Software Foundation, with special exception defined in the file
// OCCT_LGPL_EXCEPTION.txt. Consult the file LICENSE_LGPL_21.txt included in OCCT
// distribution for complete text of the license and disclaimer of any warranty.
//
// Alternatively, this file may be used under the terms of Open CASCADE
// commercial license or contractual agreement.

#include <inspector/DFBrowserPane_TDataStdTreeNodeModel.hxx>
#include <inspector/DFBrowserPane_TDataStdTreeNodeItem.hxx>

#include <TDataStd_TreeNode.hxx>

#include <Standard_WarningsDisable.hxx>
#include <QAbstractItemModel>
#include <Standard_WarningsRestore.hxx>

// =======================================================================
// function : Constructor
// purpose :
// =======================================================================
DFBrowserPane_TDataStdTreeNodeModel::DFBrowserPane_TDataStdTreeNodeModel(QObject* theParent)
    : TreeModel_ModelBase(theParent)
{
}

// =======================================================================
// function : InitColumns
// purpose :
// =======================================================================
void DFBrowserPane_TDataStdTreeNodeModel::InitColumns()
{
  setHeaderItem(0, TreeModel_HeaderSection("Name"));
}

// =======================================================================
// function : createRootItem
// purpose :
// =======================================================================
TreeModel_ItemBasePtr DFBrowserPane_TDataStdTreeNodeModel::createRootItem(const int theColumnId)
{
  return DFBrowserPane_TDataStdTreeNodeItem::CreateItem(TreeModel_ItemBasePtr(), 0, theColumnId);
}

// =======================================================================
// function : SetAttribute
// purpose :
// =======================================================================
void DFBrowserPane_TDataStdTreeNodeModel::SetAttribute(const Handle(TDF_Attribute)& theAttribute)
{
  DFBrowserPane_TDataStdTreeNodeItemPtr aRootItem =
    itemDynamicCast<DFBrowserPane_TDataStdTreeNodeItem>(RootItem(0));
  Reset();
  aRootItem->SetAttribute(theAttribute);
  EmitLayoutChanged();
}

// =======================================================================
// function : FindIndex
// purpose :
// =======================================================================
QModelIndex DFBrowserPane_TDataStdTreeNodeModel::FindIndex(
  const Handle(TDF_Attribute)& theAttribute,
  const QModelIndex            theParentIndex)
{
  QModelIndex aParentIndex = theParentIndex;

  if (!aParentIndex.isValid())
    aParentIndex = index(0, 0);

  DFBrowserPane_TDataStdTreeNodeItemPtr aParentItem =
    itemDynamicCast<DFBrowserPane_TDataStdTreeNodeItem>(
      TreeModel_ModelBase::GetItemByIndex(aParentIndex));

  if (aParentItem->GetAttribute() == theAttribute)
    return aParentIndex;

  for (int aChildId = 0, aCount = aParentItem->rowCount(); aChildId < aCount; aChildId++)
  {
    QModelIndex                           anIndex    = index(aChildId, 0, aParentIndex);
    TreeModel_ItemBasePtr                 anItemBase = TreeModel_ModelBase::GetItemByIndex(anIndex);
    DFBrowserPane_TDataStdTreeNodeItemPtr anItem =
      itemDynamicCast<DFBrowserPane_TDataStdTreeNodeItem>(anItemBase);

    if (anItem->GetAttribute() == theAttribute)
      return anIndex;

    QModelIndex aSubIndex = FindIndex(theAttribute, anIndex);
    if (aSubIndex.isValid())
      return aSubIndex;
  }
  return QModelIndex();
}