ResInsight/ApplicationLibCode/GeoMech/GeoMechDataModel/RigFemPartCollection.cpp

/////////////////////////////////////////////////////////////////////////////////
//
//  Copyright (C) 2015-     Statoil ASA
//  Copyright (C) 2015-     Ceetron Solutions AS
//
//  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 "RigFemPartCollection.h"

#include "RigHexIntersectionTools.h"

#include "cvfBoundingBox.h"

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
RigFemPartCollection::RigFemPartCollection()
{
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
RigFemPartCollection::~RigFemPartCollection()
{
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RigFemPartCollection::addFemPart( RigFemPart* part )
{
    size_t globalElementOffset      = 0;
    size_t globalNodeOffset         = 0;
    size_t globalConnectivityOffset = 0;
    if ( !m_femParts.empty() )
    {
        size_t lastIndex = m_femParts.size() - 1;
        globalElementOffset += m_femParts[lastIndex]->elementCount();
        globalElementOffset += m_partElementOffset[lastIndex];
        globalNodeOffset += m_femParts[lastIndex]->nodes().nodeIds.size();
        globalNodeOffset += m_partNodeOffset[lastIndex];
        globalConnectivityOffset += m_femParts[lastIndex]->allConnectivitiesCount();
        globalConnectivityOffset += m_partConnectivityOffset[lastIndex];
    }

    m_femParts.push_back( part );
    m_partElementOffset.push_back( globalElementOffset );
    m_partNodeOffset.push_back( globalNodeOffset );
    m_partConnectivityOffset.push_back( globalConnectivityOffset );
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
RigFemPart* RigFemPartCollection::part( size_t index )
{
    return m_femParts[index].p();
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
const RigFemPart* RigFemPartCollection::part( size_t index ) const
{
    return m_femParts[index].p();
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
int RigFemPartCollection::partCount() const
{
    return static_cast<int>( m_femParts.size() );
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
size_t RigFemPartCollection::totalElementCount() const
{
    size_t elementCount = 0;

    for ( int i = 0; i < partCount(); i++ )
    {
        elementCount += part( i )->elementCount();
    }

    return elementCount;
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
float RigFemPartCollection::characteristicElementSize() const
{
    if ( partCount() )
    {
        return part( 0 )->characteristicElementSize();
    }
    else
    {
        return 0;
    }
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
cvf::BoundingBox RigFemPartCollection::boundingBox() const
{
    cvf::BoundingBox bBox;
    for ( int i = 0; i < partCount(); i++ )
    {
        bBox.add( part( i )->boundingBox() );
    }
    return bBox;
}

//--------------------------------------------------------------------------------------------------
/// Get part id and local part element index from global element index
//--------------------------------------------------------------------------------------------------
std::pair<int, size_t> RigFemPartCollection::partIdAndElementIndex( size_t globalIndex ) const
{
    const size_t nParts = m_partElementOffset.size();

    CVF_ASSERT( nParts > 0 );

    for ( size_t i = 1; i < nParts; i++ )
    {
        if ( globalIndex < m_partElementOffset[i] )
        {
            return std::make_pair( (int)( i - 1 ), globalIndex - m_partElementOffset[i - 1] );
        }
    }

    return std::make_pair( (int)( nParts - 1 ), globalIndex - m_partElementOffset.back() );
}

//--------------------------------------------------------------------------------------------------
/// Get part and local part element index from global element index
//--------------------------------------------------------------------------------------------------
std::pair<const RigFemPart*, size_t> RigFemPartCollection::partAndElementIndex( size_t globalIndex ) const
{
    auto [partId, elementIdx] = partIdAndElementIndex( globalIndex );
    return std::make_pair( part( partId ), elementIdx );
}

//--------------------------------------------------------------------------------------------------
/// Get global element index from part id and local part element index
//--------------------------------------------------------------------------------------------------
size_t RigFemPartCollection::globalElementIndex( int partId, size_t localIndex ) const
{
    return localIndex + m_partElementOffset[partId];
}

//--------------------------------------------------------------------------------------------------
/// Find intersecting global element indexes for a given bounding box
//--------------------------------------------------------------------------------------------------
std::vector<size_t> RigFemPartCollection::findIntersectingGlobalElementIndices( const cvf::BoundingBox& intersectingBB ) const
{
    std::vector<size_t> intersectedGlobalElementIndices;
    for ( const auto& part : m_femParts )
    {
        std::vector<size_t> foundElements = part->findIntersectingElementIndices( intersectingBB );
        for ( const auto& foundElement : foundElements )
        {
            const size_t globalIdx = globalElementIndex( part->elementPartId(), foundElement );
            intersectedGlobalElementIndices.push_back( globalIdx );
        }
    }

    return intersectedGlobalElementIndices;
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
int RigFemPartCollection::getPartIndexFromPoint( const cvf::Vec3d& point ) const
{
    const int idx = -1;

    // Find candidates for intersected global elements
    const cvf::BoundingBox intersectingBb( point, point );
    std::vector<size_t>    intersectedGlobalElementIndexCandidates = findIntersectingGlobalElementIndices( intersectingBb );

    if ( intersectedGlobalElementIndexCandidates.empty() ) return idx;

    // Iterate through global element candidates and check if point is in hexCorners
    for ( const auto& globalElementIndex : intersectedGlobalElementIndexCandidates )
    {
        const auto [part, elementIndex] = partAndElementIndex( globalElementIndex );

        // Find nodes from element
        std::array<cvf::Vec3d, 8> coordinates;
        if ( part->fillElementCoordinates( elementIndex, coordinates ) )
        {
            if ( RigHexIntersectionTools::isPointInCell( point, coordinates.data() ) )
            {
                return part->elementPartId();
            }
        }
    }

    // Utilize first part to have an id
    return idx;
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
int RigFemPartCollection::nodeIdxFromElementNodeResultIdx( size_t globalIndex ) const
{
    const size_t nParts = m_partConnectivityOffset.size();
    CVF_ASSERT( nParts > 0 );

    int    partId  = (int)( nParts - 1 );
    size_t partIdx = globalIndex - m_partConnectivityOffset.back();

    for ( size_t i = 1; i < nParts; i++ )
    {
        if ( globalIndex < m_partConnectivityOffset[i] )
        {
            partId  = (int)( i - 1 );
            partIdx = globalIndex - m_partConnectivityOffset[i - 1];
            break;
        }
    }

    const RigFemPart* part = this->part( partId );

    return (int)m_partNodeOffset[partId] + part->nodeIdxFromElementNodeResultIdx( partIdx );
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
size_t RigFemPartCollection::globalElementNodeResultIdx( int partId, int elementIdx, int elmLocalNodeIdx ) const
{
    return m_partElementOffset[partId] * 8 + part( partId )->elementNodeResultIdx( elementIdx, elmLocalNodeIdx );
}