Heikin Ashi Buy Sell Arrows

// This source code is subject to the terms of the Mozilla Public License 2.0 at https://mozilla.org/MPL/2.0/
// © goodgains


//@version=5
indicator("Heikin Ashi Buy Sell Arrows", overlay = true)

// @keywords: heikin-ashi, buy-sell-arrows, ha-candles, primary-signals, consecutive-count, arrow-labels, trend-reversal, ha-open-close, signal-arrows, alert-conditions, bar-highlight, ha-high-low, entry-markers, exit-markers, smoothed-candles, direction-change, plotshape-labels, overlay-arrows, ha-logic, streak-count

// def HAclose = ohlc4;
// rec HAopen = compoundValue( 1, ( HAopen[1] + HAclose[1] ) / 2, HAclose );
// def HAhigh = Max( high, Max( HAopen, HAclose ) );
// def HAlow = Min( low, Min( HAopen, HAclose ) );
// def haBuy = HAopen <= HAlow;
// def haSell = HAopen >= HAhigh;
// rec count=if haBuy==1 and count[1]==0 then 1 else if haSell==0 and count[1]>=1 then count[1]+1 else 0;
// rec count1=if haSell==1 and count1[1]==0 then 1 else if haBuy==0 and count1[1]>=1 then count1[1]+1 else 0;
// def PrimaryUP = if HAopen <= HAlow and habuy==1 and count[1]==0 then yes else no;
// def PrimaryDOWN = if HAopen >= HAhigh and hasell==1 and count1[1]==0 then yes else no;

// plot ArrowUp = if PrimaryUP then HAlow else Double.NaN;
// ArrowUp.SetDefaultColor(Color.Dark_Orange);
// ArrowUp.SetPaintingStrategy(PaintingStrategy.BOOLEAN_ARROW_Up);
// ArrowUp.SetLineWeight(5);
// ArrowUp.HideBubble();


// plot ArrowDn = if PrimaryDOWN then HAhigh else Double.NaN;
// ArrowDn.SetDefaultColor(Color.Dark_Orange);
// ArrowDn.SetPaintingStrategy(PaintingStrategy.BOOLEAN_ARROW_DOWN);
// ArrowDn.SetLineWeight(5);
// ArrowDn.HideBubble();


[src_o, src_h, src_l, src_c, src_ohlc4] = request.security(syminfo.ticker, '', [open, high, low, close, ohlc4])

var float haclose = 0.00
var float haopen = 0.00
var float hahigh = 0.00
var float halow = 0.00

haclose := src_ohlc4
haopen  := nz((haopen[1] + haclose[1]) / 2, haclose)
hahigh  := math.max(src_h, math.max(haopen, haclose))
halow   := math.min(src_l, math.min(haopen, haclose))

ha_buy  = haopen <= halow
ha_sell = haopen >= hahigh

var int count   = 0
var int count1  = 0

count := ha_buy==1 and count[1]==0 ? 1 
 : ha_sell==0 and count[1]>=1 ? count[1]+1
 : 0

count1 := ha_sell==1 and count1[1]==0 ? 1 
 : ha_buy==0 and count1[1]>=1 ? count1[1]+1
 : 0

PrimaryUP   = haopen <= halow and ha_buy==1 and count[1]==0 ? true : false
PrimaryDOWN = haopen >= hahigh and ha_sell==1 and count1[1]==0 ? true : false

plotshape(PrimaryUP, title = 'Arrow Up', style = shape.labelup, color = color.green,
 text = 'BUY', textcolor = color.white, location = location.belowbar)

plotshape(PrimaryDOWN, title = 'Arrow Down', style = shape.labeldown, color = color.red,
 text = 'SELL', textcolor = color.white, location = location.abovebar)

var color last_color = color.white
last_color := PrimaryUP ? color.green : PrimaryDOWN ? color.red : last_color[1]
barcolor(color = last_color, display = display.none)


alertcondition(PrimaryUP, "BUY Signal", "BUY Signal")
alertcondition(PrimaryUP, "SELL Signal", "SELL Signal")