From 2577eee15935608d5d61ddb70d4d93435f539048 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Atgeirr=20Fl=C3=B8=20Rasmussen?= <atgeirr@sintef.no>
Date: Thu, 23 Feb 2017 10:35:11 +0100
Subject: [PATCH] Add wellsToState() overload for dense well state class.

---
 opm/autodiff/Compat.cpp | 16 ++++++++++++++++
 opm/autodiff/Compat.hpp |  5 +++++
 2 files changed, 21 insertions(+)

diff --git a/opm/autodiff/Compat.cpp b/opm/autodiff/Compat.cpp
index 161bdcab0..2b9d8c00b 100644
--- a/opm/autodiff/Compat.cpp
+++ b/opm/autodiff/Compat.cpp
@@ -28,6 +28,7 @@
 #include <opm/core/props/BlackoilPhases.hpp>
 #include <opm/core/simulator/BlackoilState.hpp>
 #include <opm/autodiff/WellStateFullyImplicitBlackoil.hpp>
+#include <opm/autodiff/WellStateFullyImplicitBlackoilDense.hpp>
 #include <opm/autodiff/BlackoilSolventState.hpp>
 #include <opm/output/data/Cells.hpp>
 #include <opm/output/data/Solution.hpp>
@@ -238,4 +239,19 @@ void wellsToState( const data::Wells& wells,
 
 
 
+
+
+
+void wellsToState( const data::Wells& wells,
+                   PhaseUsage phases,
+                   WellStateFullyImplicitBlackoilDense& state )
+{
+    // Set base class variables.
+    wellsToState(wells, phases, static_cast<WellStateFullyImplicitBlackoil&>(state));
+
+    // Set wellSolution() variable.
+    state.setWellSolutions(phases);
+}
+
+
 } // namespace Opm
diff --git a/opm/autodiff/Compat.hpp b/opm/autodiff/Compat.hpp
index c83af1f9c..fdd18752d 100644
--- a/opm/autodiff/Compat.hpp
+++ b/opm/autodiff/Compat.hpp
@@ -31,6 +31,7 @@ namespace Opm {
     // Forward declarations
     class SimulationDataContainer;
     class WellStateFullyImplicitBlackoil;
+    class WellStateFullyImplicitBlackoilDense;
 
     std::vector< double > destripe( const std::vector< double >& v,
                                     size_t stride,
@@ -52,6 +53,10 @@ namespace Opm {
                        PhaseUsage phases,
                        WellStateFullyImplicitBlackoil& state );
 
+    void wellsToState( const data::Wells& wells,
+                       PhaseUsage phases,
+                       WellStateFullyImplicitBlackoilDense& state );
+
 }
 
 #endif //OPM_SIMULATORS_COMPAT_HPP