Add import of fracture surfaces from VTK text files

ApplicationLibCode/CMakeLists.txt

@@ -233,6 +233,7 @@ endif()
 if(CMAKE_CXX_COMPILER_ID STREQUAL "GNU")
   target_compile_options(
     ApplicationLibCode PRIVATE -Wall -Wno-reorder -Wno-parentheses -Wno-switch
+                               -Wno-implicit-fallthrough
   )
 endif()
 
@@ -249,6 +250,7 @@ if(CMAKE_CXX_COMPILER_ID STREQUAL "Clang")
             -Wno-delete-non-abstract-non-virtual-dtor
             -Wno-ambiguous-reversed-operator
             -Wno-deprecated-declarations
+            -Wno-implicit-fallthrough
   )
 endif()
 

ApplicationLibCode/Commands/SurfaceCommands/RicImportSurfacesFeature.cpp

@@ -45,7 +45,7 @@ void RicImportSurfacesFeature::onActionTriggered( bool isChecked )
     QStringList     fileNames  = RiuFileDialogTools::getOpenFileNames( Riu3DMainWindowTools::mainWindowWidget(),
                                                                   "Import Surfaces",
                                                                   defaultDir,
-                                                                  "Surface files (*.ptl *.ts *.dat *.xyz);;All Files (*.*)" );
+                                                                  "Surface files (*.ptl *.ts *.dat *.vtu *.pvd *.xyz);;All Files (*.*)" );
 
     if ( fileNames.isEmpty() ) return;
 

ApplicationLibCode/FileInterface/CMakeLists_files.cmake

@@ -103,6 +103,7 @@ set(SOURCE_GROUP_HEADER_FILES
     ${CMAKE_CURRENT_LIST_DIR}/RifByteArrayArrowRandomAccessFile.h
     ${CMAKE_CURRENT_LIST_DIR}/RifArrowTools.h
     ${CMAKE_CURRENT_LIST_DIR}/RifReaderRegularGridModel.h
+    ${CMAKE_CURRENT_LIST_DIR}/RifVtkSurfaceImporter.h
 )
 
 set(SOURCE_GROUP_SOURCE_FILES
@@ -203,6 +204,7 @@ set(SOURCE_GROUP_SOURCE_FILES
     ${CMAKE_CURRENT_LIST_DIR}/RifByteArrayArrowRandomAccessFile.cpp
     ${CMAKE_CURRENT_LIST_DIR}/RifArrowTools.cpp
     ${CMAKE_CURRENT_LIST_DIR}/RifReaderRegularGridModel.cpp
+    ${CMAKE_CURRENT_LIST_DIR}/RifVtkSurfaceImporter.cpp
 )
 
 list(APPEND CODE_HEADER_FILES ${SOURCE_GROUP_HEADER_FILES})

ApplicationLibCode/FileInterface/RifSurfaceImporter.cpp

@@ -207,7 +207,7 @@ void RifSurfaceImporter::readGocadFile( const QString& filename, RigGocadData* g
     if ( gocadData )
     {
         gocadData->setGeometryData( vertices, triangleIndices );
-        gocadData->addPropertyData( propertyNames, propertyValues );
+        gocadData->setPropertyData( propertyNames, propertyValues );
     }
 }
 

ApplicationLibCode/FileInterface/RifVtkSurfaceImporter.cpp

@@ -0,0 +1,263 @@
+/////////////////////////////////////////////////////////////////////////////////
+//
+//  Copyright (C) 2025-     Equinor ASA
+//
+//  ResInsight is free software: you can redistribute it and/or modify
+//  it under the terms of the GNU General Public License as published by
+//  the Free Software Foundation, either version 3 of the License, or
+//  (at your option) any later version.
+//
+//  ResInsight is distributed in the hope that it will be useful, but WITHOUT ANY
+//  WARRANTY; without even the implied warranty of MERCHANTABILITY or
+//  FITNESS FOR A PARTICULAR PURPOSE.
+//
+//  See the GNU General Public License at <http://www.gnu.org/licenses/gpl.html>
+//  for more details.
+//
+/////////////////////////////////////////////////////////////////////////////////
+
+#include "RifVtkSurfaceImporter.h"
+
+#include "../../Fwk/VizFwk/LibIo/cvfTinyXmlFused.hpp"
+#include "RigGocadData.h"
+
+#include <filesystem>
+#include <memory>
+#include <sstream>
+#include <string>
+#include <vector>
+
+namespace RifVtkSurfaceImporter
+{
+using namespace cvf_tinyXML;
+
+bool importFromFile( std::string filename, RigGocadData* gocadData )
+{
+    TiXmlDocument doc;
+    if ( !doc.LoadFile( filename.c_str() ) )
+    {
+        return false;
+    }
+
+    return importFromXmlDoc( doc, gocadData );
+}
+
+//--------------------------------------------------------------------------------------------------
+///
+//--------------------------------------------------------------------------------------------------
+bool importFromPvdFile( const std::string& filename, RigGocadData* gocadData )
+{
+    auto datasets = parsePvdDatasets( filename );
+
+    if ( datasets.empty() ) return false;
+
+    // Sort and import the most recent dataset
+    std::sort( datasets.begin(), datasets.end(), []( const PvdDataset& a, const PvdDataset& b ) { return a.timestep < b.timestep; } );
+
+    return importDataset( datasets.back(), gocadData );
+}
+
+bool importFromXmlDoc( const TiXmlDocument& doc, RigGocadData* gocadData )
+{
+    auto* root = doc.FirstChildElement( "VTKFile" );
+    if ( !root ) return false;
+
+    auto* grid = root->FirstChildElement( "UnstructuredGrid" );
+    if ( !grid ) return false;
+
+    auto* piece = grid->FirstChildElement( "Piece" );
+    if ( !piece ) return false;
+
+    // Read points
+    std::vector<cvf::Vec3d> vertices;
+    if ( !readPoints( piece, vertices ) )
+    {
+        return false;
+    }
+
+    // Read connectivity
+    std::vector<unsigned> connectivity;
+    if ( !readConnectivity( piece, connectivity ) )
+    {
+        return false;
+    }
+
+    // Avoid shared nodes
+    std::vector<cvf::Vec3d> nonSharedVertices;
+    std::vector<unsigned>   nonSharedConnectivity;
+
+    const auto numTriangles = connectivity.size() / 3;
+    for ( size_t triangleIdx = 0; triangleIdx < numTriangles; triangleIdx++ )
+    {
+        nonSharedVertices.push_back( vertices[connectivity[3 * triangleIdx + 0]] );
+        nonSharedVertices.push_back( vertices[connectivity[3 * triangleIdx + 1]] );
+        nonSharedVertices.push_back( vertices[connectivity[3 * triangleIdx + 2]] );
+
+        nonSharedConnectivity.push_back( static_cast<unsigned int>( 3 * triangleIdx + 0 ) );
+        nonSharedConnectivity.push_back( static_cast<unsigned int>( 3 * triangleIdx + 1 ) );
+        nonSharedConnectivity.push_back( static_cast<unsigned int>( 3 * triangleIdx + 2 ) );
+    }
+
+    // Set geometry data
+    gocadData->setGeometryData( nonSharedVertices, nonSharedConnectivity );
+
+    // Read properties
+    std::vector<std::string>        propertyNamesOnFile;
+    std::vector<std::vector<float>> propertyValuesOnFile;
+
+    readProperties( piece, propertyNamesOnFile, propertyValuesOnFile );
+
+    if ( propertyNamesOnFile.size() == propertyValuesOnFile.size() )
+    {
+        for ( size_t i = 0; i < propertyValuesOnFile.size(); i++ )
+        {
+            // These values are per element, so we need to duplicate them for each node
+            auto values = propertyValuesOnFile[i];
+            if ( values.size() * 3 == nonSharedVertices.size() )
+            {
+                std::vector<float> valuesForEachNode;
+                for ( auto value : values )
+                {
+                    valuesForEachNode.push_back( value );
+                    valuesForEachNode.push_back( value );
+                    valuesForEachNode.push_back( value );
+                }
+
+                gocadData->addPropertyData( QString::fromStdString( propertyNamesOnFile[i] ), valuesForEachNode );
+            }
+        }
+    }
+
+    return true;
+}
+
+bool readPoints( const TiXmlElement* piece, std::vector<cvf::Vec3d>& vertices )
+{
+    auto* points = piece->FirstChildElement( "Points" );
+    if ( !points ) return false;
+
+    auto* coords = points->FirstChildElement( "DataArray" );
+    if ( !coords || strcmp( coords->Attribute( "Name" ), "Coordinates" ) != 0 ) return false;
+
+    std::string        coordsText = coords->GetText();
+    std::istringstream iss( coordsText );
+
+    double x, y, z;
+    while ( iss >> x >> y >> z )
+    {
+        vertices.push_back( cvf::Vec3d( x, y, -z ) );
+    }
+
+    return !vertices.empty();
+}
+
+bool readConnectivity( const TiXmlElement* piece, std::vector<unsigned>& connectivity )
+{
+    auto* cells = piece->FirstChildElement( "Cells" );
+    if ( !cells ) return false;
+
+    auto* connectivityArray = cells->FirstChildElement( "DataArray" );
+    while ( connectivityArray )
+    {
+        if ( strcmp( connectivityArray->Attribute( "Name" ), "connectivity" ) == 0 )
+        {
+            std::string        connectivityText = connectivityArray->GetText();
+            std::istringstream iss( connectivityText );
+
+            unsigned index;
+            while ( iss >> index )
+            {
+                connectivity.push_back( index );
+            }
+            break;
+        }
+        connectivityArray = connectivityArray->NextSiblingElement( "DataArray" );
+    }
+
+    return !connectivity.empty();
+}
+
+void readProperties( const TiXmlElement* piece, std::vector<std::string>& propertyNames, std::vector<std::vector<float>>& propertyValues )
+{
+    auto* cellData = piece->FirstChildElement( "CellData" );
+    if ( !cellData ) return;
+
+    auto* dataArray = cellData->FirstChildElement( "DataArray" );
+    while ( dataArray )
+    {
+        const char* name = dataArray->Attribute( "Name" );
+        if ( name )
+        {
+            std::vector<float> values;
+            std::string        valuesText = dataArray->GetText();
+            std::istringstream iss( valuesText );
+
+            float value;
+            while ( iss >> value )
+            {
+                values.push_back( value );
+            }
+
+            if ( !values.empty() )
+            {
+                propertyNames.push_back( name );
+                propertyValues.push_back( values );
+            }
+        }
+        dataArray = dataArray->NextSiblingElement( "DataArray" );
+    }
+}
+
+//--------------------------------------------------------------------------------------------------
+///
+//--------------------------------------------------------------------------------------------------
+std::vector<RifVtkSurfaceImporter::PvdDataset> parsePvdDatasets( const std::string& filename )
+{
+    std::vector<PvdDataset> datasets;
+    TiXmlDocument           doc;
+
+    if ( !doc.LoadFile( filename.c_str() ) ) return datasets;
+
+    auto* root = doc.FirstChildElement( "VTKFile" );
+    if ( !root ) return datasets;
+
+    auto* collection = root->FirstChildElement( "Collection" );
+    if ( !collection ) return datasets;
+
+    std::string baseDir = std::filesystem::path( filename ).parent_path().string();
+
+    auto* datasetElem = collection->FirstChildElement( "DataSet" );
+    while ( datasetElem )
+    {
+        const char* file        = datasetElem->Attribute( "file" );
+        const char* timestepStr = datasetElem->Attribute( "timestep" );
+
+        if ( file && timestepStr )
+        {
+            double      timestep = std::stod( timestepStr );
+            std::string fullPath = std::filesystem::absolute( std::filesystem::path( baseDir ) / file ).string();
+
+            datasets.push_back( { timestep, fullPath } );
+        }
+
+        datasetElem = datasetElem->NextSiblingElement( "DataSet" );
+    }
+
+    return datasets;
+}
+
+//--------------------------------------------------------------------------------------------------
+///
+//--------------------------------------------------------------------------------------------------
+bool importDataset( const PvdDataset& dataset, RigGocadData* gocadData )
+{
+    TiXmlDocument doc;
+    if ( !doc.LoadFile( dataset.filename.c_str() ) )
+    {
+        return false;
+    }
+
+    return importFromXmlDoc( doc, gocadData );
+}
+
+}; // namespace RifVtkSurfaceImporter

ApplicationLibCode/FileInterface/RifVtkSurfaceImporter.h

@@ -0,0 +1,62 @@
+/////////////////////////////////////////////////////////////////////////////////
+//
+//  Copyright (C) 2025-     Equinor ASA
+//
+//  ResInsight is free software: you can redistribute it and/or modify
+//  it under the terms of the GNU General Public License as published by
+//  the Free Software Foundation, either version 3 of the License, or
+//  (at your option) any later version.
+//
+//  ResInsight is distributed in the hope that it will be useful, but WITHOUT ANY
+//  WARRANTY; without even the implied warranty of MERCHANTABILITY or
+//  FITNESS FOR A PARTICULAR PURPOSE.
+//
+//  See the GNU General Public License at <http://www.gnu.org/licenses/gpl.html>
+//  for more details.
+//
+/////////////////////////////////////////////////////////////////////////////////
+
+#pragma once
+
+#include <map>
+#include <memory>
+#include <string>
+#include <vector>
+
+#include "cvfVector3.h"
+
+class RigGocadData;
+
+namespace cvf_tinyXML
+{
+class TiXmlDocument;
+class TiXmlElement;
+} // namespace cvf_tinyXML
+
+//==================================================================================================
+///
+//==================================================================================================
+namespace RifVtkSurfaceImporter
+{
+
+struct PvdDataset
+{
+    double      timestep;
+    std::string filename;
+};
+
+bool importFromFile( std::string filename, RigGocadData* gocadData );
+
+bool importFromPvdFile( const std::string& filename, RigGocadData* gocadData );
+bool importFromXmlDoc( const cvf_tinyXML::TiXmlDocument& doc, RigGocadData* gocadData );
+
+bool readPoints( const cvf_tinyXML::TiXmlElement* piece, std::vector<cvf::Vec3d>& vertices );
+bool readConnectivity( const cvf_tinyXML::TiXmlElement* piece, std::vector<unsigned>& connectivity );
+void readProperties( const cvf_tinyXML::TiXmlElement* piece,
+                     std::vector<std::string>&        propertyNames,
+                     std::vector<std::vector<float>>& propertyValues );
+
+std::vector<PvdDataset> parsePvdDatasets( const std::string& filename );
+bool                    importDataset( const PvdDataset& dataset, RigGocadData* gocadData );
+
+}; // namespace RifVtkSurfaceImporter

ApplicationLibCode/ProjectDataModel/GeoMech/RimGeoMechView.cpp

@@ -195,7 +195,7 @@ void RimGeoMechView::onLoadDataAndUpdate()
     geoMechPropertyFilterCollection()->loadAndInitializePropertyFilters();
     m_wellMeasurementCollection->syncWithChangesInWellMeasurementCollection();
 
-    if ( m_surfaceCollection ) m_surfaceCollection->loadData();
+    if ( m_surfaceCollection ) m_surfaceCollection->loadData( m_currentTimeStep );
 
     if ( m_partsCollection ) m_partsCollection->syncWithCase( m_geomechCase );