双色趋势通道箭头指标

#property copyright   "XAUOF 指标工坊"
#property link        "https://www.xauof.com"
#property version     "1.00"
#property strict
#property indicator_chart_window
#property indicator_buffers 4
#property indicator_plots   4

// 蓝色上轨（多头线）
#property indicator_type1   DRAW_LINE
#property indicator_color1  clrAqua
#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_ARROW
#property indicator_color3  clrAqua
#property indicator_width3  2
#property indicator_label3  "买入"

// 卖出箭头
#property indicator_type4   DRAW_ARROW
#property indicator_color4  clrHotPink
#property indicator_width4  2
#property indicator_label4  "卖出"

//--- 输入参数
input int    InpATRPeriod   = 10;    // ATR 周期（推荐 10-14）
input double InpMultiplier  = 3.0;   // ATR 倍数（推荐 2.5-3.5，越大信号越少）
input int    InpArrowGap    = 50;    // 箭头与 K 线距离（点数，推荐 30-100）

//--- 指标缓冲区
double UpLineBuffer[];      // 蓝色多头线
double DnLineBuffer[];      // 粉色空头线
double BuyArrowBuffer[];    // 买入箭头
double SellArrowBuffer[];   // 卖出箭头

//--- 全局变量
int    atrHandle;           // ATR 指标句柄
double atrData[];           // ATR 数值缓冲

//+------------------------------------------------------------------+
//| 初始化函数                                                         |
//+------------------------------------------------------------------+
int OnInit()
{
   // 创建 ATR 指标句柄
   atrHandle = iATR(_Symbol, PERIOD_CURRENT, InpATRPeriod);
   if(atrHandle == INVALID_HANDLE)
   {
      Print("ATR 指标创建失败");
      return INIT_FAILED;
   }

   // 绑定指标缓冲区
   SetIndexBuffer(0, UpLineBuffer, INDICATOR_DATA);
   SetIndexBuffer(1, DnLineBuffer, INDICATOR_DATA);
   SetIndexBuffer(2, BuyArrowBuffer, INDICATOR_DATA);
   SetIndexBuffer(3, SellArrowBuffer, INDICATOR_DATA);

   // 设置箭头样式
   PlotIndexSetInteger(2, PLOT_ARROW, 233);  // 买入向上箭头
   PlotIndexSetInteger(3, PLOT_ARROW, 234);  // 卖出向下箭头

   // 设置空值
   PlotIndexSetDouble(0, PLOT_EMPTY_VALUE, EMPTY_VALUE);
   PlotIndexSetDouble(1, PLOT_EMPTY_VALUE, EMPTY_VALUE);
   PlotIndexSetDouble(2, PLOT_EMPTY_VALUE, EMPTY_VALUE);
   PlotIndexSetDouble(3, PLOT_EMPTY_VALUE, EMPTY_VALUE);

   // 设置指标名称
   IndicatorSetString(INDICATOR_SHORTNAME, StringFormat("趋势通道(ATR:%d×%.1f)", InpATRPeriod, InpMultiplier));
   IndicatorSetInteger(INDICATOR_DIGITS, _Digits);

   return INIT_SUCCEEDED;
}

//+------------------------------------------------------------------+
//| 反初始化函数                                                       |
//+------------------------------------------------------------------+
void OnDeinit(const int reason)
{
   if(atrHandle != INVALID_HANDLE)
      IndicatorRelease(atrHandle);
}

//+------------------------------------------------------------------+
//| 主计算函数                                                         |
//+------------------------------------------------------------------+
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(atrHandle) < rates_total)
      return 0;

   // 拷贝 ATR 数据
   if(CopyBuffer(atrHandle, 0, 0, rates_total, atrData) < rates_total)
      return 0;

   // 确定起始位置
   int start = (prev_calculated > InpATRPeriod) ? prev_calculated - 1 : InpATRPeriod;

   // 首次运行初始化
   if(prev_calculated == 0)
   {
      ArrayInitialize(UpLineBuffer, EMPTY_VALUE);
      ArrayInitialize(DnLineBuffer, EMPTY_VALUE);
      ArrayInitialize(BuyArrowBuffer, EMPTY_VALUE);
      ArrayInitialize(SellArrowBuffer, EMPTY_VALUE);

      // 初始化前几根为空
      for(int i = 0; i < InpATRPeriod; i++)
      {
         UpLineBuffer[i] = EMPTY_VALUE;
         DnLineBuffer[i] = EMPTY_VALUE;
      }

      // 第一根有效 K 线初始化：根据收盘价与中线关系决定初始趋势
      if(rates_total > InpATRPeriod)
      {
         double hl2 = (high[InpATRPeriod] + low[InpATRPeriod]) / 2.0;
         double atr = atrData[InpATRPeriod];
         double upBand = hl2 - InpMultiplier * atr;
         double dnBand = hl2 + InpMultiplier * atr;

         if(close[InpATRPeriod] > hl2)
         {
            // 初始多头
            UpLineBuffer[InpATRPeriod] = upBand;
            DnLineBuffer[InpATRPeriod] = EMPTY_VALUE;
         }
         else
         {
            // 初始空头
            UpLineBuffer[InpATRPeriod] = EMPTY_VALUE;
            DnLineBuffer[InpATRPeriod] = dnBand;
         }
      }

      start = InpATRPeriod + 1;
   }

   // 计算主逻辑
   for(int i = start; i < rates_total && !IsStopped(); i++)
   {
      // 先清空箭头
      BuyArrowBuffer[i] = EMPTY_VALUE;
      SellArrowBuffer[i] = EMPTY_VALUE;

      if(i < 1) continue;

      double hl2 = (high[i] + low[i]) / 2.0;
      double atr = atrData[i];

      // 计算基础上下轨
      double upBand = hl2 - InpMultiplier * atr;
      double dnBand = hl2 + InpMultiplier * atr;

      // 判断前一根趋势状态
      bool prevWasUp = (UpLineBuffer[i-1] != EMPTY_VALUE && UpLineBuffer[i-1] > 0);

      if(prevWasUp)
      {
         // 前一根是多头（蓝线）
         // 上轨只能上升不能下降（台阶效果）
         if(upBand > UpLineBuffer[i-1])
            UpLineBuffer[i] = upBand;
         else
            UpLineBuffer[i] = UpLineBuffer[i-1];

         // 检查是否跌破蓝线 → 翻空
         if(close[i] < UpLineBuffer[i])
         {
            // 翻空：启动粉线，打卖出箭头
            DnLineBuffer[i] = dnBand;
            UpLineBuffer[i] = EMPTY_VALUE;
            SellArrowBuffer[i] = high[i] + InpArrowGap * _Point;
         }
         else
         {
            // 维持多头
            DnLineBuffer[i] = EMPTY_VALUE;
         }
      }
      else
      {
         // 前一根是空头（粉线）
         // 下轨只能下降不能上升（台阶效果）
         if(dnBand < DnLineBuffer[i-1])
            DnLineBuffer[i] = dnBand;
         else
            DnLineBuffer[i] = DnLineBuffer[i-1];

         // 检查是否突破粉线 → 翻多
         if(close[i] > DnLineBuffer[i])
         {
            // 翻多：启动蓝线，打买入箭头
            UpLineBuffer[i] = upBand;
            DnLineBuffer[i] = EMPTY_VALUE;
            BuyArrowBuffer[i] = low[i] - InpArrowGap * _Point;
         }
         else
         {
            // 维持空头
            UpLineBuffer[i] = EMPTY_VALUE;
         }
      }
   }

   return rates_total;
}
//+------------------------------------------------------------------+