TNO Intern

Commit fb71b2d4 authored by Florian Knappers's avatar Florian Knappers
Browse files

Merge branch '61-update-pythermogis-after-java-updates' into 'main' 

Resolve "update pythermogis after java updates"

Closes #61

See merge request AGS/pythermogis!76
parents 1302e75e 50143a48

src/pythermogis/thermogis_classes/utc_properties.py

+25 −0
Original line number Diff line number Diff line 
from pathlib import Path

import typing as t

import xml.etree.ElementTree as ET

from jpype import JClass
@@ -37,3 +38,27 @@ def instantiate_utc_properties_builder() -> JClass: 
    """

    start_jvm()

    return JClass("thermogis.properties.builders.UTCPropertiesBuilder")()



def get_viscosity_mode(mode: t.Literal["kestin", "batzlewang"]) -> JClass:

    """

    Map a string to the Java ViscosityMode enum.



    Parameters

    ----------

    mode : Literal["kestin", "batzlewang"]

        The viscosity mode to select.



    Returns

    -------

    viscosity_mode_enum : jpype.JClass

        The corresponding Java enum instance.

    """

    start_jvm()

    viscosity_mode = JClass("tno.geoenergy.ViscosityMode")



    if mode == "kestin":

        return viscosity_mode.KESTIN

    elif mode == "batzlewang":

        return viscosity_mode.BATZLEWANG

    else:

        raise ValueError(f"Unknown viscosity mode: {mode}")
 
 No newline at end of file