mirror of
https://github.com/OPM/ResInsight.git
synced 2025-02-25 18:55:39 -06:00
#4683 clang-format on all files in ApplicationCode
This commit is contained in:
Magne Sjaastad
@@ -1,17 +1,17 @@
|
||||
/////////////////////////////////////////////////////////////////////////////////
|
||||
//
|
||||
// Copyright (C) 2016- 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>
|
||||
//
|
||||
// See the GNU General Public License at <http://www.gnu.org/licenses/gpl.html>
|
||||
// for more details.
|
||||
//
|
||||
/////////////////////////////////////////////////////////////////////////////////
|
||||
@@ -25,8 +25,8 @@
|
||||
|
||||
#include "RimEclipseCase.h"
|
||||
#include "RimEclipseView.h"
|
||||
#include "RimSimWellInViewCollection.h"
|
||||
#include "RimSimWellInView.h"
|
||||
#include "RimSimWellInViewCollection.h"
|
||||
|
||||
#include "RiuViewer.h"
|
||||
|
||||
@@ -34,195 +34,193 @@
|
||||
#include "cafEffectGenerator.h"
|
||||
#include "cafPdmFieldCvfColor.h"
|
||||
#include "cafPdmFieldCvfMat4d.h"
|
||||
|
||||
|
||||
#include "cvfDrawableGeo.h"
|
||||
#include "cvfDrawableVectors.h"
|
||||
#include "cvfGeometryBuilderFaceList.h"
|
||||
#include "cvfGeometryBuilderTriangles.h"
|
||||
#include "cvfGeometryUtils.h"
|
||||
#include "cvfModelBasicList.h"
|
||||
#include "cvfObject.h"
|
||||
#include "cvfPart.h"
|
||||
#include "cvfDrawableVectors.h"
|
||||
#include "cvfGeometryBuilderTriangles.h"
|
||||
#include "cvfOpenGLResourceManager.h"
|
||||
#include "cvfPart.h"
|
||||
#include "cvfShaderProgram.h"
|
||||
|
||||
//--------------------------------------------------------------------------------------------------
|
||||
///
|
||||
///
|
||||
//--------------------------------------------------------------------------------------------------
|
||||
RivWellSpheresPartMgr::RivWellSpheresPartMgr(RimEclipseView* reservoirView, RimSimWellInView* well)
|
||||
RivWellSpheresPartMgr::RivWellSpheresPartMgr( RimEclipseView* reservoirView, RimSimWellInView* well )
|
||||
{
|
||||
m_rimReservoirView = reservoirView;
|
||||
m_rimWell = well;
|
||||
m_rimWell = well;
|
||||
}
|
||||
|
||||
//--------------------------------------------------------------------------------------------------
|
||||
///
|
||||
///
|
||||
//--------------------------------------------------------------------------------------------------
|
||||
RivWellSpheresPartMgr::~RivWellSpheresPartMgr()
|
||||
RivWellSpheresPartMgr::~RivWellSpheresPartMgr() {}
|
||||
|
||||
//--------------------------------------------------------------------------------------------------
|
||||
///
|
||||
//--------------------------------------------------------------------------------------------------
|
||||
void RivWellSpheresPartMgr::appendDynamicGeometryPartsToModel( cvf::ModelBasicList* model, size_t frameIndex )
|
||||
{
|
||||
if ( m_rimReservoirView.isNull() ) return;
|
||||
if ( !m_rimReservoirView->eclipseCase() ) return;
|
||||
if ( !m_rimReservoirView->eclipseCase()->eclipseCaseData() ) return;
|
||||
|
||||
}
|
||||
|
||||
|
||||
//--------------------------------------------------------------------------------------------------
|
||||
///
|
||||
//--------------------------------------------------------------------------------------------------
|
||||
void RivWellSpheresPartMgr::appendDynamicGeometryPartsToModel(cvf::ModelBasicList* model, size_t frameIndex)
|
||||
{
|
||||
if (m_rimReservoirView.isNull()) return;
|
||||
if (!m_rimReservoirView->eclipseCase()) return;
|
||||
if (!m_rimReservoirView->eclipseCase()->eclipseCaseData()) return;
|
||||
|
||||
const RigMainGrid* mainGrid = m_rimReservoirView->mainGrid();
|
||||
CVF_ASSERT(mainGrid);
|
||||
CVF_ASSERT( mainGrid );
|
||||
|
||||
RigSimWellData* rigWellResult = m_rimWell->simWellData();
|
||||
if (!rigWellResult) return;
|
||||
if ( !rigWellResult ) return;
|
||||
|
||||
if (!rigWellResult->hasWellResult(frameIndex)) return;
|
||||
if ( !rigWellResult->hasWellResult( frameIndex ) ) return;
|
||||
|
||||
const RigWellResultFrame& wellResultFrame = rigWellResult->wellResultFrame(frameIndex);
|
||||
const RigWellResultFrame& wellResultFrame = rigWellResult->wellResultFrame( frameIndex );
|
||||
|
||||
std::vector<std::pair<cvf::Vec3f, cvf::Color3f> > centerColorPairs;
|
||||
std::vector<std::pair<cvf::Vec3f, cvf::Color3f>> centerColorPairs;
|
||||
|
||||
for (const RigWellResultBranch& wellResultBranch : wellResultFrame.m_wellResultBranches)
|
||||
for ( const RigWellResultBranch& wellResultBranch : wellResultFrame.m_wellResultBranches )
|
||||
{
|
||||
for (const RigWellResultPoint& wellResultPoint : wellResultBranch.m_branchResultPoints)
|
||||
for ( const RigWellResultPoint& wellResultPoint : wellResultBranch.m_branchResultPoints )
|
||||
{
|
||||
size_t gridIndex = wellResultPoint.m_gridIndex;
|
||||
|
||||
if (gridIndex >= mainGrid->gridCount()) continue;
|
||||
|
||||
const RigGridBase* rigGrid = rigGrid = mainGrid->gridByIndex(gridIndex);
|
||||
if ( gridIndex >= mainGrid->gridCount() ) continue;
|
||||
|
||||
const RigGridBase* rigGrid = rigGrid = mainGrid->gridByIndex( gridIndex );
|
||||
|
||||
size_t gridCellIndex = wellResultPoint.m_gridCellIndex;
|
||||
if (gridCellIndex >= rigGrid->cellCount()) continue;
|
||||
if ( gridCellIndex >= rigGrid->cellCount() ) continue;
|
||||
|
||||
const RigCell& rigCell = rigGrid->cell(gridCellIndex);
|
||||
|
||||
cvf::Vec3d center = rigCell.center();
|
||||
cvf::ref<caf::DisplayCoordTransform> transForm = m_rimReservoirView->displayCoordTransform();
|
||||
cvf::Vec3d displayCoord = transForm->transformToDisplayCoord(center);
|
||||
const RigCell& rigCell = rigGrid->cell( gridCellIndex );
|
||||
|
||||
cvf::Color3f color = wellCellColor(wellResultFrame, wellResultPoint);
|
||||
cvf::Vec3d center = rigCell.center();
|
||||
cvf::ref<caf::DisplayCoordTransform> transForm = m_rimReservoirView->displayCoordTransform();
|
||||
cvf::Vec3d displayCoord = transForm->transformToDisplayCoord( center );
|
||||
|
||||
centerColorPairs.push_back(std::make_pair(cvf::Vec3f(displayCoord), color));
|
||||
cvf::Color3f color = wellCellColor( wellResultFrame, wellResultPoint );
|
||||
|
||||
centerColorPairs.push_back( std::make_pair( cvf::Vec3f( displayCoord ), color ) );
|
||||
}
|
||||
}
|
||||
|
||||
cvf::ref<cvf::Part> part = createPart(centerColorPairs, wellResultFrame.m_isOpen);
|
||||
cvf::ref<cvf::Part> part = createPart( centerColorPairs, wellResultFrame.m_isOpen );
|
||||
|
||||
model->addPart(part.p());
|
||||
model->addPart( part.p() );
|
||||
}
|
||||
|
||||
//--------------------------------------------------------------------------------------------------
|
||||
///
|
||||
///
|
||||
//--------------------------------------------------------------------------------------------------
|
||||
cvf::ref<cvf::Part> RivWellSpheresPartMgr::createPart(std::vector<std::pair<cvf::Vec3f, cvf::Color3f> >& centerColorPairs, bool isWellOpen)
|
||||
cvf::ref<cvf::Part> RivWellSpheresPartMgr::createPart( std::vector<std::pair<cvf::Vec3f, cvf::Color3f>>& centerColorPairs,
|
||||
bool isWellOpen )
|
||||
{
|
||||
cvf::ref<cvf::Vec3fArray> vertices = new cvf::Vec3fArray;
|
||||
cvf::ref<cvf::Vec3fArray> vecRes = new cvf::Vec3fArray;
|
||||
cvf::ref<cvf::Color3fArray> colors = new cvf::Color3fArray;
|
||||
cvf::ref<cvf::Vec3fArray> vertices = new cvf::Vec3fArray;
|
||||
cvf::ref<cvf::Vec3fArray> vecRes = new cvf::Vec3fArray;
|
||||
cvf::ref<cvf::Color3fArray> colors = new cvf::Color3fArray;
|
||||
|
||||
size_t numVecs = centerColorPairs.size();
|
||||
vertices->reserve(numVecs);
|
||||
vecRes->reserve(numVecs);
|
||||
colors->reserve(numVecs);
|
||||
vertices->reserve( numVecs );
|
||||
vecRes->reserve( numVecs );
|
||||
colors->reserve( numVecs );
|
||||
|
||||
for (auto centerColorPair : centerColorPairs)
|
||||
for ( auto centerColorPair : centerColorPairs )
|
||||
{
|
||||
vertices->add(centerColorPair.first);
|
||||
vecRes->add(cvf::Vec3f::X_AXIS);
|
||||
colors->add(centerColorPair.second);
|
||||
vertices->add( centerColorPair.first );
|
||||
vecRes->add( cvf::Vec3f::X_AXIS );
|
||||
colors->add( centerColorPair.second );
|
||||
}
|
||||
|
||||
cvf::ref<cvf::DrawableVectors> vectorDrawable;
|
||||
if (RiaGuiApplication::instance()->useShaders())
|
||||
if ( RiaGuiApplication::instance()->useShaders() )
|
||||
{
|
||||
// NOTE: Drawable vectors must be rendered using shaders when the rest of the application is rendered using shaders
|
||||
// Drawing vectors using fixed function when rest of the application uses shaders causes visual artifacts
|
||||
vectorDrawable = new cvf::DrawableVectors("u_transformationMatrix", "u_color");
|
||||
vectorDrawable = new cvf::DrawableVectors( "u_transformationMatrix", "u_color" );
|
||||
}
|
||||
else
|
||||
{
|
||||
vectorDrawable = new cvf::DrawableVectors();
|
||||
}
|
||||
|
||||
vectorDrawable->setVectors(vertices.p(), vecRes.p());
|
||||
vectorDrawable->setColors(colors.p());
|
||||
vectorDrawable->setVectors( vertices.p(), vecRes.p() );
|
||||
vectorDrawable->setColors( colors.p() );
|
||||
|
||||
cvf::GeometryBuilderTriangles builder;
|
||||
double characteristicCellSize = m_rimReservoirView->mainGrid()->characteristicIJCellSize();
|
||||
double characteristicCellSize = m_rimReservoirView->mainGrid()->characteristicIJCellSize();
|
||||
double cellRadius = m_rimReservoirView->wellCollection()->spheresScaleFactor() * characteristicCellSize;
|
||||
|
||||
if (isWellOpen)
|
||||
if ( isWellOpen )
|
||||
{
|
||||
// Increase radius to make sure open connection are slightly larger than closed connections
|
||||
cellRadius = 1.1 * cellRadius;
|
||||
}
|
||||
|
||||
cvf::GeometryUtils::createSphere(cellRadius, 15, 15, &builder);
|
||||
cvf::GeometryUtils::createSphere( cellRadius, 15, 15, &builder );
|
||||
|
||||
vectorDrawable->setGlyph(builder.trianglesUShort().p(), builder.vertices().p());
|
||||
vectorDrawable->setGlyph( builder.trianglesUShort().p(), builder.vertices().p() );
|
||||
|
||||
cvf::ref<cvf::Part> part = new cvf::Part;
|
||||
part->setDrawable(vectorDrawable.p());
|
||||
part->setDrawable( vectorDrawable.p() );
|
||||
|
||||
cvf::ref<cvf::Effect> eff = new cvf::Effect;
|
||||
if (RiaGuiApplication::instance()->useShaders())
|
||||
if ( RiaGuiApplication::instance()->useShaders() )
|
||||
{
|
||||
if (m_rimReservoirView->viewer())
|
||||
if ( m_rimReservoirView->viewer() )
|
||||
{
|
||||
cvf::ref<cvf::OpenGLContext> oglContext = m_rimReservoirView->viewer()->cvfOpenGLContext();
|
||||
cvf::OpenGLResourceManager* resourceManager = oglContext->resourceManager();
|
||||
cvf::ref<cvf::ShaderProgram> vectorProgram = resourceManager->getLinkedVectorDrawerShaderProgram(oglContext.p());
|
||||
cvf::ref<cvf::OpenGLContext> oglContext = m_rimReservoirView->viewer()->cvfOpenGLContext();
|
||||
cvf::OpenGLResourceManager* resourceManager = oglContext->resourceManager();
|
||||
cvf::ref<cvf::ShaderProgram> vectorProgram = resourceManager->getLinkedVectorDrawerShaderProgram(
|
||||
oglContext.p() );
|
||||
|
||||
eff->setShaderProgram(vectorProgram.p());
|
||||
eff->setShaderProgram( vectorProgram.p() );
|
||||
}
|
||||
}
|
||||
|
||||
part->setEffect(eff.p());
|
||||
part->setEffect( eff.p() );
|
||||
|
||||
return part;
|
||||
}
|
||||
|
||||
//--------------------------------------------------------------------------------------------------
|
||||
///
|
||||
///
|
||||
//--------------------------------------------------------------------------------------------------
|
||||
cvf::Color3f RivWellSpheresPartMgr::wellCellColor(const RigWellResultFrame& wellResultFrame, const RigWellResultPoint& wellResultPoint)
|
||||
cvf::Color3f RivWellSpheresPartMgr::wellCellColor( const RigWellResultFrame& wellResultFrame,
|
||||
const RigWellResultPoint& wellResultPoint )
|
||||
{
|
||||
// Colours should be synchronized with RivWellPipesPartMgr::updatePipeResultColor
|
||||
|
||||
cvf::Color3f cellColor(cvf::Color3f::GRAY);
|
||||
cvf::Color3f cellColor( cvf::Color3f::GRAY );
|
||||
|
||||
RimSimWellInViewCollection* wellColl = nullptr;
|
||||
if (m_rimWell)
|
||||
if ( m_rimWell )
|
||||
{
|
||||
m_rimWell->firstAncestorOrThisOfType(wellColl);
|
||||
m_rimWell->firstAncestorOrThisOfType( wellColl );
|
||||
}
|
||||
|
||||
if (wellColl)
|
||||
if ( wellColl )
|
||||
{
|
||||
if (wellResultPoint.m_isOpen)
|
||||
if ( wellResultPoint.m_isOpen )
|
||||
{
|
||||
switch (wellResultFrame.m_productionType)
|
||||
switch ( wellResultFrame.m_productionType )
|
||||
{
|
||||
case RigWellResultFrame::PRODUCER:
|
||||
cellColor = cvf::Color3f::GREEN;
|
||||
break;
|
||||
case RigWellResultFrame::OIL_INJECTOR:
|
||||
cellColor = cvf::Color3f::RED;
|
||||
break;
|
||||
case RigWellResultFrame::GAS_INJECTOR:
|
||||
cellColor = cvf::Color3f::RED;
|
||||
break;
|
||||
case RigWellResultFrame::WATER_INJECTOR:
|
||||
cellColor = cvf::Color3f::BLUE;
|
||||
break;
|
||||
case RigWellResultFrame::PRODUCER:
|
||||
cellColor = cvf::Color3f::GREEN;
|
||||
break;
|
||||
case RigWellResultFrame::OIL_INJECTOR:
|
||||
cellColor = cvf::Color3f::RED;
|
||||
break;
|
||||
case RigWellResultFrame::GAS_INJECTOR:
|
||||
cellColor = cvf::Color3f::RED;
|
||||
break;
|
||||
case RigWellResultFrame::WATER_INJECTOR:
|
||||
cellColor = cvf::Color3f::BLUE;
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return cellColor;
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user