/************************************************************************
* file name         : blocks_tree_widget.h
* ----------------- : 
* creation time     : 2016/06/26
* copyright         : (c) 2016 Victor Zarubkin
* author            : Victor Zarubkin
* email             : v.s.zarubkin@gmail.com
* ----------------- : 
* description       : The file contains declaration of EasyTreeWidget and it's auxiliary classes
*                   : for displyaing EasyProfiler blocks tree.
* ----------------- : 
* change log        : * 2016/06/26 Victor Zarubkin: moved sources from tree_view.h
*                   :       and renamed classes from My* to Prof*.
*                   :
*                   : * 2016/06/27 Victor Zarubkin: Added possibility to colorize rows
*                   :       with profiler blocks' colors.
*                   :
*                   : * 2016/06/29 Victor Zarubkin: Added clearSilent() method.
*                   :
*                   : * 2016/08/18 Victor Zarubkin: Added loading blocks hierarchy in separate thread;
*                   :       Moved sources of TreeWidgetItem into tree_widget_item.h/.cpp
* ----------------- : 
* license           : TODO: add license text
************************************************************************/

#ifndef EASY__TREE_WIDGET__H_
#define EASY__TREE_WIDGET__H_

#include <QTreeWidget>
#include <QAction>
#include <QTimer>
#include "tree_widget_loader.h"
#include "profiler/reader.h"

//////////////////////////////////////////////////////////////////////////

#define DECLARE_QACTION(ClassName, DataType) \
class ClassName : public QAction { \
    Q_OBJECT \
private: \
    DataType m_item; \
public: \
    ClassName(const char* _label, DataType _item) : QAction(_label, nullptr), m_item(_item) { \
        connect(this, &QAction::triggered, this, &ClassName::onToggle); } \
    ClassName(const QString& _label, DataType _item) : QAction(_label, nullptr), m_item(_item) { \
        connect(this, &QAction::triggered, this, &ClassName::onToggle); } \
    virtual ~ClassName() {} \
private: \
    void onToggle(bool) { emit clicked(m_item); }

DECLARE_QACTION(EasyItemAction, unsigned int) signals: void clicked(unsigned int _item); };
DECLARE_QACTION(EasyHideShowColumnAction, int) signals: void clicked(int _item); };

#undef DECLARE_QACTION

//////////////////////////////////////////////////////////////////////////

class EasyTreeWidget : public QTreeWidget
{
    Q_OBJECT

    typedef QTreeWidget    Parent;
    typedef EasyTreeWidget   This;

protected:

    EasyTreeWidgetLoader  m_hierarchyBuilder;
    Items                            m_items;
    RootsMap                         m_roots;
    ::profiler_gui::TreeBlocks m_inputBlocks;
    QTimer                       m_fillTimer;
    ::profiler::timestamp_t      m_beginTime;
    class QProgressDialog*        m_progress;
    bool                        m_bColorRows;
    bool                           m_bLocked;
    bool             m_bSilentExpandCollapse;

public:

    explicit EasyTreeWidget(QWidget* _parent = nullptr);
    virtual ~EasyTreeWidget();

    void clearSilent(bool _global = false);

public slots:

    void setTree(const unsigned int _blocksNumber, const ::profiler::thread_blocks_tree_t& _blocksTree);

    void setTreeBlocks(const ::profiler_gui::TreeBlocks& _blocks, ::profiler::timestamp_t _session_begin_time, ::profiler::timestamp_t _left, ::profiler::timestamp_t _right, bool _strict);

protected:

    void contextMenuEvent(QContextMenuEvent* _event) override;
    void resizeEvent(QResizeEvent* _event) override;
    void moveEvent(QMoveEvent* _event) override;

private slots:

    void onJumpToItemClicked(unsigned int _block_index);

    void onCollapseAllClicked(bool);

    void onExpandAllClicked(bool);

    void onCollapseAllChildrenClicked(bool);

    void onExpandAllChildrenClicked(bool);

    void onItemExpand(QTreeWidgetItem* _item);
    void onItemCollapse(QTreeWidgetItem* _item);

    void onColorizeRowsTriggered(bool _colorize);

    void onSelectedThreadChange(::profiler::thread_id_t _id);

    void onSelectedBlockChange(unsigned int _block_index);

    void resizeColumnsToContents();

    void onHideShowColumn(int _column);

    void onFillTimerTimeout();

protected:

    void loadSettings();
	void saveSettings();
    void alignProgressBar();

}; // END of class EasyTreeWidget.

//////////////////////////////////////////////////////////////////////////

#endif // EASY__TREE_WIDGET__H_