package index

import (
	"log"

	"git.apinb.com/quant/strategy/internal/core/trade"
	"github.com/markcheno/go-talib"
)

// LineRegSlope 根据线性回归斜率分析最新4条K线是否存在反转情况
// 改进版本：修复了逻辑错误，增加了边界检查，优化了性能
func LineRegSlope(a *Args) (action string) {
	// 边界检查：确保有足够的数据
	if len(a.InReal) < a.Period {
		if a.Debug {
			log.Println("LineRegSlope: 数据不足，需要至少", a.Period, "条数据")
		}
		return
	}

	result := talib.LinearRegSlope(a.InReal, a.Period)

	// 边界检查：确保结果数组有足够的数据
	if len(result) < 4 {
		if a.Debug {
			log.Println("LineRegSlope: 线性回归结果不足4条")
		}
		return
	}

	// 获取最新4条K线的斜率值
	origin := result[len(result)-4:]

	// 处理斜率值，保留所有值（移除过于严格的过滤）
	var k4 []float64
	for _, val := range origin {
		if val >= 0.001 || val <= -0.001 {
			roundedVal := trade.FloatRound(val, 6)
			k4 = append(k4, roundedVal)
		}
	}

	if a.Debug {
		log.Println("Origin", result)
		log.Println("Kline4", k4)
	}

	// 检查是否有足够的数据
	if len(k4) < 4 {
		return
	}

	// 判断反转信号
	// 下降反转：从正斜率转为负斜率，且斜率递减
	if k4[0] > 0 && k4[3] < 0 {
		// 检查斜率是否呈现递减趋势（从正到负的平滑过渡）
		if isDecreasingTrend(k4) {
			action = "DOWN"
		}
	}

	// 上升反转：从负斜率转为正斜率，且斜率递增
	if k4[0] < 0 && k4[3] > 0 {
		// 检查斜率是否呈现递增趋势（从负到正的平滑过渡）
		if isIncreasingTrend(k4) {
			action = "UP"
		}
	}

	return
}

// isDecreasingTrend 检查斜率是否呈现递减趋势
func isDecreasingTrend(slopes []float64) bool {
	// 简化的递减判断：确保整体趋势是递减的
	// 允许中间有小的波动，但整体方向是向下的
	for i := 1; i < len(slopes); i++ {
		if slopes[i] > slopes[i-1] {
			// 如果发现上升，检查是否只是小幅波动
			if slopes[i]-slopes[i-1] > 0.01 { // 允许小幅波动
				return false
			}
		}
	}
	return true
}

// isIncreasingTrend 检查斜率是否呈现递增趋势
func isIncreasingTrend(slopes []float64) bool {
	// 简化的递增判断：确保整体趋势是递增的
	// 允许中间有小的波动，但整体方向是向上的
	for i := 1; i < len(slopes); i++ {
		if slopes[i] < slopes[i-1] {
			// 如果发现下降，检查是否只是小幅波动
			if slopes[i-1]-slopes[i] > 0.01 { // 允许小幅波动
				return false
			}
		}
	}
	return true
}

// CurrentSlopeDirection 基于最新线性回归斜率给出当前方向（非反转）
// 返回："UP" / "DOWN" / ""(无明显方向)
func CurrentSlopeDirection(a *Args) string {
    if len(a.InReal) < a.Period {
        if a.Debug {
            log.Println("CurrentSlopeDirection: 数据不足，需要至少", a.Period, "条数据")
        }
        return ""
    }

    s := talib.LinearRegSlope(a.InReal, a.Period)
    if len(s) == 0 {
        return ""
    }

    latest := s[len(s)-1]
    // 轻微阈值，避免噪声抖动；可按回测调整
    const threshold = 0.0002
    if latest > threshold {
        return "UP"
    }
    if latest < -threshold {
        return "DOWN"
    }
    return ""
}