#ifndef NMWXCHARTWIDGET_H
#define NMWXCHARTWIDGET_H

#include <QWidget>
#include <QVector>
#include <QPointF>
#include <QRectF>
#include "nmDataPerforation.h"

class QPainter;
class QMouseEvent;
class QPaintEvent;

class nmWxChartWidget : public QWidget
{
	Q_OBJECT

public:
	explicit nmWxChartWidget(QWidget *parent = 0);
	virtual ~nmWxChartWidget();

protected:
	virtual void paintEvent(QPaintEvent *event);
	virtual void mousePressEvent(QMouseEvent *event);
	virtual void mouseMoveEvent(QMouseEvent *event);
	virtual void mouseReleaseEvent(QMouseEvent *event);

public:
	// 设置skin值
	void setSkinValues(double bValue, double kValue);

	// 设置和获取线条位置
	void setLinePositions(double x1, double y1, double x2, double y2);
	void getLinePositions(double& x1, double& y1, double& x2, double& y2) const;

	// 获取计算出的skin值
	double getSkin0Value() const;
	double getDSkinDqValue() const;

	bool getLassoMode() const;
	void setLassoMode(bool enabled);

	// 重置线条到初始位置
	void resetLinesToInitialPositions();

	// 新增：设置流动段数据和流量数据
	void setFlowSegmentData(const nmDataPerforation& perforationData);
	void setFlowData(const QVector<QPointF>& flowData);
	void setRawFlowData(const QVector<QPointF>& rawFlowData);
	void setSnapToRateChanges(bool snapToRateChanges);

	// 新增：更新×标记点数据
	void updateCrossMarkPoints();
	// 重置所有×标记点
	void resetAllCrossMarks();

	void updateCrossMarkPointsData();      // 只更新现有×标记点的数据
    void regenerateCrossMarkPoints();      // 完全重新生成×标记点

	// 计算方法
	void calculateAverageValue();
	bool shouldShowAverageLine();
	void calculateDataRange();  // 动态计算坐标范围

signals:
	void skinValuesChanged(double skin0Value, double dSkinDqValue);
	void crossMarksFiltered();  // 当×标记被过滤后发出的信号
	void lassoOperationCompleted(); // 新增：套索操作完成信号

private:
	// 绘制方法
	void drawGrid(QPainter &painter);
	void drawAxes(QPainter &painter);
	void drawLine(QPainter &painter);
	void drawPoints(QPainter &painter);
	void drawAverageLine(QPainter &painter);  // 绘制skin0线
	void drawCrossMarks(QPainter &painter);  // 新增：绘制×标记

	// 坐标转换
	QPointF dataToScreen(const QPointF &dataPoint);
	QPointF screenToData(const QPointF &screenPoint);

	// 交互相关
	int findNearestPoint(const QPointF &screenPos);
	bool isNearLine(const QPointF &screenPos);

	// 辅助方法
	void setLinePositionsFromSkinValues(double skin0Value, double dSkinDqValue);
	double getAverageFlowRateForSegment(double segmentStartTime, double segmentEndTime) const;

	// 最小二乘法拟合×标记点，返回拟合是否成功
	bool fitLineToPoints();

	// 计算拟合直线的参数 y = ax + b
	void calculateLinearRegression(const QVector<QPointF>& points, double& slope, double& intercept);

	// 根据拟合结果设置红色线的端点位置
	void setFittedLinePositions(double slope, double intercept);

	// 套索相关方法
	bool isPointInPolygon(const QPointF& point, const QPolygonF& polygon);
	void deleteSelectedCrossMarks();
	void drawLasso(QPainter& painter);
	void clearLassoSelection();
	void exitLassoMode();

	// 常量定义
	static const double POINT_RADIUS;
	static const double POINT_CLICK_RADIUS;
	static const double LINE_CLICK_DISTANCE;
	static const double CROSS_MARK_SIZE;  // 新增：×标记大小

	QVector<QPointF> linePoints;        // 红线端点
	QVector<QPointF> initialPositions;  // 初始位置（现在动态计算）
	QVector<QPointF> crossMarkPoints;   // 新增：×标记点位置
	QVector<int> crossMarkSegmentIndices; // 新增：×标记对应的段索引

	QRectF dataRect;
	QRectF chartRect;

	bool isDragging;
	bool isDraggingLine;
	int dragPointIndex;
	QPointF lastMousePos;
	QPointF dragOffset;

	bool showAverageLine;      // 是否显示skin0线
	double averageValue;       // skin0值（y轴截距）
	double dSkinDqValue;		// 斜率
	double selectionRangeX1;   // 选择范围的左端点x值
	double selectionRangeX2;   // 选择范围的右端点x值

    // 数据源
    nmDataPerforation m_perforationData;
	QVector<QPointF> m_flowData;  // 流量数据
	QVector<QPointF> m_rawFlowData;

	bool m_snapToRateChanges;  // 保存对齐到流量段的状态

	// 套索选择相关
	bool m_lassoMode;           // 是否处于套索模式
	bool m_lassoDrawing;        // 是否正在绘制套索
	QPolygonF m_lassoPath;      // 套索路径
	QVector<int> m_selectedCrossMarkIndices;  // 选中的×标记索引
	bool m_hasPerformedInitialFit; // 是否已经进行过初始拟合

};

#endif // NMWXCHARTWIDGET_H