REX indicator

-- Author of haasscript version: Bunka
-- REX indicator

-- Best used as an exit indicator at default settings.
-- Also doing well in entering positions by using TEMA(14) for REX and SMA(14) as signal.

local o = OpenPrices()
local h = HighPrices()
local l = LowPrices()
local c = ClosePrices()
local rex_ma_type = InputOptions("REX MA type", "SMA", {"EMA","SMA", "Tenkan","TEMA","TRIMA"})
local smoothing = Input("Smoothing length", 14)

local sig_ma_type = InputOptions("Signal MA type", "SMA", {"EMA","SMA", "Tenkan","TEMA","TRIMA"})
local smoothing_sig = Input("Signal smoothing length",14)

--tenkan
local function tenkan_f(sig, len)
    local tenkan_x = 0.5 * (GetHighs(sig, len) + GetLows(sig, len))
    return tenkan_x
end

-- triple ema
local function tema_f(src, len)
    local v2 = EMA(src, len)
    local v = 3 * (v2 - EMA(v2, len)) + EMA(EMA(v2, len), len)
    return v
end

local function ma_f(t, sig, len)
    local sss = {}
    if t == "SMA" then
        sss = SMA(sig, len)
    elseif t == "EMA" then
        sss = EMA(sig, len)
    elseif t == "TEMA" then
        sss = tema_f(sig, len)
    elseif t == "Tenkan" then
        sss = tenkan_f(sig, len)
    elseif t == "TRIMA" then
        sss = TRIMA(sig, len)
    end
    return sss
end

local tvb = (3*c) - o - h - l

local rex = ma_f(rex_ma_type, tvb, smoothing)
local signal = ma_f(sig_ma_type, rex, smoothing_sig)

Plot(1,"rex",rex, Cyan)
Plot(1,"signal",signal, Orange)