package com.xcong.excoin.modules.okxNewPrice.indicator.macdAndMatrategy;

import java.math.BigDecimal;
import java.util.ArrayList;
import java.util.List;

/**
 * MACD（Moving Average Convergence Divergence）指标计算器
 * <p>
 * MACD指标由三部分组成：
 * 1. DIF（Difference）：短期EMA与长期EMA的差值
 * 2. DEA（Signal Line）：DIF的指数移动平均线，作为MACD的信号线
 * 3. MACD柱状图（Histogram）：DIF与DEA的差值，反映市场动量
 * <p>
 * 默认参数：短期周期=12，长期周期=26，信号周期=9
 */
public class MACDCalculator {

    /**
     * 计算MACD指标
     *
     * @param closePrices 收盘价列表（使用BigDecimal确保计算精度）
     * @param shortPeriod 短期EMA周期（通常为12）
     * @param longPeriod  长期EMA周期（通常为26）
     * @param signalPeriod DEA的周期（通常为9）
     * @return 包含MACD各部分数据的PriceData列表
     * @throws IllegalArgumentException 如果数据点不足或参数无效
     */
    public static MACDResult calculateMACD(List<BigDecimal> closePrices, int shortPeriod, int longPeriod, int signalPeriod) {
        // 参数校验：确保数据点足够
        if (closePrices == null || closePrices.isEmpty()) {
            throw new IllegalArgumentException("Close prices list cannot be null or empty.");
        }
        if (shortPeriod <= 0 || longPeriod <= 0 || signalPeriod <= 0) {
            throw new IllegalArgumentException("All periods must be positive integers.");
        }
        if (shortPeriod >= longPeriod) {
            throw new IllegalArgumentException("Short period must be less than long period.");
        }
        if (closePrices.size() < Math.max(shortPeriod, longPeriod)) {
            throw new IllegalArgumentException("Insufficient data points for the specified periods.");
        }

        // 1. 计算短期和长期EMA（使用SMA作为初始值，提高计算准确性）
        List<BigDecimal> emaShort = EMACalculator.calculateEMA(closePrices, shortPeriod, true);
        List<BigDecimal> emaLong = EMACalculator.calculateEMA(closePrices, longPeriod, true);

        // 2. 确定公共有效起始点（从较长周期的EMA开始计算）
        int startIdx = Math.max(shortPeriod, longPeriod) - 1; // 因为EMA从第period个数据点开始有效
        int validLength = closePrices.size() - startIdx;      // 有效数据点数量

        // 3. 计算DIF（仅在有效区间内计算）
        List<BigDecimal> difValues = new ArrayList<>(validLength);
        for (int i = 0; i < validLength; i++) {
            // 计算EMA的有效索引
            int idxEmaShort = startIdx - shortPeriod + 1 + i; // 短期EMA的当前索引
            int idxEmaLong = startIdx - longPeriod + 1 + i;   // 长期EMA的当前索引

            // DIF = 短期EMA - 长期EMA
            BigDecimal dif = emaShort.get(idxEmaShort).subtract(emaLong.get(idxEmaLong));
            difValues.add(dif);
        }

        // 4. 计算DEA（基于有效DIF数据的EMA）
        List<BigDecimal> deaValues = EMACalculator.calculateEMA(difValues, signalPeriod, false);

        // 5. 构建并填充结果（包含所有MACD数据）
        List<PriceData> result = new ArrayList<>(deaValues.size());

        // 从第一个DEA值开始构建结果
        for (int i = 0; i < deaValues.size(); i++) {
            int closeIdx = startIdx + i; // 对应原收盘价列表的索引

            // 创建价格数据对象
            PriceData data = new PriceData(closePrices.get(closeIdx));

            // 设置EMA（使用正确的偏移位置）
            data.setEmaShort(emaShort.get(closeIdx - shortPeriod + 1));
            data.setEmaLong(emaLong.get(closeIdx - longPeriod + 1));

            // 设置DIF、DEA和MACD柱状图
            data.setDif(difValues.get(i));
            data.setDea(deaValues.get(i)); // DEA索引直接对应
            data.setMacdHist(data.getDif().subtract(data.getDea())); // MACD柱状图 = DIF - DEA

            result.add(data);
        }


        return new MACDResult(result, startIdx);
    }

    /**
     * 使用默认参数计算MACD指标
     * <p>
     * 默认参数：短期周期=12，长期周期=26，信号周期=9
     *
     * @param closePrices 收盘价列表
     * @return 包含MACD各部分数据的PriceData列表
     */
    public static MACDResult calculateMACD(List<BigDecimal> closePrices) {
        // 默认参数：短期周期12，长期周期26，信号周期9
        return calculateMACD(closePrices, 12, 26, 9);
    }

    /**
     * 判断是否出现顶背离
     * <p>
     * 顶背离：价格创新高，但DIF未创新高，且与价格走势背离
     * 增强空头信号可靠性
     *
     * @param closePrices 原始收盘价列表
     * @param macdResult MACD计算结果
     * @return 是否出现顶背离
     */
    public static boolean isTopDivergence(List<BigDecimal> closePrices, MACDResult macdResult) {
        List<PriceData> macdData = macdResult.getMacdData();
        int startIdx = macdResult.getStartIndex();
        
        // 确保有足够的数据点进行判断（至少需要2个高点）
        if (macdData.size() < 10) {
            return false;
        }
        
        // 找到最近的价格高点和对应的DIF值
        int recentPriceHighIdx = IndicatorUtils.findRecentHighIndex(closePrices, startIdx);
        if (recentPriceHighIdx < startIdx + 2 || recentPriceHighIdx == -1) {
            return false;
        }
        
        // 找到之前的价格高点和对应的DIF值
        int previousPriceHighIdx = IndicatorUtils.findPreviousHighIndex(closePrices, startIdx, recentPriceHighIdx);
        if (previousPriceHighIdx < startIdx || previousPriceHighIdx == -1) {
            return false;
        }
        
        // 获取对应位置的DIF值
        int recentDifIdx = recentPriceHighIdx - startIdx;
        int previousDifIdx = previousPriceHighIdx - startIdx;
        
        // 边界检查
        if (recentDifIdx >= macdData.size() || previousDifIdx >= macdData.size()) {
            return false;
        }
        
        BigDecimal recentPrice = closePrices.get(recentPriceHighIdx);
        BigDecimal previousPrice = closePrices.get(previousPriceHighIdx);
        BigDecimal recentDif = macdData.get(recentDifIdx).getDif();
        BigDecimal previousDif = macdData.get(previousDifIdx).getDif();
        
        // 顶背离条件：价格创新高，但DIF未创新高
        return recentPrice.compareTo(previousPrice) > 0 && 
               recentDif.compareTo(previousDif) < 0;
    }

    /**
     * 判断是否出现底背离
     * <p>
     * 底背离：价格创新低，但DIF未创新低，且与价格走势背离
     * 增强多头信号可靠性
     *
     * @param closePrices 原始收盘价列表
     * @param macdResult MACD计算结果
     * @return 是否出现底背离
     */
    public static boolean isBottomDivergence(List<BigDecimal> closePrices, MACDResult macdResult) {
        List<PriceData> macdData = macdResult.getMacdData();
        int startIdx = macdResult.getStartIndex();
        
        // 确保有足够的数据点进行判断（至少需要2个低点）
        if (macdData.size() < 10) {
            return false;
        }
        
        // 找到最近的价格低点和对应的DIF值
        int recentPriceLowIdx = IndicatorUtils.findRecentLowIndex(closePrices, startIdx);
        if (recentPriceLowIdx < startIdx + 2 || recentPriceLowIdx == -1) {
            return false;
        }
        
        // 找到之前的价格低点和对应的DIF值
        int previousPriceLowIdx = IndicatorUtils.findPreviousLowIndex(closePrices, startIdx, recentPriceLowIdx);
        if (previousPriceLowIdx < startIdx || previousPriceLowIdx == -1) {
            return false;
        }
        
        // 获取对应位置的DIF值
        int recentDifIdx = recentPriceLowIdx - startIdx;
        int previousDifIdx = previousPriceLowIdx - startIdx;
        
        // 边界检查
        if (recentDifIdx >= macdData.size() || previousDifIdx >= macdData.size()) {
            return false;
        }
        
        BigDecimal recentPrice = closePrices.get(recentPriceLowIdx);
        BigDecimal previousPrice = closePrices.get(previousPriceLowIdx);
        BigDecimal recentDif = macdData.get(recentDifIdx).getDif();
        BigDecimal previousDif = macdData.get(previousDifIdx).getDif();
        
        // 底背离条件：价格创新低，但DIF未创新低
        return recentPrice.compareTo(previousPrice) < 0 && 
               recentDif.compareTo(previousDif) > 0;
    }
}