Momentum Breakout Strategy

//@version=5
strategy("Momentum Breakout Strategy - Full Version", overlay=true)

// === Input: Position Sizing Options ===
position_type = input.string("Percent of Equity", title="Position Size Type", options=["Percent of Equity", "Fixed Quantity", "Fixed Cash Amount"])
qty_percent = input.float(10.0, title="Qty % of Equity", minval=0.1)
fixed_qty = input.float(1, title="Fixed Quantity", minval=0.0001)
cash_per_trade = input.float(100, title="Fixed Cash Amount ($)", minval=1)

// === Function: Determine Order Size ===
get_order_size() =>
    if position_type == "Percent of Equity"
        na
    else if position_type == "Fixed Quantity"
        fixed_qty
    else
        cash_per_trade / close

order_size = get_order_size()

// === Inputs for Strategy Logic ===
breakoutLen = input.int(20, title="Breakout Lookback Length")
rangeLen = input.int(10, title="Range Confirmation Length")
use_volume_filter = input.bool(false, title="Use Volume Filter?")
volume_mult = input.float(1.5, title="Volume Multiplier (vs. SMA)")
use_body_filter = input.bool(true, title="Use Candle Body Size Filter?")
body_min_mult = input.float(0.7, title="Min Body Size (x ATR)")

use_trailing_stop = input.bool(false, title="Use Trailing Stop?")
trail_atr_mult = input.float(1.5, title="Trailing ATR Multiplier")

use_time_exit = input.bool(false, title="Use Time-Based Exit?")
time_exit_bars = input.int(20, title="Exit After N Bars")

trade_direction = input.string("Both", title="Trade Direction", options=["Long Only", "Short Only", "Both"])
stop_loss_atr_mult = input.float(1.5, title="Stop Loss ATR Multiplier")
take_profit_atr_mult = input.float(3.0, title="Take Profit ATR Multiplier")

// === Indicators and Calculations ===
atr = ta.atr(14)
body_size = math.abs(close - open)
body_condition = not use_body_filter or body_size > (atr * body_min_mult)
breakoutHigh = ta.highest(high, breakoutLen)
breakoutLow = ta.lowest(low, breakoutLen)
rangeHigh = ta.highest(high, rangeLen)
rangeLow = ta.lowest(low, rangeLen)
smaVolume = ta.sma(volume, breakoutLen)
volume_condition = not use_volume_filter or volume > (smaVolume * volume_mult)

// === Long Entry ===
long_breakout = close > breakoutHigh[1]
range_confirm_long = close > rangeHigh[1]
allow_long = trade_direction == "Long Only" or trade_direction == "Both"

if long_breakout and range_confirm_long and volume_condition and body_condition and allow_long
    if position_type == "Percent of Equity"
        strategy.entry("Long", strategy.long, qty=qty_percent)
    else
        strategy.entry("Long", strategy.long, qty=order_size)

    if use_trailing_stop
        strategy.exit("Exit Long", from_entry="Long", trail_points=trail_atr_mult * atr, trail_offset=trail_atr_mult * atr)
    else
        strategy.exit("Exit Long", from_entry="Long", stop=close - stop_loss_atr_mult * atr, limit=close + take_profit_atr_mult * atr)

// === Short Entry ===
short_breakout = close < breakoutLow[1]
range_confirm_short = close < rangeLow[1]
allow_short = trade_direction == "Short Only" or trade_direction == "Both"

if short_breakout and range_confirm_short and volume_condition and body_condition and allow_short
    if position_type == "Percent of Equity"
        strategy.entry("Short", strategy.short, qty=qty_percent)
    else
        strategy.entry("Short", strategy.short, qty=order_size)

    if use_trailing_stop
        strategy.exit("Exit Short", from_entry="Short", trail_points=trail_atr_mult * atr, trail_offset=trail_atr_mult * atr)
    else
        strategy.exit("Exit Short", from_entry="Short", stop=close + stop_loss_atr_mult * atr, limit=close - take_profit_atr_mult * atr)

// === Time-based Exit ===
if use_time_exit
    strategy.close("Long", when=strategy.opentrades > 0 and strategy.opentrades.entry_bar_index(0) + time_exit_bars <= bar_index)
    strategy.close("Short", when=strategy.opentrades > 0 and strategy.opentrades.entry_bar_index(0) + time_exit_bars <= bar_index)

// === Plot Breakout and Range Levels ===
plot(breakoutHigh, title="Breakout High", color=color.green)
plot(breakoutLow, title="Breakout Low", color=color.red)
plot(rangeHigh, title="Range High", color=color.teal)
plot(rangeLow, title="Range Low", color=color.orange)