mirror of
https://github.com/OPM/ResInsight.git
synced 2026-10-03 12:16:45 -05:00
117 lines
6.6 KiB
C++
117 lines
6.6 KiB
C++
/////////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
// 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 "RifGeoMechReaderInterface.h"
|
|
|
|
#include "RigFemResultPosEnum.h"
|
|
#include "RigFemTypes.h"
|
|
|
|
#include <expected>
|
|
#include <filesystem>
|
|
#include <map>
|
|
#include <string>
|
|
#include <utility>
|
|
|
|
class RigFemPartCollection;
|
|
|
|
//==================================================================================================
|
|
//
|
|
//==================================================================================================
|
|
class RifVtkReader : public RifGeoMechReaderInterface
|
|
{
|
|
public:
|
|
RifVtkReader();
|
|
~RifVtkReader() override;
|
|
|
|
bool openFile( const std::string& fileName, std::string* errorMessage ) override;
|
|
bool isOpen() const override;
|
|
bool readFemParts( RigFemPartCollection* geoMechCase ) override;
|
|
std::vector<std::string> allStepNames() const override;
|
|
std::vector<std::string> filteredStepNames() const override;
|
|
std::vector<double> frameTimes( int stepIndex ) const override;
|
|
int frameCount( int stepIndex ) const override;
|
|
|
|
std::vector<std::string> elementSetNames( int partIndex, std::string partName ) override;
|
|
std::vector<size_t> elementSet( int partIndex, std::string partName, int setIndex ) override;
|
|
|
|
std::map<std::string, std::vector<std::string>> scalarNodeFieldAndComponentNames() override;
|
|
std::map<std::string, std::vector<std::string>> scalarElementFieldAndComponentNames() override;
|
|
std::map<std::string, std::vector<std::string>> scalarElementNodeFieldAndComponentNames() override;
|
|
std::map<std::string, std::vector<std::string>> scalarIntegrationPointFieldAndComponentNames() override;
|
|
|
|
void readDisplacements( int partIndex, int stepIndex, int frameIndex, std::vector<cvf::Vec3f>* displacements ) override;
|
|
|
|
void readNodeField( const std::string& fieldName,
|
|
int partIndex,
|
|
int stepIndex,
|
|
int frameIndex,
|
|
std::vector<std::vector<float>*>* resultValues ) override;
|
|
void readElementField( const std::string& fieldName,
|
|
int partIndex,
|
|
int stepIndex,
|
|
int frameIndex,
|
|
std::vector<std::vector<float>*>* resultValues ) override;
|
|
void readElementNodeField( const std::string& fieldName,
|
|
int partIndex,
|
|
int stepIndex,
|
|
int frameIndex,
|
|
std::vector<std::vector<float>*>* resultValues ) override;
|
|
void readIntegrationPointField( const std::string& fieldName,
|
|
int partIndex,
|
|
int stepIndex,
|
|
int frameIndex,
|
|
std::vector<std::vector<float>*>* resultValues ) override;
|
|
|
|
bool populateDerivedResultNames() const override;
|
|
|
|
private:
|
|
void close();
|
|
|
|
void readField( RigFemResultPosEnum resultType,
|
|
const std::string& fieldName,
|
|
int partIndex,
|
|
int stepIndex,
|
|
std::vector<std::vector<float>*>* resultValues );
|
|
|
|
void readScalarData( RigFemPartCollection* femParts,
|
|
std::map<int, std::string>& parts,
|
|
const std::map<int, std::vector<std::map<std::string, std::vector<float>>>>& properties,
|
|
size_t numElements );
|
|
|
|
std::map<std::string, std::map<int, std::map<int, std::vector<double>>>>* propertyDataMap( RigFemResultPosEnum resultType );
|
|
|
|
static std::expected<RigElementType, std::string> read( const std::filesystem::path& filePath,
|
|
std::map<int, std::string>& parts,
|
|
std::map<int, std::vector<std::pair<int, cvf::Vec3d>>>& nodes,
|
|
std::map<int, std::vector<std::pair<int, std::vector<int>>>>& elements,
|
|
std::map<int, std::vector<std::pair<std::string, std::vector<size_t>>>>& elementSets,
|
|
std::map<int, std::vector<std::map<std::string, std::vector<float>>>>& properties,
|
|
std::map<int, std::vector<std::vector<cvf::Vec3f>>>& displacements,
|
|
std::vector<std::string>& stepNames );
|
|
|
|
private:
|
|
std::map<int, std::vector<std::string>> m_partElementSetNames;
|
|
std::vector<std::string> m_stepNames;
|
|
std::filesystem::path m_inputPath;
|
|
std::map<std::string, std::map<int, std::map<int, std::vector<double>>>> m_propertyPartDataNodes;
|
|
std::map<std::string, std::map<int, std::map<int, std::vector<double>>>> m_propertyPartDataElements;
|
|
std::map<int, std::vector<std::vector<cvf::Vec3f>>> m_displacements;
|
|
};
|