luminarycloud 0.15.4__py3-none-any.whl → 0.16.0__py3-none-any.whl

luminarycloud/mesh.py

@@ -44,10 +44,17 @@ class Mesh(ProtoWrapperBase):
         """
         return lc.get_project(ProjectID(self._proto.project_id))
 
-    def geometry_version(self) -> GeometryVersion:
+    def geometry_version(self) -> GeometryVersion | None:
         """
         Get the geometry version associated with this mesh.
+
+        Returns
+        -------
+        GeometryVersion | None
+            The geometry version associated with this mesh, or None if the mesh has no geometry version.
         """
+        if self._proto.geometry_version_id == "":
+            return None
         return get_geometry_version(self._proto.geometry_version_id)
 
     def update(

luminarycloud/params/simulation/physics/fluid/solution_controls/fluid_relaxation_method/fluid_implicit_relaxation/robust_startup/__init__.py

@@ -1,2 +1,3 @@
+from .robust_startup_auto_ import RobustStartupAuto
 from .robust_startup_off_ import RobustStartupOff
 from .robust_startup_on_ import RobustStartupOn

luminarycloud/params/simulation/physics/fluid/solution_controls/fluid_relaxation_method/fluid_implicit_relaxation/robust_startup/robust_startup_auto_.py

@@ -0,0 +1,30 @@
+# Generated by generate_sdk_wrappers.py. DO NOT EDIT
+
+from abc import ABC, ABCMeta
+from dataclasses import dataclass, field
+from typing import Any
+from uuid import uuid4
+
+from google.protobuf.message import Message as _Message
+from luminarycloud.tables import RectilinearTable, _param_name_to_table_type
+from luminarycloud.types import Vector3, LcFloat
+from luminarycloud.types.adfloat import _to_ad_proto, _from_ad_proto
+from luminarycloud._helpers._entity_identifier import _create_entity_identifier
+from luminarycloud._proto.client import simulation_pb2 as clientpb
+from luminarycloud._proto.client.entity_pb2 import EntityIdentifier
+from luminarycloud._helpers import CodeRepr
+import luminarycloud.params.enum._enum_wrappers as enum
+
+from luminarycloud.params.simulation._lib import ParamGroupWrapper, create_unique_id
+
+from luminarycloud.params.simulation.physics.fluid.solution_controls.fluid_relaxation_method.fluid_implicit_relaxation.robust_startup_ import (
+    RobustStartup,
+)
+from luminarycloud.params.simulation.physics.fluid.solution_controls.fluid_relaxation_method.fluid_implicit_relaxation.robust_startup_ import *
+
+
+@dataclass(kw_only=True)
+class RobustStartupAuto(RobustStartup):
+    """Enable robust startup mode and automatically determine the settings."""
+
+    pass

luminarycloud/params/simulation/physics/fluid/solution_controls/fluid_relaxation_method/fluid_implicit_relaxation/robust_startup/robust_startup_on_.py

@@ -25,7 +25,7 @@ from luminarycloud.params.simulation.physics.fluid.solution_controls.fluid_relax
 
 @dataclass(kw_only=True)
 class RobustStartupOn(RobustStartup):
-    """Enable robust startup mode."""
+    """Enable robust startup mode and manually specify the settings."""
 
     robust_startup_initial_cfl: LcFloat = 1.0
     "Initial CFL number for robust startup mode. The CFL is ramped toward the target value during startup."

luminarycloud/params/simulation/physics/fluid/solution_controls/fluid_relaxation_method/fluid_implicit_relaxation_.py

@@ -37,6 +37,10 @@ from luminarycloud.params.simulation.physics.fluid.solution_controls.fluid_relax
     RobustStartup,
 )
 from luminarycloud.params.simulation.physics.fluid.solution_controls.fluid_relaxation_method.fluid_implicit_relaxation.robust_startup_ import *
+from luminarycloud.params.simulation.physics.fluid.solution_controls.fluid_relaxation_method.fluid_implicit_relaxation.robust_startup.robust_startup_auto_ import (
+    RobustStartupAuto,
+)
+from luminarycloud.params.simulation.physics.fluid.solution_controls.fluid_relaxation_method.fluid_implicit_relaxation.robust_startup.robust_startup_auto_ import *
 from luminarycloud.params.simulation.physics.fluid.solution_controls.fluid_relaxation_method.fluid_implicit_relaxation.robust_startup.robust_startup_off_ import (
     RobustStartupOff,
 )
@@ -54,7 +58,7 @@ class FluidImplicitRelaxation(FluidRelaxationMethod):
     implicit_method: enum.ImplicitMethod = enum.ImplicitMethod.BACKWARD_EULER
     "Scheme for implicit relaxation of the governing equations."
     linear_solver_type: LinearSolverType = field(default_factory=GaussSeidel)
-    "Type of linear solver used for implicit relaxation. Possible types: ``GaussSeidel``, ``GsAmgx``, ``KrylovAmg``, ``Amg`` from the ``linear_solver_type`` module."
+    "Type of linear solver used for implicit relaxation. Possible types: ``GaussSeidel``, ``KrylovAmg`` from the ``linear_solver_type`` module."
     jacobian_update_method: enum.JacobianUpdateMethod = (
         enum.JacobianUpdateMethod.EXPLICIT_INTERVAL_AND_WARMUP
     )
@@ -63,8 +67,8 @@ class FluidImplicitRelaxation(FluidRelaxationMethod):
     "How many iterations in between updating the Jacobian values for implicit solving."
     jacobian_warmup_threshold: int = 300
     "How many iterations to update Jacobians every iteration before switching to the specified 'Jacobian Update Interval'."
-    robust_startup: RobustStartup = field(default_factory=RobustStartupOn)
-    "Applies a robust startup process during the initial transients of a simulation. Applicable to steady problems only. Possible types: ``RobustStartupOn``, ``RobustStartupOff`` from the ``robust_startup`` module."
+    robust_startup: RobustStartup = field(default_factory=RobustStartupAuto)
+    "Applies a robust startup process during the initial transients of a simulation. Applicable to steady problems only. Possible types: ``RobustStartupOn``, ``RobustStartupAuto``, ``RobustStartupOff`` from the ``robust_startup`` module."
     relaxation_flow: LcFloat = 1.0
     "Under-relaxation factor in [0,1] applied to the mean flow solution update with each implicit nonlinear iteration. Default of 1.0."
     relaxation_turb: LcFloat = 0.5

luminarycloud/params/simulation/physics/fluid/solution_controls_fluid_.py

@@ -122,6 +122,8 @@ class SolutionControlsFluid(CodeRepr, ParamGroupWrapper[clientpb.SolutionControl
                 _proto.robust_startup_iterations.value = (
                     self.fluid_relaxation_method.robust_startup.robust_startup_iterations
                 )
+            if isinstance(self.fluid_relaxation_method.robust_startup, RobustStartupAuto):
+                _proto.robust_startup = clientpb.ROBUST_STARTUP_AUTO
             if isinstance(self.fluid_relaxation_method.robust_startup, RobustStartupOff):
                 _proto.robust_startup = clientpb.ROBUST_STARTUP_OFF
             _proto.relax_flow.CopyFrom(_to_ad_proto(self.fluid_relaxation_method.relaxation_flow))
@@ -211,6 +213,8 @@ class SolutionControlsFluid(CodeRepr, ParamGroupWrapper[clientpb.SolutionControl
                 self.fluid_relaxation_method.robust_startup.robust_startup_iterations = (
                     proto.robust_startup_iterations.value
                 )
+            elif proto.robust_startup == clientpb.ROBUST_STARTUP_AUTO:
+                self.fluid_relaxation_method.robust_startup = RobustStartupAuto()
             elif proto.robust_startup == clientpb.ROBUST_STARTUP_OFF:
                 self.fluid_relaxation_method.robust_startup = RobustStartupOff()
             self.fluid_relaxation_method.relaxation_flow = _from_ad_proto(proto.relax_flow)

luminarycloud/params/simulation/physics/fluid/turbulence_.py

@@ -54,7 +54,7 @@ class Turbulence(CodeRepr, ParamGroupWrapper[clientpb.Turbulence]):
     """Turbulence settings for a fluid flow physics solver."""
 
     des_formulation: DesFormulation = field(default_factory=DdesVtm)
-    "Select a Detached Eddy Simulation (DES) formulation. Possible types: ``DdesVtm``, ``DdesVtmSigma``, ``Iddes``, ``Ddes``, ``Des97``, ``Zdes`` from the ``des_formulation`` module."
+    "Select a Detached Eddy Simulation (DES) formulation. Possible types: ``DdesVtm``, ``Iddes``, ``Ddes`` from the ``des_formulation`` module."
     sub_grid_scale_model: SubGridScaleModel | None = field(default_factory=Vreman)
     "Sub-grid scale models available for Large Eddy Simulation (LES). Possible types: ``Smagorinsky``, ``Vreman``, ``Wale``, ``Sigma``, ``Amd`` from the ``sub_grid_scale_model`` module."
     turbulent_prandtl_number: LcFloat = 0.85

luminarycloud/params/simulation/physics/fluid_.py

@@ -124,7 +124,7 @@ class Fluid(CodeRepr, ParamGroupWrapper[clientpb.Fluid]):
     periodic_pair: list[PeriodicPair] = field(default_factory=list)
     "Defines the input options needed for periodic boundaries. We assume each periodic BC can have translational OR rotational periodicity. To transform a point on boundary A to its periodically matching point on boundary B, we first subtract the center of rotation from the point coordinates to get the distance vector from the center to the point of interest, then we apply rotation around the periodicity axis and add back the center of rotation to get the coordinates of the transformed point."
     initialization: InitializationFluid = field(default_factory=FluidPrescribedValues)
-    "Type of initial condition for the field variables. Possible types: ``FluidPrescribedValues``, ``FluidFarfieldValues``, ``FluidPotentialFlow``, ``FluidVerificationSolution``, ``FluidExistingSolution`` from the ``initialization`` module."
+    "Type of initial condition for the field variables. Possible types: ``FluidPrescribedValues``, ``FluidFarfieldValues``, ``FluidExistingSolution`` from the ``initialization`` module."
     physical_behavior: list[PhysicalBehavior] = field(default_factory=list)
     "Physical behavior settings for a fluid flow physics solver."
     porous_behavior: list[PorousBehavior] = field(default_factory=list)

luminarycloud/params/simulation/physics/heat/solution_controls/heat_relaxation_method/heat_implicit_relaxation_.py

@@ -42,4 +42,4 @@ class HeatImplicitRelaxation(HeatRelaxationMethod):
     implicit_method: enum.ImplicitMethod = enum.ImplicitMethod.BACKWARD_EULER
     "Scheme for implicit relaxation of the governing equations."
     linear_solver_type: LinearSolverType = field(default_factory=GaussSeidel)
-    "Type of linear solver used for implicit relaxation. Possible types: ``GaussSeidel``, ``GsAmgx``, ``KrylovAmg``, ``Amg`` from the ``linear_solver_type`` module."
+    "Type of linear solver used for implicit relaxation. Possible types: ``GaussSeidel``, ``KrylovAmg`` from the ``linear_solver_type`` module."

luminarycloud/params/simulation/physics/heat/solution_controls_heat_.py

@@ -32,7 +32,7 @@ class SolutionControlsHeat(CodeRepr, ParamGroupWrapper[clientpb.SolutionControls
     """Solution controls for a heat transfer physics solver."""
 
     heat_relaxation_method: HeatRelaxationMethod = field(default_factory=HeatImplicitRelaxation)
-    "Relaxation scheme for steady-state simulations or time implicit transient simulations. Possible types: ``HeatImplicitRelaxation``, ``HeatExplicitRelaxation`` from the ``heat_relaxation_method`` module."
+    "Relaxation scheme for steady-state simulations or time implicit transient simulations. Possible types: ``HeatImplicitRelaxation`` from the ``heat_relaxation_method`` module."
     preset: enum.SolutionControlsHeatPreset = (
         enum.SolutionControlsHeatPreset.DEFAULT_SOLUTION_CONTROLS_HEAT
     )

luminarycloud/physics_ai/__init__.py

@@ -8,3 +8,7 @@ from .models import (
     PhysicsAiModel as PhysicsAiModel,
     PhysicsAiModelVersion as PhysicsAiModelVersion,
 )
+
+from .solution import (
+    _download_processed_solution_physics_ai as _download_processed_solution_physics_ai,
+)

luminarycloud/physics_ai/inference.py

@@ -1,30 +1,165 @@
 # File: python/sdk/luminarycloud/inference/inference.py
 # Copyright 2025 Luminary Cloud, Inc. All Rights Reserved.
 from datetime import datetime
-from typing import Any
+from typing import Any, Callable
 from json import loads as json_loads
+from dataclasses import dataclass
+import base64
+import os
+import urllib.request
 
 from .._client import get_default_client
 from .._helpers._timestamp_to_datetime import timestamp_to_datetime
 from .._proto.api.v0.luminarycloud.inference import inference_pb2 as inferencepb
 from .._proto.inferenceservice import inferenceservice_pb2 as inferenceservicepb
 from .._wrapper import ProtoWrapper, ProtoWrapperBase
+from ..project import Project
 from .._helpers.warnings import experimental
+from ..project import Project
+from .._helpers import upload_file
+from .._proto.upload import upload_pb2 as uploadpb
+
+
+@dataclass
+class ExtAeroInferenceResult:
+    """Result of an external aerodynamic inference job.
+
+    Attributes
+    ----------
+    drag_force: float
+        The drag force returned from the inference.
+    lift_force: float
+        The lift force returned from the inference.
+    wall_shear_stress:
+        A dict containing wall shear stress data, or None if not available.
+    pressure_surface:
+        A dict containing pressure surface stress data, or None if not available.
+    """
+
+    drag_force: float
+    lift_force: float
+    wall_shear_stress: dict[str, Any] | None
+    pressure_surface: dict[str, Any] | None
+
+    def __init__(self, inference_result: dict[str, Any]) -> None:
+        self.drag_force = inference_result["drag_force"]
+        self.lift_force = inference_result["lift_force"]
+        self.wall_shear_stress = inference_result.get("wall-shear-stress", None)
+        self.pressure_surface = inference_result.get("pressure_surface", None)
+
+
+@experimental
+def external_aero_inference(
+    project: Project, stl_file: str, checkpoint_file: str, config_name: str, stencil_size: int
+) -> ExtAeroInferenceResult:
+    """Performs an inference job returning external aerodynamic results.
+    Parameters
+    ----------
+    project : Project
+        The project to which the inference files will be added.
+    stl_file : str
+        Fullpath the STL file to be used for inference.
+    checkpoint_file : str
+        Fullpath of the model to be used for inference.
+    config_name :str
+        Name of the configuration to be used for inference.
+    stencil_size :int
+        Size of the stencil to be used for inference.
+
+    Returns
+    ExtAeroInferenceResult
+        Result of the external aerodynamic inference job.
+
+    warning:: This feature is experimental and may change or be removed without notice.
+    """
+
+    result = perform_inference(project, stl_file, checkpoint_file, config_name, stencil_size)
+    return ExtAeroInferenceResult(result)
+
+
+@experimental
+def perform_inference(
+    project: Project, stl_file: str, checkpoint_file: str, config_name: str, stencil_size: int
+) -> dict[str, Any]:
+    """Creates an inference service job.
+    Parameters
+    ----------
+    project : Project
+        The project to which the inference files will be added.
+    stl_file : str
+        Fullpath the STL file to be used for inference.
+    checkpoint_file : str
+        Fullpath of the model to be used for inference.
+    config_name :str
+        Name of the configuration to be used for inference.
+    stencil_size :int
+        Size of the stencil to be used for inference.
+
+
+    Returns
+    dict[str, Any]
+        Response from the server as key-value pairs.
+
+    warning:: This feature is experimental and may change or be removed without notice.
+    """
+
+    client = get_default_client()
+
+    def upload_if_file(fname: str) -> str:
+        if os.path.exists(fname) and os.path.isfile(fname):
+            params = uploadpb.ResourceParams()
+            result = upload_file(client, project.id, params, fname)
+            return result[1].url
+        if fname.startswith("gs://"):
+            return fname
+        raise RuntimeError("Unsupported file for inference")
+
+    def future_file(url: str) -> Callable[[], dict[str, Any]]:
+        def download_file() -> dict[str, Any]:
+            with urllib.request.urlopen(url) as f:
+                serialized = f.read()
+                jsondata = json_loads(serialized)
+                data = base64.b64decode(jsondata["data"])
+                jsondata["data"] = data
+                return jsondata
+
+        return download_file
+
+    stl_url = upload_if_file(stl_file)
+    check_url = upload_if_file(checkpoint_file)
+
+    raw = start_inference_job(project, stl_url, check_url, config_name, stencil_size)
+    currated: dict[str, Any] = {}
+    for k, v in raw.items():
+        if isinstance(v, str) and v.startswith("https://"):
+            tmp = future_file(v)
+            if k.endswith("_url"):
+                currated[k[:-4]] = tmp
+                currated[k] = v
+            else:
+                currated[k] = tmp
+                currated[k + "_url"] = v
+        else:
+            currated[k] = v
+    return currated
 
 
 @experimental
 def start_inference_job(
+    project: Project,
     stl_url: str,
-    model_url: str,
+    checkpoint_url: str,
     config_name: str,
     stencil_size: int,
 ) -> dict[str, Any]:
     """Creates an inference service job.
     Parameters
     ----------
+    project : Project
+        Reference to a project.
     stl_url : str
         URL of the STL file to be used for inference.
-    model_url : str
+    checkpoint_url : str
         URL of the model to be used for inference.
     config_name :str
         Name of the configuration to be used for inference.
@@ -41,9 +176,10 @@ def start_inference_job(
 
     req = inferencepb.CreateInferenceServiceJobRequest(
         stl_url=stl_url,
-        model_url=model_url,
+        checkpoint_url=checkpoint_url,
         config_name=config_name,
         stencil_size=stencil_size,
+        project_id=project.id,
     )
 
     res: inferencepb.CreateInferenceServiceJobResponse = (

luminarycloud/physics_ai/solution.py

@@ -0,0 +1,60 @@
+from __future__ import annotations
+
+import tarfile
+from typing import List, Optional, BinaryIO, cast
+
+from .._client import get_default_client
+from .._helpers.warnings import experimental
+from .._helpers.download import download_solution_physics_ai as _download_solution_physics_ai
+from ..enum.quantity_type import QuantityType
+
+
+@experimental
+def _download_processed_solution_physics_ai(  # noqa: F841
+    solution_id: str,
+    exclude_surfaces: Optional[List[str]] = None,
+    fill_holes: float = -1.0,
+    surface_fields_to_keep: Optional[List[QuantityType]] = None,
+    volume_fields_to_keep: Optional[List[QuantityType]] = None,
+    process_volume: bool = False,
+    single_precision: bool = True,
+) -> tarfile.TarFile:
+    """
+    Download solution data with physics AI processing applied.
+
+    Returns a compressed archive containing processed solution files including
+    merged surfaces (VTP/STL) and optionally volume data (VTU).
+
+    Args:
+        solution_id: ID of the solution to download
+        exclude_surfaces: List of surface names to exclude from processing
+        fill_holes: Sets the maximum size of the hole to be filled for the STL file, measured as the radius of the bounding circumsphere.
+            If fill_holes is negative or zero, no holes will be filled.
+        surface_fields_to_keep: List of QuantityType enum values for surface fields to keep in output.
+            If None, all available surface fields are included.
+        volume_fields_to_keep: List of QuantityType enum values for volume fields to keep in output.
+            If None, all available volume fields are included.
+        process_volume: Whether to process volume data
+        single_precision: Whether to use single precision for floating point fields
+
+    Raises:
+        ValueError: If invalid field names are provided
+    """
+
+    stream = _download_solution_physics_ai(
+        get_default_client(),
+        solution_id,
+        exclude_surfaces=exclude_surfaces,
+        fill_holes=fill_holes,
+        surface_fields_to_keep=surface_fields_to_keep,
+        volume_fields_to_keep=volume_fields_to_keep,
+        process_volume=process_volume,
+        single_precision=single_precision,
+    )
+
+    assert stream is not None, "Failed to download solution data"
+    return tarfile.open(
+        name=stream.filename,
+        fileobj=cast(BinaryIO, stream),
+        mode="r|gz",
+    )

luminarycloud/tables.py

@@ -121,11 +121,11 @@ def create_rectilinear_table(
             table.header.record_label[-1].name = label
         else:
             table.header.record_label[-1].quantity = QuantityType(label).value
-        table.record.append(tablepb.Record())
 
     types = data_types(table_type, len(header))
 
     for row in rows:
+        record = tablepb.Record()
         for i, val in enumerate(row):
             # Axis coordinates are always adfloats and cannot be missing.
             if i == 0 and has_axis(table_type):
@@ -133,25 +133,23 @@ def create_rectilinear_table(
                 table.axis[0].coordinate[-1].adfloat.value = float(val)
                 continue
 
-            j = i - has_axis(table_type)
             if val == "":
                 if allow_missing_entries(table_type):
-                    table.record[j].entry.append(
-                        tablepb.Record.Entry(empty=tablepb.Record.Entry.Empty())
-                    )
+                    record.entry.append(tablepb.Record.Entry(empty=tablepb.Record.Entry.Empty()))
                     pass
                 else:
                     raise ValueError(f"Entry {i} in row {row} is missing.")
                 continue
 
-            table.record[j].entry.append(tablepb.Record.Entry())
+            record.entry.append(tablepb.Record.Entry())
             try:
                 if types[i] == float:
-                    table.record[j].entry[-1].adfloat.value = float(val)
+                    record.entry[-1].adfloat.value = float(val)
                 else:
-                    table.record[j].entry[-1].string = val
+                    record.entry[-1].string = val
             except ValueError:
                 raise ValueError(f"Expected type {types[i]} for entry {i} in row {row}.")
+        table.record.append(record)
 
     return table
 

luminarycloud/vis/data_extraction.py

@@ -381,12 +381,11 @@ class DataExtractor:
         simulation = get_simulation(self._solution.simulation_id)
         mesh_meta = get_mesh_metadata(simulation.mesh_id)
         mesh = get_mesh(simulation.mesh_id)
-        try:
-            geom = mesh.geometry_version().geometry()
-        except NotFoundError:
+        geo_ver = mesh.geometry_version()
+        if geo_ver is None:
             self._has_tags = False
-        except Exception as e:
-            raise ValueError("An unknow error occurred retrieving tags")
+        else:
+            geom = geo_ver.geometry()
 
         self._surface_ids: List[str] = []
         for zone in mesh_meta.zones:

luminarycloud/vis/display.py

@@ -4,10 +4,11 @@ from luminarycloud.enum import Representation, ColorMapPreset, FieldComponent, V
 from .._proto.api.v0.luminarycloud.vis import vis_pb2
 from typing import Optional
 import luminarycloud.enum.quantity_type as quantity_type
+from .._helpers._code_representation import CodeRepr
 
 
 @dc.dataclass
-class Field:
+class Field(CodeRepr):
     """
     The field controls the field displayed on the object. If the field doesn't
     exist, we show a solid color.
@@ -16,8 +17,8 @@ class Field:
 
     """
 
-    quantity: VisQuantity = VisQuantity.ABSOLUTE_PRESSURE
-    """The quantity to color by."""
+    quantity: VisQuantity = VisQuantity.NONE
+    """The quantity to color by. Default: NONE."""
     component: FieldComponent = FieldComponent.MAGNITUDE
     """
     The component of the field to use, applicable to vector fields. If the field is a
@@ -45,9 +46,25 @@ class Field:
                 self.component = FieldComponent(field.component)
         # If its a scalar, just ignore the component.
 
+    def _to_code(self, hide_defaults: bool = True, use_tmp_objs: bool = True) -> str:
+        # We have to handle this case specially because we need to omit the field
+        # component if the quantity is a scalar. Otherwise, it could confuse the user.
+        # Also we omit the instantiation line because all classes that use a Field already
+        # instantiate a default one in their own constructor. (Hopefully that invariant continues to
+        # hold)
+        def enum_to_string(val: VisQuantity | FieldComponent) -> str:
+            str_val = val.__repr__()
+            return str_val.split(": ")[0][1:]
+
+        code = ""
+        code += f".quantity = {enum_to_string(self.quantity)}\n"
+        if quantity_type._is_vector(self.quantity):
+            code += f".component = {enum_to_string(self.component)}\n"
+        return code
+
 
 @dc.dataclass
-class DisplayAttributes:
+class DisplayAttributes(CodeRepr):
     """
     Display attributes specify how objects such as meshes, geometries, and
     filters appear in the scene.
@@ -86,7 +103,7 @@ class DisplayAttributes:
 
 
 @dc.dataclass
-class DataRange:
+class DataRange(CodeRepr):
     """
     The data range represents a range of values. Ranges are only valid if the
     max value is greater than the or equal to the min_value. The default is
@@ -106,7 +123,7 @@ class DataRange:
 
 
 @dc.dataclass
-class ColorMapAppearance:
+class ColorMapAppearance(CodeRepr):
     """
     ColorMapAppearance controls how the color maps appear in the image, including
     visibility, position and size.
@@ -141,7 +158,7 @@ class ColorMapAppearance:
 
 
 @dc.dataclass
-class ColorMap:
+class ColorMap(CodeRepr):
     """
     The color map allows user control over how field values are mapped to
     colors. Color maps are assigned to fields (e.g., the quantity and component)
@@ -213,10 +230,8 @@ class ColorMap:
         return res
 
     def _from_proto(self, color_map: vis_pb2.ColorMap) -> None:
-        self.field = Field(
-            component=FieldComponent(color_map.field.component),
-            quantity=VisQuantity(color_map.field.quantity_typ),
-        )
+        self.field._from_proto(color_map.field)
+
         if color_map.HasField("range"):
             self.data_range = DataRange(
                 min_value=color_map.range.min,