Q-j:ddlmZddlmZmZmZddlmZmZddl m Z m Z m Z ddl mZddlmZddlmZddlmZmZmZmZmZmZed d Z dd ededededededede de de dedede de fdZy ))njit)isnannan zeros_like) DataFrameSeries)DictLikeIntIntFloat)rsi)atr)mfi)v_mamodev_offset v_pos_defaultv_seriesv_strv_talibT)cachec|j}t|}td|D]L}||r#||||dz kr ||dz ||<"||||<+||||dz kDr ||dz ||<E||||<Nt|d<|S)Nr)sizerranger)low_atrhigh_atrmomo_thresholdmresultis h/Users/jameslopez/projects/TradingBot25/.venv/lib/python3.12/site-packages/pandas_ta/trend/alphatrend.pynb_alphar!sA  F 1a[ ! qzF1q5M)"1q5Mq #AJq {VAE]*"1q5Mq $QKq F1I MNopen_highlowclosevolumesrclength multiplier thresholdlagmamodetaliboffsetkwargsc  &t|d}t||}t||}t||}t||}||||y||||d}t|d}||jvr|nd}t|d}t|d}t| d} t | d} t | }t | } |t|}|yt||||| | }|tt|ry|||zz }|||zz}d}|t|||| | }nt|||||| }|y|j|j}}t|||j|k\}t||j }|j!| }tt|stt|ry| d k7r"|j!| }|j!| }d| vr,|j#| dd|j#| ddd|d|d|}d||_d|d| |_dx|_|_|j$||j$|i}t)||j }|j$|_|j&|_|S)aAlpha Trend This indicator attempts to filter sideways movement for accurate signals. Sources: * [OnlyFibonacci](https://github.com/OnlyFibonacci/AlgoSeyri/blob/main/alphaTrendIndicator.py) * [tradingview](https://www.tradingview.com/script/o50NYLAZ-AlphaTrend/) Parameters: open_ (Series): ```open``` Series high (Series): ```high``` Series low (Series): ```low``` Series close (Series): ```close``` Series volume (Series): ```volume``` Series. Default: ```None``` src (str): One of: "open", "high", "low" or "close". Default: ```"close"``` length (int): ATR, MFI, or RSI period. Default: ```14``` multiplier (float): Trailing ATR multiple. Default: ```1``` threshold (float): Momentum threshold. Default: ```50``` lag (int): Lag period of main trend. Default: ```2``` mamode (str): See ```help(ta.ma)```. Default: ```"sma"``` talib (bool): If installed, use TA Lib. Default: ```True``` offset (int): Post shift. Default: ```0``` Other Parameters: fillna (value): ```pd.DataFrame.fillna(value)``` Returns: (DataFrame): 2 columns N)openr$r%r&r&r2sma)r$r%r&r)r-r.)r&r)r-r.)r$r%r&r'r)r.)indexrfillnaT)inplace_ALPHATALPHATltrend)rrrkeysrrrr allrr rto_numpyr!rr7shiftr8namecategoryr)r#r$r%r&r'r(r)r*r+r,r-r.r/r0_srcmode_talatr_ lower_atr upper_atrmomo np_upper_atr np_lower_atratatl_propsdatadfs r alphatrendrQ)sN62 &F UF #E D& !D 3 C UF #E }  u}4e DD W C ##Cz1-Ji,I Q C fe $Fu~H f F &! >   s%X D  |s5;'dZ''Itj((I D ~c6&Q3eF   |!*!3!3!5y7I7I7K,L , dmmo.J KB %++ &B ((3-C 59~U3Z{ XXf ii6 &"D 1 6(#T 2*Qyk 2FvhBG#'CH!((BK#, GGR3 'D 4u{{ +BggBG++BK Ir") NNNNNNNNN)numbarnumpyrrrpandasrrpandas_ta._typingr r r pandas_ta.momentumr pandas_ta.volatilityr pandas_ta.volume.mfirpandas_ta.utilsrrrrrrr!strintboolrQr"r r^s(($55"$$D,'+/3+/&* x xx&,x5;x x #x x%-x x&) x  x $ x  x #+ xr"