mirrors/JKQtPlotter
1
0
Fork 0
mirror of https://github.com/jkriege2/JKQtPlotter.git synced 2025-01-13 01:10:33 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity
4e7431ff77
JKQtPlotter/lib/jkqtplotter/graphs/jkqtpvectorfield.cpp

474 lines
18 KiB
C++
Raw Normal View History

NEW: Base class JKQTPXYAndVectorGraph for graphs like vector fields, e.g. (x,y,dx,dy) or (x,y,angle,length) NEW: ... and an actual implementation JKQTPVectorFieldGraph (+example)
2024-02-09 04:52:30 +08:00
/*
Copyright (c) 2008-2024 Jan W. Krieger (<jan@jkrieger.de>)
This software is free software: you can redistribute it and/or modify
it under the terms of the GNU Lesser General Public License (LGPL) as published by
the Free Software Foundation, either version 2.1 of the License, or
(at your option) any later version.
This program 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 Lesser General Public License (LGPL) for more details.
You should have received a copy of the GNU Lesser General Public License (LGPL)
along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
#include "jkqtplotter/graphs/jkqtpvectorfield.h"
#include "jkqtplotter/jkqtpbaseplotter.h"
#include "jkqtplotter/jkqtplotter.h"
#include <QDebug>
#include <QDateTime>
#include "jkqtcommon/jkqtpdrawingtools.h"
JKQTPVectorFieldGraph gained the option to also scale the line width with the vector length/magnitude
2024-02-09 23:45:40 +08:00
#include "jkqtmath/jkqtpstatbasics.h"
NEW: Base class JKQTPXYAndVectorGraph for graphs like vector fields, e.g. (x,y,dx,dy) or (x,y,angle,length) NEW: ... and an actual implementation JKQTPVectorFieldGraph (+example)
2024-02-09 04:52:30 +08:00
#include "jkqtplotter/jkqtptools.h"
#define SmallestGreaterZeroCompare_xvsgz() if ((xvsgz>10.0*DBL_MIN)&&((smallestGreaterZero<10.0*DBL_MIN) || (xvsgz<smallestGreaterZero))) smallestGreaterZero=xvsgz;
JKQTPVectorFieldGraph::JKQTPVectorFieldGraph(JKQTBasePlotter *parent):
JKQTPXYAndVectorGraph(parent),
replace JKQTPVectorFieldGraph::setAutoscaleLengthFactor() by an enum-property JKQTPVectorFieldGraph::setVectorLengthMode() and added a mode that ignores the length completely, the vectorfield example was also modified to better show this new property
2024-02-09 20:48:41 +08:00
m_vectorLengthMode(AutoscaleLength),
m_autoscaleLengthFactor(0.8),
m_lengthScaleFactor(1.0),
JKQTPVectorFieldGraph gained the option to also scale the line width with the vector length/magnitude
2024-02-09 23:45:40 +08:00
m_anchorPoint(AnchorBottom),
m_vectorLineWidthMode(DefaultVectorLineWidth),
NEW add JKQTPParametrizedVectorFieldGraph, which draws color-coded vector fields (color from length, angle, or user-supplied data column) + example
2024-02-10 07:26:45 +08:00
m_minLineWidth(0.001),
m_minVecLen(0),
m_maxVecLen(0)
NEW: Base class JKQTPXYAndVectorGraph for graphs like vector fields, e.g. (x,y,dx,dy) or (x,y,angle,length) NEW: ... and an actual implementation JKQTPVectorFieldGraph (+example)
2024-02-09 04:52:30 +08:00
{
initDecoratedLineStyle(parent, parentPlotStyle, JKQTPPlotStyleType::Default);
setTailDecoratorStyle(JKQTPNoDecorator);
setHeadDecoratorStyle(JKQTPFilledArrow);
setHeadDecoratorSizeFactor(getHeadDecoratorSizeFactor()/1.6);
setTailDecoratorSizeFactor(getTailDecoratorSizeFactor()/1.6);
}
JKQTPVectorFieldGraph::JKQTPVectorFieldGraph(JKQTPlotter *parent):
JKQTPVectorFieldGraph(parent->getPlotter())
{
}
void JKQTPVectorFieldGraph::draw(JKQTPEnhancedPainter &painter)
{
#ifdef JKQTBP_AUTOTIMER
JKQTPAutoOutputTimer jkaaot("JKQTPVectorFieldGraph::draw");
#endif
if (parent==nullptr) return;
const JKQTPDatastore* datastore=parent->getDatastore();
if (datastore==nullptr) return;
drawErrorsBefore(painter);
{
painter.save(); auto __finalpaint=JKQTPFinally([&painter]() {painter.restore();});
const QPen p=getLinePen(painter, parent);
JKQTPVectorFieldGraph gained the option to also scale the line width with the vector length/magnitude
2024-02-09 23:45:40 +08:00
const double lw=getLineWidth();
NEW: Base class JKQTPXYAndVectorGraph for graphs like vector fields, e.g. (x,y,dx,dy) or (x,y,angle,length) NEW: ... and an actual implementation JKQTPVectorFieldGraph (+example)
2024-02-09 04:52:30 +08:00
painter.setPen(p);
painter.setBrush(p.color());
int imax=0;
int imin=0;
double scale=1;
if (getIndexRange(imin, imax)) {
NEW add JKQTPParametrizedVectorFieldGraph, which draws color-coded vector fields (color from length, angle, or user-supplied data column) + example
2024-02-10 07:26:45 +08:00
m_minVecLen=m_maxVecLen=0;
NEW: Base class JKQTPXYAndVectorGraph for graphs like vector fields, e.g. (x,y,dx,dy) or (x,y,angle,length) NEW: ... and an actual implementation JKQTPVectorFieldGraph (+example)
2024-02-09 04:52:30 +08:00
// first determine (auto-scale) factor
replace JKQTPVectorFieldGraph::setAutoscaleLengthFactor() by an enum-property JKQTPVectorFieldGraph::setVectorLengthMode() and added a mode that ignores the length completely, the vectorfield example was also modified to better show this new property
2024-02-09 20:48:41 +08:00
if (m_vectorLengthMode==AutoscaleLength || m_vectorLengthMode==IgnoreLength) {
NEW: Base class JKQTPXYAndVectorGraph for graphs like vector fields, e.g. (x,y,dx,dy) or (x,y,angle,length) NEW: ... and an actual implementation JKQTPVectorFieldGraph (+example)
2024-02-09 04:52:30 +08:00
double avgVecLength=0;
double NDatapoints=0;
double xmin=0, xmax=0,ymin=0,ymax=0;
NEW add JKQTPParametrizedVectorFieldGraph, which draws color-coded vector fields (color from length, angle, or user-supplied data column) + example
2024-02-10 07:26:45 +08:00
std::vector<double> lengths;
replace JKQTPVectorFieldGraph::setAutoscaleLengthFactor() by an enum-property JKQTPVectorFieldGraph::setVectorLengthMode() and added a mode that ignores the length completely, the vectorfield example was also modified to better show this new property
2024-02-09 20:48:41 +08:00
lengths.reserve(imax-imin);
NEW: Base class JKQTPXYAndVectorGraph for graphs like vector fields, e.g. (x,y,dx,dy) or (x,y,angle,length) NEW: ... and an actual implementation JKQTPVectorFieldGraph (+example)
2024-02-09 04:52:30 +08:00
for (int iii=imin; iii<imax; iii++) {
const int i=qBound(imin, getDataIndex(iii), imax);
const double xv=datastore->get(static_cast<size_t>(xColumn),static_cast<size_t>(i));
const double yv=datastore->get(static_cast<size_t>(yColumn),static_cast<size_t>(i));
const QPointF vecv=getVectorDxDy(i);
if (JKQTPIsOKFloat(xv) && JKQTPIsOKFloat(yv) && JKQTPIsOKFloat(vecv)) {
NEW add JKQTPParametrizedVectorFieldGraph, which draws color-coded vector fields (color from length, angle, or user-supplied data column) + example
2024-02-10 07:26:45 +08:00
const double l=getVectorMagnitude(vecv);
lengths.push_back(l);
replace JKQTPVectorFieldGraph::setAutoscaleLengthFactor() by an enum-property JKQTPVectorFieldGraph::setVectorLengthMode() and added a mode that ignores the length completely, the vectorfield example was also modified to better show this new property
2024-02-09 20:48:41 +08:00
avgVecLength+=l;
NEW: Base class JKQTPXYAndVectorGraph for graphs like vector fields, e.g. (x,y,dx,dy) or (x,y,angle,length) NEW: ... and an actual implementation JKQTPVectorFieldGraph (+example)
2024-02-09 04:52:30 +08:00
if (NDatapoints==0) {
xmin=xmax=xv;
ymin=ymax=yv;
NEW add JKQTPParametrizedVectorFieldGraph, which draws color-coded vector fields (color from length, angle, or user-supplied data column) + example
2024-02-10 07:26:45 +08:00
m_minVecLen=m_maxVecLen=l;
NEW: Base class JKQTPXYAndVectorGraph for graphs like vector fields, e.g. (x,y,dx,dy) or (x,y,angle,length) NEW: ... and an actual implementation JKQTPVectorFieldGraph (+example)
2024-02-09 04:52:30 +08:00
} else {
xmin=qMin(xmin,xv);
xmax=qMax(xmax,xv);
ymin=qMin(ymin,yv);
ymax=qMax(ymax,yv);
NEW add JKQTPParametrizedVectorFieldGraph, which draws color-coded vector fields (color from length, angle, or user-supplied data column) + example
2024-02-10 07:26:45 +08:00
m_minVecLen=qMin(m_minVecLen,l);
m_maxVecLen=qMax(m_maxVecLen,l);
NEW: Base class JKQTPXYAndVectorGraph for graphs like vector fields, e.g. (x,y,dx,dy) or (x,y,angle,length) NEW: ... and an actual implementation JKQTPVectorFieldGraph (+example)
2024-02-09 04:52:30 +08:00
}
NDatapoints++;
}
}
replace JKQTPVectorFieldGraph::setAutoscaleLengthFactor() by an enum-property JKQTPVectorFieldGraph::setVectorLengthMode() and added a mode that ignores the length completely, the vectorfield example was also modified to better show this new property
2024-02-09 20:48:41 +08:00
//avgVecLength/=NDatapoints;
avgVecLength=jkqtpstatQuantile(lengths.begin(), lengths.end(), 0.9);
NEW: Base class JKQTPXYAndVectorGraph for graphs like vector fields, e.g. (x,y,dx,dy) or (x,y,angle,length) NEW: ... and an actual implementation JKQTPVectorFieldGraph (+example)
2024-02-09 04:52:30 +08:00
const double plotsize=qMax(fabs(xmax-xmin),fabs(ymax-ymin));
replace JKQTPVectorFieldGraph::setAutoscaleLengthFactor() by an enum-property JKQTPVectorFieldGraph::setVectorLengthMode() and added a mode that ignores the length completely, the vectorfield example was also modified to better show this new property
2024-02-09 20:48:41 +08:00
const double aproxNPerSide=sqrt(NDatapoints);
if (m_vectorLengthMode==IgnoreLength) scale=plotsize/aproxNPerSide*m_autoscaleLengthFactor; // we can assume that the elngths of all vectors have been normalized beforehand
else scale=plotsize/aproxNPerSide/avgVecLength*m_autoscaleLengthFactor;
NEW: Base class JKQTPXYAndVectorGraph for graphs like vector fields, e.g. (x,y,dx,dy) or (x,y,angle,length) NEW: ... and an actual implementation JKQTPVectorFieldGraph (+example)
2024-02-09 04:52:30 +08:00
} else {
scale=m_lengthScaleFactor;
}
// now draw
for (int iii=imin; iii<imax; iii++) {
const int i=qBound(imin, getDataIndex(iii), imax);
const double xv=datastore->get(static_cast<size_t>(xColumn),static_cast<size_t>(i));
const double yv=datastore->get(static_cast<size_t>(yColumn),static_cast<size_t>(i));
const double x=transformX(xv);
const double y=transformY(yv);
JKQTPVectorFieldGraph gained the option to also scale the line width with the vector length/magnitude
2024-02-09 23:45:40 +08:00
const QPointF vec_orig=getVectorDxDy(i);
const QPointF vecv=[&](QPointF vec) {
replace JKQTPVectorFieldGraph::setAutoscaleLengthFactor() by an enum-property JKQTPVectorFieldGraph::setVectorLengthMode() and added a mode that ignores the length completely, the vectorfield example was also modified to better show this new property
2024-02-09 20:48:41 +08:00
if (m_vectorLengthMode==IgnoreLength) {
fixtypo in code
2024-02-10 07:43:18 +08:00
const double veclen=getVectorMagnitude(vec);
replace JKQTPVectorFieldGraph::setAutoscaleLengthFactor() by an enum-property JKQTPVectorFieldGraph::setVectorLengthMode() and added a mode that ignores the length completely, the vectorfield example was also modified to better show this new property
2024-02-09 20:48:41 +08:00
if (qFuzzyIsNull(veclen)) vec=QPointF(0,0);
else vec/=veclen; // normalize vector
}
return vec;
JKQTPVectorFieldGraph gained the option to also scale the line width with the vector length/magnitude
2024-02-09 23:45:40 +08:00
}(vec_orig);
NEW: Base class JKQTPXYAndVectorGraph for graphs like vector fields, e.g. (x,y,dx,dy) or (x,y,angle,length) NEW: ... and an actual implementation JKQTPVectorFieldGraph (+example)
2024-02-09 04:52:30 +08:00
const QLineF l=[&]() {
switch (m_anchorPoint) {
case AnchorBottom: return QLineF(x,y,transformX(xv+scale*vecv.x()),transformY(yv+scale*vecv.y()));
case AnchorMid: return QLineF(transformX(xv-0.5*scale*vecv.x()),transformY(yv-0.5*scale*vecv.y()),transformX(xv+0.5*scale*vecv.x()),transformY(yv+0.5*scale*vecv.y()));
case AnchorTip: return QLineF(transformX(xv-scale*vecv.x()),transformY(yv-scale*vecv.y()), x,y);
}
return QLineF(JKQTP_NAN,JKQTP_NAN,JKQTP_NAN,JKQTP_NAN);
}();
replace JKQTPVectorFieldGraph::setAutoscaleLengthFactor() by an enum-property JKQTPVectorFieldGraph::setVectorLengthMode() and added a mode that ignores the length completely, the vectorfield example was also modified to better show this new property
2024-02-09 20:48:41 +08:00
if (JKQTPIsOKFloat(l) && l.length()>0) {
JKQTPVectorFieldGraph gained the option to also scale the line width with the vector length/magnitude
2024-02-09 23:45:40 +08:00
double actualLW=p.widthF();
NEW add JKQTPParametrizedVectorFieldGraph, which draws color-coded vector fields (color from length, angle, or user-supplied data column) + example
2024-02-10 07:26:45 +08:00
QPen plw=p;
JKQTPVectorFieldGraph gained the option to also scale the line width with the vector length/magnitude
2024-02-09 23:45:40 +08:00
if (m_vectorLineWidthMode==AutoscaleLineWidthFromLength) {
NEW add JKQTPParametrizedVectorFieldGraph, which draws color-coded vector fields (color from length, angle, or user-supplied data column) + example
2024-02-10 07:26:45 +08:00
const double vec_origlen=getVectorMagnitude(vec_orig);
plw.setWidthF(m_minLineWidth+(vec_origlen-m_minVecLen)/(m_maxVecLen-m_minVecLen)*(lw-m_minLineWidth));
JKQTPVectorFieldGraph gained the option to also scale the line width with the vector length/magnitude
2024-02-09 23:45:40 +08:00
actualLW=plw.widthF();
}
NEW add JKQTPParametrizedVectorFieldGraph, which draws color-coded vector fields (color from length, angle, or user-supplied data column) + example
2024-02-10 07:26:45 +08:00
plw.setColor(getLocalVectorColor(i,xv,yv,vec_orig.x(),vec_orig.y()));
painter.setPen(plw);
painter.setBrush(plw.color());
JKQTPVectorFieldGraph gained the option to also scale the line width with the vector length/magnitude
2024-02-09 23:45:40 +08:00
JKQTPPlotDecoratedLine(painter,l, getTailDecoratorStyle(), calcTailDecoratorSize(actualLW), getHeadDecoratorStyle(), calcHeadDecoratorSize(actualLW));
NEW: Base class JKQTPXYAndVectorGraph for graphs like vector fields, e.g. (x,y,dx,dy) or (x,y,angle,length) NEW: ... and an actual implementation JKQTPVectorFieldGraph (+example)
2024-02-09 04:52:30 +08:00
}
}
}
}
drawErrorsAfter(painter);
}
void JKQTPVectorFieldGraph::drawKeyMarker(JKQTPEnhancedPainter &painter, const QRectF &rect)
{
painter.save(); auto __finalpaint=JKQTPFinally([&painter]() {painter.restore();});
const QPen p=getKeyLinePen(painter, rect, parent);
painter.setPen(p);
painter.setBrush(p.color());
const QLineF l(rect.left(), rect.bottom(), rect.right(), rect.top());
JKQTPPlotDecoratedLine(painter,l, getTailDecoratorStyle(), calcTailDecoratorSize(p.widthF()), getHeadDecoratorStyle(), calcHeadDecoratorSize(p.widthF()));
}
QColor JKQTPVectorFieldGraph::getKeyLabelColor() const
{
return getLineColor();
}
replace JKQTPVectorFieldGraph::setAutoscaleLengthFactor() by an enum-property JKQTPVectorFieldGraph::setVectorLengthMode() and added a mode that ignores the length completely, the vectorfield example was also modified to better show this new property
2024-02-09 20:48:41 +08:00
JKQTPVectorFieldGraph::VectorLengthMode JKQTPVectorFieldGraph::getVectorLengthMode() const
NEW: Base class JKQTPXYAndVectorGraph for graphs like vector fields, e.g. (x,y,dx,dy) or (x,y,angle,length) NEW: ... and an actual implementation JKQTPVectorFieldGraph (+example)
2024-02-09 04:52:30 +08:00
{
replace JKQTPVectorFieldGraph::setAutoscaleLengthFactor() by an enum-property JKQTPVectorFieldGraph::setVectorLengthMode() and added a mode that ignores the length completely, the vectorfield example was also modified to better show this new property
2024-02-09 20:48:41 +08:00
return m_vectorLengthMode;
NEW: Base class JKQTPXYAndVectorGraph for graphs like vector fields, e.g. (x,y,dx,dy) or (x,y,angle,length) NEW: ... and an actual implementation JKQTPVectorFieldGraph (+example)
2024-02-09 04:52:30 +08:00
}
replace JKQTPVectorFieldGraph::setAutoscaleLengthFactor() by an enum-property JKQTPVectorFieldGraph::setVectorLengthMode() and added a mode that ignores the length completely, the vectorfield example was also modified to better show this new property
2024-02-09 20:48:41 +08:00
void JKQTPVectorFieldGraph::setVectorLengthMode(VectorLengthMode newMode)
NEW: Base class JKQTPXYAndVectorGraph for graphs like vector fields, e.g. (x,y,dx,dy) or (x,y,angle,length) NEW: ... and an actual implementation JKQTPVectorFieldGraph (+example)
2024-02-09 04:52:30 +08:00
{
replace JKQTPVectorFieldGraph::setAutoscaleLengthFactor() by an enum-property JKQTPVectorFieldGraph::setVectorLengthMode() and added a mode that ignores the length completely, the vectorfield example was also modified to better show this new property
2024-02-09 20:48:41 +08:00
m_vectorLengthMode = newMode;
NEW: Base class JKQTPXYAndVectorGraph for graphs like vector fields, e.g. (x,y,dx,dy) or (x,y,angle,length) NEW: ... and an actual implementation JKQTPVectorFieldGraph (+example)
2024-02-09 04:52:30 +08:00
}
JKQTPVectorFieldGraph: added scaling factor for length-autoscaling
2024-02-09 18:41:19 +08:00
double JKQTPVectorFieldGraph::getAutoscaleLengthFactor() const
{
return m_autoscaleLengthFactor;
}
void JKQTPVectorFieldGraph::setAutoscaleLengthFactor(double newAutoscaleLengthFactor)
{
m_autoscaleLengthFactor=newAutoscaleLengthFactor;
}
NEW: Base class JKQTPXYAndVectorGraph for graphs like vector fields, e.g. (x,y,dx,dy) or (x,y,angle,length) NEW: ... and an actual implementation JKQTPVectorFieldGraph (+example)
2024-02-09 04:52:30 +08:00
double JKQTPVectorFieldGraph::getLengthScaleFactor() const
{
return m_lengthScaleFactor;
}
void JKQTPVectorFieldGraph::setLengthScaleFactor(double newLengthScaleFactor)
{
m_lengthScaleFactor = newLengthScaleFactor;
}
JKQTPVectorFieldGraph::VectorAnchorPoint JKQTPVectorFieldGraph::getAnchorPoint() const
{
return m_anchorPoint;
}
void JKQTPVectorFieldGraph::setAnchorPoint(VectorAnchorPoint newAnchorPoint)
{
m_anchorPoint = newAnchorPoint;
}
JKQTPVectorFieldGraph gained the option to also scale the line width with the vector length/magnitude
2024-02-09 23:45:40 +08:00
void JKQTPVectorFieldGraph::setVectorLineWidthMode(VectorLineWidthMode m)
{
m_vectorLineWidthMode=m;
}
JKQTPVectorFieldGraph::VectorLineWidthMode JKQTPVectorFieldGraph::getVectorLineWidthMode() const
{
return m_vectorLineWidthMode;
}
void JKQTPVectorFieldGraph::setMinLineWidth(double lw)
{
m_minLineWidth=lw;
}
double JKQTPVectorFieldGraph::getMinLineWIdth() const
{
return m_minLineWidth;
}
NEW add JKQTPParametrizedVectorFieldGraph, which draws color-coded vector fields (color from length, angle, or user-supplied data column) + example
2024-02-10 07:26:45 +08:00
QColor JKQTPVectorFieldGraph::getLocalVectorColor(int /*i*/, double /*x*/, double /*y*/, double /*dx*/, double /*dy*/) const
{
return getLineColor();
}
JKQTPParametrizedVectorFieldGraph::JKQTPParametrizedVectorFieldGraph(JKQTBasePlotter *parent):
JKQTPVectorFieldGraph(parent),
JKQTPColorPaletteStyleAndToolsMixin(parent),
m_colorColumn(-1),
m_colorColumnContainsRGB(false),
fix security issues (members were not initialized)
2024-02-13 05:27:24 +08:00
m_intColMin(0),
m_intColMax(0),
NEW add JKQTPParametrizedVectorFieldGraph, which draws color-coded vector fields (color from length, angle, or user-supplied data column) + example
2024-02-10 07:26:45 +08:00
m_vectorColorMode(ColorFromMagnitude)
{
palette=JKQTPMathImageMATLAB;
if (parent) {
palette=parent->getCurrentPlotterStyle().graphsStyle.defaultPalette;
}
}
JKQTPParametrizedVectorFieldGraph::JKQTPParametrizedVectorFieldGraph(JKQTPlotter *parent):
JKQTPParametrizedVectorFieldGraph(parent->getPlotter())
{
}
void JKQTPParametrizedVectorFieldGraph::drawKeyMarker(JKQTPEnhancedPainter &painter, const QRectF &rect)
{
if (m_vectorColorMode==DefaultColor) {
JKQTPVectorFieldGraph::drawKeyMarker(painter,rect);
} else {
QColor color1=getKeyLabelColor();
QColor color2=getKeyLabelColor();
if (m_colorColumn>=0) {
if (m_colorColumnContainsRGB) {
color1=QColor("red");
color2=QColor("blue");
} else {
QImage img;
double colorval[]={0,1};
JKQTPImageTools::array2image<double>(colorval, 2, 1, img, getColorPalette(), double(0.0), double(1.0));
color1=img.pixel(0,0);
color2=img.pixel(1,0);
}
}
painter.save(); auto __finalpaint=JKQTPFinally([&painter]() {painter.restore();});
QPen p=getKeyLinePen(painter, rect, parent);
p.setColor(color1);
painter.setPen(p);
painter.setBrush(p.color());
const QLineF l(rect.left(), rect.bottom(), rect.right(), (rect.top()+rect.bottom())/2.0);
JKQTPPlotDecoratedLine(painter,l, getTailDecoratorStyle(), calcTailDecoratorSize(p.widthF()), getHeadDecoratorStyle(), calcHeadDecoratorSize(p.widthF()));
p=getKeyLinePen(painter, rect, parent);
p.setColor(color2);
painter.setPen(p);
painter.setBrush(p.color());
const QLineF l2(rect.left(), (rect.top()+rect.bottom())/2.0, rect.right(), rect.top());
JKQTPPlotDecoratedLine(painter,l2, getTailDecoratorStyle(), calcTailDecoratorSize(p.widthF()), getHeadDecoratorStyle(), calcHeadDecoratorSize(p.widthF()));
}
}
void JKQTPParametrizedVectorFieldGraph::draw(JKQTPEnhancedPainter &painter)
{
cbGetDataMinMax(m_intColMin, m_intColMax);
JKQTPVectorFieldGraph::draw(painter);
}
void JKQTPParametrizedVectorFieldGraph::setParent(JKQTBasePlotter *parent)
{
JKQTPVectorFieldGraph::setParent(parent);
cbSetParent(parent);
}
void JKQTPParametrizedVectorFieldGraph::getOutsideSize(JKQTPEnhancedPainter &painter, int &leftSpace, int &rightSpace, int &topSpace, int &bottomSpace)
{
JKQTPVectorFieldGraph::getOutsideSize(painter, leftSpace, rightSpace, topSpace, bottomSpace);
if (showColorBar&& m_colorColumn>=0 && !m_colorColumnContainsRGB) cbGetOutsideSize(painter, leftSpace, rightSpace, topSpace, bottomSpace);
}
void JKQTPParametrizedVectorFieldGraph::drawOutside(JKQTPEnhancedPainter &painter, QRect leftSpace, QRect rightSpace, QRect topSpace, QRect bottomSpace)
{
JKQTPVectorFieldGraph::drawOutside(painter, leftSpace, rightSpace, topSpace, bottomSpace);
if (showColorBar&& m_colorColumn>=0 && !m_colorColumnContainsRGB) cbDrawOutside(painter, leftSpace, rightSpace, topSpace, bottomSpace);
}
void JKQTPParametrizedVectorFieldGraph::cbGetDataMinMax(double &dmin, double &dmax)
{
dmin=dmax=0;
if (autoImageRange) {
if (parent==nullptr) return;
JKQTPDatastore* datastore=parent->getDatastore();
if (datastore==nullptr) return;
int imin=0, imax=0;
if (getIndexRange(imin, imax)) {
if (m_vectorColorMode==ColorFromCustomColumn) {
if (m_colorColumn<0) return;
bool first=true;
for (int iii=imin; iii<imax; iii++) {
const int i=qBound(imin, getDataIndex(iii), imax);
const double xv=datastore->get(m_colorColumn,i);
if (JKQTPIsOKFloat(xv)) {
if (first) {
dmin=dmax=xv;
first=false;
} else {
dmin=qMin(xv, dmin);
dmax=qMax(xv, dmax);
}
}
}
} else if (m_vectorColorMode==ColorFromMagnitude) {
bool first=true;
for (int iii=imin; iii<imax; iii++) {
const int i=qBound(imin, getDataIndex(iii), imax);
const double vecLen=getVectorMagnitude(i);
if (JKQTPIsOKFloat(vecLen)) {
if (first) {
dmin=dmax=vecLen;
first=false;
} else {
dmin=qMin(vecLen, dmin);
dmax=qMax(vecLen, dmax);
}
}
}
} else if (m_vectorColorMode==ColorFromAngle) {
bool first=true;
for (int iii=imin; iii<imax; iii++) {
const int i=qBound(imin, getDataIndex(iii), imax);
const double vecAngle=getVectorAngle(i);
if (JKQTPIsOKFloat(vecAngle)) {
if (first) {
dmin=dmax=vecAngle;
first=false;
} else {
dmin=qMin(vecAngle, dmin);
dmax=qMax(vecAngle, dmax);
}
}
}
}
}
} else {
dmin=imageMin;
dmax=imageMax;
}
}
bool JKQTPParametrizedVectorFieldGraph::usesColumn(int c) const
{
return (c==m_colorColumn) || JKQTPVectorFieldGraph::usesColumn(c);
}
void JKQTPParametrizedVectorFieldGraph::setColorColumn(int __value)
{
m_colorColumn = __value;
if (__value<0) m_vectorColorMode=ColorFromMagnitude;
else m_vectorColorMode=ColorFromCustomColumn;
}
int JKQTPParametrizedVectorFieldGraph::getColorColumn() const
{
return m_colorColumn;
}
void JKQTPParametrizedVectorFieldGraph::setColorColumn(size_t __value)
{
m_colorColumn = static_cast<int>(__value);
m_vectorColorMode=ColorFromCustomColumn;
}
void JKQTPParametrizedVectorFieldGraph::setColorColumnContainsRGB(bool __value)
{
m_colorColumnContainsRGB=__value;
}
bool JKQTPParametrizedVectorFieldGraph::getColorColumnContainsRGB() const
{
return m_colorColumnContainsRGB;
}
void JKQTPParametrizedVectorFieldGraph::setVectorColorMode(VectorColorMode __value)
{
m_vectorColorMode=__value;
}
JKQTPParametrizedVectorFieldGraph::VectorColorMode JKQTPParametrizedVectorFieldGraph::getVectorColorMode() const
{
return m_vectorColorMode;
}
QColor JKQTPParametrizedVectorFieldGraph::getLocalVectorColor(int i, double x, double y, double dx, double dy) const
{
if (parent==nullptr) return getLineColor();
const JKQTPDatastore* datastore=parent->getDatastore();
if (datastore==nullptr) return getLineColor();
if (m_colorColumn<0 && m_vectorColorMode==ColorFromCustomColumn) return getLineColor();
if (m_colorColumn>=0 && m_vectorColorMode==ColorFromCustomColumn && (i<0 || i>=(int64_t)datastore->getRows(m_colorColumn))) return getLineColor();
double colValue=0;
double colMin=m_intColMin;
double colMax=m_intColMax;
switch(m_vectorColorMode) {
case ColorFromCustomColumn:
colValue=datastore->get(m_colorColumn,i);
if (m_intColMin==m_intColMax) {
colMin=0;
colMax=datastore->getRows(m_colorColumn)-1;
}
break;
case ColorFromMagnitude:
colValue=getVectorMagnitude(QPointF(dx,dy));
break;
case ColorFromAngle:
colValue=getVectorAngle(QPointF(dx,dy));
break;
case DefaultColor:
return getLineColor();
}
if (m_colorColumnContainsRGB && m_vectorColorMode==ColorFromCustomColumn) {
return QRgb(round(colValue));
} else {
QImage img;
JKQTPImageTools::array2image(&colValue, 1, 1, img, palette, colMin, colMax);
return img.pixel(0,0);
}
}
Reference in New Issue Copy Permalink