ResInsight/ApplicationCode/ModelVisualization/GridBox/RivPatchGenerator.cpp
2018-10-17 13:52:48 +02:00

107 lines
3.6 KiB
C++

/////////////////////////////////////////////////////////////////////////////////
//
// 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 "RivPatchGenerator.h"
#include "cvfGeometryUtils.h"
#include "cvfArray.h"
//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
RivPatchGenerator::RivPatchGenerator()
: m_origin(0, 0, 0),
m_axisU(cvf::Vec3d::X_AXIS),
m_axisV(cvf::Vec3d::Y_AXIS),
m_useQuads(true),
m_windingCCW(true)
{
}
//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RivPatchGenerator::setOrigin(const cvf::Vec3d& origin)
{
m_origin = origin;
}
//--------------------------------------------------------------------------------------------------
/// Set the axes
///
/// The specified axes will be normalized
//--------------------------------------------------------------------------------------------------
void RivPatchGenerator::setAxes(const cvf::Vec3d& axisU, const cvf::Vec3d& axisV)
{
m_axisU = axisU.getNormalized();
m_axisV = axisV.getNormalized();
}
//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RivPatchGenerator::setSubdivisions(const std::vector<double>& uValues, const std::vector<double>& vValues)
{
m_uValues = uValues;
m_vValues = vValues;
}
//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RivPatchGenerator::generate(cvf::GeometryBuilder* builder)
{
CVF_ASSERT(m_uValues.size() > 0);
CVF_ASSERT(m_vValues.size() > 0);
size_t numVertices = m_uValues.size() * m_vValues.size();
cvf::Vec3fArray vertices;
vertices.reserve(numVertices);
for (size_t v = 0; v < m_vValues.size(); v++)
{
cvf::Vec3d rowOrigo(m_origin + m_axisV * m_vValues[v]);
for (size_t u = 0; u < m_uValues.size(); u++)
{
vertices.add(cvf::Vec3f(rowOrigo + m_axisU * m_uValues[u]));
}
}
cvf::uint baseNodeIdx = builder->addVertices(vertices);
if (m_useQuads)
{
cvf::UIntArray conn;
cvf::GeometryUtils::tesselatePatchAsQuads(static_cast<int>(m_uValues.size()), static_cast<int>(m_vValues.size()), baseNodeIdx, m_windingCCW, &conn);
builder->addQuads(conn);
}
else
{
cvf::UIntArray conn;
cvf::GeometryUtils::tesselatePatchAsTriangles(static_cast<int>(m_uValues.size()), static_cast<int>(m_vValues.size()), baseNodeIdx, m_windingCCW, &conn);
builder->addTriangles(conn);
}
}