ResInsight/ApplicationCode/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 "RimAnnotationCollection.h"
#include "RimReachCircleAnnotation.h"

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

#include "cafEffectGenerator.h"

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

#include <cmath>

//--------------------------------------------------------------------------------------------------
/// 
//--------------------------------------------------------------------------------------------------
RivReachCircleAnnotationPartMgr::RivReachCircleAnnotationPartMgr(RimReachCircleAnnotation* annotation)
: m_rimAnnotation(annotation)
{
}

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

}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RivReachCircleAnnotationPartMgr::buildParts(const caf::DisplayCoordTransform* displayXf, bool doFlatten, double xOffset)
{
    clearAllGeometry();

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

    Vec3d centerPositionInDomain = m_rimAnnotation->centerPoint();

    {
        auto* collection = dynamic_cast<RimAnnotationCollection*>(annotationCollection());
        if (collection && collection->snapAnnotations())
        {
            centerPositionInDomain.z() = collection->annotationPlaneZ();
        }
    }

    Vec3d   centerPosition = displayXf->transformToDisplayCoord(centerPositionInDomain);
    double  radius         = m_rimAnnotation->radius();
    auto    lineColor      = m_rimAnnotation->appearance()->color();
    auto    isDashedLine   = m_rimAnnotation->appearance()->isDashed();
    auto    lineThickness = m_rimAnnotation->appearance()->thickness();

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

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

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

        caf::MeshEffectGenerator effgen(lineColor);
        effgen.setLineWidth(lineThickness);
        if (isDashedLine) effgen.setLineStipple(true);
        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
    {
        double symbolSize = 20;
        double xMin = centerPosition.x() - symbolSize / 2.0;
        double xMax = xMin + symbolSize;
        double yMin = centerPosition.y() - symbolSize / 2.0;
        double yMax = yMin + symbolSize;
        double z = centerPosition.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->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)
{
    if (m_rimAnnotation.isNull()) return;
    if (!m_rimAnnotation->isActive()) return;

    if (!validateAnnotation(m_rimAnnotation)) return;

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

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

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
RimAnnotationCollectionBase* RivReachCircleAnnotationPartMgr::annotationCollection() const
{
    RimAnnotationCollectionBase* coll;
    m_rimAnnotation->firstAncestorOrThisOfType(coll);
    return coll;
}