Explicit validator for IMAS core_profiles IDS.

PiperOrigin-RevId: 846714287

Author: Nush395

torax/_src/imas_tools/input/core_profiles.py

@@ -12,14 +12,49 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 """Useful functions to load IMAS core_profiles or plasma_profiles IDSs."""
+
 from collections.abc import Collection, Mapping
+import logging
 from typing import Any
 
+from imas import ids_structure
 from imas import ids_toplevel
 import numpy as np
 from torax._src import constants
 
 
+def _validate_profiles_1d(profiles_1d: ids_structure.IDSStructure):
+  grid = profiles_1d.grid
+  if not grid.rho_tor_norm.has_value:
+    raise ValueError("The IDS is missing the grid.rho_tor_norm quantity.")
+  if not grid.psi.has_value:
+    logging.warning("The IDS is missing the grid.psi quantity.")
+
+  if not profiles_1d.time.has_value:
+    logging.warning("The IDS is missing the time quantity.")
+
+  electrons_ids = profiles_1d.electrons
+  if not electrons_ids.temperature.has_value:
+    logging.warning("The IDS is missing the electrons.temperature quantity.")
+  if not electrons_ids.density.has_value:
+    logging.warning("The IDS is missing the electrons.density quantity.")
+
+  if not profiles_1d.t_i_average.has_value:
+    logging.warning("The IDS is missing the t_i_average quantity.")
+
+
+def _validate_profiles_conditions(ids: ids_toplevel.IDSToplevel):
+  """Validates the profiles_conditions IDS."""
+  for profiles_1d in ids.profiles_1d:
+    _validate_profiles_1d(profiles_1d)
+
+  global_quantities = ids.global_quantities
+  if not global_quantities.v_loop.has_value:
+    logging.warning("The IDS is missing the v_loop quantity.")
+  if not global_quantities.ip.has_value:
+    logging.warning("The IDS is missing the ip quantity.")
+
+
 # pylint: disable=invalid-name
 def profile_conditions_from_IMAS(
     ids: ids_toplevel.IDSToplevel,
@@ -38,6 +73,7 @@ def profile_conditions_from_IMAS(
     The updated fields read from the IDS that can be used to completely or
     partially fill the `profile_conditions` section of a TORAX `CONFIG`.
   """
+  _validate_profiles_conditions(ids)
   profiles_1d, rhon_array, time_array = _get_time_and_radial_arrays(
       ids, t_initial
   )