ResInsight/ApplicationLibCode/ModelVisualization/GridBox/RivPatchGenerator.cpp
2021-01-11 15:27:45 +01:00

111 lines
4.1 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 "cvfArray.h"
#include "cvfGeometryUtils.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 );
}
}