mirror of
https://github.com/OPM/ResInsight.git
synced 2025-01-01 03:37:15 -06:00
405 lines
18 KiB
C++
405 lines
18 KiB
C++
/////////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
// Copyright (C) 2017- 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 "RiuPlotAnnotationTool.h"
|
|
|
|
#include "RimPlotAxisAnnotation.h"
|
|
#include "RiuGuiTheme.h"
|
|
|
|
#include "cafCategoryMapper.h"
|
|
#include "cvfMath.h"
|
|
|
|
#include "qwt_plot.h"
|
|
#include "qwt_plot_zoneitem.h"
|
|
#include "qwt_text.h"
|
|
|
|
#include <QString>
|
|
|
|
#include <algorithm>
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
RiuPlotAnnotationTool::~RiuPlotAnnotationTool()
|
|
{
|
|
detachAllAnnotations();
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
void RiuPlotAnnotationTool::attachNamedRegions( QwtPlot* plot,
|
|
const std::vector<QString>& names,
|
|
RiaDefines::Orientation depthOrientation,
|
|
const std::vector<std::pair<double, double>>& regionRanges,
|
|
RiaDefines::RegionDisplay regionDisplay,
|
|
const caf::ColorTable& colorTable,
|
|
int shadingAlphaByte,
|
|
bool showNames /*= true */,
|
|
RiaDefines::TrackSpan trackSpan /*= FULL_WIDTH*/,
|
|
const std::vector<Qt::BrushStyle>& brushStyles /* = {}*/,
|
|
int fontSize )
|
|
{
|
|
if ( names.size() != regionRanges.size() ) return;
|
|
m_plot = plot;
|
|
|
|
double delta = 0.5;
|
|
|
|
std::vector<int> categoryIndices( names.size() );
|
|
std::iota( categoryIndices.begin(), categoryIndices.end(), 0 );
|
|
|
|
caf::CategoryMapper catMapper;
|
|
catMapper.setCategories( categoryIndices );
|
|
catMapper.setInterpolateColors( colorTable.color3ubArray() );
|
|
|
|
RiaDefines::Orientation annotationOrientation = RiaDefines::Orientation::HORIZONTAL;
|
|
if ( depthOrientation == RiaDefines::Orientation::HORIZONTAL ) annotationOrientation = RiaDefines::Orientation::VERTICAL;
|
|
|
|
for ( size_t i = 0; i < names.size(); i++ )
|
|
{
|
|
auto* line( new QwtPlotMarker() );
|
|
|
|
QString name;
|
|
if ( showNames && !names[i].isEmpty() )
|
|
{
|
|
name = names[i];
|
|
if ( ( regionDisplay & RiaDefines::COLOR_SHADING ) == 0 && names[i].toLower().indexOf( "top" ) == -1 )
|
|
{
|
|
name += " Top";
|
|
}
|
|
}
|
|
if ( regionDisplay & RiaDefines::COLOR_SHADING )
|
|
{
|
|
cvf::Color3ub cvfColor = catMapper.mapToColor( static_cast<double>( i ) );
|
|
QColor shadingColor( cvfColor.r(), cvfColor.g(), cvfColor.b(), shadingAlphaByte );
|
|
|
|
auto* shading = new QwtPlotZoneItem();
|
|
|
|
if ( depthOrientation == RiaDefines::Orientation::HORIZONTAL )
|
|
shading->setOrientation( Qt::Vertical );
|
|
else
|
|
shading->setOrientation( Qt::Horizontal );
|
|
|
|
shading->setInterval( regionRanges[i].first, regionRanges[i].second );
|
|
shading->setPen( shadingColor, 0.0, Qt::NoPen );
|
|
QBrush brush( shadingColor );
|
|
if ( i < brushStyles.size() )
|
|
{
|
|
brush.setStyle( brushStyles[i] );
|
|
}
|
|
shading->setBrush( brush );
|
|
shading->attach( m_plot );
|
|
shading->setZ( -100.0 );
|
|
shading->setXAxis( QwtAxis::XTop );
|
|
m_plotItems.push_back( shading );
|
|
}
|
|
|
|
QColor lineColor( 0, 0, 0, 0 );
|
|
QColor textColor( 0, 0, 0, 255 );
|
|
if ( regionDisplay & RiaDefines::DARK_LINES || regionDisplay & RiaDefines::COLORED_LINES || regionDisplay & RiaDefines::LIGHT_LINES )
|
|
{
|
|
cvf::Color3ub cvfColor = catMapper.mapToColor( static_cast<double>( i ) );
|
|
QColor cycledColor( cvfColor.r(), cvfColor.g(), cvfColor.b() );
|
|
|
|
if ( regionDisplay & RiaDefines::DARK_LINES )
|
|
{
|
|
lineColor = QColor( 50, 50, 100 );
|
|
}
|
|
else if ( regionDisplay & RiaDefines::LIGHT_LINES )
|
|
{
|
|
lineColor = QColor( 200, 200, 200 );
|
|
}
|
|
else
|
|
{
|
|
lineColor = cycledColor;
|
|
}
|
|
textColor = lineColor;
|
|
}
|
|
Qt::Alignment horizontalAlignment = trackTextAlignment( trackSpan );
|
|
|
|
RiuPlotAnnotationTool::setLineProperties( line,
|
|
name,
|
|
annotationOrientation,
|
|
regionRanges[i].first,
|
|
Qt::DashLine,
|
|
lineColor,
|
|
textColor,
|
|
horizontalAlignment,
|
|
fontSize );
|
|
line->attach( m_plot );
|
|
m_plotItems.push_back( line );
|
|
|
|
if ( ( i != names.size() - 1 ) && cvf::Math::abs( regionRanges[i].second - regionRanges[i + 1].first ) > delta )
|
|
{
|
|
auto* bottomLine( new QwtPlotMarker() );
|
|
|
|
RiuPlotAnnotationTool::setLineProperties( bottomLine,
|
|
QString(),
|
|
annotationOrientation,
|
|
regionRanges[i].second,
|
|
Qt::DashLine,
|
|
lineColor,
|
|
textColor,
|
|
Qt::AlignRight,
|
|
fontSize );
|
|
bottomLine->attach( m_plot );
|
|
m_plotItems.push_back( bottomLine );
|
|
}
|
|
}
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
void RiuPlotAnnotationTool::attachWellPicks( QwtPlot* plot, const std::vector<QString>& names, const std::vector<double>& yPositions )
|
|
{
|
|
detachAllAnnotations();
|
|
|
|
if ( names.size() != yPositions.size() ) return;
|
|
m_plot = plot;
|
|
|
|
for ( size_t i = 0; i < names.size(); i++ )
|
|
{
|
|
auto* line( new QwtPlotMarker() );
|
|
RiuPlotAnnotationTool::setLineProperties( line, names[i], RiaDefines::Orientation::HORIZONTAL, yPositions[i] );
|
|
line->attach( m_plot );
|
|
m_plotItems.push_back( line );
|
|
}
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
void RiuPlotAnnotationTool::attachAnnotationLine( QwtPlot* plot,
|
|
const QColor& color,
|
|
const QString& annotationText,
|
|
Qt::PenStyle penStyle,
|
|
const double position,
|
|
RiaDefines::Orientation orientation )
|
|
{
|
|
m_plot = plot;
|
|
|
|
auto* line( new QwtPlotMarker() );
|
|
|
|
auto textColor = color;
|
|
if ( orientation == RiaDefines::Orientation::VERTICAL )
|
|
{
|
|
textColor = RiuGuiTheme::getColorByVariableName( "textColor" );
|
|
}
|
|
|
|
RiuPlotAnnotationTool::setLineProperties( line, annotationText, orientation, position, penStyle, color, textColor );
|
|
m_plotItems.push_back( line );
|
|
line->attach( m_plot );
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
void RiuPlotAnnotationTool::attachAnnotation( QwtPlot* plot, RimPlotAxisAnnotation* annotation, RiaDefines::Orientation orientation )
|
|
{
|
|
if ( annotation->annotationType() == RimPlotAxisAnnotation::AnnotationType::LINE )
|
|
{
|
|
attachAnnotationLine( plot, annotation->color(), annotation->name(), annotation->penStyle(), annotation->value(), orientation );
|
|
}
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
void RiuPlotAnnotationTool::attachAnnotationRange( QwtPlot* plot,
|
|
const QColor& color,
|
|
const QString& annotationText,
|
|
const double rangeStart,
|
|
const double rangeEnd,
|
|
RiaDefines::Orientation orientation )
|
|
{
|
|
m_plot = plot;
|
|
if ( orientation == RiaDefines::Orientation::HORIZONTAL )
|
|
{
|
|
RiuPlotAnnotationTool::horizontalRange( annotationText,
|
|
std::make_pair( rangeStart, rangeEnd ),
|
|
color,
|
|
RiuGuiTheme::getColorByVariableName( "textColor" ) );
|
|
}
|
|
else if ( orientation == RiaDefines::Orientation::VERTICAL )
|
|
{
|
|
RiuPlotAnnotationTool::verticalRange( annotationText,
|
|
std::make_pair( rangeStart, rangeEnd ),
|
|
color,
|
|
RiuGuiTheme::getColorByVariableName( "textColor" ),
|
|
Qt::AlignHCenter );
|
|
}
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
void RiuPlotAnnotationTool::detachAllAnnotations()
|
|
{
|
|
if ( m_plot )
|
|
{
|
|
for ( auto& plotItem : m_plotItems )
|
|
{
|
|
plotItem->detach();
|
|
delete plotItem;
|
|
}
|
|
}
|
|
m_plotItems.clear();
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
Qt::Alignment RiuPlotAnnotationTool::trackTextAlignment( RiaDefines::TrackSpan trackSpan )
|
|
{
|
|
switch ( trackSpan )
|
|
{
|
|
case RiaDefines::TrackSpan::FULL_WIDTH:
|
|
return Qt::AlignRight;
|
|
case RiaDefines::TrackSpan::LEFT_COLUMN:
|
|
return Qt::AlignLeft;
|
|
case RiaDefines::TrackSpan::CENTRE_COLUMN:
|
|
return Qt::AlignCenter;
|
|
case RiaDefines::TrackSpan::RIGHT_COLUMN:
|
|
return Qt::AlignRight;
|
|
}
|
|
return Qt::AlignRight;
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
void RiuPlotAnnotationTool::horizontalRange( const QString& name,
|
|
const std::pair<double, double> yRange,
|
|
const QColor& color /*= QColor( 0, 0, 100 )*/,
|
|
const QColor& textColor /*= QColor( 0, 0, 100 )*/,
|
|
Qt::Alignment horizontalAlignment /*= Qt::AlignRight */ )
|
|
{
|
|
QColor shadingColor = color;
|
|
shadingColor.setAlpha( 10 );
|
|
|
|
auto* shading = new QwtPlotZoneItem();
|
|
shading->setOrientation( Qt::Horizontal );
|
|
shading->setInterval( yRange.first, yRange.second );
|
|
shading->setPen( shadingColor, 0.0, Qt::NoPen );
|
|
QBrush brush( shadingColor );
|
|
shading->setBrush( brush );
|
|
shading->attach( m_plot );
|
|
shading->setZ( -100.0 );
|
|
shading->setXAxis( QwtAxis::XBottom );
|
|
m_plotItems.push_back( shading );
|
|
|
|
auto* line( new QwtPlotMarker() );
|
|
RiuPlotAnnotationTool::setLineProperties( line, name, RiaDefines::Orientation::HORIZONTAL, yRange.first, Qt::DashLine, color, color, horizontalAlignment );
|
|
line->attach( m_plot );
|
|
m_plotItems.push_back( line );
|
|
|
|
auto* bottomLine( new QwtPlotMarker() );
|
|
RiuPlotAnnotationTool::setLineProperties( bottomLine, QString(), RiaDefines::Orientation::HORIZONTAL, yRange.second, Qt::DashLine, color, color );
|
|
|
|
bottomLine->attach( m_plot );
|
|
m_plotItems.push_back( bottomLine );
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
void RiuPlotAnnotationTool::verticalRange( const QString& name,
|
|
const std::pair<double, double> xRange,
|
|
const QColor& color /*= QColor( 0, 0, 100 )*/,
|
|
const QColor& textColor /*= QColor( 0, 0, 100 )*/,
|
|
Qt::Alignment horizontalAlignment /*= Qt::AlignRight */ )
|
|
{
|
|
QColor shadingColor = color;
|
|
shadingColor.setAlpha( 50 );
|
|
|
|
auto* shading = new QwtPlotZoneItem();
|
|
shading->setOrientation( Qt::Vertical );
|
|
shading->setInterval( xRange.first, xRange.second );
|
|
shading->setPen( shadingColor, 0.0, Qt::NoPen );
|
|
QBrush brush( shadingColor );
|
|
shading->setBrush( brush );
|
|
shading->attach( m_plot );
|
|
shading->setZ( -100.0 );
|
|
shading->setXAxis( QwtAxis::XBottom );
|
|
m_plotItems.push_back( shading );
|
|
|
|
QStringList labels = name.split( " - " );
|
|
|
|
auto* line( new QwtPlotMarker() );
|
|
RiuPlotAnnotationTool::setLineProperties( line,
|
|
labels[0],
|
|
RiaDefines::Orientation::VERTICAL,
|
|
xRange.first,
|
|
Qt::SolidLine,
|
|
color,
|
|
textColor,
|
|
Qt::AlignRight | horizontalAlignment );
|
|
line->attach( m_plot );
|
|
m_plotItems.push_back( line );
|
|
|
|
auto* rightLine( new QwtPlotMarker() );
|
|
RiuPlotAnnotationTool::setLineProperties( rightLine,
|
|
labels.size() == 2 ? labels[1] : QString(),
|
|
RiaDefines::Orientation::VERTICAL,
|
|
xRange.second,
|
|
Qt::SolidLine,
|
|
color,
|
|
textColor,
|
|
Qt::AlignLeft | horizontalAlignment );
|
|
rightLine->attach( m_plot );
|
|
m_plotItems.push_back( rightLine );
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
void RiuPlotAnnotationTool::setLineProperties( QwtPlotMarker* line,
|
|
const QString& name,
|
|
RiaDefines::Orientation orientation,
|
|
double linePosition,
|
|
Qt::PenStyle lineStyle,
|
|
const QColor& color /*= QColor( 0, 0, 100 )*/,
|
|
const QColor& textColor /*= QColor( 0, 0, 100 )*/,
|
|
Qt::Alignment horizontalAlignment /*= Qt::AlignRight*/,
|
|
int fontSize /*= 0 */ )
|
|
{
|
|
QPen curvePen;
|
|
curvePen.setStyle( lineStyle );
|
|
curvePen.setColor( color );
|
|
curvePen.setWidth( 1 );
|
|
|
|
line->setLinePen( curvePen );
|
|
QwtText label( name );
|
|
label.setColor( textColor );
|
|
if ( fontSize > 0 ) label.setFont( QFont( label.font().key(), fontSize ) );
|
|
line->setLabel( label );
|
|
line->setLabelAlignment( horizontalAlignment | Qt::AlignBottom );
|
|
|
|
if ( orientation == RiaDefines::Orientation::HORIZONTAL )
|
|
{
|
|
line->setLineStyle( QwtPlotMarker::HLine );
|
|
line->setYValue( linePosition );
|
|
}
|
|
else
|
|
{
|
|
line->setLineStyle( QwtPlotMarker::VLine );
|
|
line->setXValue( linePosition );
|
|
}
|
|
}
|