PartSet_TreeNodes.h

// Copyright (C) 2014-2026  CEA, EDF
//
// This library is free software; you can redistribute it and/or
// modify it under the terms of the GNU Lesser General Public
// License as published by the Free Software Foundation; either
// version 2.1 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
// Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public
// License along with this library; if not, write to the Free Software
// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
//
// See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
//
 
#ifndef PartSet_TreeNodes_H
#define PartSet_TreeNodes_H
 
#include "PartSet.h"
 
#include <ModuleBase_ITreeNode.h>
#include <ModelAPI_Feature.h>
 
 
class PartSet_TreeNode : public ModuleBase_ITreeNode
{
public:
  PartSet_TreeNode(ModuleBase_ITreeNode* theParent = 0) : ModuleBase_ITreeNode(theParent) {}
 
  virtual QVariant data(int theColumn, int theRole) const;
 
  // Returns color of the Item when it is active
  virtual QColor activeItemColor() const;
};
 
class PartSet_ObjectNode : public PartSet_TreeNode
{
public:
  PartSet_ObjectNode(const ObjectPtr& theObj, ModuleBase_ITreeNode* theParent = 0)
    : PartSet_TreeNode(theParent), myObject(theObj) {}
 
  static std::string typeId()
  {
    static std::string myType = "Object";
    return myType;
  }
 
  virtual std::string type() const { return typeId(); }
 
  virtual QVariant data(int theColumn, int theRole) const;
 
  virtual Qt::ItemFlags flags(int theColumn) const;
 
  virtual ObjectPtr object() const { return myObject; }
 
  void setObject(ObjectPtr theObj) { myObject = theObj; }
 
  virtual VisibilityState visibilityState() const;
 
  virtual void update();
 
  virtual QTreeNodesList objectCreated(const QObjectPtrList& theObjects);
 
  virtual QTreeNodesList objectsDeleted(const DocumentPtr& theDoc, const QString& theGroup);
 
  virtual int numberOfSubs() const;
 
  virtual ObjectPtr subObject(int theId) const;
 
  // Returns color of the Item when it is active
  virtual QColor activeItemColor() const;
 
protected:
  ObjectPtr myObject;
};
 
class PartSet_FolderNode : public PartSet_TreeNode
{
public:
  enum FolderType {
    ParametersFolder,
    ConstructionFolder,
    PartsFolder,
    ResultsFolder,
    FieldsFolder,
    GroupsFolder
  };
 
  PartSet_FolderNode(ModuleBase_ITreeNode* theParent, FolderType theType);
 
  static std::string typeId()
  {
    static std::string myType = "Folder";
    return myType;
  }
 
  virtual std::string type() const { return typeId(); }
 
  virtual QVariant data(int theColumn, int theRole) const;
 
  virtual Qt::ItemFlags flags(int theColumn) const;
 
  virtual void update();
 
  virtual QTreeNodesList objectCreated(const QObjectPtrList& theObjects);
 
  virtual QTreeNodesList objectsDeleted(const DocumentPtr& theDoc, const QString& theGroup);
 
  QString name() const;
 
  virtual ModuleBase_ITreeNode* findParent(const DocumentPtr& theDoc, QString theGroup)
  {
    if ((theDoc == document()) && (theGroup.toStdString() == groupName()))
      return this;
    return 0;
  }
 
private:
  std::string groupName() const;
 
  ModuleBase_ITreeNode* createNode(const ObjectPtr& theObj);
 
  FolderType myType;
};
 
 
class PartSet_FeatureFolderNode : public PartSet_TreeNode
{
public:
  PartSet_FeatureFolderNode(ModuleBase_ITreeNode* theParent = 0) : PartSet_TreeNode(theParent) {}
 
  virtual QTreeNodesList objectCreated(const QObjectPtrList& theObjects);
 
  virtual QTreeNodesList objectsDeleted(const DocumentPtr& theDoc, const QString& theGroup);
 
  virtual ModuleBase_ITreeNode* findParent(const DocumentPtr& theDoc, QString theGroup);
 
protected:
  virtual ModuleBase_ITreeNode* createNode(const ObjectPtr& theObj) = 0;
 
  virtual int numberOfFolders() const { return 0; }
};
 
 
 
class PartSet_RootNode : public PartSet_FeatureFolderNode
{
public:
  PartSet_RootNode();
 
  static std::string typeId()
  {
    static std::string myType = "PartSetRoot";
    return myType;
  }
 
  virtual std::string type() const { return typeId(); }
 
  virtual void update();
 
  virtual ModuleBase_IWorkshop* workshop() const { return myWorkshop; }
 
  virtual DocumentPtr document() const;
 
  void setWorkshop(ModuleBase_IWorkshop* theWork) { myWorkshop = theWork; }
 
protected:
  virtual ModuleBase_ITreeNode* createNode(const ObjectPtr& theObj);
 
  virtual int numberOfFolders() const { return 3; }
 
private:
  PartSet_FolderNode* myParamsFolder;
  PartSet_FolderNode* myConstrFolder;
  PartSet_FolderNode* myPartsFolder;
 
  ModuleBase_IWorkshop* myWorkshop;
};
 
 
class PartSet_PartRootNode : public PartSet_FeatureFolderNode
{
public:
  PartSet_PartRootNode(const ObjectPtr& theObj, ModuleBase_ITreeNode* theParent);
 
  static std::string typeId()
  {
    static std::string myType = "PartRoot";
    return myType;
  }
 
  virtual std::string type() const { return typeId(); }
 
  virtual ObjectPtr object() const { return myObject; }
 
  virtual DocumentPtr document() const;
 
  virtual void update();
 
  virtual QVariant data(int theColumn, int theRole) const;
 
  virtual Qt::ItemFlags flags(int theColumn) const;
 
  virtual QTreeNodesList objectCreated(const QObjectPtrList& theObjects);
 
  virtual QTreeNodesList objectsDeleted(const DocumentPtr& theDoc, const QString& theGroup);
 
protected:
  virtual ModuleBase_ITreeNode* createNode(const ObjectPtr& theObj);
 
  virtual int numberOfFolders() const;
 
  virtual void deleteChildren();
 
private:
  PartSet_FolderNode* myParamsFolder;
  PartSet_FolderNode* myConstrFolder;
  PartSet_FolderNode* myResultsFolder;
  PartSet_FolderNode* myFieldsFolder;
  PartSet_FolderNode* myGroupsFolder;
 
  ObjectPtr myObject;
};
 
 
class PartSet_ObjectFolderNode : public PartSet_ObjectNode
{
public:
  PartSet_ObjectFolderNode(const ObjectPtr& theObj, ModuleBase_ITreeNode* theParent)
    : PartSet_ObjectNode(theObj, theParent) {}
 
  static std::string typeId()
  {
    static std::string myType = "ObjectFolder";
    return myType;
  }
 
  virtual std::string type() const { return typeId(); }
 
  virtual void update();
 
  virtual QTreeNodesList objectCreated(const QObjectPtrList& theObjects);
 
  virtual QTreeNodesList objectsDeleted(const DocumentPtr& theDoc, const QString& theGroup);
 
  virtual QVariant data(int theColumn, int theRole) const;
};
 
 
 
class PartSet_StepNode : public PartSet_ObjectNode
{
public:
  PartSet_StepNode(const ObjectPtr& theObj, ModuleBase_ITreeNode* theParent) :
    PartSet_ObjectNode(theObj, theParent) {}
 
  static std::string typeId()
  {
    static std::string myType = "FieldStep";
    return myType;
  }
 
  virtual std::string type() const { return typeId(); }
 
  virtual QVariant data(int theColumn, int theRole) const;
 
  virtual VisibilityState visibilityState() const;
};
 
#endif