Merge pull request #622 from joakim-hove/udq-units

UDQ - hook the configuration up in the Schedule object

src/opm/parser/eclipse/EclipseState/Schedule/Schedule.cpp

@@ -49,6 +49,7 @@
 
 #include <opm/parser/eclipse/EclipseState/Schedule/OilVaporizationProperties.hpp>
 #include <opm/parser/eclipse/EclipseState/Schedule/ScheduleEnums.hpp>
+#include <opm/parser/eclipse/EclipseState/Schedule/UDQ.hpp>
 #include <opm/parser/eclipse/EclipseState/Schedule/Schedule.hpp>
 #include <opm/parser/eclipse/EclipseState/Schedule/TimeMap.hpp>
 #include <opm/parser/eclipse/EclipseState/Schedule/Tuning.hpp>
@@ -81,7 +82,8 @@ namespace Opm {
         m_messageLimits( this->m_timeMap ),
         m_runspec( runspec ),
         wtest_config(this->m_timeMap, std::make_shared<WellTestConfig>() ),
-        wlist_manager( this->m_timeMap, std::make_shared<WListManager>())
+        wlist_manager( this->m_timeMap, std::make_shared<WListManager>()),
+        udq_config(this->m_timeMap, std::make_shared<UDQ>())
     {
         m_controlModeWHISTCTL = WellProducer::CMODE_UNDEFINED;
         addGroup( "FIELD", 0 );
@@ -209,6 +211,9 @@ namespace Opm {
             currentStep += keyword.getRecord(0).getItem(0).size(); // This is a bit weird API.
         }
 
+        else if (keyword.name() == "UDQ")
+            handleUDQ(keyword, currentStep);
+
         else if (keyword.name() == "WLIST")
             handleWLIST( keyword, currentStep );
 
@@ -936,6 +941,17 @@ namespace Opm {
         this->wlist_manager.update(currentStep, new_wlm);
     }
 
+    void Schedule::handleUDQ(const DeckKeyword& keyword, size_t currentStep) {
+        const auto& current = *this->udq_config.get(currentStep);
+        std::shared_ptr<UDQ> new_udq = std::make_shared<UDQ>(current);
+
+        for (const auto& record : keyword)
+            new_udq->add_record(record);
+
+        this->udq_config.update(currentStep, new_udq);
+    }
+
+
     void Schedule::handleWTEST(const DeckKeyword& keyword, size_t currentStep, const ParseContext& parseContext, ErrorGuard& errors) {
         const auto& current = *this->wtest_config.get(currentStep);
         std::shared_ptr<WellTestConfig> new_config(new WellTestConfig(current));
@@ -2147,6 +2163,11 @@ namespace Opm {
         return *ptr;
     }
 
+    const UDQ& Schedule::getUDQConfig(size_t timeStep) const {
+        const auto& ptr = this->udq_config.get(timeStep);
+        return *ptr;
+    }
+
 
     size_t Schedule::size() const {
         return this->m_timeMap.size();