occt/tools/TreeModel/TreeModel_ItemBase.cxx

// 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/TreeModel_ColumnType.hxx>
#include <inspector/TreeModel_ItemBase.hxx>
#include <inspector/TreeModel_ItemProperties.hxx>
#include <inspector/TreeModel_ItemRole.hxx>
#include <inspector/TreeModel_ItemStream.hxx>

#include <Standard_Dump.hxx>
#include <Standard_WarningsDisable.hxx>
#include <QStringList>
#include <Standard_WarningsRestore.hxx>

// =======================================================================
// function : Constructor
// purpose :
// =======================================================================
TreeModel_ItemBase::TreeModel_ItemBase(TreeModel_ItemBasePtr theParent,
                                       const int             theRow,
                                       const int             theColumn)
    : m_iStreamChildren(0),
      m_bInitialized(false)
{
  m_pParent = theParent;
  m_iRow    = theRow;
  m_iColumn = theColumn;
}

// =======================================================================
// function :  Reset
// purpose :
// =======================================================================
void TreeModel_ItemBase::Reset()
{
  for (PositionToItemHash::const_iterator aChildrenIt = m_ChildItems.begin();
       aChildrenIt != m_ChildItems.end();
       aChildrenIt++)
  {
    TreeModel_ItemBasePtr anItem = aChildrenIt.value();
    if (anItem)
      anItem->Reset();
  }
  m_bInitialized = false;
  if (!myProperties.IsNull())
  {
    myProperties->Reset();
  }
  myCachedValues.clear();
  myStream.str("");
}

// =======================================================================
// function :  Reset
// purpose :
// =======================================================================
void TreeModel_ItemBase::Reset(int theRole)
{
  if (!myCachedValues.contains(theRole))
    return;

  myCachedValues.remove(theRole);
}

// =======================================================================
// function :  child
// purpose :
// =======================================================================
TreeModel_ItemBasePtr TreeModel_ItemBase::Child(int theRow, int theColumn, const bool isToCreate)
{
  QPair<int, int> aPos = qMakePair(theRow, theColumn);

  if (m_ChildItems.contains(aPos))
    return m_ChildItems[aPos];

  TreeModel_ItemBasePtr anItem;
  if (isToCreate)
  {
    if (theRow < m_iStreamChildren)
      anItem = TreeModel_ItemStream::CreateItem(currentItem(), theRow, theColumn);
    else
      anItem = createChild(theRow - m_iStreamChildren, theColumn);

    if (anItem)
      m_ChildItems[aPos] = anItem;
  }
  return anItem;
}

// =======================================================================
// function :  Presentations
// purpose :
// =======================================================================
void TreeModel_ItemBase::Presentations(
  NCollection_List<Handle(Standard_Transient)>& thePresentations)
{
  if (Column() != 0)
    return;

  const Handle(TreeModel_ItemProperties)& anItemProperties = Properties();
  if (anItemProperties)
  {
    anItemProperties->Presentations(thePresentations);
  }
}

// =======================================================================
// function :  currentItem
// purpose :
// =======================================================================
const TreeModel_ItemBasePtr TreeModel_ItemBase::currentItem()
{
  return TreeModel_ItemBasePtr(this);
}

// =======================================================================
// function :  cachedValue
// purpose :
// =======================================================================
QVariant TreeModel_ItemBase::cachedValue(const int theItemRole) const
{
  if (myCachedValues.contains(theItemRole))
    return myCachedValues[theItemRole];

  QVariant aValueToCache;
  if (theItemRole == TreeModel_ItemRole_RowCountRole)
    aValueToCache = initRowCount() + const_cast<TreeModel_ItemBase*>(this)->initStreamRowCount();
  else
    aValueToCache = initValue(theItemRole);

  myCachedValues.insert(theItemRole, aValueToCache);
  return myCachedValues.contains(theItemRole) ? myCachedValues[theItemRole] : QVariant();
}

// =======================================================================
// function : Init
// purpose :
// =======================================================================
void TreeModel_ItemBase::Init()
{
  m_bInitialized = true;

  initStream(myStream);
  initStreamRowCount();
}

// =======================================================================
// function : Object
// purpose :
// =======================================================================
const Handle(Standard_Transient)& TreeModel_ItemBase::Object() const
{
  static Handle(Standard_Transient) aNullObject;
  return aNullObject;
}

// =======================================================================
// function : initStreamRowCount
// purpose :
// =======================================================================
int TreeModel_ItemBase::initStreamRowCount()
{
  int aStreamChildrenCount = 0;
  if (Column() == 0)
  {
    Standard_SStream aStream;
    initStream(aStream);
    if (!Standard_Dump::Text(aStream).IsEmpty())
    {
      if (!myProperties)
      {
        myProperties = new TreeModel_ItemProperties();
        myProperties->SetItem(currentItem());
      }
      myProperties->Init();
      aStreamChildrenCount = myProperties->Children().Extent();
    }
  }
  m_iStreamChildren = aStreamChildrenCount;
  return m_iStreamChildren;
}

// =======================================================================
// function : initValue
// purpose :
// =======================================================================
QVariant TreeModel_ItemBase::initValue(const int theItemRole) const
{
  if (theItemRole != Qt::DisplayRole && theItemRole != Qt::ToolTipRole)
    return QVariant();

  switch (Column())
  {
    case TreeModel_ColumnType_Row: {
      return Row();
    }
  }

  return QVariant();
}