Q-j^ ddlmZddlmZmZddlmZmZmZddl m Z ddl m Z m Z mZmZddlmZ dded ed ed ed ed ededededefdZy))nan) DataFrameSeries)DictLikeIntIntFloat)hl2)v_mamodev_offset v_pos_defaultv_series)atrNhighlowcloselength atr_length multiplier atr_mamodeoffsetkwargsreturnc t|d}t||}t||dz}t||dz}t||dz}|||yt|d}t|d}t|}|j} dg| zdg| z} } t g| zt g| z} } t ||}|t|||||z}||z }||z}td| D](}|j||j|dz kDrd| |<n|j||j|dz krd| |<n| |dz | |<| |dkDrA|j||j|dz kr|j|dz |j|<| |dkrA|j||j|dz kDr|j|dz |j|<| |dkDr|j|x| |<| |<|j|x| |<| |<+t | d<t g|z| d|d |d |}d || d || d || d || i}t||j}d ||_ d|_ |dk7r|j|}d|vr|j|dd|S)aSupertrend This indicator attempts to identify trend direction as well as support and resistance levels. Sources: * [freebsensetips](http://www.freebsensetips.com/blog/detail/7/What-is-supertrend-indicator-its-calculation) Parameters: high (Series): ```high``` Series low (Series): ```low``` Series close (Series): ```close``` Series length (int): The period. Default: ```7``` atr_length (int): ATR period. Default: ```length``` multiplier (float): Coefficient for upper and lower band distance to midrange. Default: ```3.0``` atr_mamode (str) : MA type to be used for ATR calculation. See ```help(ta.ma)```. Default: ```"rma"``` offset (int): Post shift. Default: ```0``` Other Parameters: fillna (value): ```pd.DataFrame.fillna(value)``` Returns: (DataFrame): 4 columns Ng@rmar)mamode_SUPERTSUPERTdSUPERTlSUPERTs)indexoverlapfillnaT)inplace)r r r r sizerr rrangeiatrr$namecategoryshiftr&)rrrrrrrrrmdir_trendlongshorthl2_matrlbubi_propsdatadfs j/Users/jameslopez/projects/TradingBot25/.venv/lib/python3.12/site-packages/pandas_ta/overlap/supertrend.py supertrendr< sD61 %Fz62J D&1* %D 3 #C UFQJ 'E |s{emz3/J*e,J f F  A#'A37%D%!)cUQY%D tS>D D#ujL LD B B 1a[ 99Q<"&&Q- 'DG YYq\BFF1q5M )DG1q5kDGAw{rvvay266!a%=8FF1q5Mq Aw{rvvay266!a%=8FF1q5Mq 7Q;!# *E!HtAw"$&&) +E!HuQx"E!HEFND&M* &F 5 &D &D &E  D 4u{{ +BvhBGBK{ XXf 6 &"D 1 I)NNNNN)numpyrpandasrrpandas_ta._typingrrrpandas_ta.overlapr pandas_ta.utilsr r r r pandas_ta.volatilityrstrr<r=r;rFs$55!GG$ +/ a aa&,a a$'aa a  a #+ a  ar=