HierarchyObject.h

#pragma once


#include <qpoint.h>
#include <qstring.h>
#include <qabstractitemmodel.h>

#include <vector>
#include <set>
#include <algorithm>
#include <glm/glm.hpp>
#include <glm/gtc/matrix_transform.hpp>
#include <glm/gtc/type_ptr.hpp>

#include <functional>
#include <queue>

#include "nesteddefs.h"


class HierarchyObject : public QAbstractItemModel {

private:
    std::vector<HierarchyObject*> children;
    std::vector<Component*> components;
    HierarchyObject* parentObj = NULL;

public:
    glm::mat4 transform;
    QString name;
    bool enabled;

    int childrenCount() { return children.size(); }
    HierarchyObject* getChildren(int i) { return (children.size() > i && i >= 0) ? children[i] : NULL; }
    HierarchyObject* getChildren(const QString& name);
    int findChild(HierarchyObject* child);
    int componentsCount() { return components.size(); }
    HierarchyObject* getParent() { return parentObj; }

    // ctor
    HierarchyObject(const QString& name, HierarchyObject* parent = NULL);
    ~HierarchyObject();

    glm::mat4 localToWorld();
    glm::mat4 worldToLocal() {
        return glm::inverse(localToWorld());
    }
    
    
    HierarchyObject* popChild(int index);
    void insertChild(int index, HierarchyObject* child);
    void moveChild(int oldIndex, int newIndex);
    
    
    template <class T>
    T* getComponent(int i) {
        if (i < 0 || i >= components.size()) return NULL;
        return dynamic_cast<T*>(components[i]);
    }

    template <class T>
    T* getComponent() {
        for (auto component : components) {
            if (dynamic_cast<T*>(component)) {
                return dynamic_cast<T*>(component);
            }
        }
        return NULL;
    }

    Component * addComponent(Component* component);


    void updateRecursively();

    void callRecursively(const std::function<void(HierarchyObject*)>& func, bool requireEnabled = false);


    // -------------实现QAbstractItemModel的必选接口-----------------  
    // 获取给定元素的第row个子元素 
    QModelIndex index(int row, int column, const QModelIndex &parent = QModelIndex()) const;
    // 获取给定元素的父元素 
    QModelIndex parent(const QModelIndex &index) const;
    // 获取某元素的子节点数 
    int rowCount(const QModelIndex &parent = QModelIndex()) const;
    // 2
    int columnCount(const QModelIndex &parent = QModelIndex()) const;
    // 取出数据 
    QVariant data(const QModelIndex &index, int role = Qt::DisplayRole) const;

    // ----------------------重写可选函数----------------------------
    // 接收编辑name事件  
    bool setData(const QModelIndex & index, const QVariant & value, int role);
    // 返回指定节点的各种显示属性  
    Qt::ItemFlags flags(const QModelIndex &index) const;
    // 返回标题  
    QVariant headerData(int section, Qt::Orientation orientation, int role = Qt::DisplayRole) const;

};