EntityTree.hh

Go to the documentation of this file.

/*
 * Copyright (C) 2019 Open Source Robotics Foundation
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *
*/
 
#ifndef GZ_SIM_GUI_ENTITYTREE_HH_
#define GZ_SIM_GUI_ENTITYTREE_HH_
 
#include <map>
#include <memory>
#include <vector>
 
#include <gz/sim/Entity.hh>
#include <gz/sim/gui/GuiSystem.hh>
 
namespace gz
{
namespace sim
{
  class EntityTreePrivate;
 
  class TreeModel : public QStandardItemModel
  {
    Q_OBJECT
 
    public: explicit TreeModel();
 
    public: ~TreeModel() override = default;
 
    // Documentation inherited
    public: QHash<int, QByteArray> roleNames() const override;
 
    public slots: void AddEntity(Entity _entity,
        const QString &_entityName,
        Entity _parentEntity = kNullEntity,
        const QString &_type = QString());
 
    public slots: void RemoveEntity(Entity _entity);
 
    public: Q_INVOKABLE QString EntityType(const QModelIndex &_index) const;
 
    public: Q_INVOKABLE QString ScopedName(const QModelIndex &_index) const;
 
    public: Q_INVOKABLE Entity EntityId(const QModelIndex &_index) const;
 
    private: std::map<Entity, QStandardItem *> entityItems;
 
    struct EntityInfo
    {
      // cppcheck-suppress unmatchedSuppression
      // cppcheck-suppress unusedStructMember
      Entity entity;
 
      QString name;
 
      // cppcheck-suppress unmatchedSuppression
      // cppcheck-suppress unusedStructMember
      Entity parentEntity;
 
      QString type;
    };
 
    private: std::vector<EntityInfo> pendingEntities;
  };
 
  class EntityTree : public gz::sim::GuiSystem
  {
    Q_OBJECT
 
    public: EntityTree();
 
    public: ~EntityTree() override;
 
    // Documentation inherited
    public: void LoadConfig(const tinyxml2::XMLElement *_pluginElem) override;
 
    // Documentation inherited
    public: void Update(const UpdateInfo &, EntityComponentManager &) override;
 
    public: Q_INVOKABLE void OnEntitySelectedFromQml(Entity _entity);
 
    public: Q_INVOKABLE void DeselectAllEntities();
 
    public: Q_INVOKABLE void OnInsertEntity(const QString &_type);
 
    public: Q_INVOKABLE void OnLoadMesh(const QString &_mesh);
 
    // Documentation inherited
    protected: bool eventFilter(QObject *_obj, QEvent *_event) override;
 
    private: std::unique_ptr<EntityTreePrivate> dataPtr;
  };
}
}
 
#endif