first commit

evaluation.py

@@ -0,0 +1,39 @@
+import json
+from pathlib import Path
+
+from constellaration import problems
+from constellaration.geometry import surface_rz_fourier
+
+PROBLEM_TYPES = ["geometrical", "simple_to_build", "mhd_stable"]
+
+
+def load_boundary(data: str) -> surface_rz_fourier.SurfaceRZFourier:
+    return surface_rz_fourier.SurfaceRZFourier.model_validate_json(data)
+
+
+def load_boubdaries(data: str) -> list[surface_rz_fourier.SurfaceRZFourier]:
+    data_json = json.loads(data)
+    return [
+        surface_rz_fourier.SurfaceRZFourier.model_validate_json(b) for b in data_json
+    ]
+
+
+def evaluate_problem(
+    problem_type: str, input_file: str
+) -> problems.EvaluationSingleObjective | problems.EvaluationMultiObjective:
+    with Path(input_file).open("r") as f:
+        data = f.read()
+
+    match problem_type:
+        case "geometrical":
+            boundary = load_boundary(data)
+            result = problems.GeometricalProblem().evaluate(boundary)
+        case "simple_to_build":
+            boundary = load_boundary(data)
+            result = problems.SimpleToBuildQIStellarator().evaluate(boundary)
+        case "mhd_stable":
+            boundaries = load_boubdaries(data)
+            result = problems.MHDStableQIStellarator().evaluate(boundaries)
+        case _:
+            raise ValueError(f"Unknown problem type: {problem_type}")
+    return result