From ba906fc286f5e6cc6c253565bb819d50f1a3c852 Mon Sep 17 00:00:00 2001
From: Tor Harald Sandve <tosa@norceresearch.no>
Date: Wed, 15 Mar 2023 14:00:16 +0100
Subject: [PATCH] add molar fraction of co2 in brine to RST output

---
 ebos/eclgenericoutputblackoilmodule.cc | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/ebos/eclgenericoutputblackoilmodule.cc b/ebos/eclgenericoutputblackoilmodule.cc
index 8abbdbdfb..e253df507 100644
--- a/ebos/eclgenericoutputblackoilmodule.cc
+++ b/ebos/eclgenericoutputblackoilmodule.cc
@@ -789,6 +789,15 @@ assignToSolution(data::Solution& sol)
         sol.insert("SGAS", UnitSystem::measure::identity, std::move(saturation_[gasPhaseIdx]), data::TargetType::RESTART_SOLUTION);
     }
 
+    if (eclState_.runspec().co2Storage() && !rsw_.empty()) {
+        std::vector<double> mfrac(rsw_.size(), 0.0);
+        const auto& pvtnum = eclState_.fieldProps().get_int("PVTNUM");
+        for (size_t i = 0; i < rsw_.size(); ++i) {
+            mfrac[i] = FluidSystem::convertXoGToxoG(FluidSystem::convertRswToXwG(rsw_[i], pvtnum[i]-1), pvtnum[i]-1);
+        }
+        sol.insert("XMFCO2", UnitSystem::measure::identity, std::move(mfrac), data::TargetType::RESTART_AUXILIARY);
+    }
+
     // Fluid in place
     for (const auto& phase : Inplace::phases()) {
         if (outputFipRestart_ && !fip_[phase].empty()) {