超级趋势双通道指标

#property copyright   "XAUOF 指标工坊"
#property version     "1.00"
#property strict
#property indicator_chart_window
#property indicator_buffers 6
#property indicator_plots   4

// 绿色支撑通道
#property indicator_type1   DRAW_LINE
#property indicator_color1  clrLimeGreen
#property indicator_width1  2
#property indicator_label1  "支撑通道"

// 粉色阻力通道
#property indicator_type2   DRAW_LINE
#property indicator_color2  clrHotPink
#property indicator_width2  2
#property indicator_label2  "阻力通道"

// 多头区域填充（支撑线下边界到支撑线）
#property indicator_type3   DRAW_FILLING
#property indicator_color3  clrPaleGreen
#property indicator_label3  "多头区域"

// 空头区域填充（阻力线到阻力线上边界）
#property indicator_type4   DRAW_FILLING
#property indicator_color4  clrMistyRose
#property indicator_label4  "空头区域"

//--- 输入参数
input int      ATR_Period = 10;          // ATR 周期（控制通道灵敏度，推荐 10-20）
input double   ATR_Multiplier = 3.0;     // ATR 倍数（控制通道宽度，推荐 2.5-4.0）
input int      Channel_Smooth = 5;       // 通道平滑周期（数值越大台阶越宽，推荐 3-10）

//--- 指标缓冲区
double SupportBuffer[];        // 绿色支撑线
double ResistanceBuffer[];     // 粉色阻力线
double SupportFillBuffer[];    // 支撑线填充（绿色区域下边界）
double ResistanceFillBuffer[]; // 阻力线填充（粉色区域上边界）
double TempBuffer1[];          // 辅助缓冲区
double TempBuffer2[];          // 辅助缓冲区

//--- 全局变量
int atr_handle;

//+------------------------------------------------------------------+
//| 自定义指标初始化函数                                               |
//+------------------------------------------------------------------+
int OnInit()
{
   // 创建 ATR 指标句柄
   atr_handle = iATR(_Symbol, PERIOD_CURRENT, ATR_Period);
   if(atr_handle == INVALID_HANDLE)
   {
      Print("创建 ATR 指标失败，错误代码：", GetLastError());
      return INIT_FAILED;
   }
   
   // 绑定指标缓冲区（绘图缓冲区必须先绑定）
   SetIndexBuffer(0, SupportBuffer, INDICATOR_DATA);
   SetIndexBuffer(1, ResistanceBuffer, INDICATOR_DATA);
   SetIndexBuffer(2, SupportFillBuffer, INDICATOR_DATA);
   SetIndexBuffer(3, ResistanceFillBuffer, INDICATOR_DATA);
   SetIndexBuffer(4, TempBuffer1, INDICATOR_CALCULATIONS);
   SetIndexBuffer(5, TempBuffer2, INDICATOR_CALCULATIONS);
   
   // 设置填充关系
   // plot3（索引2）的下边界与 plot1（索引0）的支撑线之间填充绿色
   PlotIndexSetInteger(2, PLOT_DRAW_BEGIN, Channel_Smooth + ATR_Period);
   
   // plot4（索引3）的上边界与 plot2（索引1）的阻力线之间填充粉色
   PlotIndexSetInteger(3, PLOT_DRAW_BEGIN, Channel_Smooth + ATR_Period);
   
   // 设置指标精度
   IndicatorSetInteger(INDICATOR_DIGITS, _Digits);
   
   return INIT_SUCCEEDED;
}

//+------------------------------------------------------------------+
//| 自定义指标迭代函数                                                 |
//+------------------------------------------------------------------+
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[])
{
   // 检查 ATR 指标是否计算完毕
   if(BarsCalculated(atr_handle) < rates_total)
      return 0;
   
   // 拷贝 ATR 数据
   double atr[];
   ArrayResize(atr, rates_total);
   if(CopyBuffer(atr_handle, 0, 0, rates_total, atr) < rates_total)
      return 0;
   
   // 计算起始位置
   int start = (prev_calculated > 0) ? prev_calculated - 1 : ATR_Period + Channel_Smooth;
   
   // 首次运行时初始化缓冲区
   if(prev_calculated == 0)
   {
      ArrayInitialize(SupportBuffer, 0);
      ArrayInitialize(ResistanceBuffer, 0);
      ArrayInitialize(SupportFillBuffer, 0);
      ArrayInitialize(ResistanceFillBuffer, 0);
   }
   
   // 主循环：计算每根 K 线的通道值
   for(int i = start; i < rates_total && !IsStopped(); i++)
   {
      // 安全检查：确保有足够的历史数据
      if(i < Channel_Smooth)
      {
         SupportBuffer[i] = 0;
         ResistanceBuffer[i] = 0;
         SupportFillBuffer[i] = 0;
         ResistanceFillBuffer[i] = 0;
         continue;
      }
      
      // 计算平滑周期内的最高价和最低价
      double period_high = high[i];
      double period_low = low[i];
      for(int j = 1; j < Channel_Smooth && (i - j) >= 0; j++)
      {
         if(high[i - j] > period_high) period_high = high[i - j];
         if(low[i - j] < period_low)   period_low = low[i - j];
      }
      
      // 计算通道中轴（最高最低价的中点）
      double middle = (period_high + period_low) / 2.0;
      
      // 计算通道带宽（ATR 的倍数）
      double channel_width = atr[i] * ATR_Multiplier;
      
      // 计算支撑线和阻力线（从中轴向上下扩展）
      double temp_support = middle - channel_width / 2.0;
      double temp_resistance = middle + channel_width / 2.0;
      
      // 台阶效果：只有当新值与旧值差距足够大时才跳跃
      if(i > start)
      {
         double support_diff = MathAbs(temp_support - SupportBuffer[i - 1]);
         double resistance_diff = MathAbs(temp_resistance - ResistanceBuffer[i - 1]);
         
         // 如果变化小于 0.5 倍 ATR，保持上一根的值（形成台阶）
         if(support_diff < atr[i] * 0.5)
            temp_support = SupportBuffer[i - 1];
         if(resistance_diff < atr[i] * 0.5)
            temp_resistance = ResistanceBuffer[i - 1];
      }
      
      SupportBuffer[i] = temp_support;
      ResistanceBuffer[i] = temp_resistance;
      
      // 设置填充边界
      // 多头区域：从支撑线向下延伸 2 倍通道宽度
      SupportFillBuffer[i] = SupportBuffer[i] - channel_width * 2.0;
      
      // 空头区域：从阻力线向上延伸 2 倍通道宽度
      ResistanceFillBuffer[i] = ResistanceBuffer[i] + channel_width * 2.0;
   }
   
   return rates_total;
}

//+------------------------------------------------------------------+
//| 指标卸载函数                                                       |
//+------------------------------------------------------------------+
void OnDeinit(const int reason)
{
   // 释放指标句柄
   if(atr_handle != INVALID_HANDLE)
      IndicatorRelease(atr_handle);
}