Files
ResInsight/ApplicationCode/ModelVisualization/Riv3dWellLogPlanePartMgr.cpp
2018-03-09 15:42:23 +01:00

141 lines
5.3 KiB
C++

/////////////////////////////////////////////////////////////////////////////////
//
// Copyright (C) 2018- Statoil 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.
//
/////////////////////////////////////////////////////////////////////////////////
#include "Riv3dWellLogPlanePartMgr.h"
#include "RigWellPath.h"
#include "Riv3dWellLogCurveGeomertyGenerator.h"
#include "cafDisplayCoordTransform.h"
#include "cafEffectGenerator.h"
#include "cvfModelBasicList.h"
#include "cvfPart.h"
//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
Riv3dWellLogPlanePartMgr::Riv3dWellLogPlanePartMgr(RigWellPath* wellPathGeometry)
:m_wellPathGeometry(wellPathGeometry)
{
}
//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void Riv3dWellLogPlanePartMgr::append3dWellLogCurvesToModel(cvf::ModelBasicList* model,
const caf::DisplayCoordTransform* displayCoordTransform,
std::vector<Rim3dWellLogCurve*> rim3dWellLogCurves)
{
if (rim3dWellLogCurves.empty()) return;
m_3dWellLogCurveGeometryGenerator = new Riv3dWellLogCurveGeometryGenerator;
std::vector<cvf::uint> indices = createPolylineIndices(m_wellPathGeometry->m_wellPathPoints.size());
for (Rim3dWellLogCurve* rim3dWellLogCurve : rim3dWellLogCurves)
{
std::vector<cvf::Vec3f> vertices = createCurveVertices(rim3dWellLogCurve, displayCoordTransform);
cvf::ref<cvf::Drawable> drawable = m_3dWellLogCurveGeometryGenerator->createDrawable(vertices, indices);
caf::SurfaceEffectGenerator surfaceGen(cvf::Color4f(255, 0, 0, 0.5), caf::PO_1);
cvf::ref<cvf::Effect> effect = surfaceGen.generateCachedEffect();
cvf::ref<cvf::Part> part = new cvf::Part;
part->setDrawable(drawable.p());
part->setEffect(effect.p());
if (part.notNull())
{
model->addPart(part.p());
}
}
}
//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
std::vector<cvf::Vec3f> Riv3dWellLogPlanePartMgr::createCurveVertices(const Rim3dWellLogCurve* rim3dWellLogCurve,
const caf::DisplayCoordTransform* displayCoordTransform)
{
std::vector<cvf::Vec3d> wellPathPoints = m_wellPathGeometry->m_wellPathPoints;
std::vector<cvf::Vec3f> vertices;
vertices.resize(wellPathPoints.size());
std::vector<cvf::Vec3d> curveNormals;
curveNormals.reserve(wellPathPoints.size());
for (size_t i = 0; i < wellPathPoints.size() - 1; i++)
{
cvf::Vec3d wellPathSegment = wellPathPoints[i + 1] - wellPathPoints[i];
cvf::Vec3d normVec;
if (rim3dWellLogCurve->drawPlane() == Rim3dWellLogCurve::HORIZONTAL_LEFT)
{
normVec = -wellPathSegment.perpendicularVector();
}
else
{
normVec = wellPathSegment.perpendicularVector();
}
curveNormals.push_back((wellPathSegment ^ normVec).getNormalized()*30);
}
cvf::Vec3d wellPathSegment = wellPathPoints[wellPathPoints.size()-1] - wellPathPoints[wellPathPoints.size() - 2];
cvf::Vec3d normVec = wellPathSegment.perpendicularVector();
curveNormals.push_back((wellPathSegment ^ normVec).getNormalized() * 30);
for (size_t i = 0; i < curveNormals.size(); i++)
{
cvf::Mat4d transMat;
transMat.setTranslation(curveNormals[i]);
vertices[i] = cvf::Vec3f(displayCoordTransform->transformToDisplayCoord(wellPathPoints[i].getTransformedPoint(transMat)));
}
// cvf::Mat4d transMat;
// transMat.setTranslation(cvf::Vec3d(0, 0, 10));
//
// for (size_t i = 0; i < wellPathPoints.size(); i++)
// {
// vertices[i] = cvf::Vec3f(displayCoordTransform->transformToDisplayCoord(wellPathPoints[i].getTransformedPoint(transMat)));
// }
return vertices;
}
//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
std::vector<cvf::uint> Riv3dWellLogPlanePartMgr::createPolylineIndices(size_t vertexCount)
{
std::vector<cvf::uint> indices;
indices.resize((vertexCount - 1) * 2);
cvf::uint counter = 0;
for (size_t i = 0; i < indices.size(); i++)
{
indices[i] = counter;
if (i % 2 == 0) counter++;
}
return indices;
}