/////////////////////////////////////////////////////////////////////////////////
//
//  Copyright (C) Statoil ASA
//  Copyright (C) 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 "RivExtrudedCurveIntersectionPartMgr.h"

#include "RiaGuiApplication.h"
#include "RiaOffshoreSphericalCoords.h"
#include "RiaPreferences.h"

#include "RigCaseCellResultsData.h"
#include "RigFemPartCollection.h"
#include "RigFemPartResultsCollection.h"
#include "RigGeoMechCaseData.h"
#include "RigResultAccessor.h"
#include "RigResultAccessorFactory.h"

#include "Rim2dIntersectionView.h"
#include "RimEclipseCase.h"
#include "RimEclipseCellColors.h"
#include "RimEclipseView.h"
#include "RimExtrudedCurveIntersection.h"
#include "RimFaultInView.h"
#include "RimFaultInViewCollection.h"
#include "RimGeoMechCase.h"
#include "RimGeoMechCellColors.h"
#include "RimGeoMechView.h"
#include "RimSimWellInView.h"
#include "RimSimWellInViewCollection.h"
#include "RimWellPath.h"
#include "RimWellPathCollection.h"

#include "RiuGeoMechXfTensorResultAccessor.h"

#include "RivExtrudedCurveIntersectionGeometryGenerator.h"
#include "RivExtrudedCurveIntersectionSourceInfo.h"
#include "RivHexGridIntersectionTools.h"
#include "RivIntersectionResultsColoringTools.h"
#include "RivMeshLinesSourceInfo.h"
#include "RivObjectSourceInfo.h"
#include "RivPartPriority.h"
#include "RivPipeGeometryGenerator.h"
#include "RivResultToTextureMapper.h"
#include "RivScalarMapperUtils.h"
#include "RivSimWellPipeSourceInfo.h"
#include "RivTernaryScalarMapper.h"
#include "RivTernaryTextureCoordsCreator.h"
#include "RivWellPathSourceInfo.h"

#include "cafTensor3.h"

#include "cvfDrawableGeo.h"
#include "cvfDrawableText.h"
#include "cvfGeometryTools.h"
#include "cvfModelBasicList.h"
#include "cvfPart.h"
#include "cvfPrimitiveSetDirect.h"
#include "cvfRenderStateDepth.h"
#include "cvfRenderStatePoint.h"
#include "cvfRenderState_FF.h"
#include "cvfStructGridGeometryGenerator.h"
#include "cvfTransform.h"
#include "cvfqtUtils.h"

#include <functional>

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
RivExtrudedCurveIntersectionPartMgr::RivExtrudedCurveIntersectionPartMgr( RimExtrudedCurveIntersection* rimCrossSection,
                                                                          bool                          isFlattened )
    : m_rimIntersection( rimCrossSection )
    , m_isFlattened( isFlattened )
{
    CVF_ASSERT( m_rimIntersection );

    m_intersectionFacesTextureCoords = new cvf::Vec2fArray;

    cvf::Vec3d flattenedPolylineStartPoint;

    std::vector<std::vector<cvf::Vec3d>> polyLines = m_rimIntersection->polyLines( &flattenedPolylineStartPoint );
    if ( polyLines.size() > 0 )
    {
        cvf::Vec3d                                direction = m_rimIntersection->extrusionDirection();
        cvf::ref<RivIntersectionHexGridInterface> hexGrid   = m_rimIntersection->createHexGridInterface();
        m_intersectionGenerator = new RivExtrudedCurveIntersectionGeometryGenerator( m_rimIntersection,
                                                                                     polyLines,
                                                                                     direction,
                                                                                     hexGrid.p(),
                                                                                     m_isFlattened,
                                                                                     flattenedPolylineStartPoint );
    }
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RivExtrudedCurveIntersectionPartMgr::applySingleColorEffect()
{
    if ( m_intersectionGenerator.isNull() ) return;

    caf::SurfaceEffectGenerator geometryEffgen( cvf::Color3f::OLIVE, caf::PO_1 );

    cvf::ref<cvf::Effect> geometryOnlyEffect = geometryEffgen.generateCachedEffect();

    if ( m_intersectionFaces.notNull() )
    {
        m_intersectionFaces->setEffect( geometryOnlyEffect.p() );
    }

    // Update mesh colors as well, in case of change
    RiaPreferences* prefs = RiaApplication::instance()->preferences();

    if ( m_intersectionGridLines.notNull() )
    {
        cvf::ref<cvf::Effect>    eff;
        caf::MeshEffectGenerator CrossSectionEffGen( prefs->defaultGridLineColors() );
        eff = CrossSectionEffGen.generateCachedEffect();

        m_intersectionGridLines->setEffect( eff.p() );
    }

    if ( m_intersectionFaultGridLines.notNull() )
    {
        cvf::ref<cvf::Effect>    eff;
        caf::MeshEffectGenerator CrossSectionEffGen( prefs->defaultFaultGridLineColors() );
        eff = CrossSectionEffGen.generateCachedEffect();

        m_intersectionFaultGridLines->setEffect( eff.p() );
    }
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RivExtrudedCurveIntersectionPartMgr::updateCellResultColor( size_t                   timeStepIndex,
                                                                 const cvf::ScalarMapper* explicitScalarColorMapper,
                                                                 const RivTernaryScalarMapper* explicitTernaryColorMapper )
{
    RivIntersectionResultsColoringTools::calculateIntersectionResultColors( timeStepIndex,
                                                                            !m_isFlattened,
                                                                            m_rimIntersection,
                                                                            m_intersectionGenerator.p(),
                                                                            explicitScalarColorMapper,
                                                                            explicitTernaryColorMapper,
                                                                            m_intersectionFaces.p(),
                                                                            m_intersectionFacesTextureCoords.p() );
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RivIntersectionResultsColoringTools::calculateNodeOrElementNodeBasedGeoMechTextureCoords(
    cvf::Vec2fArray*                                 textureCoords,
    const std::vector<RivIntersectionVertexWeights>& vertexWeights,
    const std::vector<float>&                        resultValues,
    bool                                             isElementNodalResult,
    const RigFemPart*                                femPart,
    const cvf::ScalarMapper*                         mapper )
{
    textureCoords->resize( vertexWeights.size() );

    if ( resultValues.size() == 0 )
    {
        textureCoords->setAll( cvf::Vec2f( 0.0, 1.0f ) );
    }
    else
    {
        cvf::Vec2f* rawPtr = textureCoords->ptr();

        int vxCount = static_cast<int>( vertexWeights.size() );

#pragma omp parallel for schedule( dynamic )
        for ( int triangleVxIdx = 0; triangleVxIdx < vxCount; ++triangleVxIdx )
        {
            float resValue = HUGE_VAL;

            int weightCount = vertexWeights[triangleVxIdx].size();
            if ( weightCount )
            {
                resValue = 0;

                for ( int wIdx = 0; wIdx < weightCount; ++wIdx )
                {
                    size_t resIdx;
                    if ( isElementNodalResult )
                    {
                        resIdx = vertexWeights[triangleVxIdx].vxId( wIdx );
                    }
                    else
                    {
                        resIdx = femPart->nodeIdxFromElementNodeResultIdx( vertexWeights[triangleVxIdx].vxId( wIdx ) );
                    }

                    resValue += resultValues[resIdx] * vertexWeights[triangleVxIdx].weight( wIdx );
                }
            }

            if ( resValue == HUGE_VAL || resValue != resValue ) // a != a is true for NAN's
            {
                rawPtr[triangleVxIdx][1] = 1.0f;
            }
            else
            {
                rawPtr[triangleVxIdx] = mapper->mapToTextureCoord( resValue );
            }
        }
    }
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RivExtrudedCurveIntersectionPartMgr::generatePartGeometry()
{
    if ( m_intersectionGenerator.isNull() ) return;

    bool useBufferObjects = true;
    // Surface geometry
    {
        cvf::ref<cvf::DrawableGeo> geo = m_intersectionGenerator->generateSurface();
        if ( geo.notNull() )
        {
            geo->computeNormals();

            if ( useBufferObjects )
            {
                geo->setRenderMode( cvf::DrawableGeo::BUFFER_OBJECT );
            }

            cvf::ref<cvf::Part> part = new cvf::Part;
            part->setName( "Cross Section" );
            part->setDrawable( geo.p() );

            // Set mapping from triangle face index to cell index
            cvf::ref<RivExtrudedCurveIntersectionSourceInfo> si =
                new RivExtrudedCurveIntersectionSourceInfo( m_intersectionGenerator.p() );
            part->setSourceInfo( si.p() );

            part->updateBoundingBox();
            part->setEnableMask( intersectionCellFaceBit );
            part->setPriority( RivPartPriority::PartType::Intersection );

            m_intersectionFaces = part;
        }
    }

    // Cell Mesh geometry
    {
        cvf::ref<cvf::DrawableGeo> geoMesh = m_intersectionGenerator->createMeshDrawable();
        if ( geoMesh.notNull() )
        {
            if ( useBufferObjects )
            {
                geoMesh->setRenderMode( cvf::DrawableGeo::BUFFER_OBJECT );
            }

            cvf::ref<cvf::Part> part = new cvf::Part;
            part->setName( "Cross Section mesh" );
            part->setDrawable( geoMesh.p() );

            part->updateBoundingBox();
            part->setEnableMask( intersectionCellMeshBit );
            part->setPriority( RivPartPriority::PartType::MeshLines );

            part->setSourceInfo( new RivMeshLinesSourceInfo( m_rimIntersection ) );

            m_intersectionGridLines = part;
        }
    }

    // Fault Mesh geometry
    {
        cvf::ref<cvf::DrawableGeo> geoMesh = m_intersectionGenerator->createFaultMeshDrawable();
        if ( geoMesh.notNull() )
        {
            if ( useBufferObjects )
            {
                geoMesh->setRenderMode( cvf::DrawableGeo::BUFFER_OBJECT );
            }

            cvf::ref<cvf::Part> part = new cvf::Part;
            part->setName( "Cross Section faultmesh" );
            part->setDrawable( geoMesh.p() );

            part->updateBoundingBox();
            part->setEnableMask( intersectionFaultMeshBit );
            part->setPriority( RivPartPriority::PartType::FaultMeshLines );

            part->setSourceInfo( new RivMeshLinesSourceInfo( m_rimIntersection ) );

            m_intersectionFaultGridLines = part;
        }
    }
    createPolyLineParts( useBufferObjects );

    createExtrusionDirParts( useBufferObjects );

    if ( m_isFlattened ) createFaultLabelParts( m_intersectionGenerator->faultMeshLabelAndAnchorPositions() );

    applySingleColorEffect();
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RivExtrudedCurveIntersectionPartMgr::createFaultLabelParts( const std::vector<std::pair<QString, cvf::Vec3d>>& labelAndAnchors )
{
    m_faultMeshLabels     = nullptr;
    m_faultMeshLabelLines = nullptr;

    if ( !labelAndAnchors.size() ) return;

    RimFaultInViewCollection* faultInViewColl = nullptr;

    if ( !m_rimIntersection->activeSeparateResultDefinition() )
    {
        RimEclipseView* eclipseView = nullptr;
        m_rimIntersection->firstAncestorOrThisOfType( eclipseView );
        if ( eclipseView )
        {
            faultInViewColl = eclipseView->faultCollection();

            if ( faultInViewColl && !faultInViewColl->showFaultLabel() ) return;
        }
    }

    cvf::Color3f faultLabelColor = RiaApplication::instance()->preferences()->defaultWellLabelColor();
    if ( faultInViewColl ) faultLabelColor = faultInViewColl->faultLabelColor();

    cvf::Font* font = RiaGuiApplication::instance()->defaultSceneFont();

    std::vector<cvf::Vec3f> lineVertices;

    cvf::ref<cvf::DrawableText> drawableText = new cvf::DrawableText;
    {
        drawableText->setFont( font );
        drawableText->setCheckPosVisible( false );
        drawableText->setDrawBorder( false );
        drawableText->setDrawBackground( false );
        drawableText->setVerticalAlignment( cvf::TextDrawer::BASELINE );
        drawableText->setTextColor( faultLabelColor );
    }

    cvf::BoundingBox bb                    = m_intersectionFaces->boundingBox();
    double           labelZOffset          = bb.extent().z() / 10;
    int              visibleFaultNameCount = 0;

    for ( const auto& labelAndAnchorPair : labelAndAnchors )
    {
        if ( faultInViewColl )
        {
            RimFaultInView* fault = faultInViewColl->findFaultByName( labelAndAnchorPair.first );
            if ( !( fault && fault->showFault() ) ) continue;
        }

        cvf::String cvfString = cvfqt::Utils::toString( labelAndAnchorPair.first );
        cvf::Vec3f  textCoord( labelAndAnchorPair.second );

        textCoord.z() += labelZOffset;
        drawableText->addText( cvfString, textCoord );

        lineVertices.push_back( cvf::Vec3f( labelAndAnchorPair.second ) );
        lineVertices.push_back( textCoord );
        visibleFaultNameCount++;
    }

    if ( visibleFaultNameCount == 0 ) return;

    // Labels part
    {
        cvf::ref<cvf::Part> part = new cvf::Part;
        part->setName( "Fault mesh label : text " );
        part->setDrawable( drawableText.p() );

        cvf::ref<cvf::Effect> eff = new cvf::Effect;

        part->setEffect( eff.p() );
        part->setPriority( RivPartPriority::PartType::Text );
        part->updateBoundingBox();

        m_faultMeshLabels = part;
    }

    // Lines to labels part
    {
        cvf::ref<cvf::Vec3fArray> vertices = new cvf::Vec3fArray;
        vertices->assign( lineVertices );
        cvf::ref<cvf::DrawableGeo> geo = new cvf::DrawableGeo;
        geo->setVertexArray( vertices.p() );

        cvf::ref<cvf::PrimitiveSetDirect> primSet = new cvf::PrimitiveSetDirect( cvf::PT_LINES );
        primSet->setStartIndex( 0 );
        primSet->setIndexCount( vertices->size() );
        geo->addPrimitiveSet( primSet.p() );

        cvf::ref<cvf::Part> part = new cvf::Part;
        part->setName( "Anchor lines for fault mesh labels" );
        part->setDrawable( geo.p() );

        part->updateBoundingBox();

        caf::MeshEffectGenerator gen( RiaApplication::instance()->preferences()->defaultFaultGridLineColors() );
        cvf::ref<cvf::Effect>    eff = gen.generateCachedEffect();

        part->setEffect( eff.p() );
        m_faultMeshLabelLines = part;
    }
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RivExtrudedCurveIntersectionPartMgr::createPolyLineParts( bool useBufferObjects )
{
    // Highlight line

    m_highlightLineAlongPolyline = nullptr;
    m_highlightPointsForPolyline = nullptr;

    if ( m_rimIntersection->type == RimExtrudedCurveIntersection::CS_POLYLINE ||
         m_rimIntersection->type == RimExtrudedCurveIntersection::CS_AZIMUTHLINE )
    {
        {
            cvf::ref<cvf::DrawableGeo> polylineGeo = m_intersectionGenerator->createLineAlongPolylineDrawable();
            if ( polylineGeo.notNull() )
            {
                if ( useBufferObjects )
                {
                    polylineGeo->setRenderMode( cvf::DrawableGeo::BUFFER_OBJECT );
                }

                cvf::ref<cvf::Part> part = new cvf::Part;
                part->setName( "Cross Section Polyline" );
                part->setDrawable( polylineGeo.p() );

                part->updateBoundingBox();
                part->setPriority( RivPartPriority::PartType::Highlight );

                // Always show this part, also when mesh is turned off
                // part->setEnableMask(meshFaultBit);

                cvf::ref<cvf::Effect>    eff;
                caf::MeshEffectGenerator lineEffGen( cvf::Color3::MAGENTA );
                eff = lineEffGen.generateUnCachedEffect();

                cvf::ref<cvf::RenderStateDepth> depth = new cvf::RenderStateDepth;
                depth->enableDepthTest( false );
                eff->setRenderState( depth.p() );

                part->setEffect( eff.p() );

                m_highlightLineAlongPolyline = part;
            }
        }

        cvf::ref<cvf::DrawableGeo> polylinePointsGeo = m_intersectionGenerator->createPointsFromPolylineDrawable();
        if ( polylinePointsGeo.notNull() )
        {
            if ( useBufferObjects )
            {
                polylinePointsGeo->setRenderMode( cvf::DrawableGeo::BUFFER_OBJECT );
            }

            cvf::ref<cvf::Part> part = new cvf::Part;
            part->setName( "Cross Section Polyline" );
            part->setDrawable( polylinePointsGeo.p() );

            part->updateBoundingBox();
            part->setPriority( RivPartPriority::PartType::Highlight );

            // Always show this part, also when mesh is turned off
            // part->setEnableMask(meshFaultBit);

            cvf::ref<cvf::Effect>    eff;
            caf::MeshEffectGenerator lineEffGen( cvf::Color3::MAGENTA );
            eff = lineEffGen.generateUnCachedEffect();

            cvf::ref<cvf::RenderStateDepth> depth = new cvf::RenderStateDepth;
            depth->enableDepthTest( false );
            eff->setRenderState( depth.p() );

            cvf::ref<cvf::RenderStatePoint> pointRendState = new cvf::RenderStatePoint( cvf::RenderStatePoint::FIXED_SIZE );
            pointRendState->setSize( 5.0f );
            eff->setRenderState( pointRendState.p() );

            part->setEffect( eff.p() );

            m_highlightPointsForPolyline = part;
        }
    }
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RivExtrudedCurveIntersectionPartMgr::createExtrusionDirParts( bool useBufferObjects )
{
    m_highlightLineAlongExtrusionDir = nullptr;
    m_highlightPointsForExtrusionDir = nullptr;

    if ( m_rimIntersection->direction() == RimExtrudedCurveIntersection::CS_TWO_POINTS )
    {
        {
            cvf::ref<cvf::DrawableGeo> polylineGeo = m_intersectionGenerator->createLineAlongExtrusionLineDrawable(
                m_rimIntersection->polyLinesForExtrusionDirection() );
            if ( polylineGeo.notNull() )
            {
                if ( useBufferObjects )
                {
                    polylineGeo->setRenderMode( cvf::DrawableGeo::BUFFER_OBJECT );
                }

                cvf::ref<cvf::Part> part = new cvf::Part;
                part->setName( "Cross Section Polyline" );
                part->setDrawable( polylineGeo.p() );

                part->updateBoundingBox();
                part->setPriority( RivPartPriority::PartType::Highlight );

                // Always show this part, also when mesh is turned off
                // part->setEnableMask(meshFaultBit);

                cvf::ref<cvf::Effect>    eff;
                caf::MeshEffectGenerator lineEffGen( cvf::Color3::MAGENTA );
                eff = lineEffGen.generateUnCachedEffect();

                cvf::ref<cvf::RenderStateDepth> depth = new cvf::RenderStateDepth;
                depth->enableDepthTest( false );
                eff->setRenderState( depth.p() );

                part->setEffect( eff.p() );

                m_highlightLineAlongExtrusionDir = part;
            }
        }

        cvf::ref<cvf::DrawableGeo> polylinePointsGeo = m_intersectionGenerator->createPointsFromExtrusionLineDrawable(
            m_rimIntersection->polyLinesForExtrusionDirection() );
        if ( polylinePointsGeo.notNull() )
        {
            if ( useBufferObjects )
            {
                polylinePointsGeo->setRenderMode( cvf::DrawableGeo::BUFFER_OBJECT );
            }

            cvf::ref<cvf::Part> part = new cvf::Part;
            part->setName( "Cross Section Polyline" );
            part->setDrawable( polylinePointsGeo.p() );

            part->updateBoundingBox();
            part->setPriority( RivPartPriority::PartType::Highlight );

            // Always show this part, also when mesh is turned off
            // part->setEnableMask(meshFaultBit);

            cvf::ref<cvf::Effect>    eff;
            caf::MeshEffectGenerator lineEffGen( cvf::Color3::MAGENTA );
            eff = lineEffGen.generateUnCachedEffect();

            cvf::ref<cvf::RenderStateDepth> depth = new cvf::RenderStateDepth;
            depth->enableDepthTest( false );
            eff->setRenderState( depth.p() );

            cvf::ref<cvf::RenderStatePoint> pointRendState = new cvf::RenderStatePoint( cvf::RenderStatePoint::FIXED_SIZE );
            pointRendState->setSize( 5.0f );
            eff->setRenderState( pointRendState.p() );

            part->setEffect( eff.p() );

            m_highlightPointsForExtrusionDir = part;
        }
    }
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RivExtrudedCurveIntersectionPartMgr::appendIntersectionFacesToModel( cvf::ModelBasicList* model,
                                                                          cvf::Transform*      scaleTransform )
{
    if ( m_intersectionFaces.isNull() )
    {
        generatePartGeometry();
    }

    if ( m_intersectionFaces.notNull() )
    {
        m_intersectionFaces->setTransform( scaleTransform );
        model->addPart( m_intersectionFaces.p() );
    }
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RivExtrudedCurveIntersectionPartMgr::appendMeshLinePartsToModel( cvf::ModelBasicList* model,
                                                                      cvf::Transform*      scaleTransform )
{
    if ( m_intersectionGridLines.isNull() )
    {
        generatePartGeometry();
    }

    if ( m_intersectionGridLines.notNull() )
    {
        m_intersectionGridLines->setTransform( scaleTransform );
        model->addPart( m_intersectionGridLines.p() );
    }

    if ( m_intersectionFaultGridLines.notNull() )
    {
        m_intersectionFaultGridLines->setTransform( scaleTransform );
        model->addPart( m_intersectionFaultGridLines.p() );
    }

    if ( m_faultMeshLabelLines.notNull() )
    {
        m_faultMeshLabelLines->setTransform( scaleTransform );
        model->addPart( m_faultMeshLabelLines.p() );
    }

    if ( m_faultMeshLabels.notNull() )
    {
        m_faultMeshLabels->setTransform( scaleTransform );
        model->addPart( m_faultMeshLabels.p() );
    }
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RivExtrudedCurveIntersectionPartMgr::appendPolylinePartsToModel( Rim3dView&           view,
                                                                      cvf::ModelBasicList* model,
                                                                      cvf::Transform*      scaleTransform )
{
    Rim2dIntersectionView* curr2dView = dynamic_cast<Rim2dIntersectionView*>( &view );

    if ( m_rimIntersection->inputPolyLineFromViewerEnabled || ( curr2dView && curr2dView->showDefiningPoints() ) )
    {
        if ( m_highlightLineAlongPolyline.notNull() )
        {
            m_highlightLineAlongPolyline->setTransform( scaleTransform );
            model->addPart( m_highlightLineAlongPolyline.p() );
        }

        if ( m_highlightPointsForPolyline.notNull() )
        {
            m_highlightPointsForPolyline->setTransform( scaleTransform );
            model->addPart( m_highlightPointsForPolyline.p() );
        }
    }

    if ( m_rimIntersection->inputExtrusionPointsFromViewerEnabled )
    {
        if ( m_highlightLineAlongExtrusionDir.notNull() )
        {
            m_highlightLineAlongExtrusionDir->setTransform( scaleTransform );
            model->addPart( m_highlightLineAlongExtrusionDir.p() );
        }

        if ( m_highlightPointsForExtrusionDir.notNull() )
        {
            m_highlightPointsForExtrusionDir->setTransform( scaleTransform );
            model->addPart( m_highlightPointsForExtrusionDir.p() );
        }
    }

    if ( m_rimIntersection->inputTwoAzimuthPointsFromViewerEnabled || ( curr2dView && curr2dView->showDefiningPoints() ) )
    {
        if ( m_highlightLineAlongPolyline.notNull() )
        {
            m_highlightLineAlongPolyline->setTransform( scaleTransform );
            model->addPart( m_highlightLineAlongPolyline.p() );
        }

        if ( m_highlightPointsForPolyline.notNull() )
        {
            m_highlightPointsForPolyline->setTransform( scaleTransform );
            model->addPart( m_highlightPointsForPolyline.p() );
        }
    }
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
const RimExtrudedCurveIntersection* RivExtrudedCurveIntersectionPartMgr::intersection() const
{
    return m_rimIntersection.p();
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
cvf::Mat4d RivExtrudedCurveIntersectionPartMgr::unflattenTransformMatrix( const cvf::Vec3d& intersectionPointFlat ) const
{
    return m_intersectionGenerator->unflattenTransformMatrix( intersectionPointFlat );
}