ResInsight/ApplicationLibCode/ModelVisualization/RivReachCircleAnnotationPartMgr.cpp

/////////////////////////////////////////////////////////////////////////////////
//
//  Copyright (C) 2011-     Statoil ASA
//  Copyright (C) 2013-     Ceetron Solutions AS
//  Copyright (C) 2011-2012 Ceetron 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 "RivReachCircleAnnotationPartMgr.h"

#include "RiaBoundingBoxTools.h"

#include "Rim3dView.h"
#include "RimAnnotationInViewCollection.h"
#include "RimReachCircleAnnotation.h"
#include "RimReachCircleAnnotationInView.h"

#include "RivPartPriority.h"
#include "RivPolylineGenerator.h"
#include "RivReachCircleAnnotationSourceInfo.h"

#include "cafEffectGenerator.h"

#include "cafDisplayCoordTransform.h"
#include "cvfDrawableGeo.h"
#include "cvfModelBasicList.h"
#include "cvfPart.h"

#include <cmath>

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
RivReachCircleAnnotationPartMgr::RivReachCircleAnnotationPartMgr( Rim3dView* view, RimReachCircleAnnotationInView* annotationInView )
    : m_rimView( view )
    , m_rimAnnotationInView( annotationInView )
{
}

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

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RivReachCircleAnnotationPartMgr::buildParts( const caf::DisplayCoordTransform* displayXf, bool doFlatten, double xOffset )
{
    auto rimAnnotation = m_rimAnnotationInView->sourceAnnotation();
    clearAllGeometry();

    cvf::ref<RivReachCircleAnnotationSourceInfo> sourceInfo = new RivReachCircleAnnotationSourceInfo( rimAnnotation );

    auto lineColor     = rimAnnotation->appearance()->color();
    auto isDashedLine  = rimAnnotation->appearance()->isDashed();
    auto lineThickness = rimAnnotation->appearance()->thickness();

    // Circle part
    auto* collection = annotationCollection();
    if ( collection )
    {
        std::vector<Vec3d> pointsInDomain = computeCirclePointsInDomain( collection->snapAnnotations(), collection->annotationPlaneZ() );
        std::vector<cvf::Vec3d> points    = displayXf->transformToDisplayCoords( pointsInDomain );

        cvf::ref<cvf::DrawableGeo> drawableGeo = RivPolylineGenerator::createLineAlongPolylineDrawable( points );

        cvf::ref<cvf::Part> part = new cvf::Part;
        part->setName( "RivReachCircleAnnotationPartMgr" );
        part->setDrawable( drawableGeo.p() );

        caf::MeshEffectGenerator effgen( lineColor );
        effgen.setLineWidth( lineThickness );
        if ( isDashedLine ) effgen.setLineStipple( true ); // Currently, dashed lines are not supported
        cvf::ref<cvf::Effect> eff = effgen.generateUnCachedEffect();

        part->setEffect( eff.p() );
        part->setPriority( RivPartPriority::PartType::MeshLines );
        part->setSourceInfo( sourceInfo.p() );

        m_circlePart = part;
    }

    // Center point part
    {
        auto                            centerPos   = displayXf->transformToDisplayCoord( rimAnnotation->centerPoint() );
        double                          symbolSize  = 20;
        double                          xMin        = centerPos.x() - symbolSize / 2.0;
        double                          xMax        = xMin + symbolSize;
        double                          yMin        = centerPos.y() - symbolSize / 2.0;
        double                          yMax        = yMin + symbolSize;
        double                          z           = centerPos.z();
        std::vector<Vec3d>              line1       = { { xMin, yMin, z }, { xMax, yMax, z } };
        std::vector<Vec3d>              line2       = { { xMax, yMin, z }, { xMin, yMax, z } };
        std::vector<std::vector<Vec3d>> symbol      = { line1, line2 };
        cvf::ref<cvf::DrawableGeo>      drawableGeo = RivPolylineGenerator::createLineAlongPolylineDrawable( symbol );

        cvf::ref<cvf::Part> part = new cvf::Part;
        part->setName( "RivReachCircleAnnotationPartMgr" );
        part->setDrawable( drawableGeo.p() );

        caf::MeshEffectGenerator effgen( lineColor );
        effgen.setLineWidth( 2 );
        cvf::ref<cvf::Effect> eff = effgen.generateUnCachedEffect();

        part->setEffect( eff.p() );
        part->setPriority( RivPartPriority::PartType::MeshLines );
        part->setSourceInfo( sourceInfo.p() );

        m_centerPointPart = part;
    }
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RivReachCircleAnnotationPartMgr::clearAllGeometry()
{
    m_circlePart      = nullptr;
    m_centerPointPart = nullptr;
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RivReachCircleAnnotationPartMgr::appendDynamicGeometryPartsToModel( cvf::ModelBasicList*              model,
                                                                         const caf::DisplayCoordTransform* displayXf,
                                                                         const cvf::BoundingBox&           boundingBox )
{
    if ( m_rimAnnotationInView.isNull() || !m_rimAnnotationInView->sourceAnnotation() ) return;
    if ( !m_rimAnnotationInView->isVisible() ) return;

    // Check bounding box
    if ( !isCircleInBoundingBox( boundingBox ) ) return;

    if ( !validateAnnotation( m_rimAnnotationInView->sourceAnnotation() ) ) return;

    buildParts( displayXf, false, 0.0 );
    model->addPart( m_circlePart.p() );
    model->addPart( m_centerPointPart.p() );
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
bool RivReachCircleAnnotationPartMgr::validateAnnotation( const RimReachCircleAnnotation* annotation ) const
{
    auto a = m_rimAnnotationInView->sourceAnnotation();
    return a->centerPoint() != cvf::Vec3d::ZERO && a->radius() > 0.0;
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
bool RivReachCircleAnnotationPartMgr::isCircleInBoundingBox( const cvf::BoundingBox& boundingBox )
{
    auto coll = annotationCollection();
    if ( !coll ) return false;

    auto effectiveBoundingBox = RiaBoundingBoxTools::inflate( boundingBox, 3 );
    auto points               = computeCirclePointsInDomain( coll->snapAnnotations(), coll->annotationPlaneZ() );
    for ( const auto& pt : points )
    {
        if ( effectiveBoundingBox.contains( pt ) ) return true;
    }
    return false;
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
std::vector<cvf::Vec3d> RivReachCircleAnnotationPartMgr::computeCirclePointsInDomain( bool snapToPlaneZ, double planeZ )
{
    int  numPoints = 36;
    auto centerPos = m_rimAnnotationInView->sourceAnnotation()->centerPoint();
    auto radius    = m_rimAnnotationInView->sourceAnnotation()->radius();

    if ( snapToPlaneZ )
    {
        centerPos.z() = planeZ;
    }

    std::vector<Vec3d> points;
    for ( int i = 0; i < numPoints; i++ )
    {
        double rad = 2 * cvf::PI_D * (double)i / (double)numPoints;
        Vec3d  pt( centerPos.x() + cos( rad ) * radius, centerPos.y() + sin( rad ) * radius, centerPos.z() );
        points.push_back( pt );
    }
    points.push_back( points.front() );
    return points;
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
RimAnnotationInViewCollection* RivReachCircleAnnotationPartMgr::annotationCollection() const
{
    auto colls = m_rimView->descendantsIncludingThisOfType<RimAnnotationInViewCollection>();
    return !colls.empty() ? colls.front() : nullptr;
}