LEGSTECH Optimizer
d5a8973c05
NEW FILES: ea/__init__.py - EA module package ea/schema.py - ParameterDef + ParameterSchema (replaces YAML manifest) ea/set_parser.py - Parse ANY MT5 .set file → ParameterSchema ea/registry.py - EAProfile + EARegistry (ea_registry.yaml) ea_registry.yaml - LEGSTECH_EA_V2 pre-registered with automation overrides MODIFIED: mt5/ini_builder.py - Accepts ParameterSchema OR legacy manifest_path mt5/runner.py - Accepts optional EAProfile for EA-agnostic validation config.yaml - Removed hardcoded ea: block; added ea_registry path KEY DESIGN: - .set file IS the manifest: value|min|max|step auto-detects float/int/bool/enum/fixed - automation_overrides in EAProfile forces headless-safe values (InpShowPanel=0) - Phase A optimization INI: Optimization=2 (genetic), ranges from schema - Phase B single backtest: Optimization=0 (unchanged behavior) - LEGSTECH advanced mode: fully backwards compatible via legacy manifest fallback - runner.py still works for legacy callers (no EAProfile needed) TESTED: - 49/49 LEGSTECH params parsed correctly - InpShowPanel=0 automation override applied - INI output correct for both Phase A (Optimization=2) and Phase B (Optimization=0) - All imports pass
279 lines
9.5 KiB
Python
279 lines
9.5 KiB
Python
"""
|
||
ea/set_parser.py
|
||
Parse any MT5 .set file into a ParameterSchema.
|
||
|
||
Handles both .set formats:
|
||
value|min|max|step (single pipe — most common)
|
||
value||min||max||step||Y/N (double pipe — some MT5 builds)
|
||
|
||
Fixed detection:
|
||
min == max → type="fixed" (e.g. InpMagicNumber=202402|202402|202402|1)
|
||
min == 0 AND max == 0 → type="fixed" (zeroed range = "don't optimize")
|
||
No range at all → type="fixed"
|
||
|
||
Type detection (non-fixed only):
|
||
min==0, max==1, step==1 → bool
|
||
"." in step string → float
|
||
max - min <= 8, step==1 → enum (small discrete integer set)
|
||
otherwise → int
|
||
"""
|
||
from __future__ import annotations
|
||
|
||
import re
|
||
from pathlib import Path
|
||
from typing import Any, Optional
|
||
|
||
from loguru import logger
|
||
|
||
from ea.schema import ParameterDef, ParameterSchema
|
||
|
||
|
||
# ── Tester section keys to skip (not EA inputs) ───────────────────────────────
|
||
|
||
_TESTER_KEYS = {
|
||
"expert", "symbol", "period", "optimization", "model",
|
||
"fromdate", "todate", "forwardmode", "report", "replacereport",
|
||
"shutdownterminal", "deposit", "currency", "leverage",
|
||
"optimizationmode", "forwarddate", "optimizationiterations",
|
||
}
|
||
|
||
# EA params that should always be fixed even if they have a range
|
||
_FORCE_FIXED_PATTERNS = [
|
||
"testermode", "testeri", "testerinit", "showpanel",
|
||
"magicnumber", "magic",
|
||
]
|
||
|
||
|
||
class SetParser:
|
||
"""
|
||
Parses a MT5 .set file into a ParameterSchema.
|
||
|
||
Usage:
|
||
parser = SetParser()
|
||
schema = parser.parse(
|
||
path=Path("C:/MT5 Set files/LEGSTECH_EA_V2.set"),
|
||
ea_name="LEGSTECH_EA_V2",
|
||
default_optimize=False, # user chooses via UI
|
||
)
|
||
"""
|
||
|
||
def parse(
|
||
self,
|
||
path: Path,
|
||
ea_name: str,
|
||
default_optimize: bool = False,
|
||
force_optimize: Optional[set[str]] = None,
|
||
force_fixed: Optional[set[str]] = None,
|
||
) -> ParameterSchema:
|
||
"""
|
||
Parse a .set file and return a ParameterSchema.
|
||
|
||
Args:
|
||
path: Path to the .set file.
|
||
ea_name: Display name for the EA.
|
||
default_optimize: Whether to mark all optimizable params as optimize=True by default.
|
||
If False (default), the user selects via UI.
|
||
force_optimize: Set of param names that are always optimize=True regardless.
|
||
force_fixed: Set of param names that are always type="fixed".
|
||
"""
|
||
force_optimize = force_optimize or set()
|
||
force_fixed = force_fixed or set()
|
||
|
||
path = Path(path)
|
||
if not path.exists():
|
||
raise FileNotFoundError(f".set file not found: {path}")
|
||
|
||
try:
|
||
text = path.read_text(encoding="utf-16")
|
||
except UnicodeError:
|
||
text = path.read_text(encoding="utf-8", errors="replace")
|
||
|
||
parameters: dict[str, ParameterDef] = {}
|
||
current_section = ""
|
||
|
||
for raw_line in text.splitlines():
|
||
line = raw_line.strip()
|
||
if not line or line.startswith(";"):
|
||
continue
|
||
|
||
# Section header
|
||
if line.startswith("[") and line.endswith("]"):
|
||
current_section = line[1:-1].lower()
|
||
continue
|
||
|
||
# Skip lines without "="
|
||
if "=" not in line:
|
||
continue
|
||
|
||
name, _, rest = line.partition("=")
|
||
name = name.strip()
|
||
rest = rest.strip()
|
||
|
||
# Skip tester-section metadata keys
|
||
if name.lower() in _TESTER_KEYS:
|
||
continue
|
||
|
||
# Parse the value + optional range
|
||
param = self._parse_param(name, rest)
|
||
if param is None:
|
||
logger.debug(f"SetParser: skipped unrecognised line: {line!r}")
|
||
continue
|
||
|
||
# Apply force-fixed overrides
|
||
if name in force_fixed or self._is_force_fixed(name):
|
||
param.type = "fixed"
|
||
param.optimize = False
|
||
elif name in force_optimize:
|
||
param.optimize = True
|
||
elif default_optimize and param.type != "fixed":
|
||
param.optimize = True
|
||
|
||
parameters[name] = param
|
||
|
||
if not parameters:
|
||
raise ValueError(f"No EA input parameters found in .set file: {path}")
|
||
|
||
schema = ParameterSchema(ea_name=ea_name, source_set=path, parameters=parameters)
|
||
logger.info(f"SetParser: parsed {schema.summary()} from {path.name}")
|
||
return schema
|
||
|
||
# ── Internal ─────────────────────────────────────────────────────────────
|
||
|
||
def _parse_param(self, name: str, rest: str) -> Optional[ParameterDef]:
|
||
"""
|
||
Parse a single parameter line.
|
||
rest is everything after the first "=" on the line.
|
||
"""
|
||
# Normalise: double-pipe "||" → single "|"
|
||
rest = re.sub(r"\|\|", "|", rest)
|
||
|
||
# Strip trailing Y/N optimize flag if present
|
||
yn_match = re.search(r"\|([YN])$", rest, re.IGNORECASE)
|
||
if yn_match:
|
||
rest = rest[:yn_match.start()]
|
||
|
||
parts = [p.strip() for p in rest.split("|")]
|
||
|
||
if len(parts) == 1:
|
||
# No range info → fixed
|
||
value = self._cast_value(parts[0])
|
||
return ParameterDef(
|
||
name=name, default=value, type="fixed",
|
||
min=None, max=None, step=None, optimize=False,
|
||
)
|
||
|
||
if len(parts) < 4:
|
||
# Incomplete range — treat as fixed
|
||
value = self._cast_value(parts[0])
|
||
return ParameterDef(
|
||
name=name, default=value, type="fixed",
|
||
min=None, max=None, step=None, optimize=False,
|
||
)
|
||
|
||
raw_val, raw_min, raw_max, raw_step = parts[0], parts[1], parts[2], parts[3]
|
||
|
||
try:
|
||
default_f = float(raw_val)
|
||
min_f = float(raw_min)
|
||
max_f = float(raw_max)
|
||
step_f = float(raw_step)
|
||
except ValueError:
|
||
value = self._cast_value(raw_val)
|
||
return ParameterDef(
|
||
name=name, default=value, type="fixed",
|
||
min=None, max=None, step=None, optimize=False,
|
||
)
|
||
|
||
# Fixed detection
|
||
is_fixed = (
|
||
abs(min_f - max_f) < 1e-9 # min == max
|
||
or (abs(min_f) < 1e-9 and abs(max_f) < 1e-9) # both zero (zeroed range)
|
||
)
|
||
if is_fixed:
|
||
return ParameterDef(
|
||
name=name,
|
||
default=self._typed_default(raw_val, raw_step),
|
||
type="fixed",
|
||
min=min_f, max=max_f, step=step_f,
|
||
optimize=False,
|
||
)
|
||
|
||
# Type detection
|
||
ptype = self._detect_type(min_f, max_f, step_f, raw_step, raw_val)
|
||
default = self._typed_cast(ptype, default_f, raw_val)
|
||
|
||
return ParameterDef(
|
||
name=name,
|
||
default=default,
|
||
type=ptype,
|
||
min=min_f,
|
||
max=max_f,
|
||
step=step_f,
|
||
optimize=False, # user sets this via UI; can be overridden by caller
|
||
)
|
||
|
||
@staticmethod
|
||
def _detect_type(min_f: float, max_f: float, step_f: float,
|
||
raw_step: str, raw_val: str) -> str:
|
||
"""Infer parameter type from its range."""
|
||
# Bool: exactly 0–1 with step 1
|
||
if abs(min_f) < 1e-9 and abs(max_f - 1.0) < 1e-9 and abs(step_f - 1.0) < 1e-9:
|
||
return "bool"
|
||
|
||
# Float: step has decimal component
|
||
if "." in raw_step and not raw_step.endswith(".0") and float(raw_step) % 1 != 0:
|
||
return "float"
|
||
# Also float if default value has meaningful decimal
|
||
if "." in raw_val and float(raw_val) % 1 != 0:
|
||
return "float"
|
||
|
||
# Enum: small integer set (≤ 8 distinct values, step 1)
|
||
n_values = int(round((max_f - min_f) / step_f)) + 1 if step_f > 0 else 1
|
||
if abs(step_f - 1.0) < 1e-9 and n_values <= 8:
|
||
return "enum"
|
||
|
||
return "int"
|
||
|
||
@staticmethod
|
||
def _typed_cast(ptype: str, value_f: float, raw: str) -> Any:
|
||
if ptype == "bool":
|
||
return value_f != 0 or raw.lower() in ("true", "1")
|
||
if ptype == "int":
|
||
return int(round(value_f))
|
||
if ptype == "enum":
|
||
return int(round(value_f))
|
||
return value_f # float
|
||
|
||
@staticmethod
|
||
def _typed_default(raw: str, raw_step: str) -> Any:
|
||
"""Cast a fixed-param value without range context."""
|
||
lower = raw.lower()
|
||
if lower in ("true", "false"):
|
||
return lower == "true"
|
||
try:
|
||
f = float(raw)
|
||
# Return int if it's a whole number and step is integer-like
|
||
if "." not in raw_step or raw_step.endswith(".0"):
|
||
if f == int(f):
|
||
return int(f)
|
||
return f
|
||
except ValueError:
|
||
return raw
|
||
|
||
@staticmethod
|
||
def _cast_value(raw: str) -> Any:
|
||
lower = raw.lower()
|
||
if lower in ("true", "false"):
|
||
return lower == "true"
|
||
try:
|
||
f = float(raw)
|
||
return int(f) if f == int(f) and "." not in raw else f
|
||
except ValueError:
|
||
return raw
|
||
|
||
@staticmethod
|
||
def _is_force_fixed(name: str) -> bool:
|
||
"""Return True for params that are always fixed regardless of their range."""
|
||
lower = name.lower()
|
||
return any(pat in lower for pat in _FORCE_FIXED_PATTERNS)
|