Usability fixes to table generator

Author: moyner

src/PVTExperiments/interface.jl

@@ -51,6 +51,9 @@ function generate_pvt_tables(eos, z, T_res;
         n_pvdo = 20,
         n_undersaturated = 5
     )
+    if p_range[1] <= p_range[2]
+        p_range = (p_range[2], p_range[1])
+    end
     z = collect(Float64, z)
     z ./= sum(z)
 

src/PVTExperiments/tables.jl

@@ -13,14 +13,16 @@ Generate a PVTO (live oil) table from DLE or MSS experiment data.
 - `p_sc`: Standard condition pressure (Pa). Default: 101325.0
 - `T_sc`: Standard condition temperature (K). Default: 288.706
 - `separator_stages`: Optional separator stages. If provided, MSS is used for surface conditions.
+- `extend`: Whether to extend the table with an entry at p_range[1] if it's above bubble point. Default: true.
 """
 function pvto_table(eos, z, T;
         p_range = (50e6, 1e5),
         n_rs = 15,
         n_undersaturated = 5,
         p_sc = 101325.0,
         T_sc = 288.706,
-        separator_stages = nothing
+        separator_stages = nothing,
+        extend = true
     )
     z = collect(Float64, z)
     z ./= sum(z)
@@ -124,6 +126,15 @@ function pvto_table(eos, z, T;
                 push!(Bo_entries, Bo_us)
                 push!(mu_entries, props_us.μ_l)
             end
+
+            if p_range[1] > p_bubble + 1e5 && extend
+                # Add an entry at p_range[1] if it's above bubble point
+                props_us = flash_and_properties(eos, p_range[1], T, z_oil_at_level)
+                Bo_us = props_us.V_mol_l / (V_oil_st / liq_frac)
+                push!(p_entries, p_range[1])
+                push!(Bo_entries, Bo_us)
+                push!(mu_entries, props_us.μ_l)
+            end
         end
 
         push!(Rs_values, Rs)