# SPDX-License-Identifier: LicenseRef-PolyForm-Noncommercial-1.0.0 (see LICENSE)
"""Compare what the indicator drew (DumpIndicator CSVs on MT5, Dump4 CSVs on MT4) with tests/reference.py, bar by bar.

    python3 tests/check_parity.py <dump dir>          (needs numpy and pandas; looks in <dump dir>/dump)
    python3 tests/check_parity.py <dump dir> --mt4    (the MT4 edition, dumps from the MT4 tester)

Buffers: b0 bullish arrow, b1 bearish arrow, b2/b3 their pattern bit masks (1 ABCD, 2 Gartley, 4 Bat).
The indicator scans the last InpHistory closed bars (start = bars - 1 - history) and never the forming bar.
Pass: arrows on the same bars with the same pattern masks."""
import sys
from pathlib import Path

import numpy as np
import pandas as pd

sys.path.insert(0, str(Path(__file__).parent))
from reference import Params, run  # noqa: E402

FREE = (True, True, True)
HEAD = "s:Patterns|b:{a}|b:{g}|b:{b}|s:Swings and ratios|i:{l}|i:{r}|d:{t}|d:{z}|s:Display|i:{h}|b:true|b:true|b:true|c:6465062|c:4606690|s:Alerts|b:false|b:false|s:alert.wav|b:false|b:false"
CONFIGS = {  # tag: (Params, history, DumpIndicator InpParams for MT5 in input order; each group counts) - as tests/configs.tsv
    "hm_default": (Params(use=FREE), 3000, HEAD.format(a="true", g="true", b="true", l=10, r=5, t="5.0", z="3.0", h=3000)),
    "hm_all_loose": (Params(left=6, right=3, tol=0.08, prz=0.04, use=FREE), 5000,
                     HEAD.format(a="true", g="true", b="true", l=6, r=3, t="8.0", z="4.0", h=5000)),
    "hm_gartley": (Params(left=5, right=2, use=(False, True, False)), 3000,
                   HEAD.format(a="false", g="true", b="false", l=5, r=2, t="5.0", z="3.0", h=3000)),
}


def num(col):
    return pd.to_numeric(col, errors="coerce").to_numpy(float)


def check(csv: Path, pr: Params, hist: int, mt4=False):
    d = pd.read_csv(csv)
    n = len(d)
    # MT4 tester: the indicator was attached with far fewer than InpHistory bars, so it scans from bar 0 and keeps feeding
    start = 0 if mt4 else max(0, n - 1 - hist)
    h, l, c = (d[k].to_numpy(float)[start:n - 1] for k in ("high", "low", "close"))
    ev = run(h, l, c, pr)
    msgs, ok = [], True
    for b, mb, dr, name in (("b0", "b2", 1, "bull"), ("b1", "b3", -1, "bear")):
        ref = {}
        for e in ev:
            if e["dir"] == dr:
                ref[start + e["bar"]] = ref.get(start + e["bar"], 0) | (1 << e["pat"])
        arr = num(d[b])
        mt_bars = [int(i) for i in np.flatnonzero(~np.isnan(arr)) if start <= i < n - 1]
        mt = {i: int(num(d[mb])[i]) for i in mt_bars}
        same = mt == ref
        ok &= same
        msgs.append(f"{name} {len(ref):>3} {'OK' if same else f'DIFFER (mt5 {len(mt)})'}")
        if not same:
            diff = sorted(set(mt.items()) ^ set(ref.items()))[:5]
            msgs.append(f"[first differences {diff}]")
    return ok, f"{csv.stem:38} bars {n:>6} scanned {n - 1 - start:>5}  " + "  ".join(msgs)


if __name__ == "__main__":
    mt4 = "--mt4" in sys.argv          # dumps of the MT4 Strategy Tester run
    root = Path([a for a in sys.argv[1:] if a != "--mt4"][0])
    allok, k = True, 0
    for tag, (pr, hist, _) in CONFIGS.items():
        for f in sorted(root.glob(f"dump/{tag}_*.csv")):
            ok, msg = check(f, pr, hist, mt4)
            allok &= ok
            k += 1
            print(("PASS " if ok else "FAIL ") + msg)
    print(f"\n{'ALL PASS' if allok and k else 'FAILURES'} ({k} files)")
    sys.exit(0 if allok and k else 1)