Add some friends to protect direct access to internal cell storage

OPM · Oct 25, 2024 · 2228b67 · 2228b67
1 parent 19bf09d
commit 2228b67
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/ApplicationLibCode/ReservoirDataModel/RigActiveCellGrid.h b/ApplicationLibCode/ReservoirDataModel/RigActiveCellGrid.h
@@ -34,7 +34,8 @@ class RigActiveCellGrid : public RigMainGrid
     size_t totalCellCount() const override;
     size_t totalActiveCellCount() const;
 
-public: // only for use by file readers!
+protected: // only for use by file readers!
+    friend class RifReaderOpmCommonActive;
     std::map<size_t, RigCell>&       nativeCells();
     const std::map<size_t, RigCell>& nativeCells() const;
     void                             setTotalCellCount( size_t totalCellCount );

diff --git a/ApplicationLibCode/ReservoirDataModel/RigMainGrid.h b/ApplicationLibCode/ReservoirDataModel/RigMainGrid.h
@@ -113,7 +113,16 @@ class RigMainGrid : public RigGridBase
     bool isDualPorosity() const;
     void setDualPorosity( bool enable );
 
-public: // only for use by file readers!
+protected: // only for use by file readers and internal services. TODO: replace with a better API
+    friend class RigGridBase;
+    friend class RigReservoirBuilder;
+    friend class RifRoffFileTools;
+    friend class RifEclipseOutputFileTools;
+    friend class RifReaderEclipseOutput;
+    friend class RifReaderOpmCommon;
+    friend class RiaGrpcCaseService;
+    friend class RiaActiveCellInfoStateHandler;
+    friend class RicCreateTemporaryLgrFeature;
     std::vector<RigCell>&       reservoirCells();
     const std::vector<RigCell>& reservoirCells() const;