SuperTrend 回血止损带（精准复刻版）

#property copyright   "Adapted from LuxAlgo CC BY-NC-SA 4.0"
#property version     "1.00"
#property strict
#property indicator_chart_window
#property indicator_buffers 5
#property indicator_plots   3

// 主线（动态颜色）
#property indicator_type1   DRAW_COLOR_LINE
#property indicator_color1  C'8,153,129', C'242,54,69'  // 青绿、粉红
#property indicator_width1  2
#property indicator_label1  "SuperTrend 带"

// 多头圆点
#property indicator_type2   DRAW_ARROW
#property indicator_color2  C'8,153,129'
#property indicator_width2  3
#property indicator_label2  "多头切换"

// 空头圆点
#property indicator_type3   DRAW_ARROW
#property indicator_color3  C'242,54,69'
#property indicator_width3  3
#property indicator_label3  "空头切换"

//--- 输入参数
input group "SuperTrend 设置"
input int    ATR_Period      = 10;     // ATR 周期
input double Base_Multiplier = 3.0;    // 基础倍数

input group "回血逻辑"
input double Recovery_Alpha     = 5.0;  // 回血权重 (%)
input double Recovery_Threshold = 1.0;  // 回血阈值 (xATR)

input group "显示设置"
input bool   Show_Fills   = true;      // 显示渐变填充
input bool   Show_Labels  = true;      // 显示信号标签

//--- 指标缓冲区
double BandBuffer[];       // 主线值
double BandColors[];       // 主线颜色索引
double BullDotBuffer[];    // 多头圆点
double BearDotBuffer[];    // 空头圆点
double ATR_Buffer[];       // ATR 值

//--- 全局变量
int atrHandle;
int trend = 1;             
double switchPrice = 0;    

//+------------------------------------------------------------------+
int OnInit()
{
    atrHandle = iATR(_Symbol, PERIOD_CURRENT, ATR_Period);
    if(atrHandle == INVALID_HANDLE)
    {
        Print("创建 ATR 指标失败");
        return INIT_FAILED;
    }
    
    SetIndexBuffer(0, BandBuffer, INDICATOR_DATA);
    SetIndexBuffer(1, BandColors, INDICATOR_COLOR_INDEX);
    SetIndexBuffer(2, BullDotBuffer, INDICATOR_DATA);
    SetIndexBuffer(3, BearDotBuffer, INDICATOR_DATA);
    SetIndexBuffer(4, ATR_Buffer, INDICATOR_CALCULATIONS);
    
    PlotIndexSetInteger(2, PLOT_ARROW, 159);
    PlotIndexSetInteger(3, PLOT_ARROW, 159);
    PlotIndexSetDouble(2, PLOT_EMPTY_VALUE, EMPTY_VALUE);
    PlotIndexSetDouble(3, PLOT_EMPTY_VALUE, EMPTY_VALUE);
    
    IndicatorSetString(INDICATOR_SHORTNAME, "SuperTrend Recovery");
    
    return INIT_SUCCEEDED;
}

//+------------------------------------------------------------------+
void OnDeinit(const int reason)
{
    if(atrHandle != INVALID_HANDLE)
        IndicatorRelease(atrHandle);
    
    ObjectsDeleteAll(0, "STR_");
}

//+------------------------------------------------------------------+
int OnCalculate(const int rates_total,
                const int prev_calculated,
                const datetime &time[],
                const double &open[],
                const double &high[],
                const double &low[],
                const double &close[],
                const long &tick_volume[],
                const long &volume[],
                const int &spread[])
{
    if(BarsCalculated(atrHandle) < rates_total)
        return 0;
    
    if(CopyBuffer(atrHandle, 0, 0, rates_total, ATR_Buffer) < rates_total)
        return 0;
    
    int start = (prev_calculated > 1) ? prev_calculated - 1 : ATR_Period;
    
    if(prev_calculated == 0)
    {
        ArrayInitialize(BandBuffer, EMPTY_VALUE);
        ArrayInitialize(BandColors, 0);
        ArrayInitialize(BullDotBuffer, EMPTY_VALUE);
        ArrayInitialize(BearDotBuffer, EMPTY_VALUE);
        trend = 1;
        switchPrice = close[start];
    }
    
    // 主循环
    for(int i = start; i < rates_total && !IsStopped(); i++)
    {
        if(i < ATR_Period) continue;
        
        double src = (high[i] + low[i]) / 2.0;
        double atr = ATR_Buffer[i];
        double alpha = Recovery_Alpha / 100.0;
        
        double upperBase = src + Base_Multiplier * atr;
        double lowerBase = src - Base_Multiplier * atr;
        
        double prevBand = (i > 0 && BandBuffer[i-1] != EMPTY_VALUE) ? BandBuffer[i-1] : 
                          (trend == 1 ? lowerBase : upperBase);
        
        double deviation = Recovery_Threshold * atr;
        bool isAtLoss = false;
        
        if(trend == 1)
            isAtLoss = (switchPrice - close[i]) > deviation;
        else
            isAtLoss = (close[i] - switchPrice) > deviation;
        
        BullDotBuffer[i] = EMPTY_VALUE;
        BearDotBuffer[i] = EMPTY_VALUE;
        
        if(trend == 1)
        {
            double targetBand = isAtLoss ? (alpha * close[i] + (1.0 - alpha) * prevBand) : lowerBase;
            BandBuffer[i] = MathMax(targetBand, prevBand);
            BandColors[i] = 0;
            
            if(close[i] < BandBuffer[i])
            {
                trend = -1;
                BandBuffer[i] = upperBase;
                BandColors[i] = 1;
                switchPrice = close[i];
                BearDotBuffer[i] = BandBuffer[i];
                
                if(Show_Labels)
                {
                    string name = "STR_Arrow_" + IntegerToString(i);
                    ObjectCreate(0, name, OBJ_ARROW_DOWN, 0, time[i], BandBuffer[i] + 5 * _Point);
                    ObjectSetInteger(0, name, OBJPROP_COLOR, C'242,54,69');
                    ObjectSetInteger(0, name, OBJPROP_WIDTH, 2);
                    
                    string txtName = "STR_Text_" + IntegerToString(i);
                    ObjectCreate(0, txtName, OBJ_TEXT, 0, time[i], BandBuffer[i] + 8 * _Point);
                    ObjectSetString(0, txtName, OBJPROP_TEXT, "BEAR");
                    ObjectSetInteger(0, txtName, OBJPROP_COLOR, clrWhite);
                    ObjectSetInteger(0, txtName, OBJPROP_FONTSIZE, 7);
                    ObjectSetInteger(0, txtName, OBJPROP_ANCHOR, ANCHOR_UPPER);
                }
            }
        }
        else
        {
            double targetBand = isAtLoss ? (alpha * close[i] + (1.0 - alpha) * prevBand) : upperBase;
            BandBuffer[i] = MathMin(targetBand, prevBand);
            BandColors[i] = 1;
            
            if(close[i] > BandBuffer[i])
            {
                trend = 1;
                BandBuffer[i] = lowerBase;
                BandColors[i] = 0;
                switchPrice = close[i];
                BullDotBuffer[i] = BandBuffer[i];
                
                if(Show_Labels)
                {
                    string name = "STR_Arrow_" + IntegerToString(i);
                    ObjectCreate(0, name, OBJ_ARROW_UP, 0, time[i], BandBuffer[i] - 5 * _Point);
                    ObjectSetInteger(0, name, OBJPROP_COLOR, C'8,153,129');
                    ObjectSetInteger(0, name, OBJPROP_WIDTH, 2);
                    
                    string txtName = "STR_Text_" + IntegerToString(i);
                    ObjectCreate(0, txtName, OBJ_TEXT, 0, time[i], BandBuffer[i] - 8 * _Point);
                    ObjectSetString(0, txtName, OBJPROP_TEXT, "BULL");
                    ObjectSetInteger(0, txtName, OBJPROP_COLOR, clrWhite);
                    ObjectSetInteger(0, txtName, OBJPROP_FONTSIZE, 7);
                    ObjectSetInteger(0, txtName, OBJPROP_ANCHOR, ANCHOR_LOWER);
                }
            }
        }
        
        // 创建精细填充：只在价格和止损带之间，减少透明度让颜色淡一些
        if(Show_Fills && i > 0 && i < rates_total - 1)
        {
            // 确保只在带和价格之间填充，不要扩大范围
            double bandPrice = BandBuffer[i];
            double priceNow = close[i];
            
            // 只有当价格和带之间有明显距离才画填充
            if(MathAbs(bandPrice - priceNow) > 2 * _Point)
            {
                string name = "STR_Fill_" + IntegerToString(i);
                
                if(ObjectFind(0, name) < 0)
                {
                    double top = MathMax(bandPrice, priceNow);
                    double bottom = MathMin(bandPrice, priceNow);
                    
                    // 颜色更淡，透明度更高
                    color fillColor = (trend == 1) ? C'8,153,129' : C'242,54,69';
                    
                    ObjectCreate(0, name, OBJ_RECTANGLE, 0, time[i], top, time[i+1], bottom);
                    ObjectSetInteger(0, name, OBJPROP_COLOR, fillColor);
                    ObjectSetInteger(0, name, OBJPROP_FILL, true);
                    ObjectSetInteger(0, name, OBJPROP_BACK, true);
                    ObjectSetInteger(0, name, OBJPROP_SELECTABLE, false);
                    ObjectSetInteger(0, name, OBJPROP_HIDDEN, true);
                    ObjectSetInteger(0, name, OBJPROP_WIDTH, 0);
                }
            }
        }
    }
    
    return rates_total;
}