#ifndef XTREEWIDGETITEM_H
#define XTREEWIDGETITEM_H

#include "xassettype.h"
#include <QTreeWidget>
#include <QTreeWidgetItem>

class XAsset;

// Custom item class
class XTreeWidgetItem : public QTreeWidgetItem
{
public:
    XTreeWidgetItem(QTreeWidget *parent, bool group = false);
    XTreeWidgetItem(QTreeWidgetItem *parent, bool group = false);
    ~XTreeWidgetItem() = default;

    // Override the less-than operator to customize sorting.
    bool operator<(const QTreeWidgetItem &other) const override;
    XTreeWidgetItem &operator =(const XTreeWidgetItem &other);

    XAssetType AssetType() const;
    void SetAssetType(XAssetType aAssetType);

    bool GetIsGroup() const;
    void SetIsGroup(bool aIsGroup);
    XAsset *AssetPtr() const;
    void SetAssetPtr(XAsset *aAssetPtr);

private:
    bool isGroup;
    XAssetType mAssetType;
    XAsset* mAssetPtr;
};


#endif // XTREEWIDGETITEM_H