o j5C@sldZddlmZmZddlmZddlmZmZm Z m Z m Z m Z m Z dZdZdZdZd Zd Zd Zd Zd Zeee e hZe e hZddddddddddddddd ZddddddZddd dd!Zdddd"Zdddddd#ZeGd$d%d%Zd&d'Zd(d)Z d*d+Z!d,d-Z"d.d/Z#d0d1Z$d2d3Z%d4d5Z&d6d7hZ'd8d9hZ(d:d;Z) < e+d?e,d@e,dAef dBdCZ-dboolEXTENSION_NORMALr?rlistrArBrNrN5/var/www/screener/premarket/fast_prediction_engine.pyr3]s             r3cC<|durdS|duo |dk}|dkr|rdSdS|rdSdS)aslope_5m = price slope over the last 5 one-minute bars (already computed by stream_features.py -- see this module's docstring). momentum_acceleration = stream_features' cross-horizon waterfall (shorter horizons moving further/faster than the 5m one = building).Nr+rr'r)r(r*rN)slope_5mmomentum_acceleration acceleratingrNrNrO_momentum_stateps   rTcCs<|dus|dur dS||dkr|dkrdS|dkrdSdS)Nr+rg?r,ffffff?r.r-rN)volume_accelerationrelative_volumecfgrNrNrO _volume_state}srYcCs|tvrdS|tvr dSdS)Nr/r0r+) _VWAP_BULLISH _VWAP_BEARISH)classificationrNrNrO _vwap_states r]cCrP)a1Three-tier EMA relationship per explicit instruction: price-vs-EMA9 is the immediate state, EMA9's own slope is the momentum tell (price can still read above a FLATTENING EMA9 -- that's "Fading", the exact "still green but momentum already deteriorating" case this whole review was about).Nr+rr/r2r1r0rN)price_vs_ema9_pct ema9_sloperisingrNrNrO _ema9_states   rac Cs|r tdd|Dddnd}|pid}|pid}d|fd|fd |ffD]\}}|r?||kr?|||||d fSq(d S) aKNearest active resistance ABOVE current price, checked in the explicit priority order: session high -> premarket high -> previous-day high. previous-day high is optional/fault-tolerant -- its absence just removes it from consideration, never blocks anything (see premarket_scanner.py's prev_day_high_out docstring).cs|]}|dVqdShNrN.0brNrNrO z&_resistance_context..N)defaultpm_high prev_day_high session_highpremarket_highprevious_day_highY@)NNN)maxget) pricebarspremarket_resultrXrmrkrlnamelevelrNrNrO_resistance_contexts  rxcCsf|dus|sdSt|dt|}tdd|| dD}||dk}|duo.||dk}|o2|S)NFr!csrbrcrNrerNrNrOrhriz(_breakout_in_progress..g+?r")minlenrq)rtr;rVrXlookback recent_hightouched_or_throughrSrNrNrO_breakout_in_progresss  r~cCsX|durtS|r tS|r|d|nd}||dks |r"||kr"tS||dkr*tStS)Nrrr )rJRESISTANCE_CAUTIONRESISTANCE_TOO_CLOSE) distance_pctatr_pctr>rX atr_safe_pctrNrNrO_resistance_classifications rcCsNdd||fD}|s tSt|}||dkr|dvrtS||dkr%tStS)NcSsg|] }|durt|qS)N)abs)rfxrNrNrO sz-_extension_classification..r$)r)r*r#)rLrqEXTENSION_SEVEREEXTENSION_EXTENDED)price_vs_ema9_atrvwap_distance_atrrrXdistsmax_distrNrNrO_extension_classifications rr'r)r,r-c Cs|d}t||dt||dt||dt||d}|tkr-|d9}n|tkr5|d9}|tkr>|d9}n|tkrF|d9}t|d d } |t voX|t voX|t k} | rat } | | fS|d krkt } | | fSt} | | fS) aJdirection is now a strict unanimity gate over momentum, volume, and resistance only (see comment above -- vwap/ema9 intentionally excluded as of 2026-09-14). confidence stays a blended score over all four inputs, purely descriptive (shortlist ranking/logging only, see module docstring), unaffected by this change.r%rrrrrg333333?rUrpgffffff?)_MOMENTUM_SCORE _VOLUME_SCORE _VWAP_SCORE _EMA_SCORErrrrround_POSITIVE_MOMENTUM_POSITIVE_VOLUMErJ DIRECTION_UPDIRECTION_DOWNrH) rr7r8r9r?r:rXwrawr6 all_positiver5rNrNrO_direction_and_confidences8  rNr4rtrurXreturncCs*it|p tdi}t|ddrt|ddgdSt||dkr5t|dd|dd t|gdS|jd }|jd }t||}|j d }|j d } t || |} |j d} t | } |j d} |j d}t| |}|jd}t||||\}}}t||||}|jd}t||||}|j d}|jd}t||||}t|| | ||||\}}|durdn|d|dd|dd|rdnd}d|d|dd | d!| d"|dd#| d| dd$|d%| d&|dd'|d|dd(|g}t||||| | ||||||||d)S)*a features: a stream_features.StreamFeatures reading, computed by the caller exactly as prediction_pipeline.py already does today -- this module doesn't change how features are computed, only how they're read. bars: the same live 1-min bars features was computed from (needed here directly for session_high / breakout-in-progress). premarket_result: this symbol's premarket_scanner.py result dict (pm_high/pm_low/prev_day_high) -- optional, fail-soft throughout. fast_predictionrBFTz*insufficient upstream stream_features data)r4rBrArzwarm-up: need z one-minute bars, have rQrRrVrWr\r^r_lastrrrNzclear overheadz $z.2fz (z% away)z2 [breakout in progress -- distance penalty waived]z momentum=z (slope_5m=)zvolume=z (rvol=z, accel=zvwap=zema9=z (price_vs_ema9=z%, ema9_slope=z resistance=z extension=)r4r5r6rr7r8r9r:r;r<r=r>r?rA)DEFAULT_CONFIGrrrgetattrr3rzslope accelerationrTrrYrr]rrarsrxr~ volatilityrrr)r4featuresrtrurXrQrRrrVrWr7vwap_classificationr8r^r_r9rsres_name res_levelres_distance_pctbreakoutrresistance_staterrextension_stater5r6resistance_descrArNrNrOcompute_fast_prediction%sf                  r)NN).__doc__ dataclassesrr config_loaderrstream_featuresrrrrr r r rrrHrJrrrLrrrZr[rrrrrr3rTrYr]rarxr~rrrrrrFrMdictrrNrNrNrOsd$ $       9#