#pragma once

#include "TreeWxMain.h"

class M_GUI_TREE_EXPORT TreeWxMainRs : public TreeWxMain
{
    Q_OBJECT

public:

    TreeWxMainRs(QWidget* parent = 0);
    ~TreeWxMainRs();

    // 初始化
    virtual void fillTree();
    virtual bool isDataObjConsidered(ZxDataObject* pObj);
    virtual void initSubLevels(TreeItemMain* pItem);
    void initThirdLevel_WellRs(TreeItemMain*& pParent);
    TreeItemMain* initThirdLevel_WellRs_One(TreeItemMain*& pParent, ZxDataObject* p);

    // 作业保存后，刷新树节点
    bool updateRsNodes(ZxDataWell* pWell,
                        ZxRstJobUtils* pJobRst);
    // 获取能够Rst Fit 的Item
    // 允许不同井不同类型的成果参数进行对比
    // 成果对比也借用了该函数
    bool getRstFitItems(QList<TreeItemMain*>& listRst, int nMinCount);

protected:

    // 公共Action初始化
    virtual void initCommonActions();

    // 右键菜单功能_节点通用
    virtual void initContextMenuCommonNode(QMenu* pMenu);

    // 空节点通用
    virtual void initContextMenuCommonAll(QMenu* pMenu);

    // 右键节点级别的菜单
    virtual void initContextMenuWell(QMenu* pMenu);
    void initContextMenuRs(QMenu* pMenu);

    virtual void changeNodeAttrs(TreeItemMain*& p);
    virtual QString getTipsOf(TreeItemMain*& p);

Q_SIGNALS:

    // 对于作业的处理,获取能否加载或保存
    void sigCanLoadSaveRss(TreeItemMain* pItemSel, bool bForLoad, bool& b);

public slots:

    virtual void slotSaveNodeJobs();
    virtual void slotLoadNodeJobs();

    virtual void slotRstFit();//参数对比（敏感性分析）
    virtual void slotRstCp();//成果对比

protected:

    QAction* m_pActionRsLoad;
    QAction* m_pActionRsSave;

    QAction* m_pActionRstFit;
    QAction* m_pActionRstCp;
};