From d1b1cebdf56be4752841edbe466b25d9a3219f78 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?B=C3=A5rd=20Skaflestad?= <Bard.Skaflestad@sintef.no>
Date: Tue, 22 Jun 2021 16:46:06 +0200
Subject: [PATCH] Support Retrieving Non-Zero Restart Vector Occurrence

Needed to load connection information for analytic aquifers (xCAQ).
---
 opm/io/eclipse/RestartFileView.hpp     |  2 ++
 src/opm/io/eclipse/RestartFileView.cpp | 10 ++++++++++
 2 files changed, 12 insertions(+)

diff --git a/opm/io/eclipse/RestartFileView.hpp b/opm/io/eclipse/RestartFileView.hpp
index d4d3c9d64..44ad59a17 100644
--- a/opm/io/eclipse/RestartFileView.hpp
+++ b/opm/io/eclipse/RestartFileView.hpp
@@ -47,6 +47,8 @@ public:
     std::size_t simStep() const;
     int reportStep() const;
 
+    int occurrenceCount(const std::string& vector) const;
+
     template <typename ElmType>
     bool hasKeyword(const std::string& vector) const;
 
diff --git a/src/opm/io/eclipse/RestartFileView.cpp b/src/opm/io/eclipse/RestartFileView.cpp
index 72baa91ba..985d92a4c 100644
--- a/src/opm/io/eclipse/RestartFileView.cpp
+++ b/src/opm/io/eclipse/RestartFileView.cpp
@@ -96,6 +96,11 @@ public:
         return this->report_step_;
     }
 
+    int occurrenceCount(const std::string& vector) const
+    {
+        return this->rst_file_->occurrence_count(vector, this->report_step_);
+    }
+
     template <typename ElmType>
     bool hasKeyword(const std::string& vector) const
     {
@@ -225,6 +230,11 @@ int Opm::EclIO::RestartFileView::reportStep() const
     return this->pImpl_->reportStep();
 }
 
+int Opm::EclIO::RestartFileView::occurrenceCount(const std::string& vector) const
+{
+    return this->pImpl_->occurrenceCount(vector);
+}
+
 const std::vector<int>& Opm::EclIO::RestartFileView::intehead() const
 {
     return this->pImpl_->intehead();