mirror of
https://github.com/OPM/ResInsight.git
synced 2025-01-23 23:13:39 -06:00
136 lines
5.3 KiB
C++
136 lines
5.3 KiB
C++
/////////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
// Copyright (C) Statoil ASA, 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 "RivScalarMapperUtils.h"
|
|
|
|
#include "RimCellEdgeResultSlot.h"
|
|
#include "RimLegendConfig.h"
|
|
#include "RimReservoirView.h"
|
|
#include "RimResultSlot.h"
|
|
#include "RimTernaryLegendConfig.h"
|
|
|
|
#include "RivCellEdgeEffectGenerator.h"
|
|
#include "RivTernaryScalarMapper.h"
|
|
#include "RivTernaryScalarMapperEffectGenerator.h"
|
|
|
|
#include "cafEffectGenerator.h"
|
|
|
|
#include "cvfDrawableGeo.h"
|
|
#include "cvfPart.h"
|
|
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
void RivScalarMapperUtils::applyTextureResultsToPart(cvf::Part* part, cvf::Vec2fArray* textureCoords, const cvf::ScalarMapper* mapper, float opacityLevel)
|
|
{
|
|
CVF_ASSERT(part && textureCoords && mapper);
|
|
|
|
cvf::DrawableGeo* dg = dynamic_cast<cvf::DrawableGeo*>(part->drawable());
|
|
if (dg) dg->setTextureCoordArray(textureCoords);
|
|
|
|
cvf::ref<cvf::Effect> scalarEffect = RivScalarMapperUtils::createScalarMapperEffect(mapper, opacityLevel);
|
|
part->setEffect(scalarEffect.p());
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
void RivScalarMapperUtils::applyTernaryTextureResultsToPart(cvf::Part* part, cvf::Vec2fArray* textureCoords, const RivTernaryScalarMapper* mapper, float opacityLevel)
|
|
{
|
|
CVF_ASSERT(part && textureCoords && mapper);
|
|
|
|
cvf::DrawableGeo* dg = dynamic_cast<cvf::DrawableGeo*>(part->drawable());
|
|
if (dg) dg->setTextureCoordArray(textureCoords);
|
|
|
|
cvf::ref<cvf::Effect> scalarEffect = RivScalarMapperUtils::createTernaryScalarMapperEffect(mapper, opacityLevel);
|
|
part->setEffect(scalarEffect.p());
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
cvf::ref<cvf::Effect> RivScalarMapperUtils::createCellEdgeEffect(cvf::DrawableGeo* dg,
|
|
const cvf::StructGridQuadToCellFaceMapper* quadToCellFaceMapper,
|
|
size_t gridIndex,
|
|
size_t timeStepIndex,
|
|
RimResultSlot* cellResultSlot,
|
|
RimCellEdgeResultSlot* cellEdgeResultSlot,
|
|
float opacityLevel,
|
|
cvf::Color3f defaultColor)
|
|
{
|
|
CellEdgeEffectGenerator cellFaceEffectGen(cellEdgeResultSlot->legendConfig()->scalarMapper());
|
|
|
|
if (cellResultSlot->isTernarySaturationSelected())
|
|
{
|
|
RivCellEdgeGeometryGenerator::addTernaryCellEdgeResultsToDrawableGeo(timeStepIndex, cellResultSlot, cellEdgeResultSlot,
|
|
quadToCellFaceMapper, dg, gridIndex, opacityLevel);
|
|
|
|
RivTernaryScalarMapper* ternaryCellScalarMapper = cellResultSlot->ternaryLegendConfig()->scalarMapper();
|
|
cellFaceEffectGen.setTernaryScalarMapper(ternaryCellScalarMapper);
|
|
}
|
|
else
|
|
{
|
|
if (cellResultSlot->hasResult())
|
|
{
|
|
RivCellEdgeGeometryGenerator::addCellEdgeResultsToDrawableGeo(timeStepIndex, cellResultSlot, cellEdgeResultSlot,
|
|
quadToCellFaceMapper, dg, gridIndex, opacityLevel);
|
|
|
|
cvf::ScalarMapper* cellScalarMapper = cellResultSlot->legendConfig()->scalarMapper();
|
|
cellFaceEffectGen.setScalarMapper(cellScalarMapper);
|
|
}
|
|
}
|
|
|
|
cellFaceEffectGen.setOpacityLevel(opacityLevel);
|
|
cellFaceEffectGen.setDefaultCellColor(defaultColor);
|
|
|
|
cvf::ref<cvf::Effect> eff = cellFaceEffectGen.generateEffect();
|
|
return eff;
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
cvf::ref<cvf::Effect> RivScalarMapperUtils::createScalarMapperEffect(const cvf::ScalarMapper* mapper, float opacityLevel)
|
|
{
|
|
CVF_ASSERT(mapper);
|
|
|
|
caf::PolygonOffset polygonOffset = caf::PO_1;
|
|
caf::ScalarMapperEffectGenerator scalarEffgen(mapper, polygonOffset);
|
|
scalarEffgen.setOpacityLevel(opacityLevel);
|
|
cvf::ref<cvf::Effect> scalarEffect = scalarEffgen.generateEffect();
|
|
|
|
return scalarEffect;
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
cvf::ref<cvf::Effect> RivScalarMapperUtils::createTernaryScalarMapperEffect(const RivTernaryScalarMapper* mapper, float opacityLevel)
|
|
{
|
|
CVF_ASSERT(mapper);
|
|
|
|
caf::PolygonOffset polygonOffset = caf::PO_1;
|
|
RivTernaryScalarMapperEffectGenerator scalarEffgen(mapper, polygonOffset);
|
|
scalarEffgen.setOpacityLevel(opacityLevel);
|
|
cvf::ref<cvf::Effect> scalarEffect = scalarEffgen.generateEffect();
|
|
|
|
return scalarEffect;
|
|
}
|
|
|