#ifndef NMWXRESULTPARAMETERS_H
#define NMWXRESULTPARAMETERS_H
#include "nmSubWxs_global.h"
#include <QWidget>
#include <QTreeWidget>
#include <QTreeWidgetItem>
#include "nmDataAttribute.h"
#include <QVBoxLayout>
#include <QMouseEvent>

// 扩展QTreeWidget类，实现点击某一行时，触发第二列的编辑框
class CustomTreeWidget : public QTreeWidget
{
	Q_OBJECT
public:
	explicit CustomTreeWidget(QWidget *parent = 0);
protected:
	void mousePressEvent(QMouseEvent *event) override;
	void mouseMoveEvent(QMouseEvent *event) override;
	void mouseReleaseEvent(QMouseEvent *event) override;
	void paintEvent(QPaintEvent *event) override;
private:
	bool m_isResizing;
	int m_resizeColumn;
	int m_resizeStartX;
	int m_resizeStartWidth;
};

#include <QStyledItemDelegate>
#include <QPixmap>
// 自定义委托类，处理QTreeWidget类的Item交互
class CustomTreeDelegate : public QStyledItemDelegate
{
	Q_OBJECT
public:
	explicit CustomTreeDelegate(QObject *parent = nullptr);
	void paint(QPainter *painter, const QStyleOptionViewItem &option,
		const QModelIndex &index) const override;
	bool editorEvent(QEvent *event, QAbstractItemModel *model,
		const QStyleOptionViewItem &option,
		const QModelIndex &index) override;
	// 绘制背景
	void drawBackground(QPainter* painter, QStyleOptionViewItem& opt, const QModelIndex& index) const;
	// 绘制第一列（带图标）
	void drawFirstColumn(QPainter* painter, QStyleOptionViewItem& opt, const QModelIndex& index, const QFont& font) const;
	// 绘制第二列（数值+单位）
	void drawSecondColumn(QPainter* painter, const QStyleOptionViewItem& opt, const QModelIndex& index, const QFont& baseFont, bool isSelected) const;
	// 绘制网格线
	void drawGridLines(QPainter* painter, const QStyleOptionViewItem& opt, const QModelIndex& index) const;

private:
	// 判断当前节点是不是最后一个节点
	bool isLastSibling(const QModelIndex& index) const;
	// 格式化显示值的辅助方法
	QString formatDisplayValue(const QVariant& value) const;

	QSize m_iconSize; // 图标大小
	QPixmap m_expandIcon; // 左侧展开图标
	QPixmap m_collapseIcon; // 左侧折叠图标
};

class nmDataReservoir;
class nmDataWellBase;
class nmDataMixedResults;
class nmDataPvtParaForPebi;

// 结果参数Widget
class NM_SUB_WXS_EXPORT nmWxResultParameters : public QWidget
{
	Q_OBJECT
public:
	explicit nmWxResultParameters(QWidget *parent = nullptr);
	~nmWxResultParameters();
	void initUI();
	void initLayout();
	void initComponents();
	// 加载结果数据
	void loadResultDatas();
	// 添加混合结果参数到树
	void appendMixedResultsToTree(nmDataMixedResults* mixedResults,nmDataWellBase* well,nmDataReservoir* reservoir,nmDataPvtParaForPebi* pvtPara);
	// 添加井的结果参数到树控件(当前是默认井，即第一口井)
	void appendWellToTree(nmDataWellBase* well);
	// 添加储层结果参数到树控件
	void appendReservoirToTree(nmDataReservoir* reservoir);
	// 辅助函数：添加属性到树
	void addAttributeToTree(QTreeWidgetItem* parent,const nmDataAttribute& attr);

private:
	// 保留原有方法（可选，如果需要兼容性）
	QString formatAttributeValue(const QVariant& value, const QString& unit);

signals:
private:
	// 布局
	QVBoxLayout* m_pLayout;
	CustomTreeWidget* m_pTreeWidget;
};
#endif // NMWXRESULTPARAMETERS_H