def check_impulse_rules(self, waves: List[Dict]) -> bool: """ Validate Elliott Wave impulse pattern (5 waves: 1,2,3,4,5). Rules: 1. Wave 2 cannot retrace more than 100% of Wave 1. 2. Wave 3 is never the shortest (in magnitude). 3. Wave 4 does not overlap Wave 1 (in price). 4. Wave 3 often shows 1.618 extension of Wave 1. """ if len(waves) < 5: return False
if impulse_ok: pattern_type = 'impulse_5wave' elif corrective_ok: pattern_type = 'corrective_abc' else: pattern_type = 'unclear'
# Rule 3: Wave 4 price overlap with Wave 1? # For uptrend impulse: w1 up, w2 down, w3 up, w4 down, w5 up # Overlap means low of w4 < high of w1 if w1['direction'] == 'up': wave1_high = max(w1['start_price'], w1['end_price']) wave4_low = min(w4['start_price'], w4['end_price']) if wave4_low <= wave1_high: return False else: # downtrend impulse wave1_low = min(w1['start_price'], w1['end_price']) wave4_high = max(w4['start_price'], w4['end_price']) if wave4_high >= wave1_low: return False
detector = ElliottWaveDetector(swing_window=8) result = detector.detect_elliott_waves(price_series)
return { 'pattern': pattern_type, 'waves': waves, 'valid': impulse_ok or corrective_ok, 'fibonacci_levels': fibs, 'swing_points': swings_df } Example usage & visualization ------------------------------- if name == " main ": import matplotlib.pyplot as plt
def detect_elliott_waves(self, prices: np.ndarray) -> Dict: """ Main function: returns detected wave structure and validation. """ swings_df = self.find_swing_points(prices) waves = self.label_swing_waves(swings_df)
impulse_ok = self.check_impulse_rules(waves) corrective_ok = self.check_corrective_rules(waves)
